upstream/ipython Files · IPython/kernel/tests/test_kernelmanager.py

DEV: Allow CheckpointManagers to optimize for shared backends....

DEV: Allow CheckpointManagers to optimize for shared backends. On `create_checkpoint` and `restore_checkpoint`, pass a path and a `ContentsManager` to `CheckpointManager` instead of an already-loaded model. The `CheckpointManager` base class provides a correct implementation of these methods that's generic across any ContentsManager, but subclasses are free to specialize when the storage backend of `ContentsManager` is shared.

MinRK - - Load All Authors

File last commit:

r12416:8773f2f4


                r19828:adba75fe

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)