Ejemplo n.º 1
0
 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)
Ejemplo n.º 2
0
 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'])
Ejemplo n.º 3
0
 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')
Ejemplo n.º 4
0
 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'])
Ejemplo n.º 5
0
 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')
Ejemplo n.º 6
0
    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')
Ejemplo n.º 7
0
    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)