@@ -659,7 +659,7 @@ async def run_capacity_test( # pylint: disable=too-many-locals
659659 )
660660 compare_messages (msg , expected , WAIT_FOR_COMPONENT_DATA_SEC + 0.2 )
661661
662- batteries_in_pool = list (battery_pool .battery_ids )
662+ batteries_in_pool = list (battery_pool .component_ids )
663663 scenarios : list [Scenario [Sample [Energy ]]] = [
664664 Scenario (
665665 batteries_in_pool [0 ],
@@ -851,7 +851,7 @@ async def run_soc_test(setup_args: SetupArgs) -> None:
851851 )
852852 compare_messages (msg , expected , WAIT_FOR_COMPONENT_DATA_SEC + 0.2 )
853853
854- batteries_in_pool = list (battery_pool .battery_ids )
854+ batteries_in_pool = list (battery_pool .component_ids )
855855 scenarios : list [Scenario [Sample [Percentage ]]] = [
856856 Scenario (
857857 batteries_in_pool [0 ],
@@ -1007,7 +1007,7 @@ async def run_power_bounds_test( # pylint: disable=too-many-locals
10071007 )
10081008 compare_messages (msg , expected , WAIT_FOR_COMPONENT_DATA_SEC + 0.2 )
10091009
1010- batteries_in_pool = list (battery_pool .battery_ids )
1010+ batteries_in_pool = list (battery_pool .component_ids )
10111011 scenarios : list [Scenario [SystemBounds ]] = [
10121012 Scenario (
10131013 next (iter (bat_invs_map [batteries_in_pool [0 ]])),
@@ -1241,7 +1241,7 @@ async def run_temperature_test( # pylint: disable=too-many-locals
12411241 expected = Sample (now , value = Temperature .from_celsius (25.0 ))
12421242 compare_messages (msg , expected , WAIT_FOR_COMPONENT_DATA_SEC + 0.2 )
12431243
1244- batteries_in_pool = list (battery_pool .battery_ids )
1244+ batteries_in_pool = list (battery_pool .component_ids )
12451245 bat_0 , bat_1 = batteries_in_pool
12461246 scenarios : list [Scenario [Sample [Temperature ]]] = [
12471247 Scenario (
0 commit comments