Add test

USEPA · Sep 26, 2024 · 7c62655 · 7c62655
1 parent b3ec22c
commit 7c62655
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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):