test_coalescestreams.py
37 lines
| 1.4 KiB
| text/x-python
|
PythonLexer
Jonathan Frederic
|
r12021 | """ | |
Module with tests for the coalescestreams 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 | |||
#----------------------------------------------------------------------------- | |||
Jonathan Frederic
|
r12023 | from .base import TransformerTestsBase | |
Jonathan Frederic
|
r12021 | from ..coalescestreams import coalesce_streams | |
#----------------------------------------------------------------------------- | |||
# Class | |||
#----------------------------------------------------------------------------- | |||
Jonathan Frederic
|
r12023 | class TestCoalesceStreams(TransformerTestsBase): | |
Jonathan Frederic
|
r12021 | """Contains test functions for coalescestreams.py""" | |
def test_coalesce_streams(self): | |||
Jonathan Frederic
|
r12023 | """coalesce_streams transformer output test""" | |
Jonathan Frederic
|
r12039 | nb = self.build_notebook() | |
res = self.build_resources() | |||
nb, res = coalesce_streams(nb, res) | |||
Jonathan Frederic
|
r12035 | outputs = nb.worksheets[0].cells[0].outputs | |
self.assertEqual(outputs[0].text, "a") | |||
self.assertEqual(outputs[1].output_type, "text") | |||
self.assertEqual(outputs[2].text, "cd") | |||
self.assertEqual(outputs[3].text, "ef") | |||
Jonathan Frederic
|
r12021 |