@@ -227,16 +227,16 @@ def test_min_operating_cost_load_following_battery_dispatch(
227227
228228 # Test that discharge is always positive
229229 with subtests .test ("Discharge is always positive" ):
230- assert np .all (prob .get_val ("battery.battery_electricity_discharge " ) >= 0 )
230+ assert np .all (prob .get_val ("battery.storage_electricity_discharge " ) >= 0 )
231231 with subtests .test ("Charge is always negative" ):
232- assert np .all (prob .get_val ("battery.battery_electricity_charge " ) <= 0 )
232+ assert np .all (prob .get_val ("battery.storage_electricity_charge " ) <= 0 )
233233 # Set rtol lower b/c the values are in kW
234234 with subtests .test ("Charge + Discharge == battery_electricity_out" ):
235- charge_plus_discharge = prob .get_val ("battery.battery_electricity_charge " ) + prob .get_val (
236- "battery.battery_electricity_discharge "
235+ charge_plus_discharge = prob .get_val ("battery.storage_electricity_charge " ) + prob .get_val (
236+ "battery.storage_electricity_discharge "
237237 )
238238 np .testing .assert_allclose (
239- charge_plus_discharge , prob .get_val ("battery_electricity_out " ), rtol = 1e-2
239+ charge_plus_discharge , prob .get_val ("storage_electricity_out " ), rtol = 1e-2
240240 )
241241 with subtests .test ("Initial SOC is correct" ):
242242 assert pytest .approx (prob .model .get_val ("battery.SOC" )[0 ], rel = 1e-2 ) == 50
@@ -255,25 +255,25 @@ def test_min_operating_cost_load_following_battery_dispatch(
255255
256256 with subtests .test ("SOC increases when charging" ):
257257 assert np .all (
258- prob .get_val ("battery.battery_electricity_charge " , units = "kW" )[indx_soc_increase ] <= 0
258+ prob .get_val ("battery.storage_electricity_charge " , units = "kW" )[indx_soc_increase ] <= 0
259259 )
260260 assert np .all (
261- prob .get_val ("battery.battery_electricity_charge " , units = "kW" )[indx_soc_decrease ] == 0
261+ prob .get_val ("battery.storage_electricity_charge " , units = "kW" )[indx_soc_decrease ] == 0
262262 )
263263 assert np .all (
264- prob .get_val ("battery.battery_electricity_charge " , units = "kW" )[indx_soc_same ] == 0
264+ prob .get_val ("battery.storage_electricity_charge " , units = "kW" )[indx_soc_same ] == 0
265265 )
266266
267267 with subtests .test ("SOC decreases when discharging" ):
268268 assert np .all (
269- prob .get_val ("battery.battery_electricity_discharge " , units = "kW" )[indx_soc_decrease ] > 0
269+ prob .get_val ("battery.storage_electricity_discharge " , units = "kW" )[indx_soc_decrease ] > 0
270270 )
271271 assert np .all (
272- prob .get_val ("battery.battery_electricity_discharge " , units = "kW" )[indx_soc_increase ]
272+ prob .get_val ("battery.storage_electricity_discharge " , units = "kW" )[indx_soc_increase ]
273273 == 0
274274 )
275275 assert np .all (
276- prob .get_val ("battery.battery_electricity_discharge " , units = "kW" )[indx_soc_same ] == 0
276+ prob .get_val ("battery.storage_electricity_discharge " , units = "kW" )[indx_soc_same ] == 0
277277 )
278278
279279 with subtests .test ("Max SOC <= Max storage percent" ):
@@ -284,28 +284,28 @@ def test_min_operating_cost_load_following_battery_dispatch(
284284
285285 with subtests .test ("Charge never exceeds charge rate" ):
286286 assert (
287- prob .get_val ("battery.battery_electricity_charge " , units = "kW" ).min () >= - 1 * charge_rate
287+ prob .get_val ("battery.storage_electricity_charge " , units = "kW" ).min () >= - 1 * charge_rate
288288 )
289289
290290 with subtests .test ("Discharge never exceeds discharge rate" ):
291291 assert (
292- prob .get_val ("battery.battery_electricity_discharge " , units = "kW" ).max ()
292+ prob .get_val ("battery.storage_electricity_discharge " , units = "kW" ).max ()
293293 <= discharge_rate
294294 )
295295
296296 with subtests .test ("Discharge never exceeds demand" ):
297297 assert np .all (
298- prob .get_val ("battery.battery_electricity_discharge " , units = "kW" ).max () <= demand_in
298+ prob .get_val ("battery.storage_electricity_discharge " , units = "kW" ).max () <= demand_in
299299 )
300300
301301 with subtests .test ("Sometimes discharges" ):
302302 assert any (
303- k > 1e-3 for k in prob .get_val ("battery.battery_electricity_discharge " , units = "kW" )
303+ k > 1e-3 for k in prob .get_val ("battery.storage_electricity_discharge " , units = "kW" )
304304 )
305305
306306 with subtests .test ("Sometimes charges" ):
307307 assert any (
308- k < - 1e-3 for k in prob .get_val ("battery.battery_electricity_charge " , units = "kW" )
308+ k < - 1e-3 for k in prob .get_val ("battery.storage_electricity_charge " , units = "kW" )
309309 )
310310
311311 with subtests .test ("Cumulative charge/discharge does not exceed storage capacity" ):
@@ -315,15 +315,15 @@ def test_min_operating_cost_load_following_battery_dispatch(
315315 with subtests .test ("Expected discharge from hour 10-30" ):
316316 expected_discharge = np .concat ([np .zeros (8 ), np .ones (8 ) * 5000 , np .zeros (4 )])
317317 np .testing .assert_allclose (
318- prob .get_val ("battery.battery_electricity_discharge " , units = "kW" )[0 :20 ],
318+ prob .get_val ("battery.storage_electricity_discharge " , units = "kW" )[0 :20 ],
319319 expected_discharge ,
320320 rtol = 1e-2 ,
321321 )
322322
323323 with subtests .test ("Expected charge hour 0-24" ):
324324 expected_charge = - 1 * np .concat ([np .zeros (16 ), np .ones (8 ) * 4000 ])
325325 np .testing .assert_allclose (
326- prob .get_val ("battery.battery_electricity_charge " , units = "kW" )[0 :24 ],
326+ prob .get_val ("battery.storage_electricity_charge " , units = "kW" )[0 :24 ],
327327 expected_charge ,
328328 rtol = 1e-2 ,
329329 )
0 commit comments