@@ -150,6 +150,8 @@ async def test_power_distributor_one_user(self, mocker: MockerFixture) -> None:
150150 assert result .excess_power .isclose (Power .from_watts (200.0 ))
151151 assert result .request == request
152152
153+ await mockgrid .cleanup ()
154+
153155 async def test_power_distributor_exclusion_bounds (
154156 self , mocker : MockerFixture
155157 ) -> None :
@@ -242,6 +244,8 @@ async def test_power_distributor_exclusion_bounds(
242244 assert result .bounds == PowerBounds (- 1000 , - 600 , 600 , 1000 )
243245 assert result .request == request
244246
247+ await mockgrid .cleanup ()
248+
245249 # pylint: disable=too-many-locals
246250 async def test_two_batteries_one_inverters (self , mocker : MockerFixture ) -> None :
247251 """Test if power distribution works with two batteries connected to one inverter."""
@@ -312,6 +316,8 @@ async def test_two_batteries_one_inverters(self, mocker: MockerFixture) -> None:
312316 assert result .excess_power .isclose (Power .from_watts (700.0 ))
313317 assert result .request == request
314318
319+ await mockgrid .cleanup ()
320+
315321 async def test_two_batteries_one_broken_one_inverters (
316322 self , mocker : MockerFixture
317323 ) -> None :
@@ -393,6 +399,8 @@ async def test_two_batteries_one_broken_one_inverters(
393399 result .msg == "No data for at least one of the given batteries {9, 19}"
394400 )
395401
402+ await mockgrid .cleanup ()
403+
396404 async def test_battery_two_inverters (self , mocker : MockerFixture ) -> None :
397405 """Test if power distribution works with two inverters for one battery."""
398406 gen = GraphGenerator ()
@@ -464,6 +472,8 @@ async def test_battery_two_inverters(self, mocker: MockerFixture) -> None:
464472 assert result .excess_power .isclose (Power .from_watts (200.0 ))
465473 assert result .request == request
466474
475+ await mockgrid .cleanup ()
476+
467477 async def test_two_batteries_three_inverters (self , mocker : MockerFixture ) -> None :
468478 """Test if power distribution works with two batteries connected to three inverters."""
469479 gen = GraphGenerator ()
@@ -540,6 +550,8 @@ async def test_two_batteries_three_inverters(self, mocker: MockerFixture) -> Non
540550 assert result .excess_power .isclose (Power .from_watts (200.0 ))
541551 assert result .request == request
542552
553+ await mockgrid .cleanup ()
554+
543555 async def test_two_batteries_one_inverter_different_exclusion_bounds_2 (
544556 self , mocker : MockerFixture
545557 ) -> None :
@@ -625,6 +637,8 @@ async def test_two_batteries_one_inverter_different_exclusion_bounds_2(
625637 assert result .request == request
626638 assert result .bounds == PowerBounds (- 1000 , - 500 , 500 , 1000 )
627639
640+ await mockgrid .cleanup ()
641+
628642 async def test_two_batteries_one_inverter_different_exclusion_bounds (
629643 self , mocker : MockerFixture
630644 ) -> None :
@@ -712,6 +726,8 @@ async def test_two_batteries_one_inverter_different_exclusion_bounds(
712726 # each inverter is bounded at 500
713727 assert result .bounds == PowerBounds (- 500 , - 400 , 400 , 500 )
714728
729+ await mockgrid .cleanup ()
730+
715731 async def test_connected_but_not_requested_batteries (
716732 self , mocker : MockerFixture
717733 ) -> None :
@@ -785,6 +801,8 @@ async def test_connected_but_not_requested_batteries(
785801 )
786802 assert err_msg is not None
787803
804+ await mockgrid .cleanup ()
805+
788806 async def test_battery_soc_nan (self , mocker : MockerFixture ) -> None :
789807 """Test if battery with SoC==NaN is not used."""
790808 mockgrid = MockMicrogrid (grid_meter = False )
@@ -847,6 +865,8 @@ async def test_battery_soc_nan(self, mocker: MockerFixture) -> None:
847865 assert result .excess_power .isclose (Power .from_watts (700.0 ))
848866 assert result .request == request
849867
868+ await mockgrid .cleanup ()
869+
850870 async def test_battery_capacity_nan (self , mocker : MockerFixture ) -> None :
851871 """Test battery with capacity set to NaN is not used."""
852872 mockgrid = MockMicrogrid (grid_meter = False )
@@ -902,6 +922,8 @@ async def test_battery_capacity_nan(self, mocker: MockerFixture) -> None:
902922 assert result .excess_power .isclose (Power .from_watts (700.0 ))
903923 assert result .request == request
904924
925+ await mockgrid .cleanup ()
926+
905927 async def test_battery_power_bounds_nan (self , mocker : MockerFixture ) -> None :
906928 """Test battery with power bounds set to NaN is not used."""
907929 mockgrid = MockMicrogrid (grid_meter = False )
@@ -972,6 +994,8 @@ async def test_battery_power_bounds_nan(self, mocker: MockerFixture) -> None:
972994 assert result .excess_power .isclose (Power .from_watts (200.0 ))
973995 assert result .request == request
974996
997+ await mockgrid .cleanup ()
998+
975999 async def test_power_distributor_invalid_battery_id (
9761000 self , mocker : MockerFixture
9771001 ) -> None :
@@ -1017,6 +1041,8 @@ async def test_power_distributor_invalid_battery_id(
10171041 err_msg = re .search (r"No battery 100, available batteries:" , result .msg )
10181042 assert err_msg is not None
10191043
1044+ await mockgrid .cleanup ()
1045+
10201046 async def test_power_distributor_one_user_adjust_power_consume (
10211047 self , mocker : MockerFixture
10221048 ) -> None :
@@ -1067,6 +1093,8 @@ async def test_power_distributor_one_user_adjust_power_consume(
10671093 assert result .request == request
10681094 assert result .bounds .inclusion_upper == 1000
10691095
1096+ await mockgrid .cleanup ()
1097+
10701098 async def test_power_distributor_one_user_adjust_power_supply (
10711099 self , mocker : MockerFixture
10721100 ) -> None :
@@ -1117,6 +1145,8 @@ async def test_power_distributor_one_user_adjust_power_supply(
11171145 assert result .request == request
11181146 assert result .bounds .inclusion_lower == - 1000
11191147
1148+ await mockgrid .cleanup ()
1149+
11201150 async def test_power_distributor_one_user_adjust_power_success (
11211151 self , mocker : MockerFixture
11221152 ) -> None :
@@ -1167,6 +1197,8 @@ async def test_power_distributor_one_user_adjust_power_success(
11671197 assert result .excess_power .isclose (Power .zero (), abs_tol = 1e-9 )
11681198 assert result .request == request
11691199
1200+ await mockgrid .cleanup ()
1201+
11701202 async def test_not_all_batteries_are_working (self , mocker : MockerFixture ) -> None :
11711203 """Test if power distribution works if not all batteries are working."""
11721204 mockgrid = MockMicrogrid (grid_meter = False )
@@ -1215,3 +1247,5 @@ async def test_not_all_batteries_are_working(self, mocker: MockerFixture) -> Non
12151247 assert result .excess_power .isclose (Power .from_watts (700.0 ))
12161248 assert result .succeeded_power .isclose (Power .from_watts (500.0 ))
12171249 assert result .request == request
1250+
1251+ await mockgrid .cleanup ()
0 commit comments