upstream/ipython Files · IPython/kernel/zmq/tests/test_start_kernel.py

Shut down kernels in parallel...

Shut down kernels in parallel When stopping the notebook server, it currently sends a shutdown request to each kernel and then waits for the process to finish. This can be slow if you have several kernels running. This makes it issues all the shutdown requests before waiting on the processes, so shutdown happens in parallel. KernelManager (and MultiKernelManager) gain three new public API methods to allow this: * request_shutdown (promoted from a private method) * wait_shutdown (refactored out of shutdown_kernel) * cleanup (refactored out of shutdown_kernel)

Thomas Kluyver - - Load All Authors

File last commit:

r12477:4d6e0e5f


                r16510:633371e5

Download file

             test_start_kernel.py
        
                    45 lines
            
             | 1.7 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / IPython / kernel / zmq / tests / test_start_kernel.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      import nose.tools as nt

      from .test_embed_kernel import setup, teardown, setup_kernel

      TIMEOUT = 15

      def test_ipython_start_kernel_userns():

          cmd = ('from IPython import start_kernel\n'

                 'ns = {"tre": 123}\n'

                 'start_kernel(user_ns=ns)')

          with setup_kernel(cmd) as client:

              msg_id = client.object_info('tre')

              msg = client.get_shell_msg(block=True, timeout=TIMEOUT)

              content = msg['content']

              assert content['found']

              nt.assert_equal(content['string_form'], u'123')

              # user_module should be an instance of DummyMod

              msg_id = client.execute("usermod = get_ipython().user_module")

              msg = client.get_shell_msg(block=True, timeout=TIMEOUT)

              content = msg['content']

              nt.assert_equal(content['status'], u'ok')

              msg_id = client.object_info('usermod')

              msg = client.get_shell_msg(block=True, timeout=TIMEOUT)

              content = msg['content']

              assert content['found']

              nt.assert_in('DummyMod', content['string_form'])

      def test_ipython_start_kernel_no_userns():

          # Issue #4188 - user_ns should be passed to shell as None, not {}

          cmd = ('from IPython import start_kernel\n'

                 'start_kernel()')

          with setup_kernel(cmd) as client:

              # user_module should not be an instance of DummyMod

              msg_id = client.execute("usermod = get_ipython().user_module")

              msg = client.get_shell_msg(block=True, timeout=TIMEOUT)

              content = msg['content']

              nt.assert_equal(content['status'], u'ok')

              msg_id = client.object_info('usermod')

              msg = client.get_shell_msg(block=True, timeout=TIMEOUT)

              content = msg['content']

              assert content['found']

              nt.assert_not_in('DummyMod', content['string_form'])

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

				import nose.tools as nt

				from .test_embed_kernel import setup, teardown, setup_kernel

				TIMEOUT = 15

				def test_ipython_start_kernel_userns():
				cmd = ('from IPython import start_kernel\n'
				'ns = {"tre": 123}\n'
				'start_kernel(user_ns=ns)')

				with setup_kernel(cmd) as client:
				msg_id = client.object_info('tre')
				msg = client.get_shell_msg(block=True, timeout=TIMEOUT)
				content = msg['content']
				assert content['found']
				nt.assert_equal(content['string_form'], u'123')

				# user_module should be an instance of DummyMod
				msg_id = client.execute("usermod = get_ipython().user_module")
				msg = client.get_shell_msg(block=True, timeout=TIMEOUT)
				content = msg['content']
				nt.assert_equal(content['status'], u'ok')
				msg_id = client.object_info('usermod')
				msg = client.get_shell_msg(block=True, timeout=TIMEOUT)
				content = msg['content']
				assert content['found']
				nt.assert_in('DummyMod', content['string_form'])

				def test_ipython_start_kernel_no_userns():
				# Issue #4188 - user_ns should be passed to shell as None, not {}
				cmd = ('from IPython import start_kernel\n'
				'start_kernel()')

				with setup_kernel(cmd) as client:
				# user_module should not be an instance of DummyMod
				msg_id = client.execute("usermod = get_ipython().user_module")
				msg = client.get_shell_msg(block=True, timeout=TIMEOUT)
				content = msg['content']
				nt.assert_equal(content['status'], u'ok')
				msg_id = client.object_info('usermod')
				msg = client.get_shell_msg(block=True, timeout=TIMEOUT)
				content = msg['content']
				assert content['found']
				nt.assert_not_in('DummyMod', content['string_form'])