You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
21 lines
883 B
21 lines
883 B
from jupyter_client.manager import AsyncKernelManager
|
|
|
|
|
|
class FakeCustomKernelManager(AsyncKernelManager):
|
|
expected_methods = {'__init__': 0, 'client': 0, 'start_kernel': 0}
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
self.log.info('FakeCustomKernelManager initialized')
|
|
self.expected_methods['__init__'] += 1
|
|
super(FakeCustomKernelManager, self).__init__(*args, **kwargs)
|
|
|
|
async def start_kernel(self, *args, **kwargs):
|
|
self.log.info('FakeCustomKernelManager started a kernel')
|
|
self.expected_methods['start_kernel'] += 1
|
|
return await super(FakeCustomKernelManager, self).start_kernel(*args, **kwargs)
|
|
|
|
def client(self, *args, **kwargs):
|
|
self.log.info('FakeCustomKernelManager created a client')
|
|
self.expected_methods['client'] += 1
|
|
return super(FakeCustomKernelManager, self).client(*args, **kwargs)
|