Show More
@@ -0,0 +1,36 b'' | |||||
|
1 | """Tests for the notebook kernel and session manager.""" | |||
|
2 | ||||
|
3 | from unittest import TestCase | |||
|
4 | ||||
|
5 | from IPython.config.loader import Config | |||
|
6 | from IPython.zmq.kernelmanager import KernelManager | |||
|
7 | ||||
|
8 | class TestKernelManager(TestCase): | |||
|
9 | ||||
|
10 | def _get_tcp_km(self): | |||
|
11 | return KernelManager() | |||
|
12 | ||||
|
13 | def _get_ipc_km(self): | |||
|
14 | c = Config() | |||
|
15 | c.KernelManager.transport = 'ipc' | |||
|
16 | c.KernelManager.ip = 'test' | |||
|
17 | km = KernelManager(config=c) | |||
|
18 | return km | |||
|
19 | ||||
|
20 | def _run_lifecycle(self, km): | |||
|
21 | km.start_kernel() | |||
|
22 | km.start_channels(shell=True, iopub=False, stdin=False, hb=False) | |||
|
23 | # km.shell_channel.start() | |||
|
24 | km.restart_kernel() | |||
|
25 | km.interrupt_kernel() | |||
|
26 | self.assertTrue(isinstance(km, KernelManager)) | |||
|
27 | km.shutdown_kernel() | |||
|
28 | km.shell_channel.stop() | |||
|
29 | ||||
|
30 | def test_km_tcp(self): | |||
|
31 | km = self._get_tcp_km() | |||
|
32 | self._run_lifecycle(km) | |||
|
33 | ||||
|
34 | def test_km_ipc(self): | |||
|
35 | km = self._get_ipc_km() | |||
|
36 | self._run_lifecycle(km) |
General Comments 0
You need to be logged in to leave comments.
Login now