Update tests to get resampled streams for correct namespaces

... because we now have tests that compose formulas from different
namespaces, from the logical meter and the battery pool.

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

Author: shsms

tests/timeseries/_formula_engine/test_formula_composition.py

@@ -30,6 +30,7 @@ async def test_formula_composition(  # pylint: disable=too-many-locals
         logical_meter = microgrid.logical_meter()
         battery_pool = microgrid.battery_pool()
         main_meter_recv = get_resampled_stream(
+            logical_meter._namespace,  # pylint: disable=protected-access
             4,
             ComponentMetricId.ACTIVE_POWER,
         )

tests/timeseries/_formula_engine/utils.py

@@ -18,6 +18,7 @@
 
 
 def get_resampled_stream(  # pylint: disable=too-many-arguments
+    namespace: str,
     comp_id: int,
     metric_id: ComponentMetricId,
 ) -> Receiver[Sample]:
@@ -27,7 +28,7 @@ def get_resampled_stream(  # pylint: disable=too-many-arguments
 
     # pylint: disable=protected-access
     builder = ResampledFormulaBuilder(
-        _data_pipeline._get().logical_meter()._namespace,
+        namespace,
         "",
         _data_pipeline._get()._channel_registry,
         _data_pipeline._get()._resampling_request_sender(),

tests/timeseries/test_ev_charger_pool.py

@@ -93,6 +93,7 @@ async def test_ev_power(  # pylint: disable=too-many-locals
         ev_pool = microgrid.ev_charger_pool()
 
         main_meter_recv = get_resampled_stream(
+            logical_meter._namespace,  # pylint: disable=protected-access
             mockgrid.main_meter_id,
             ComponentMetricId.ACTIVE_POWER,
         )

tests/timeseries/test_logical_meter.py

@@ -34,6 +34,7 @@ async def test_grid_power_1(self, mocker: MockerFixture) -> None:
         grid_power_recv = logical_meter.grid_power.new_receiver()
 
         main_meter_recv = get_resampled_stream(
+            logical_meter._namespace,  # pylint: disable=protected-access
             mockgrid.main_meter_id,
             ComponentMetricId.ACTIVE_POWER,
         )
@@ -67,6 +68,7 @@ async def test_grid_power_2(
 
         meter_receivers = [
             get_resampled_stream(
+                logical_meter._namespace,  # pylint: disable=protected-access
                 meter_id,
                 ComponentMetricId.ACTIVE_POWER,
             )
@@ -111,6 +113,7 @@ async def test_battery_and_pv_power(  # pylint: disable=too-many-locals
 
         bat_inv_receivers = [
             get_resampled_stream(
+                battery_pool._namespace,  # pylint: disable=protected-access
                 meter_id,
                 ComponentMetricId.ACTIVE_POWER,
             )
@@ -119,6 +122,7 @@ async def test_battery_and_pv_power(  # pylint: disable=too-many-locals
 
         pv_inv_receivers = [
             get_resampled_stream(
+                logical_meter._namespace,  # pylint: disable=protected-access
                 meter_id,
                 ComponentMetricId.ACTIVE_POWER,
             )