def test_execute(self): """ Does executing code in an in-process kernel work? """ km = BlockingInProcessKernelManager() km.start_kernel() km.shell_channel.execute('foo = 1') self.assertEquals(km.kernel.shell.user_ns['foo'], 1)
def test_inteface(self): """ Does the in-process kernel manager implement the basic KM interface? """ km = BlockingInProcessKernelManager() self.assert_(not km.channels_running) self.assert_(not km.has_kernel) km.start_channels() self.assert_(km.channels_running) km.start_kernel() self.assert_(km.has_kernel) self.assert_(km.kernel is not None) old_kernel = km.kernel km.restart_kernel() self.assert_(km.kernel is not None) 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) km.stop_channels() self.assert_(not km.channels_running)
def test_pylab(self): """ Does pylab work in the in-process kernel? """ km = BlockingInProcessKernelManager() km.start_kernel() km.shell_channel.execute('%pylab') msg = get_stream_message(km) self.assert_('Welcome to pylab' in msg['content']['data'])
def test_object_info(self): """ Does requesting object information from an in-process kernel work? """ km = BlockingInProcessKernelManager() km.start_kernel() km.kernel.shell.user_ns['foo'] = 1 km.shell_channel.object_info('foo') msg = km.shell_channel.get_msg() self.assertEquals(msg['header']['msg_type'], 'object_info_reply') self.assertEquals(msg['content']['name'], 'foo') self.assertEquals(msg['content']['type_name'], 'int')
def test_complete(self): """ Does requesting completion from an in-process kernel work? """ km = BlockingInProcessKernelManager() km.start_kernel() km.kernel.shell.push({'my_bar': 0, 'my_baz': 1}) km.shell_channel.complete('my_ba', 'my_ba', 5) msg = km.shell_channel.get_msg() self.assertEquals(msg['header']['msg_type'], 'complete_reply') self.assertEquals(sorted(msg['content']['matches']), ['my_bar', 'my_baz'])
def test_history(self): """ Does requesting history from an in-process kernel work? """ km = BlockingInProcessKernelManager() km.start_kernel() km.shell_channel.execute('%who') km.shell_channel.history(hist_access_type='tail', n=1) msg = km.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_raw_input(self): """ Does the in-process kernel handle raw_input correctly? """ km = BlockingInProcessKernelManager() km.start_kernel() io = StringIO('foobar\n') sys_stdin = sys.stdin sys.stdin = io try: if py3compat.PY3: km.shell_channel.execute('x = input()') else: km.shell_channel.execute('x = raw_input()') finally: sys.stdin = sys_stdin self.assertEqual(km.kernel.shell.user_ns.get('x'), 'foobar')