Show More
@@ -37,10 +37,13 b' class PreprocessorTestsBase(TestsBase):' | |||
|
37 | 37 | nbformat.new_output(output_type="stream", stream="stderr", output_text="e"), |
|
38 | 38 | nbformat.new_output(output_type="stream", stream="stderr", output_text="f"), |
|
39 | 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 | 44 | cells=[nbformat.new_code_cell(input="$ e $", prompt_number=1,outputs=outputs), |
|
42 | 45 | nbformat.new_text_cell('markdown', source="$ e $")] |
|
43 |
worksheets = [nbformat.new_worksheet( |
|
|
46 | worksheets = [nbformat.new_worksheet(cells=cells)] | |
|
44 | 47 | |
|
45 | 48 | return nbformat.new_notebook(name="notebook1", worksheets=worksheets) |
|
46 | 49 |
@@ -29,7 +29,7 b' class TestExtractOutput(PreprocessorTestsBase):' | |||
|
29 | 29 | def build_preprocessor(self): |
|
30 | 30 | """Make an instance of a preprocessor""" |
|
31 | 31 | preprocessor = ExtractOutputPreprocessor() |
|
32 | preprocessor.extract_output_types = {'text', 'png'} | |
|
32 | preprocessor.extract_output_types = {'text', 'png', 'application/pdf'} | |
|
33 | 33 | preprocessor.enabled = True |
|
34 | 34 | return preprocessor |
|
35 | 35 | |
@@ -56,6 +56,11 b' class TestExtractOutput(PreprocessorTestsBase):' | |||
|
56 | 56 | assert 'png_filename' in output |
|
57 | 57 | png_filename = output['png_filename'] |
|
58 | 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'] | |
|
63 | ||
|
59 | 64 | # Verify text output |
|
60 | 65 | assert text_filename in res['outputs'] |
|
61 | 66 | self.assertEqual(res['outputs'][text_filename], b'b') |
@@ -63,3 +68,7 b' class TestExtractOutput(PreprocessorTestsBase):' | |||
|
63 | 68 | # Verify png output |
|
64 | 69 | assert png_filename in res['outputs'] |
|
65 | 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