@@ -102,13 +102,13 @@ def test_time_control_open_vs_closed(self):
102102 link_res ["flowrate" ].at [t , "pipe2" ], 150 / 3600.0
103103 )
104104 self .assertEqual (
105- link_res ["status" ].at [t , "pipe2" ], self .wntr .network .LinkStatus .open
105+ link_res ["status" ].at [t , "pipe2" ], self .wntr .network .LinkStatus .Open
106106 )
107107 else :
108108 self .assertAlmostEqual (link_res ["flowrate" ].at [t , "pipe2" ], 0.0 )
109109 self .assertEqual (
110110 link_res ["status" ].at [t , "pipe2" ],
111- self .wntr .network .LinkStatus .closed ,
111+ self .wntr .network .LinkStatus .Closed ,
112112 )
113113
114114
@@ -147,14 +147,14 @@ def test_close_link_by_tank_level(self):
147147 self .assertAlmostEqual (link_res ["flowrate" ].at [t , "pump1" ], 0.0 )
148148 self .assertEqual (
149149 link_res ["status" ].at [t , "pump1" ],
150- self .wntr .network .LinkStatus .closed ,
150+ self .wntr .network .LinkStatus .Closed ,
151151 )
152152 count += 1
153153 else :
154154 self .assertGreaterEqual (link_res ["flowrate" ].at [t , "pump1" ], 0.0001 )
155155 self .assertEqual (
156156 link_res ["status" ].loc [t , "pump1" ],
157- self .wntr .network .LinkStatus .open ,
157+ self .wntr .network .LinkStatus .Open ,
158158 )
159159 self .assertEqual (activated_flag , True )
160160 self .assertGreaterEqual (count , 2 )
@@ -229,20 +229,20 @@ def test_open_link_by_tank_level(self):
229229 self .assertGreaterEqual (results .link ["flowrate" ].at [t , "pipe1" ], 0.002 )
230230 self .assertEqual (
231231 results .link ["status" ].at [t , "pipe1" ],
232- self .wntr .network .LinkStatus .open ,
232+ self .wntr .network .LinkStatus .Open ,
233233 )
234234 count += 1
235235 else :
236236 self .assertAlmostEqual (results .link ["flowrate" ].at [t , "pipe1" ], 0.0 )
237237 self .assertEqual (
238238 results .link ["status" ].at [t , "pipe1" ],
239- self .wntr .network .LinkStatus .closed ,
239+ self .wntr .network .LinkStatus .Closed ,
240240 )
241241 self .assertEqual (activated_flag , True )
242242 self .assertGreaterEqual (count , 2 )
243243 self .assertEqual (
244244 results .link ["status" ].at [results .link ["status" ].index [0 ], "pipe1" ],
245- self .wntr .network .LinkStatus .closed ,
245+ self .wntr .network .LinkStatus .Closed ,
246246 ) # make sure the pipe starts closed
247247 self .assertLessEqual (
248248 results .node ["pressure" ].at [results .node ["pressure" ].index [0 ], "tank1" ],
@@ -277,7 +277,7 @@ def test_pipe_closed_for_low_level(self):
277277 self .assertLessEqual (results .link ["flowrate" ].at [t , "pipe1" ], 0.0 )
278278 self .assertEqual (
279279 results .link ["status" ].at [t , "pipe1" ],
280- self .wntr .network .LinkStatus .closed ,
280+ self .wntr .network .LinkStatus .Closed ,
281281 )
282282 tank_level_dropped_flag = True
283283 self .assertEqual (tank_level_dropped_flag , True )
@@ -384,7 +384,7 @@ def test_open_by_time_close_by_condition(self):
384384 inp_file = join (test_datadir , "control_comb.inp" )
385385 wn = self .wntr .network .WaterNetworkModel (inp_file )
386386 control_action = self .wntr .network .ControlAction (
387- wn .get_link ("pipe1" ), "status" , self .wntr .network .LinkStatus .opened
387+ wn .get_link ("pipe1" ), "status" , self .wntr .network .LinkStatus .Opened
388388 )
389389 control = self .wntr .network .controls .Control ._time_control (
390390 wn , 6 * 3600 , "SIM_TIME" , False , control_action
@@ -432,9 +432,9 @@ def test_close_by_condition_open_by_time_stay(self):
432432 tank1 .init_level = 40.0
433433 tank1 ._head = tank1 .elevation + 40.0
434434 pipe1 = wn .get_link ("pipe1" )
435- pipe1 ._user_status = self .wntr .network .LinkStatus .opened
435+ pipe1 ._user_status = self .wntr .network .LinkStatus .Opened
436436 control_action = self .wntr .network .ControlAction (
437- wn .get_link ("pipe1" ), "status" , self .wntr .network .LinkStatus .opened
437+ wn .get_link ("pipe1" ), "status" , self .wntr .network .LinkStatus .Opened
438438 )
439439 control = self .wntr .network .controls .Control ._time_control (
440440 wn , 19 * 3600 , "SIM_TIME" , False , control_action
@@ -482,9 +482,9 @@ def test_close_by_condition_open_by_time_reclose(self):
482482 tank1 .init_level = 40.0
483483 tank1 ._head = tank1 .elevation + 40.0
484484 pipe1 = wn .get_link ("pipe1" )
485- pipe1 ._user_status = self .wntr .network .LinkStatus .opened
485+ pipe1 ._user_status = self .wntr .network .LinkStatus .Opened
486486 control_action = self .wntr .network .ControlAction (
487- wn .get_link ("pipe1" ), "status" , self .wntr .network .LinkStatus .opened
487+ wn .get_link ("pipe1" ), "status" , self .wntr .network .LinkStatus .Opened
488488 )
489489 control = self .wntr .network .controls .Control ._time_control (
490490 wn , 5 * 3600 , "SIM_TIME" , False , control_action
0 commit comments