Uploaded Test files

venv/Lib/site-packages/nbformat/v3/tests/test_json.py

@@ -0,0 +1,102 @@
+import copy
+import json
+from unittest import TestCase
+
+from ..._compat import decodebytes
+from ..nbjson import reads, writes
+from ..nbbase import from_dict
+from .. import nbjson
+from .nbexamples import nb0
+
+from . import formattest
+
+from .nbexamples import nb0
+
+
+class TestJSON(formattest.NBFormatTest, TestCase):
+
+    nb0_ref = None
+    ext = 'ipynb'
+    mod = nbjson
+
+    def test_roundtrip_nosplit(self):
+        """Ensure that multiline blobs are still readable"""
+        # ensures that notebooks written prior to splitlines change
+        # are still readable.
+        s = writes(nb0, split_lines=False)
+        self.assertEqual(nbjson.reads(s),nb0)
+
+    def test_roundtrip_split(self):
+        """Ensure that splitting multiline blocks is safe"""
+        # This won't differ from test_roundtrip unless the default changes
+        s = writes(nb0, split_lines=True)
+        self.assertEqual(nbjson.reads(s),nb0)
+
+    def test_strip_transient(self):
+        """transient values aren't written to files"""
+        nb = copy.deepcopy(nb0)
+        nb.orig_nbformat = 2
+        nb.orig_nbformat_minor = 3
+        nb.worksheets[0].cells[0].metadata.trusted = False
+        nbs = nbjson.writes(nb)
+
+        nb2 = from_dict(json.loads(nbs))
+        self.assertNotIn('orig_nbformat', nb2)
+        self.assertNotIn('orig_nbformat_minor', nb2)
+        for cell in nb2.worksheets[0].cells:
+            self.assertNotIn('trusted', cell.metadata)
+
+    def test_to_json(self):
+        """to_notebook_json doesn't strip transient"""
+        nb = copy.deepcopy(nb0)
+        nb.orig_nbformat = 2
+        nb.orig_nbformat_minor = 3
+        nb.worksheets[0].cells[0].metadata.trusted = False
+        nbs = json.dumps(nb)
+        nb2 = nbjson.to_notebook(json.loads(nbs))
+
+        nb2 = from_dict(json.loads(nbs))
+        self.assertIn('orig_nbformat', nb2)
+        self.assertIn('orig_nbformat_minor', nb2)
+        cell = nb2.worksheets[0].cells[0]
+        self.assertIn('trusted', cell.metadata)
+
+    def test_read_png(self):
+        """PNG output data is b64 unicode"""
+        s = writes(nb0)
+        nb1 = nbjson.reads(s)
+        found_png = False
+        for cell in nb1.worksheets[0].cells:
+            if not 'outputs' in cell:
+                continue
+            for output in cell.outputs:
+                if 'png' in output:
+                    found_png = True
+                    pngdata = output['png']
+                    self.assertEqual(type(pngdata), str)
+                    # test that it is valid b64 data
+                    b64bytes = pngdata.encode('ascii')
+                    raw_bytes = decodebytes(b64bytes)
+        assert found_png, "never found png output"
+
+    def test_read_jpeg(self):
+        """JPEG output data is b64 unicode"""
+        s = writes(nb0)
+        nb1 = nbjson.reads(s)
+        found_jpeg = False
+        for cell in nb1.worksheets[0].cells:
+            if not 'outputs' in cell:
+                continue
+            for output in cell.outputs:
+                if 'jpeg' in output:
+                    found_jpeg = True
+                    jpegdata = output['jpeg']
+                    self.assertEqual(type(jpegdata), str)
+                    # test that it is valid b64 data
+                    b64bytes = jpegdata.encode('ascii')
+                    raw_bytes = decodebytes(b64bytes)
+        assert found_jpeg, "never found jpeg output"
+
+
+
+