base.py
53 lines
| 2.2 KiB
| text/x-python
|
PythonLexer
|
r12023 | """ | ||
|
r12219 | Module with utility functions for preprocessor tests | ||
|
r12023 | """ | ||
#----------------------------------------------------------------------------- | ||||
# 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 | ||||
|
r12031 | from ...tests.base import TestsBase | ||
from ...exporters.exporter import ResourcesDict | ||||
|
r12023 | #----------------------------------------------------------------------------- | ||
# Class | ||||
#----------------------------------------------------------------------------- | ||||
|
r12219 | class PreprocessorTestsBase(TestsBase): | ||
"""Contains test functions preprocessor tests""" | ||||
|
r12023 | |||
def build_notebook(self): | ||||
|
r12219 | """Build a notebook in memory for use with preprocessor tests""" | ||
|
r12023 | |||
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"), | ||||
|
r12027 | nbformat.new_output(output_type="stream", stream="stderr", output_text="f"), | ||
|
r12037 | nbformat.new_output(output_type="png", output_png=b'Zw==')] #g | ||
|
r12023 | |||
|
r12030 | cells=[nbformat.new_code_cell(input="$ e $", prompt_number=1,outputs=outputs), | ||
nbformat.new_text_cell('markdown', source="$ e $")] | ||||
|
r12023 | worksheets = [nbformat.new_worksheet(name="worksheet1", cells=cells)] | ||
return nbformat.new_notebook(name="notebook1", worksheets=worksheets) | ||||
|
r12031 | |||
def build_resources(self): | ||||
"""Build an empty resources dictionary.""" | ||||
res = ResourcesDict() | ||||
res['metadata'] = ResourcesDict() | ||||
|
r12219 | return res | ||