base.py
52 lines
| 2.2 KiB
| text/x-python
|
PythonLexer
Jonathan Frederic
|
r12023 | """ | |
Module with utility functions for transformer tests | |||
""" | |||
#----------------------------------------------------------------------------- | |||
# Copyright (c) 2013, the IPython Development Team. | |||
# | |||
# Distributed under the terms of the Modified BSD License. | |||
# | |||
# The full license is in the file COPYING.txt, distributed with this software. | |||
#----------------------------------------------------------------------------- | |||
#----------------------------------------------------------------------------- | |||
# Imports | |||
#----------------------------------------------------------------------------- | |||
from IPython.nbformat import current as nbformat | |||
Jonathan Frederic
|
r12031 | from ...tests.base import TestsBase | |
from ...exporters.exporter import ResourcesDict | |||
Jonathan Frederic
|
r12023 | #----------------------------------------------------------------------------- | |
# Class | |||
#----------------------------------------------------------------------------- | |||
class TransformerTestsBase(TestsBase): | |||
"""Contains test functions transformer tests""" | |||
def build_notebook(self): | |||
"""Build a notebook in memory for use with transformer tests""" | |||
outputs = [nbformat.new_output(output_type="stream", stream="stdout", output_text="a"), | |||
nbformat.new_output(output_type="text", output_text="b"), | |||
nbformat.new_output(output_type="stream", stream="stdout", output_text="c"), | |||
nbformat.new_output(output_type="stream", stream="stdout", output_text="d"), | |||
nbformat.new_output(output_type="stream", stream="stderr", output_text="e"), | |||
Jonathan Frederic
|
r12027 | nbformat.new_output(output_type="stream", stream="stderr", output_text="f"), | |
Jonathan Frederic
|
r12037 | nbformat.new_output(output_type="png", output_png=b'Zw==')] #g | |
Jonathan Frederic
|
r12023 | ||
Jonathan Frederic
|
r12030 | cells=[nbformat.new_code_cell(input="$ e $", prompt_number=1,outputs=outputs), | |
nbformat.new_text_cell('markdown', source="$ e $")] | |||
Jonathan Frederic
|
r12023 | worksheets = [nbformat.new_worksheet(name="worksheet1", cells=cells)] | |
return nbformat.new_notebook(name="notebook1", worksheets=worksheets) | |||
Jonathan Frederic
|
r12031 | ||
def build_resources(self): | |||
"""Build an empty resources dictionary.""" | |||
res = ResourcesDict() | |||
res['metadata'] = ResourcesDict() | |||
return res |