Unittest the shifting matryoshka algorithm

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

Author: shsms

tests/actor/_power_managing/test_shifting_matryoshka.py

@@ -543,3 +543,65 @@ def ensure_overlapping_bucket_request_fails() -> None:
         expected=20.0,
         batteries=overlapping_batteries,
     )
+
+
+async def test_matryoshka_shifting_limiting() -> None:
+    """Tests for the power managing actor.
+
+    With the following scenario:
+
+    |    actor | available   |   req | req bounds | s/l    | unshifted | shifted |
+    | priority | bounds      | power |            |        |     power |   power |
+    |----------|-------------|-------|------------|--------|-----------|---------|
+    |        7 | -100 to 100 |    10 | None       | shifts |        10 |      10 |
+    |----------|-------------|-------|------------|--------|-----------|---------|
+    |        6 | -110 to 90  |    10 | -110 to 80 | limits |           |         |
+    |        5 | -110 to 80  |    80 | -100 to 80 | limits |           |         |
+    |        4 | -100 to 80  |  -120 | None       | shifts |      -100 |     -90 |
+    |----------|-------------|-------|------------|--------|-----------|---------|
+    |        3 | 0 to 180    |    60 | None       | shifts |        60 |     -30 |
+    |----------|-------------|-------|------------|--------|-----------|---------|
+    |        2 | -60 to 120  |    20 | -40 to 30  | limits |           |         |
+    |        1 | -40 to 30   |    12 | None       | shifts |        12 |     -18 |
+    |----------|-------------|-------|------------|--------|-----------|---------|
+    |        0 | -52 to 18   |   -10 | -40 to -10 | limits |       -10 |     -28 |
+    |----------|-------------|-------|------------|--------|-----------|---------|
+    |          |             |       |            | target |           |         |
+    |          |             |       |            | power  |       -28 |         |
+    """
+    batteries = frozenset({2, 5})
+
+    system_bounds = _base_types.SystemBounds(
+        timestamp=datetime.now(tz=timezone.utc),
+        inclusion_bounds=timeseries.Bounds(
+            lower=Power.from_watts(-100.0), upper=Power.from_watts(100.0)
+        ),
+        exclusion_bounds=timeseries.Bounds(
+            lower=Power.from_watts(-0.0), upper=Power.from_watts(0.0)
+        ),
+    )
+
+    tester = StatefulTester(batteries, system_bounds)
+    tester.tgt_power(priority=7, power=10.0, bounds=(None, None), expected=10.0)
+    tester.bounds(priority=7, expected_power=10.0, expected_bounds=(-100.0, 100.0))
+    tester.bounds(priority=6, expected_power=10.0, expected_bounds=(-110.0, 90.0))
+
+    tester.tgt_power(priority=6, power=10.0, bounds=(-110.0, 80.0), expected=20.0)
+    tester.bounds(priority=5, expected_power=20.0, expected_bounds=(-110.0, 80.0))
+
+    tester.tgt_power(priority=5, power=80.0, bounds=(-100.0, 80.0), expected=90.0)
+    tester.bounds(priority=4, expected_power=90.0, expected_bounds=(-100.0, 80.0))
+
+    tester.tgt_power(priority=4, power=-120.0, bounds=(None, None), expected=-90.0)
+    tester.bounds(priority=3, expected_power=-90.0, expected_bounds=(0.0, 180.0))
+
+    tester.tgt_power(priority=3, power=60.0, bounds=(None, None), expected=-30.0)
+    tester.bounds(priority=2, expected_power=-30.0, expected_bounds=(-60.0, 120.0))
+
+    tester.tgt_power(priority=2, power=20.0, bounds=(-40.0, 30.0), expected=-10.0)
+    tester.bounds(priority=1, expected_power=-10.0, expected_bounds=(-40.0, 30.0))
+
+    tester.tgt_power(priority=1, power=12.0, bounds=(None, None), expected=-18.0)
+    tester.bounds(priority=0, expected_power=-18.0, expected_bounds=(-52.0, 18.0))
+
+    tester.tgt_power(priority=0, power=-10.0, bounds=(-40.0, -10.0), expected=-28.0)