Show More
@@ -14,6 +14,8 b' Module with tests for the coalescestreams preprocessor' | |||
|
14 | 14 | # Imports |
|
15 | 15 | #----------------------------------------------------------------------------- |
|
16 | 16 | |
|
17 | from IPython.nbformat import current as nbformat | |
|
18 | ||
|
17 | 19 | from .base import PreprocessorTestsBase |
|
18 | 20 | from ..coalescestreams import coalesce_streams |
|
19 | 21 | |
@@ -36,3 +38,23 b' class TestCoalesceStreams(PreprocessorTestsBase):' | |||
|
36 | 38 | self.assertEqual(outputs[2].text, "cd") |
|
37 | 39 | self.assertEqual(outputs[3].text, "ef") |
|
38 | 40 | |
|
41 | ||
|
42 | def test_coalesce_sequenced_streams(self): | |
|
43 | """Can the coalesce streams preprocessor merge a sequence of streams?""" | |
|
44 | ||
|
45 | outputs = [nbformat.new_output(output_type="stream", stream="stdout", output_text="0"), | |
|
46 | nbformat.new_output(output_type="stream", stream="stdout", output_text="1"), | |
|
47 | nbformat.new_output(output_type="stream", stream="stdout", output_text="2"), | |
|
48 | nbformat.new_output(output_type="stream", stream="stdout", output_text="3"), | |
|
49 | nbformat.new_output(output_type="stream", stream="stdout", output_text="4"), | |
|
50 | nbformat.new_output(output_type="stream", stream="stdout", output_text="5"), | |
|
51 | nbformat.new_output(output_type="stream", stream="stdout", output_text="6"), | |
|
52 | nbformat.new_output(output_type="stream", stream="stdout", output_text="7")] | |
|
53 | cells=[nbformat.new_code_cell(input="# None", prompt_number=1,outputs=outputs)] | |
|
54 | worksheets = [nbformat.new_worksheet(name="worksheet1", cells=cells)] | |
|
55 | ||
|
56 | nb = nbformat.new_notebook(name="notebook1", worksheets=worksheets) | |
|
57 | res = self.build_resources() | |
|
58 | nb, res = coalesce_streams(nb, res) | |
|
59 | outputs = nb.worksheets[0].cells[0].outputs | |
|
60 | self.assertEqual(outputs[0].text, u'01234567') |
General Comments 0
You need to be logged in to leave comments.
Login now