diff --git a/src/arcade_collection/convert/convert_to_simularium_shapes.py b/src/arcade_collection/convert/convert_to_simularium_shapes.py index a0b3d31..9e101c1 100644 --- a/src/arcade_collection/convert/convert_to_simularium_shapes.py +++ b/src/arcade_collection/convert/convert_to_simularium_shapes.py @@ -109,7 +109,7 @@ def format_patch_for_shapes( graph_tar, series_key, frame, "GRAPH", field="graph" ) - for (from_node, to_node, edge) in graph_timepoint: + for from_node, to_node, edge in graph_timepoint: edge_type, radius, _, _, _, _, flow = edge name = f"VASCULATURE##{'UNDEFINED' if isnan(flow) else EDGE_TYPES[edge_type + 2]}" diff --git a/src/arcade_collection/output/parse_growth_file.py b/src/arcade_collection/output/parse_growth_file.py index 399c1dc..c958c4e 100644 --- a/src/arcade_collection/output/parse_growth_file.py +++ b/src/arcade_collection/output/parse_growth_file.py @@ -121,7 +121,7 @@ def parse_growth_timepoint(timepoint: dict, seed: int) -> list: parsed_data = [] time = timepoint["time"] - for (location, cells) in timepoint["cells"]: + for location, cells in timepoint["cells"]: u, v, w, z = location for cell in cells: diff --git a/tests/arcade_collection/output/test_convert_model_units.py b/tests/arcade_collection/output/test_convert_model_units.py index 465c59c..89f21fa 100644 --- a/tests/arcade_collection/output/test_convert_model_units.py +++ b/tests/arcade_collection/output/test_convert_model_units.py @@ -51,3 +51,7 @@ def test_estimate_spatial_resolution_invalid_spatiall_key(self): self.assertEqual(1, estimate_spatial_resolution(f"{ds_key}_B_C")) self.assertEqual(1, estimate_spatial_resolution(f"A_{ds_key}_C")) self.assertEqual(1, estimate_spatial_resolution(f"A_B_{ds_key}")) + + +if __name__ == "__main__": + unittest.main() diff --git a/tests/arcade_collection/output/test_parse_growth_file.py b/tests/arcade_collection/output/test_parse_growth_file.py index 6706e76..f6850ea 100644 --- a/tests/arcade_collection/output/test_parse_growth_file.py +++ b/tests/arcade_collection/output/test_parse_growth_file.py @@ -123,3 +123,7 @@ def test_parse_growth_timepoint(self): expected_df = pd.DataFrame(expected_dict) self.assertTrue(expected_df.equals(returned_df)) + + +if __name__ == "__main__": + unittest.main()