def test_execute(self): """ Does executing code in an in-process kernel work? """ km = InProcessKernelManager() km.start_kernel() kc = km.client() kc.start_channels() kc.wait_for_ready() kc.execute('foo = 1') self.assertEquals(km.kernel.shell.user_ns['foo'], 1)
def test_interface(self): """ Does the in-process kernel manager implement the basic KM interface? """ km = InProcessKernelManager() self.assert_(not km.has_kernel) km.start_kernel() self.assert_(km.has_kernel) self.assert_(km.kernel is not None) kc = km.client() self.assert_(not kc.channels_running) kc.start_channels() self.assert_(kc.channels_running) old_kernel = km.kernel km.restart_kernel() self.assertIsNotNone(km.kernel) self.assertNotEquals(km.kernel, old_kernel) km.shutdown_kernel() self.assert_(not km.has_kernel) self.assertRaises(NotImplementedError, km.interrupt_kernel) self.assertRaises(NotImplementedError, km.signal_kernel, 9) kc.stop_channels() self.assert_(not kc.channels_running)
def test_complete(self): """ Does requesting completion from an in-process kernel work? """ km = InProcessKernelManager() km.start_kernel() kc = km.client() kc.start_channels() kc.wait_for_ready() km.kernel.shell.push({'my_bar': 0, 'my_baz': 1}) kc.complete('my_ba', 5) msg = kc.get_shell_msg() self.assertEqual(msg['header']['msg_type'], 'complete_reply') self.assertEqual(sorted(msg['content']['matches']), ['my_bar', 'my_baz'])
def test_history(self): """ Does requesting history from an in-process kernel work? """ km = InProcessKernelManager() km.start_kernel() kc = km.client() kc.start_channels() kc.wait_for_ready() kc.execute('%who') kc.history(hist_access_type='tail', n=1) msg = kc.shell_channel.get_msgs()[-1] self.assertEquals(msg['header']['msg_type'], 'history_reply') history = msg['content']['history'] self.assertEquals(len(history), 1) self.assertEquals(history[0][2], '%who')
def test_inspect(self): """ Does requesting object information from an in-process kernel work? """ km = InProcessKernelManager() km.start_kernel() kc = km.client() kc.start_channels() kc.wait_for_ready() km.kernel.shell.user_ns['foo'] = 1 kc.inspect('foo') msg = kc.get_shell_msg() self.assertEqual(msg['header']['msg_type'], 'inspect_reply') content = msg['content'] assert content['found'] text = content['data']['text/plain'] self.assertIn('int', text)
class InProcessKernelTestCase(unittest.TestCase): def setUp(self): self.km = InProcessKernelManager() self.km.start_kernel() self.kc = self.km.client() self.kc.start_channels() self.kc.wait_for_ready() @skipif_not_matplotlib def test_pylab(self): """Does %pylab work in the in-process kernel?""" kc = self.kc kc.execute('%pylab') out, err = assemble_output(kc.iopub_channel) self.assertIn('matplotlib', out) def test_raw_input(self): """ Does the in-process kernel handle raw_input correctly? """ io = StringIO('foobar\n') sys_stdin = sys.stdin sys.stdin = io try: if py3compat.PY3: self.kc.execute('x = input()') else: self.kc.execute('x = raw_input()') finally: sys.stdin = sys_stdin self.assertEqual(self.km.kernel.shell.user_ns.get('x'), 'foobar') def test_stdout(self): """ Does the in-process kernel correctly capture IO? """ kernel = InProcessKernel() with capture_output() as io: kernel.shell.run_cell('print("foo")') self.assertEqual(io.stdout, 'foo\n') kc = BlockingInProcessKernelClient(kernel=kernel, session=kernel.session) kernel.frontends.append(kc) kc.execute('print("bar")') out, err = assemble_output(kc.iopub_channel) self.assertEqual(out, 'bar\n')
def setUp(self): self.km = InProcessKernelManager() self.km.start_kernel() self.kc = self.km.client() self.kc.start_channels() self.kc.wait_for_ready()