Call fancy_unflatten with required arguments

Bjwebb · Bjwebb · commit 22690deb6189 · 2017-08-16T14:48:09.000+01:00
diff --git a/flattentool/tests/test_end_to_end.py b/flattentool/tests/test_end_to_end.py
@@ -1011,7 +1011,7 @@ def run(sheets, schema=None, source_maps=False):
         )
     spreadsheet_input.read_sheets()
     if source_maps:
-        result, cell_source_map_data, heading_source_map_data = spreadsheet_input.fancy_unflatten()
+        result, cell_source_map_data, heading_source_map_data = spreadsheet_input.fancy_unflatten(with_cell_source_map=True, with_heading_source_map=True)
         return result, cell_source_map_data, heading_source_map_data
     else:
         return spreadsheet_input.unflatten(), None, None

Original file line number	Diff line number	Diff line change
`@@ -1011,7 +1011,7 @@ def run(sheets, schema=None, source_maps=False):`
`1011`	`1011`	`)`
`1012`	`1012`	`spreadsheet_input.read_sheets()`
`1013`	`1013`	`if source_maps:`
`1014`		`- result, cell_source_map_data, heading_source_map_data = spreadsheet_input.fancy_unflatten()`
	`1014`	`+ result, cell_source_map_data, heading_source_map_data = spreadsheet_input.fancy_unflatten(with_cell_source_map=True, with_heading_source_map=True)`
`1015`	`1015`	`return result, cell_source_map_data, heading_source_map_data`
`1016`	`1016`	`else:`
`1017`	`1017`	`return spreadsheet_input.unflatten(), None, None`