fix: mypy

Author: FBumann

test/test_constraint.py

@@ -437,6 +437,20 @@ def test_constraint_to_polars(c: linopy.constraints.Constraint) -> None:
     assert isinstance(c.to_polars(), pl.DataFrame)
 
 
+def test_constraint_to_polars_mixed_signs(m: Model, x: linopy.Variable) -> None:
+    """Test to_polars when a constraint has mixed sign values across dims."""
+    # Create a constraint, then manually patch the sign to have mixed values
+    m.add_constraints(x >= 0, name="mixed")
+    con = m.constraints["mixed"]
+    # Replace sign data with mixed signs across the first dimension
+    n = con.data.sizes["first"]
+    signs = np.array(["<=" if i % 2 == 0 else ">=" for i in range(n)])
+    con.data["sign"] = xr.DataArray(signs, dims=con.data["sign"].dims)
+    df = con.to_polars()
+    assert isinstance(df, pl.DataFrame)
+    assert set(df["sign"].to_list()) == {"<=", ">="}
+
+
 def test_constraint_assignment_with_anonymous_constraints(
     m: Model, x: linopy.Variable, y: linopy.Variable
 ) -> None:

test/test_io.py

@@ -383,3 +383,23 @@ def test_to_file_lp_same_sign_constraints(tmp_path: Path) -> None:
     content = fn.read_text()
     assert "s.t." in content
     assert "<=" in content
+
+
+def test_to_file_lp_mixed_sign_constraints(tmp_path: Path) -> None:
+    """Test LP writing when constraints have different sign operators."""
+    m = Model()
+    N = np.arange(5)
+    x = m.add_variables(coords=[N], name="x")
+    # Mix of <= and >= constraints in the same container
+    m.add_constraints(x <= 10, name="upper")
+    m.add_constraints(x >= 1, name="lower")
+    m.add_constraints(2 * x == 8, name="eq")
+    m.add_objective(x.sum())
+
+    fn = tmp_path / "mixed_sign.lp"
+    m.to_file(fn)
+    content = fn.read_text()
+    assert "s.t." in content
+    assert "<=" in content
+    assert ">=" in content
+    assert "=" in content