Add and update tests

Author: tbittar

tests/e2e/functional/test_out_of_bounds_processing.py

@@ -140,7 +140,7 @@ def test_out_of_bounds_processing(study_id: str, expected_objective: float) -> N
 # Constraints: 2 components (gen_1, gen_2) × 3 timesteps = 6 potential instances each.
 #
 # system_cyclic_with_param_in_shift — no drop mode, all instances present:
-#   is_on_dynamics (lb + ub): 6 each
+#   is_on_dynamics (eq): 6
 #   min_up_duration (ub only): 6
 #   min_down_duration (ub only): 6
 #
@@ -151,14 +151,12 @@ def test_out_of_bounds_processing(study_id: str, expected_objective: float) -> N
 #   min_down_duration: both have d_min_down=1 → range [0,0] → never dropped → 6
 _EXPECTED_CONSTRAINT_COUNTS: Dict[str, Dict[str, int]] = {
     "system_cyclic_with_param_in_shift": {
-        f"{_GEN_PREFIX}__is_on_dynamics__lb": 6,
-        f"{_GEN_PREFIX}__is_on_dynamics__ub": 6,
+        f"{_GEN_PREFIX}__is_on_dynamics__eq": 6,
         f"{_GEN_PREFIX}__min_up_duration__ub": 6,
         f"{_GEN_PREFIX}__min_down_duration__ub": 6,
     },
     "system_drop_with_param_in_shift": {
-        f"{_GEN_PREFIX}__is_on_dynamics__lb": 4,
-        f"{_GEN_PREFIX}__is_on_dynamics__ub": 4,
+        f"{_GEN_PREFIX}__is_on_dynamics__eq": 4,
         f"{_GEN_PREFIX}__min_up_duration__ub": 5,
         f"{_GEN_PREFIX}__min_down_duration__ub": 6,
     },

tests/unittests/system/test_model.py

@@ -219,6 +219,41 @@ def test_constraint_equals() -> None:
     )
 
 
+def test_is_equality_true_for_equality_comparison() -> None:
+    c = Constraint(name="c", expression=var("x") == param("p"))
+    assert c.is_equality is True
+
+
+def test_is_equality_false_for_inequality_comparison() -> None:
+    assert Constraint(name="c", expression=var("x") <= param("p")).is_equality is False
+    assert Constraint(name="c", expression=var("x") >= param("p")).is_equality is False
+
+
+def test_is_equality_false_for_range_constraint() -> None:
+    c = Constraint(
+        name="c", expression=var("x"), lower_bound=literal(0), upper_bound=literal(10)
+    )
+    assert c.is_equality is False
+
+
+def test_is_equality_true_for_equal_explicit_bounds() -> None:
+    c = Constraint(
+        name="c", expression=var("x"), lower_bound=param("p"), upper_bound=param("p")
+    )
+    assert c.is_equality is True
+
+
+def test_is_equality_false_for_one_sided_constraint() -> None:
+    assert (
+        Constraint(name="c", expression=var("x"), lower_bound=literal(0)).is_equality
+        is False
+    )
+    assert (
+        Constraint(name="c", expression=var("x"), upper_bound=literal(0)).is_equality
+        is False
+    )
+
+
 # --- Issue #76: tolerate absence of expec() in objective contributions ---