MIPLearn v0.3

tests/problems/test_setpack.py

@@ -0,0 +1,26 @@
+#  MIPLearn: Extensible Framework for Learning-Enhanced Mixed-Integer Optimization
+#  Copyright (C) 2020-2022, UChicago Argonne, LLC. All rights reserved.
+#  Released under the modified BSD license. See COPYING.md for more details.
+
+import numpy as np
+
+from miplearn.problems.setpack import (
+    SetPackData,
+    build_setpack_model,
+)
+
+
+def test_setpack() -> None:
+    data = SetPackData(
+        costs=np.array([5, 10, 12, 6, 8]),
+        incidence_matrix=np.array(
+            [
+                [1, 0, 0, 1, 0],
+                [1, 1, 0, 0, 0],
+                [0, 0, 1, 1, 1],
+            ],
+        ),
+    )
+    model = build_setpack_model(data)
+    model.optimize()
+    assert model.inner.objval == -22.0