##// END OF EJS Templates
add pdf to extract output tests
MinRK -
Show More
@@ -36,11 +36,14 b' class PreprocessorTestsBase(TestsBase):'
36 36 nbformat.new_output(output_type="stream", stream="stdout", output_text="d"),
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 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 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(name="worksheet1", cells=cells)]
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
@@ -55,6 +55,11 b' class TestExtractOutput(PreprocessorTestsBase):'
55 55 output = nb.worksheets[0].cells[0].outputs[6]
56 56 assert 'png_filename' in output
57 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 64 # Verify text output
60 65 assert text_filename in res['outputs']
@@ -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