Send test data from secondary components

This won't be necessary as soon as the coalesce AST node is able to subscribe
to secondary streams only if primary streams are not available.

But for now, coalesce nodes need all their input streams to have data.

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

Author: shsms

tests/microgrid/test_grid.py

@@ -125,6 +125,7 @@ async def test_grid_power_2(mocker: MockerFixture) -> None:
         for count in range(10):
             await mockgrid.mock_resampler.send_meter_power([20.0 + count, 12.0, -13.0])
             await mockgrid.mock_resampler.send_bat_inverter_power([0.0, -5.0])
+            await mockgrid.mock_resampler.send_pv_inverter_power([0.0])
             meter_sum = 0.0
             for recv in component_receivers:
                 val = await recv.receive()
@@ -220,6 +221,7 @@ async def test_grid_reactive_power_2(mocker: MockerFixture) -> None:
             await mockgrid.mock_resampler.send_meter_reactive_power(
                 [20.0 + count, 12.0, -13.0]
             )
+            await mockgrid.mock_resampler.send_pv_inverter_reactive_power([-13.0])
             await mockgrid.mock_resampler.send_bat_inverter_reactive_power([0.0, -5.0])
             meter_sum = 0.0
             for recv in component_receivers:
@@ -326,9 +328,13 @@ async def test_grid_production_consumption_power_consumer_meter(
         grid_recv = grid.power.new_receiver()
 
         await mockgrid.mock_resampler.send_meter_power([1.0, 2.0, 3.0, 4.0])
+        await mockgrid.mock_resampler.send_bat_inverter_power([2.0, 3.0])
+        await mockgrid.mock_resampler.send_pv_inverter_power([4.0])
         assert (await grid_recv.receive()).value == Power.from_watts(10.0)
 
         await mockgrid.mock_resampler.send_meter_power([1.0, 2.0, -3.0, -4.0])
+        await mockgrid.mock_resampler.send_bat_inverter_power([2.0, -3.0])
+        await mockgrid.mock_resampler.send_pv_inverter_power([-4.0])
         assert (await grid_recv.receive()).value == Power.from_watts(-4.0)
 
 
@@ -348,9 +354,13 @@ async def test_grid_production_consumption_power_no_grid_meter(
         grid_recv = grid.power.new_receiver()
 
         await mockgrid.mock_resampler.send_meter_power([2.5, 3.5, 4.0])
+        await mockgrid.mock_resampler.send_bat_inverter_power([2.5, 3.5])
+        await mockgrid.mock_resampler.send_pv_inverter_power([4.0])
         assert (await grid_recv.receive()).value == Power.from_watts(10.0)
 
         await mockgrid.mock_resampler.send_meter_power([3.0, -3.0, -4.0])
+        await mockgrid.mock_resampler.send_bat_inverter_power([3.0, -3.0])
+        await mockgrid.mock_resampler.send_pv_inverter_power([-4.0])
         assert (await grid_recv.receive()).value == Power.from_watts(-4.0)
 
 

tests/timeseries/_ev_charger_pool/test_ev_charger_pool.py

@@ -26,8 +26,10 @@ async def test_ev_power(  # pylint: disable=too-many-locals
             ev_pool = microgrid.new_ev_charger_pool(priority=5)
             power_receiver = ev_pool.power.new_receiver()
 
+            await mockgrid.mock_resampler.send_meter_power([None])
             await mockgrid.mock_resampler.send_evc_power([2.0, 4.0, 10.0])
             assert (await power_receiver.receive()).value == Power.from_watts(16.0)
 
+            await mockgrid.mock_resampler.send_meter_power([None])
             await mockgrid.mock_resampler.send_evc_power([2.0, 4.0, -10.0])
             assert (await power_receiver.receive()).value == Power.from_watts(-4.0)

tests/timeseries/mock_resampler.py

@@ -75,6 +75,9 @@ def metric_senders(
         self._pv_inverter_power_senders = metric_senders(
             pv_inverter_ids, Metric.AC_ACTIVE_POWER
         )
+        self._pv_inverter_reactive_power_senders = metric_senders(
+            pv_inverter_ids, Metric.AC_REACTIVE_POWER
+        )
         self._ev_power_senders = metric_senders(evc_ids, Metric.AC_ACTIVE_POWER)
 
         self._chp_power_senders = metric_senders(chp_ids, Metric.AC_ACTIVE_POWER)
@@ -287,6 +290,13 @@ async def send_pv_inverter_power(self, values: list[float | None]) -> None:
             sample = self.make_sample(value)
             await chan.send(sample)
 
+    async def send_pv_inverter_reactive_power(self, values: list[float | None]) -> None:
+        """Send the given values as resampler output for PV Inverter power."""
+        assert len(values) == len(self._pv_inverter_reactive_power_senders)
+        for chan, value in zip(self._pv_inverter_reactive_power_senders, values):
+            sample = self.make_sample(value)
+            await chan.send(sample)
+
     async def send_meter_frequency(self, values: list[float | None]) -> None:
         """Send the given values as resampler output for meter frequency."""
         assert len(values) == len(self._meter_frequency_senders)

tests/timeseries/test_consumer.py

@@ -28,6 +28,8 @@ async def test_consumer_power_grid_meter(self, mocker: MockerFixture) -> None:
             consumer_power_receiver = consumer.power.new_receiver()
 
             await mockgrid.mock_resampler.send_meter_power([20.0, 2.0, 3.0, 4.0, 5.0])
+            await mockgrid.mock_resampler.send_bat_inverter_power([2.0, 3.0])
+            await mockgrid.mock_resampler.send_pv_inverter_power([4.0, 5.0])
             assert (await consumer_power_receiver.receive()).value == Power.from_watts(
                 6.0
             )
@@ -45,6 +47,8 @@ async def test_consumer_power_no_grid_meter(self, mocker: MockerFixture) -> None
             consumer_power_receiver = consumer.power.new_receiver()
 
             await mockgrid.mock_resampler.send_meter_power([20.0, 2.0, 3.0, 4.0, 5.0])
+            await mockgrid.mock_resampler.send_bat_inverter_power([2.0, 3.0])
+            await mockgrid.mock_resampler.send_pv_inverter_power([4.0, 5.0])
             assert (await consumer_power_receiver.receive()).value == Power.from_watts(
                 20.0
             )
@@ -63,6 +67,9 @@ async def test_consumer_power_no_grid_meter_no_consumer_meter(
             consumer_power_receiver = consumer.power.new_receiver()
 
             await mockgrid.mock_resampler.send_non_existing_component_value()
+            await mockgrid.mock_resampler.send_meter_power([2.0, 3.0, 4.0, 5.0])
+            await mockgrid.mock_resampler.send_bat_inverter_power([2.0, 3.0])
+            await mockgrid.mock_resampler.send_pv_inverter_power([4.0, 5.0])
             assert (await consumer_power_receiver.receive()).value == Power.from_watts(
                 0.0
             )

tests/timeseries/test_logical_meter.py

@@ -33,9 +33,11 @@ async def test_chp_power(self, mocker: MockerFixture) -> None:
             chp_power_receiver = logical_meter.chp_power.new_receiver()
 
             await mockgrid.mock_resampler.send_meter_power([2.0, 3.0, 4.0])
+            await mockgrid.mock_resampler.send_chp_power([2.0])
             assert (await chp_power_receiver.receive()).value == Power.from_watts(2.0)
 
             await mockgrid.mock_resampler.send_meter_power([-12.0, None, 10.2])
+            await mockgrid.mock_resampler.send_chp_power([-12.0])
             assert (await chp_power_receiver.receive()).value == Power.from_watts(-12.0)
 
     async def test_pv_power(self, mocker: MockerFixture) -> None:

tests/timeseries/test_producer.py

@@ -27,9 +27,11 @@ async def test_producer_power(self, mocker: MockerFixture) -> None:
             stack.push_async_callback(producer.stop)
             producer_power_receiver = producer.power.new_receiver()
 
-            await mockgrid.mock_resampler.send_meter_power([2.0, 3.0, 4.0, 5.0])
+            await mockgrid.mock_resampler.send_meter_power([-2.0, -3.0, -4.0, -5.0])
+            await mockgrid.mock_resampler.send_pv_inverter_power([-2.0, -3.0])
+            await mockgrid.mock_resampler.send_chp_power([-4.0, -5.0])
             assert (await producer_power_receiver.receive()).value == Power.from_watts(
-                14.0
+                -14.0
             )
 
     async def test_producer_power_no_chp(self, mocker: MockerFixture) -> None:
@@ -42,9 +44,10 @@ async def test_producer_power_no_chp(self, mocker: MockerFixture) -> None:
             stack.push_async_callback(producer.stop)
             producer_power_receiver = producer.power.new_receiver()
 
-            await mockgrid.mock_resampler.send_meter_power([2.0, 3.0])
+            await mockgrid.mock_resampler.send_meter_power([-2.0, -3.0])
+            await mockgrid.mock_resampler.send_pv_inverter_power([-2.0, -3.0])
             assert (await producer_power_receiver.receive()).value == Power.from_watts(
-                5.0
+                -5.0
             )
 
     async def test_producer_power_no_pv_no_consumer_meter(
@@ -59,10 +62,10 @@ async def test_producer_power_no_pv_no_consumer_meter(
             stack.push_async_callback(producer.stop)
             producer_power_receiver = producer.power.new_receiver()
 
-            await mockgrid.mock_resampler.send_chp_power([2.0])
+            await mockgrid.mock_resampler.send_chp_power([-2.0])
 
             assert (await producer_power_receiver.receive()).value == Power.from_watts(
-                2.0
+                -2.0
             )
 
     async def test_producer_power_no_pv(self, mocker: MockerFixture) -> None:
@@ -76,9 +79,10 @@ async def test_producer_power_no_pv(self, mocker: MockerFixture) -> None:
             stack.push_async_callback(producer.stop)
             producer_power_receiver = producer.power.new_receiver()
 
-            await mockgrid.mock_resampler.send_meter_power([20.0, 2.0])
+            await mockgrid.mock_resampler.send_meter_power([20.0, -2.0])
+            await mockgrid.mock_resampler.send_chp_power([-2.0])
             assert (await producer_power_receiver.receive()).value == Power.from_watts(
-                2.0
+                -2.0
             )
 
     async def test_no_producer_power(self, mocker: MockerFixture) -> None: