upstream/ipython Files · IPython/nbformat/v3/tests/formattest.py

Backport PR : Catch interrupted poll() in terminal console...

Backport PR : Catch interrupted poll() in terminal console Alternative to my own PR - catch ZMQError in run_cell, and if it's caused by an interrupt, ignore it. catches the exception in the blocking kernel client API, which is more complex, especially if we want to handle the timeout nicely as proposed in the comments, but it's possibly also more convenient for other users of that API. Or perhaps not - I'm not sure what makes sense for other API consumers in this case. Fixes gh-8105

Bradley M. Froehle - - Load All Authors

File last commit:

r7874:4a6836ce


                r20883:0d9a347c

Download file

             formattest.py
        
                    63 lines
            
             | 1.5 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / IPython / nbformat / v3 / tests / formattest.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      # -*- coding: utf8 -*-

      import io

      import os

      import shutil

      import tempfile

      pjoin = os.path.join

      from ..nbbase import (

          NotebookNode,

          new_code_cell, new_text_cell, new_worksheet, new_notebook

      )

      from ..nbpy import reads, writes, read, write

      from .nbexamples import nb0, nb0_py

      def open_utf8(fname, mode):

          return io.open(fname, mode=mode, encoding='utf-8')

      class NBFormatTest:

          """Mixin for writing notebook format tests"""

          # override with appropriate values in subclasses

          nb0_ref = None

          ext = None

          mod = None

          def setUp(self):

              self.wd = tempfile.mkdtemp()

          def tearDown(self):

              shutil.rmtree(self.wd)

          def assertNBEquals(self, nba, nbb):

              self.assertEqual(nba, nbb)

          def test_writes(self):

              s = self.mod.writes(nb0)

              if self.nb0_ref:

                  self.assertEqual(s, self.nb0_ref)

          def test_reads(self):

              s = self.mod.writes(nb0)

              nb = self.mod.reads(s)

          def test_roundtrip(self):

              s = self.mod.writes(nb0)

              self.assertNBEquals(self.mod.reads(s),nb0)

          def test_write_file(self):

              with open_utf8(pjoin(self.wd, "nb0.%s" % self.ext), 'w') as f:

                  self.mod.write(nb0, f)

          def test_read_file(self):

              with open_utf8(pjoin(self.wd, "nb0.%s" % self.ext), 'w') as f:

                  self.mod.write(nb0, f)

              with open_utf8(pjoin(self.wd, "nb0.%s" % self.ext), 'r') as f:

                  nb = self.mod.read(f)

	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

				# -- coding: utf8 --
				import io
				import os
				import shutil
				import tempfile

				pjoin = os.path.join

				from ..nbbase import (
				NotebookNode,
				new_code_cell, new_text_cell, new_worksheet, new_notebook
				)

				from ..nbpy import reads, writes, read, write
				from .nbexamples import nb0, nb0_py


				def open_utf8(fname, mode):
				return io.open(fname, mode=mode, encoding='utf-8')

				class NBFormatTest:
				"""Mixin for writing notebook format tests"""

				# override with appropriate values in subclasses
				nb0_ref = None
				ext = None
				mod = None

				def setUp(self):
				self.wd = tempfile.mkdtemp()

				def tearDown(self):
				shutil.rmtree(self.wd)

				def assertNBEquals(self, nba, nbb):
				self.assertEqual(nba, nbb)

				def test_writes(self):
				s = self.mod.writes(nb0)
				if self.nb0_ref:
				self.assertEqual(s, self.nb0_ref)

				def test_reads(self):
				s = self.mod.writes(nb0)
				nb = self.mod.reads(s)

				def test_roundtrip(self):
				s = self.mod.writes(nb0)
				self.assertNBEquals(self.mod.reads(s),nb0)

				def test_write_file(self):
				with open_utf8(pjoin(self.wd, "nb0.%s" % self.ext), 'w') as f:
				self.mod.write(nb0, f)

				def test_read_file(self):
				with open_utf8(pjoin(self.wd, "nb0.%s" % self.ext), 'w') as f:
				self.mod.write(nb0, f)

				with open_utf8(pjoin(self.wd, "nb0.%s" % self.ext), 'r') as f:
				nb = self.mod.read(f)