103 lines
3.3 KiB
Python
103 lines
3.3 KiB
Python
|
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"
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|