test_extractoutput.py
62 lines
| 2.1 KiB
| text/x-python
|
PythonLexer
Jonathan Frederic
|
r12029 | """ | ||
Module with tests for the extractoutput transformer | ||||
""" | ||||
#----------------------------------------------------------------------------- | ||||
# 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 .base import TransformerTestsBase | ||||
from ..extractoutput import ExtractOutputTransformer | ||||
#----------------------------------------------------------------------------- | ||||
# Class | ||||
#----------------------------------------------------------------------------- | ||||
class TestExtractOutput(TransformerTestsBase): | ||||
"""Contains test functions for extractoutput.py""" | ||||
Jonathan Frederic
|
r12035 | |||
def build_transformer(self): | ||||
"""Make an instance of a transformer""" | ||||
Jonathan Frederic
|
r12029 | transformer = ExtractOutputTransformer() | ||
transformer.enabled = True | ||||
return transformer | ||||
Jonathan Frederic
|
r12035 | |||
def test_constructor(self): | ||||
"""Can a ExtractOutputTransformer be constructed?""" | ||||
self.build_transformer() | ||||
Jonathan Frederic
|
r12029 | |||
def test_output(self): | ||||
"""Test the output of the ExtractOutputTransformer""" | ||||
Jonathan Frederic
|
r12039 | nb = self.build_notebook() | ||
res = self.build_resources() | ||||
transformer = self.build_transformer() | ||||
nb, res = transformer(nb, res) | ||||
Jonathan Frederic
|
r12029 | |||
# Check if text was extracted. | ||||
assert 'text_filename' in nb.worksheets[0].cells[0].outputs[1] | ||||
text_filename = nb.worksheets[0].cells[0].outputs[1]['text_filename'] | ||||
# Check if png was extracted. | ||||
assert 'png_filename' in nb.worksheets[0].cells[0].outputs[6] | ||||
png_filename = nb.worksheets[0].cells[0].outputs[6]['png_filename'] | ||||
# Verify text output | ||||
assert text_filename in res['outputs'] | ||||
Jonathan Frederic
|
r12038 | self.assertEqual(res['outputs'][text_filename], b'b') | ||
Jonathan Frederic
|
r12029 | |||
# Verify png output | ||||
assert png_filename in res['outputs'] | ||||
Jonathan Frederic
|
r12038 | self.assertEqual(res['outputs'][png_filename], b'g') | ||