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

DEV: Separate FileCheckpointManager and GenericFileCheckpointManager....

DEV: Separate FileCheckpointManager and GenericFileCheckpointManager. - Adds a `GenericCheckpointMixin` as a helper for implementing the two boundary-traversing Checkpoint API methods, `create_checkpoint` and `restore_checkpoint`. - `GenericFileCheckpointManager` is implemented as a subclass of `FileCheckpointManager` using `GenericCheckpointMixin`. Note that this is the safe subtyping relationship because of method signature *contra*variance: `FileCheckpointManager` accepts `FileContentsManager` in its method signatures type, whereas `GenericFileCheckpointManager` accepts any `ContentsManager`. - Moved Checkpoint-related classes to their own files.

MinRK - - Load All Authors

File last commit:

r18573:efb0afba


                r19838:4cce9bcf

Download file

             formattest.py
        
                    54 lines
            
             | 1.3 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / IPython / nbformat / v4 / 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 .nbexamples import nb0

      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 .nbexamples import nb0


				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)