Show More
@@ -55,6 +55,10 b' class TestExecute(PreprocessorTestsBase):' | |||||
55 | normalized_actual_outputs = list(map(self.normalize_output, actual_outputs)) |
|
55 | normalized_actual_outputs = list(map(self.normalize_output, actual_outputs)) | |
56 | assert normalized_expected_outputs == normalized_actual_outputs |
|
56 | assert normalized_expected_outputs == normalized_actual_outputs | |
57 |
|
57 | |||
|
58 | expected_prompt_number = expected_cell.get('prompt_number', None) | |||
|
59 | actual_prompt_number = actual_cell.get('prompt_number', None) | |||
|
60 | assert expected_prompt_number == actual_prompt_number | |||
|
61 | ||||
58 |
|
62 | |||
59 | def build_preprocessor(self): |
|
63 | def build_preprocessor(self): | |
60 | """Make an instance of a preprocessor""" |
|
64 | """Make an instance of a preprocessor""" | |
@@ -77,6 +81,10 b' class TestExecute(PreprocessorTestsBase):' | |||||
77 | input_nb = nbformat.read(f, 'ipynb') |
|
81 | input_nb = nbformat.read(f, 'ipynb') | |
78 | res = self.build_resources() |
|
82 | res = self.build_resources() | |
79 | preprocessor = self.build_preprocessor() |
|
83 | preprocessor = self.build_preprocessor() | |
80 |
|
|
84 | cleaned_input_nb = copy.deepcopy(input_nb) | |
|
85 | for cell in cleaned_input_nb.worksheets[0].cells: | |||
|
86 | if 'prompt_number' in cell: | |||
|
87 | del cell['prompt_number'] | |||
|
88 | cell['outputs'] = [] | |||
|
89 | output_nb, _ = preprocessor(cleaned_input_nb, res) | |||
81 | self.assert_notebooks_equal(output_nb, input_nb) |
|
90 | self.assert_notebooks_equal(output_nb, input_nb) | |
82 |
|
General Comments 0
You need to be logged in to leave comments.
Login now