Show More
@@ -36,11 +36,14 b' class PreprocessorTestsBase(TestsBase):' | |||||
36 | nbformat.new_output(output_type="stream", stream="stdout", output_text="d"), |
|
36 | nbformat.new_output(output_type="stream", stream="stdout", output_text="d"), | |
37 | nbformat.new_output(output_type="stream", stream="stderr", output_text="e"), |
|
37 | nbformat.new_output(output_type="stream", stream="stderr", output_text="e"), | |
38 | nbformat.new_output(output_type="stream", stream="stderr", output_text="f"), |
|
38 | nbformat.new_output(output_type="stream", stream="stderr", output_text="f"), | |
39 | nbformat.new_output(output_type="png", output_png='Zw==')] #g |
|
39 | nbformat.new_output(output_type="png", output_png='Zw==')] # g | |
|
40 | out = nbformat.new_output(output_type="application/pdf") | |||
|
41 | out['application/pdf'] = 'aA==' # h | |||
|
42 | outputs.append(out) | |||
40 |
|
43 | |||
41 | cells=[nbformat.new_code_cell(input="$ e $", prompt_number=1,outputs=outputs), |
|
44 | cells=[nbformat.new_code_cell(input="$ e $", prompt_number=1,outputs=outputs), | |
42 | nbformat.new_text_cell('markdown', source="$ e $")] |
|
45 | nbformat.new_text_cell('markdown', source="$ e $")] | |
43 |
worksheets = [nbformat.new_worksheet( |
|
46 | worksheets = [nbformat.new_worksheet(cells=cells)] | |
44 |
|
47 | |||
45 | return nbformat.new_notebook(name="notebook1", worksheets=worksheets) |
|
48 | return nbformat.new_notebook(name="notebook1", worksheets=worksheets) | |
46 |
|
49 |
@@ -29,7 +29,7 b' class TestExtractOutput(PreprocessorTestsBase):' | |||||
29 | def build_preprocessor(self): |
|
29 | def build_preprocessor(self): | |
30 | """Make an instance of a preprocessor""" |
|
30 | """Make an instance of a preprocessor""" | |
31 | preprocessor = ExtractOutputPreprocessor() |
|
31 | preprocessor = ExtractOutputPreprocessor() | |
32 | preprocessor.extract_output_types = {'text', 'png'} |
|
32 | preprocessor.extract_output_types = {'text', 'png', 'application/pdf'} | |
33 | preprocessor.enabled = True |
|
33 | preprocessor.enabled = True | |
34 | return preprocessor |
|
34 | return preprocessor | |
35 |
|
35 | |||
@@ -55,6 +55,11 b' class TestExtractOutput(PreprocessorTestsBase):' | |||||
55 | output = nb.worksheets[0].cells[0].outputs[6] |
|
55 | output = nb.worksheets[0].cells[0].outputs[6] | |
56 | assert 'png_filename' in output |
|
56 | assert 'png_filename' in output | |
57 | png_filename = output['png_filename'] |
|
57 | png_filename = output['png_filename'] | |
|
58 | ||||
|
59 | # Check that pdf was extracted | |||
|
60 | output = nb.worksheets[0].cells[0].outputs[7] | |||
|
61 | assert 'application/pdf_filename' in output | |||
|
62 | pdf_filename = output['application/pdf_filename'] | |||
58 |
|
63 | |||
59 | # Verify text output |
|
64 | # Verify text output | |
60 | assert text_filename in res['outputs'] |
|
65 | assert text_filename in res['outputs'] | |
@@ -63,3 +68,7 b' class TestExtractOutput(PreprocessorTestsBase):' | |||||
63 | # Verify png output |
|
68 | # Verify png output | |
64 | assert png_filename in res['outputs'] |
|
69 | assert png_filename in res['outputs'] | |
65 | self.assertEqual(res['outputs'][png_filename], b'g') |
|
70 | self.assertEqual(res['outputs'][png_filename], b'g') | |
|
71 | ||||
|
72 | # Verify pdf output | |||
|
73 | assert pdf_filename in res['outputs'] | |||
|
74 | self.assertEqual(res['outputs'][pdf_filename], b'h') |
General Comments 0
You need to be logged in to leave comments.
Login now