diff --git a/wntr/tests/test_gis.py b/wntr/tests/test_gis.py index 31513f26..86b2c7c1 100644 --- a/wntr/tests/test_gis.py +++ b/wntr/tests/test_gis.py @@ -131,6 +131,9 @@ def test_wn_to_gis(self): assert set(['link_type', 'start_node_name', 'end_node_name', 'geometry']).issubset(self.gis_data.pipes.columns) assert set(['link_type', 'start_node_name', 'end_node_name', 'geometry']).issubset(self.gis_data.pumps.columns) #assert set(['link_type', 'start_node_name', 'end_node_name', 'geometry']).issubset(self.gis_data.valves.columns) # Net1 has no valves + + #check base_demand and demand_pattern attrivutes + assert set(['base_demand','demand_pattern']).issubset(self.gis_data.junctions.columns) def test_gis_to_wn(self):