ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-06 01:18:52 -06:00
Code Issues Packages Projects Releases Wiki Activity

Use np.ndarray in Constraints.{basis_status,senses}

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-08-09 06:04:14 -05:00
parent 9ddda7e1e2
commit f809dd7de4
7 changed files with 42 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/features/test_extractor.py
View File

@@ -54,7 +54,7 @@ def test_knapsack() -> None:
)
assert sample.get_vector_list("static_var_features") is not None
assert_equals(
sample.get_vector("static_constr_names"),
sample.get_array("static_constr_names"),
np.array(["eq_capacity"], dtype="S"),
)
# assert_equals(
@@ -70,7 +70,10 @@ def test_knapsack() -> None:
# ],
# )
assert_equals(sample.get_vector("static_constr_rhs"), [0.0])
assert_equals(sample.get_vector("static_constr_senses"), ["="])
assert_equals(
sample.get_array("static_constr_senses"),
np.array(["="], dtype="S"),
)
assert_equals(sample.get_vector("static_constr_features"), [None])
assert_equals(
sample.get_vector("static_constr_categories"),
@@ -114,7 +117,10 @@ def test_knapsack() -> None:
assert_equals(sample.get_vector("lp_var_sa_ub_up"), [2.043478, inf, 2.2, inf, 69.0])
assert_equals(sample.get_vector("lp_var_values"), [1.0, 0.923077, 1.0, 0.0, 67.0])
assert sample.get_vector_list("lp_var_features") is not None
assert_equals(sample.get_vector("lp_constr_basis_status"), ["N"])
assert_equals(
sample.get_array("lp_constr_basis_status"),
np.array(["N"], dtype="S"),
)
assert_equals(sample.get_vector("lp_constr_dual_values"), [13.538462])
assert_equals(sample.get_vector("lp_constr_sa_rhs_down"), [-24.0])
assert_equals(sample.get_vector("lp_constr_sa_rhs_up"), [2.0])
@@ -132,7 +138,7 @@ def test_constraint_getindex() -> None:
cf = Constraints(
names=np.array(["c1", "c2", "c3"], dtype="S"),
rhs=np.array([1.0, 2.0, 3.0]),
senses=["=", "<", ">"],
senses=np.array(["=", "<", ">"], dtype="S"),
lhs=[
[
(b"x1", 1.0),
@@ -153,7 +159,7 @@ def test_constraint_getindex() -> None:
Constraints(
names=np.array(["c1", "c3"], dtype="S"),
rhs=np.array([1.0, 3.0]),
senses=["=", ">"],
senses=np.array(["=", ">"], dtype="S"),
lhs=[
[
(b"x1", 1.0),