Show More
@@ -69,7 +69,7 def coalesce_streams(cell, resources, index): | |||
|
69 | 69 | last.text += output.text |
|
70 | 70 | else: |
|
71 | 71 | new_outputs.append(output) |
|
72 | last = output | |
|
72 | last = output | |
|
73 | 73 | |
|
74 | 74 | cell.outputs = new_outputs |
|
75 | 75 | return cell, resources |
@@ -14,6 +14,8 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 | |
@@ -35,4 +37,24 class TestCoalesceStreams(PreprocessorTestsBase): | |||
|
35 | 37 | self.assertEqual(outputs[1].output_type, "text") |
|
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