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

Turn on MathJax Safe extension...

Turn on MathJax Safe extension Not having this on allows mathjax links in Markdown cells which can execute javascript that are not sanitized by caja, for example: $E \href{javascript:alert("Einstein says so!")}{=} mc^2$ See http://docs.mathjax.org/en/latest/safe-mode.html for more details.

Bradley M. Froehle - - Load All Authors

File last commit:

r7874:4a6836ce


                r21454:a6b56ebf

Download file

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

        MinRK
    
add NBFormatTestCase base class, to consolidate nbformat testing

              r6209
            
      # -*- 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

        MinRK
    
specify utf8 when calling io.open

              r6212
            
      def open_utf8(fname, mode):

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

        MinRK
    
NBFormatTest is now a mixin, rather than a base class

              r6476
            
      class NBFormatTest:

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

        MinRK
    
add NBFormatTestCase base class, to consolidate nbformat testing

              r6209
            
          # 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):

        Bradley M. Froehle
    
s/assertEquals/assertEqual/

              r7874
            
              self.assertEqual(nba, nbb)

        MinRK
    
add NBFormatTestCase base class, to consolidate nbformat testing

              r6209
            
          def test_writes(self):

              s = self.mod.writes(nb0)

              if self.nb0_ref:

        Bradley M. Froehle
    
s/assertEquals/assertEqual/

              r7874
            
                  self.assertEqual(s, self.nb0_ref)

        MinRK
    
add NBFormatTestCase base class, to consolidate nbformat testing

              r6209
            
          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):

        MinRK
    
specify utf8 when calling io.open

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

        MinRK
    
add NBFormatTestCase base class, to consolidate nbformat testing

              r6209
            
                  self.mod.write(nb0, f)

          def test_read_file(self):

        MinRK
    
specify utf8 when calling io.open

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

        MinRK
    
add NBFormatTestCase base class, to consolidate nbformat testing

              r6209
            
                  self.mod.write(nb0, f)

        MinRK
    
specify utf8 when calling io.open

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

        MinRK
    
add NBFormatTestCase base class, to consolidate nbformat testing

              r6209
            
                  nb = self.mod.read(f)

        MinRK
    
NBFormatTest is now a mixin, rather than a base class

              r6476

	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

MinRK add NBFormatTestCase base class, to consolidate nbformat testing	r6209	# -- 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


MinRK specify utf8 when calling io.open	r6212	def open_utf8(fname, mode):
		return io.open(fname, mode=mode, encoding='utf-8')

MinRK NBFormatTest is now a mixin, rather than a base class	r6476	class NBFormatTest:
		"""Mixin for writing notebook format tests"""
MinRK add NBFormatTestCase base class, to consolidate nbformat testing	r6209
		# 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):
Bradley M. Froehle s/assertEquals/assertEqual/	r7874	self.assertEqual(nba, nbb)
MinRK add NBFormatTestCase base class, to consolidate nbformat testing	r6209
		def test_writes(self):
		s = self.mod.writes(nb0)
		if self.nb0_ref:
Bradley M. Froehle s/assertEquals/assertEqual/	r7874	self.assertEqual(s, self.nb0_ref)
MinRK add NBFormatTestCase base class, to consolidate nbformat testing	r6209
		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):
MinRK specify utf8 when calling io.open	r6212	with open_utf8(pjoin(self.wd, "nb0.%s" % self.ext), 'w') as f:
MinRK add NBFormatTestCase base class, to consolidate nbformat testing	r6209	self.mod.write(nb0, f)

		def test_read_file(self):
MinRK specify utf8 when calling io.open	r6212	with open_utf8(pjoin(self.wd, "nb0.%s" % self.ext), 'w') as f:
MinRK add NBFormatTestCase base class, to consolidate nbformat testing	r6209	self.mod.write(nb0, f)

MinRK specify utf8 when calling io.open	r6212	with open_utf8(pjoin(self.wd, "nb0.%s" % self.ext), 'r') as f:
MinRK add NBFormatTestCase base class, to consolidate nbformat testing	r6209	nb = self.mod.read(f)


MinRK NBFormatTest is now a mixin, rather than a base class	r6476