fixup! Implement ListComponents

Add tests for Bounds

Author: llucax

tests/metrics/test_bounds.py

@@ -0,0 +1,132 @@
+# License: MIT
+# Copyright © 2025 Frequenz Energy-as-a-Service GmbH
+
+"""Tests for the Bounds class."""
+
+from dataclasses import dataclass
+
+import pytest
+from frequenz.api.common.v1.metrics import bounds_pb2
+
+from frequenz.client.microgrid.metrics import Bounds
+from frequenz.client.microgrid.metrics._bounds_proto import bounds_from_proto
+
+
+@dataclass(frozen=True, kw_only=True)
+class ProtoConversionTestCase:
+    """Test case for protobuf conversion."""
+
+    name: str
+    """Description of the test case."""
+
+    has_lower: bool
+    """Whether to include lower bound in the protobuf message."""
+
+    has_upper: bool
+    """Whether to include upper bound in the protobuf message."""
+
+    lower: float | None
+    """The lower bound value to set."""
+
+    upper: float | None
+    """The upper bound value to set."""
+
+
+def test_bounds_creation() -> None:
+    """Test creation of Bounds with valid values."""
+    bounds = Bounds(lower=-10.0, upper=10.0)
+    assert bounds.lower == -10.0
+    assert bounds.upper == 10.0
+
+
+def test_bounds_invalid_values() -> None:
+    """Test that Bounds creation fails with invalid values."""
+    with pytest.raises(
+        ValueError, match="Lower bound must be less than or equal to upper bound"
+    ):
+        Bounds(lower=10.0, upper=-10.0)
+
+
+def test_bounds_str_representation() -> None:
+    """Test string representation of Bounds."""
+    bounds = Bounds(lower=-10.0, upper=10.0)
+    assert str(bounds) == "[-10.0, 10.0]"
+
+
+def test_bounds_equality() -> None:
+    """Test equality comparison of Bounds objects."""
+    bounds1 = Bounds(lower=-10.0, upper=10.0)
+    bounds2 = Bounds(lower=-10.0, upper=10.0)
+    bounds3 = Bounds(lower=-5.0, upper=5.0)
+
+    assert bounds1 == bounds2
+    assert bounds1 != bounds3
+    assert bounds2 != bounds3
+
+
+def test_bounds_hash() -> None:
+    """Test that Bounds objects can be used in sets and as dictionary keys."""
+    bounds1 = Bounds(lower=-10.0, upper=10.0)
+    bounds2 = Bounds(lower=-10.0, upper=10.0)
+    bounds3 = Bounds(lower=-5.0, upper=5.0)
+
+    bounds_set = {bounds1, bounds2, bounds3}
+    assert len(bounds_set) == 2  # bounds1 and bounds2 are equal
+
+    bounds_dict = {bounds1: "test1", bounds3: "test2"}
+    assert len(bounds_dict) == 2
+
+
+@pytest.mark.parametrize(
+    "case",
+    [
+        ProtoConversionTestCase(
+            name="all fields present",
+            has_lower=True,
+            has_upper=True,
+            lower=-10.0,
+            upper=10.0,
+        ),
+        ProtoConversionTestCase(
+            name="only lower bound",
+            has_lower=True,
+            has_upper=False,
+            lower=-10.0,
+            upper=None,
+        ),
+        ProtoConversionTestCase(
+            name="only upper bound",
+            has_lower=False,
+            has_upper=True,
+            lower=None,
+            upper=10.0,
+        ),
+        ProtoConversionTestCase(
+            name="no bounds",
+            has_lower=False,
+            has_upper=False,
+            lower=None,
+            upper=None,
+        ),
+    ],
+    ids=lambda case: case.name,
+)
+def test_bounds_from_proto(case: ProtoConversionTestCase) -> None:
+    """Test conversion from protobuf message to Bounds.
+
+    Args:
+        case: Test case parameters
+
+    Raises:
+        AssertionError: If the conversion doesn't match expected values
+    """
+    proto = bounds_pb2.Bounds()
+    if case.has_lower and case.lower is not None:
+        proto.lower = case.lower
+    if case.has_upper and case.upper is not None:
+        proto.upper = case.upper
+
+    bounds = bounds_from_proto(proto)
+
+    assert bounds.lower == case.lower
+    assert bounds.upper == case.upper