Add consumption and production formula tests

Signed-off-by: Matthias Wende <[email protected]>

Author: matthias-wende-frequenz

tests/timeseries/test_formula_engine.py

@@ -318,6 +318,19 @@ async def run_test(  # pylint: disable=too-many-locals
         num_items: int,
         make_builder: (
             Callable[
+                [
+                    FormulaEngine[Quantity],
+                ],
+                HigherOrderFormulaBuilder[Quantity],
+            ]
+            | Callable[
+                [
+                    FormulaEngine[Quantity],
+                    FormulaEngine[Quantity],
+                ],
+                HigherOrderFormulaBuilder[Quantity],
+            ]
+            | Callable[
                 [
                     FormulaEngine[Quantity],
                     FormulaEngine[Quantity],
@@ -586,6 +599,95 @@ async def test_compound(self) -> None:
             ],
         )
 
+    async def test_consumption(self) -> None:
+        """Test the consumption operator."""
+        await self.run_test(
+            1,
+            lambda c1: c1.consumption(),
+            [
+                ([10.0], 10.0),
+                ([-10.0], 0.0),
+            ],
+        )
+
+    async def test_production(self) -> None:
+        """Test the production operator."""
+        await self.run_test(
+            1,
+            lambda c1: c1.production(),
+            [
+                ([10.0], 0.0),
+                ([-10.0], 10.0),
+            ],
+        )
+
+    async def test_consumption_production(self) -> None:
+        """Test the consumption and production operator combined."""
+        await self.run_test(
+            2,
+            lambda c1, c2: c1.consumption() + c2.production(),
+            [
+                ([10.0, 12.0], 10.0),
+                ([-12.0, -10.0], 10.0),
+            ],
+        )
+        await self.run_test(
+            2,
+            lambda c1, c2: c1.consumption() + c2.consumption(),
+            [
+                ([10.0, -12.0], 10.0),
+                ([-10.0, 12.0], 12.0),
+            ],
+        )
+        await self.run_test(
+            2,
+            lambda c1, c2: c1.production() + c2.production(),
+            [
+                ([10.0, -12.0], 12.0),
+                ([-10.0, 12.0], 10.0),
+            ],
+        )
+        await self.run_test(
+            2,
+            lambda c1, c2: c1.min(c2).consumption(),
+            [
+                ([10.0, -12.0], 0.0),
+                ([10.0, 12.0], 10.0),
+            ],
+        )
+        await self.run_test(
+            2,
+            lambda c1, c2: c1.max(c2).consumption(),
+            [
+                ([10.0, -12.0], 10.0),
+                ([-10.0, -12.0], 0.0),
+            ],
+        )
+        await self.run_test(
+            2,
+            lambda c1, c2: c1.min(c2).production(),
+            [
+                ([10.0, -12.0], 12.0),
+                ([10.0, 12.0], 0.0),
+            ],
+        )
+        await self.run_test(
+            2,
+            lambda c1, c2: c1.max(c2).production(),
+            [
+                ([10.0, -12.0], 0.0),
+                ([-10.0, -12.0], 10.0),
+            ],
+        )
+        await self.run_test(
+            2,
+            lambda c1, c2: c1.production() + c2,
+            [
+                ([10.0, -12.0], -12.0),
+                ([-10.0, -12.0], -2.0),
+            ],
+        )
+
     async def test_nones_are_zeros(self) -> None:
         """Test that `None`s are treated as zeros when configured."""
         await self.run_test(