def test_wrap_code_cells_2():
# Test wrap_code_cells() with two code cells.
ast = ASTNode('root')
cb0 = ASTNode('CodeBlock', lang='python', children=['a'])
cb1 = ASTNode('CodeBlock', lang='python', children=['b'])
ast.add_child(cb0)
ast.add_child(cb1)
ast.show()
ast_wrapped = wrap_code_cells(ast)
ast_wrapped.show()
ast_expected = ASTNode('root')
# First code cell.
code_cell0 = ASTNode('CodeCell')
code_cell0.add_child(cb0)
ast_expected.add_child(code_cell0)
# Second code cell.
code_cell1 = ASTNode('CodeCell')
code_cell1.add_child(cb1)
ast_expected.add_child(code_cell1)
ast_expected.show()
assert_equal(ast_wrapped, ast_expected)
def test_notebook_writer_hello():
path = get_test_file_path('ast', 'hello.json')
ast = ASTPlugin().open(path)
nb = NotebookWriter().write(ast)
# Compare the notebooks.
nb_expected = open_notebook(get_test_file_path('notebook', 'hello.ipynb'))
# Ignore some fields when comparing the notebooks.
assert_equal(nb, nb_expected, ('metadata', 'kernelspec'))
def test_wrap_code_cells_1():
# Test wrap_code_cells() with a single code cell.
ast = ASTNode('root')
ast.add_child(ASTNode('CodeBlock', lang='python', children=['']))
ast_wrapped = wrap_code_cells(ast)
ast_wrapped.show()
ast_expected = ASTNode('root')
ast_expected.add_child(ASTNode('CodeCell', children=[ast.children[0]]))
assert_equal(ast_wrapped, ast_expected)
def test_notebook_reader_notebook():
# Open a test notebook with a code cell.
path = get_test_file_path('notebook', 'notebook.ipynb')
notebook = open_notebook(path)
# Convert it to an AST.
reader = NotebookReader()
ast = reader.read(notebook)
ast.show()
# Compare with the markdown version.
path = get_test_file_path('markdown', 'notebook.md')
markdown = load_text(path)
assert_equal(MarkdownPlugin().write(ast), markdown)
assert 'output_4_1.png' in reader.resources
def test_notebook_writer_notebook():
path = get_test_file_path('ast', 'notebook.json')
ast = ASTPlugin().open(path)
# Load the image.
fn = get_test_file_path('markdown', 'output_4_1.png')
with open(fn, 'rb') as f:
img = f.read()
resources = {op.basename(fn): img}
# Convert the AST to a notebook.
nb = NotebookWriter().write(ast, resources=resources)
# Compare the notebooks.
nb_expected = open_notebook(get_test_file_path('notebook',
'notebook.ipynb'))
# Ignore some fields when comparing the notebooks.
assert_equal(nb, nb_expected, ('metadata', 'kernelspec'))
def assert_equal(self, nb0, nb1):
return assert_equal(nb0, nb1,
to_remove=('metadata', 'kernel_spec'))
def assert_equal(self, ast0, ast1):
return assert_equal(ast0, ast1, to_remove=('_visit_meta',))
