1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
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)
|
