import copy import nose.tools as nt from IPython.nbformat.current import validate from .. import convert from . import nbexamples from IPython.nbformat.v3.tests import nbexamples as v3examples from IPython.nbformat import v3, v4 def test_upgrade_notebook(): nb03 = copy.deepcopy(v3examples.nb0) validate(nb03) nb04 = convert.upgrade(nb03) validate(nb04) def test_downgrade_notebook(): nb04 = copy.deepcopy(nbexamples.nb0) validate(nb04) nb03 = convert.downgrade(nb04) validate(nb03) def test_upgrade_heading(): v3h = v3.new_heading_cell v4m = v4.new_markdown_cell for v3cell, expected in [ ( v3h(source='foo', level=1), v4m(source='# foo'), ), ( v3h(source='foo\nbar\nmulti-line\n', level=4), v4m(source='#### foo bar multi-line'), ), ]: upgraded = convert.upgrade_cell(v3cell) nt.assert_equal(upgraded, expected) def test_downgrade_heading(): v3h = v3.new_heading_cell v4m = v4.new_markdown_cell v3m = lambda source: v3.new_text_cell('markdown', source) for v4cell, expected in [ ( v4m(source='# foo'), v3h(source='foo', level=1), ), ( v4m(source='#foo'), v3h(source='foo', level=1), ), ( v4m(source='#\tfoo'), v3h(source='foo', level=1), ), ( v4m(source='# \t foo'), v3h(source='foo', level=1), ), ( v4m(source='# foo\nbar'), v3m(source='# foo\nbar'), ), ]: downgraded = convert.downgrade_cell(v4cell) nt.assert_equal(downgraded, expected)