manyi/venv/Lib/site-packages/nbformat/v4/tests/test_nbbase.py

# coding: utf-8
"""Tests for the Python API for composing notebook elements"""

from nbformat.validator import isvalid, validate, ValidationError
from ..nbbase import (
    NotebookNode, nbformat,
    new_code_cell, new_markdown_cell, new_notebook,
    new_output, new_raw_cell,
)

def test_empty_notebook():
    nb = new_notebook()
    assert nb.cells == []
    assert nb.metadata == NotebookNode()
    assert nb.nbformat == nbformat

def test_empty_markdown_cell():
    cell = new_markdown_cell()
    assert cell.cell_type == 'markdown'
    assert cell.source == ''

def test_markdown_cell():
    cell = new_markdown_cell(u'* Søme markdown')
    assert cell.source == u'* Søme markdown'

def test_empty_raw_cell():
    cell = new_raw_cell()
    assert cell.cell_type == u'raw'
    assert cell.source == ''

def test_raw_cell():
    cell = new_raw_cell('hi')
    assert cell.source == u'hi'

def test_empty_code_cell():
    cell = new_code_cell('hi')
    assert cell.cell_type == 'code'
    assert cell.source == u'hi'

def test_empty_display_data():
    output = new_output('display_data')
    assert output.output_type == 'display_data'

def test_empty_stream():
    output = new_output('stream')
    assert output.output_type == 'stream'
    assert output.name == 'stdout'
    assert output.text == ''

def test_empty_execute_result():
    output = new_output('execute_result', execution_count=1)
    assert output.output_type == 'execute_result'

mimebundle = {
    'text/plain': "some text",
    "application/json": {
        "key": "value"
    },
    "image/svg+xml": 'ABCDEF',
    "application/octet-stream": 'ABC-123',
    "application/vnd.foo+bar": "Some other stuff",
}

def test_display_data():
    output = new_output('display_data', mimebundle)
    for key, expected in mimebundle.items():
        assert output.data[key] == expected

def test_execute_result():
    output = new_output('execute_result', mimebundle, execution_count=10)
    assert output.execution_count == 10
    for key, expected in mimebundle.items():
        assert output.data[key] == expected

def test_error():
    o = new_output(output_type=u'error', ename=u'NameError',
        evalue=u'Name not found', traceback=[u'frame 0', u'frame 1', u'frame 2']
    )
    assert o.output_type == u'error'
    assert o.ename == u'NameError'
    assert o.evalue == u'Name not found'
    assert o.traceback == [u'frame 0', u'frame 1', u'frame 2']

def test_code_cell_with_outputs():
    cell = new_code_cell(execution_count=10, outputs=[
        new_output('display_data', mimebundle),
        new_output('stream', text='hello'),
        new_output('execute_result', mimebundle, execution_count=10),
    ])
    assert cell.execution_count == 10
    assert len(cell.outputs) == 3
    er = cell.outputs[-1]
    assert er.execution_count == 10
    assert er['output_type'] == 'execute_result'

def test_stream():
    output = new_output('stream', name='stderr', text='hello there')
    assert output.name == 'stderr'
    assert output.text == 'hello there'