Add a stop method to BatteryPool

Signed-off-by: Sahas Subramanian <[email protected]>

Author: shsms

src/frequenz/sdk/timeseries/battery_pool/_battery_pool.py

@@ -429,3 +429,7 @@ def _system_power_bounds(self) -> ReceiverFetcher[SystemBounds]:
             )
 
         return self._battery_pool._active_methods[method_name]
+
+    async def stop(self) -> None:
+        """Stop all tasks and channels owned by the BatteryPool."""
+        await self._battery_pool.stop()

tests/timeseries/_formula_engine/test_formula_composition.py

@@ -36,9 +36,7 @@ async def test_formula_composition(  # pylint: disable=too-many-locals
             stack.push_async_callback(logical_meter.stop)
 
             battery_pool = microgrid.battery_pool()
-            stack.push_async_callback(
-                battery_pool._battery_pool.stop  # pylint: disable=protected-access
-            )
+            stack.push_async_callback(battery_pool.stop)
 
             pv_pool = microgrid.pv_pool()
             stack.push_async_callback(pv_pool.stop)
@@ -116,9 +114,7 @@ async def test_formula_composition_missing_pv(self, mocker: MockerFixture) -> No
         count = 0
         async with mockgrid, AsyncExitStack() as stack:
             battery_pool = microgrid.battery_pool()
-            stack.push_async_callback(
-                battery_pool._battery_pool.stop  # pylint: disable=protected-access
-            )
+            stack.push_async_callback(battery_pool.stop)
 
             pv_pool = microgrid.pv_pool()
             stack.push_async_callback(pv_pool.stop)
@@ -160,9 +156,7 @@ async def test_formula_composition_missing_bat(self, mocker: MockerFixture) -> N
         count = 0
         async with mockgrid, AsyncExitStack() as stack:
             battery_pool = microgrid.battery_pool()
-            stack.push_async_callback(
-                battery_pool._battery_pool.stop  # pylint: disable=protected-access
-            )
+            stack.push_async_callback(battery_pool.stop)
 
             pv_pool = microgrid.pv_pool()
             stack.push_async_callback(pv_pool.stop)