Files
AzSuicideDataVisualization/.venv/Lib/site-packages/nbconvert/preprocessors/tests/fake_kernelmanager.py
2022-05-23 00:16:32 +04:00

25 lines
805 B
Python

from jupyter_client.manager import KernelManager
class FakeCustomKernelManager(KernelManager):
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().__init__(*args, **kwargs)
def start_kernel(self, *args, **kwargs):
self.log.info("FakeCustomKernelManager started a kernel")
self.expected_methods["start_kernel"] += 1
return super().start_kernel(*args, **kwargs)
def client(self, *args, **kwargs):
self.log.info("FakeCustomKernelManager created a client")
self.expected_methods["client"] += 1
return super().client(*args, **kwargs)