upstream/ipython Files · IPython/kernel/tests/test_kernelmanager.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)

MinRK - - Load All Authors

File last commit:

r12416:8773f2f4


                r16510:633371e5

Download file

             test_kernelmanager.py
        
                    54 lines
            
             | 1.5 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / IPython / kernel / tests / test_kernelmanager.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      """Tests for the notebook kernel and session manager"""

      from subprocess import PIPE

      import time

      from unittest import TestCase

      from IPython.testing import decorators as dec

      from IPython.config.loader import Config

      from IPython.kernel import KernelManager

      class TestKernelManager(TestCase):

          def _get_tcp_km(self):

              c = Config()

              km = KernelManager(config=c)

              return km

          def _get_ipc_km(self):

              c = Config()

              c.KernelManager.transport = 'ipc'

              c.KernelManager.ip = 'test'

              km = KernelManager(config=c)

              return km

          def _run_lifecycle(self, km):

              km.start_kernel(stdout=PIPE, stderr=PIPE)

              self.assertTrue(km.is_alive())

              km.restart_kernel(now=True)

              self.assertTrue(km.is_alive())

              km.interrupt_kernel()

              self.assertTrue(isinstance(km, KernelManager))

              km.shutdown_kernel(now=True)

          def test_tcp_lifecycle(self):

              km = self._get_tcp_km()

              self._run_lifecycle(km)

          @dec.skip_win32

          def test_ipc_lifecycle(self):

              km = self._get_ipc_km()

              self._run_lifecycle(km)

          def test_get_connect_info(self):

              km = self._get_tcp_km()

              cinfo = km.get_connection_info()

              keys = sorted(cinfo.keys())

              expected = sorted([

                  'ip', 'transport',

                  'hb_port', 'shell_port', 'stdin_port', 'iopub_port', 'control_port',

                  'key', 'signature_scheme',

              ])

              self.assertEqual(keys, expected)

	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

				"""Tests for the notebook kernel and session manager"""

				from subprocess import PIPE
				import time
				from unittest import TestCase

				from IPython.testing import decorators as dec

				from IPython.config.loader import Config
				from IPython.kernel import KernelManager

				class TestKernelManager(TestCase):

				def _get_tcp_km(self):
				c = Config()
				km = KernelManager(config=c)
				return km

				def _get_ipc_km(self):
				c = Config()
				c.KernelManager.transport = 'ipc'
				c.KernelManager.ip = 'test'
				km = KernelManager(config=c)
				return km

				def _run_lifecycle(self, km):
				km.start_kernel(stdout=PIPE, stderr=PIPE)
				self.assertTrue(km.is_alive())
				km.restart_kernel(now=True)
				self.assertTrue(km.is_alive())
				km.interrupt_kernel()
				self.assertTrue(isinstance(km, KernelManager))
				km.shutdown_kernel(now=True)

				def test_tcp_lifecycle(self):
				km = self._get_tcp_km()
				self._run_lifecycle(km)

				@dec.skip_win32
				def test_ipc_lifecycle(self):
				km = self._get_ipc_km()
				self._run_lifecycle(km)

				def test_get_connect_info(self):
				km = self._get_tcp_km()
				cinfo = km.get_connection_info()
				keys = sorted(cinfo.keys())
				expected = sorted([
				'ip', 'transport',
				'hb_port', 'shell_port', 'stdin_port', 'iopub_port', 'control_port',
				'key', 'signature_scheme',
				])
				self.assertEqual(keys, expected)