mirror of
https://github.com/ANL-CEEESA/MIPLearn.git
synced 2025-12-06 17:38:51 -06:00
Use compact variable features everywhere
This commit is contained in:
@@ -16,6 +16,7 @@ from miplearn.solvers.internal import InternalSolver
|
||||
from miplearn.solvers.learning import LearningSolver
|
||||
|
||||
# noinspection PyUnresolvedReferences
|
||||
from miplearn.solvers.tests import _round
|
||||
from tests.solvers.test_internal_solver import internal_solvers
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
@@ -39,12 +40,9 @@ def test_learning_solver(
|
||||
|
||||
after_mip = sample.after_mip
|
||||
assert after_mip is not None
|
||||
assert after_mip.variables_old is not None
|
||||
assert after_mip.variables is not None
|
||||
assert after_mip.variables.values == (1.0, 0.0, 1.0, 1.0, 61.0)
|
||||
assert after_mip.mip_solve is not None
|
||||
assert after_mip.variables_old["x[0]"].value == 1.0
|
||||
assert after_mip.variables_old["x[1]"].value == 0.0
|
||||
assert after_mip.variables_old["x[2]"].value == 1.0
|
||||
assert after_mip.variables_old["x[3]"].value == 1.0
|
||||
assert after_mip.mip_solve.mip_lower_bound == 1183.0
|
||||
assert after_mip.mip_solve.mip_upper_bound == 1183.0
|
||||
assert after_mip.mip_solve.mip_log is not None
|
||||
@@ -52,16 +50,9 @@ def test_learning_solver(
|
||||
|
||||
after_lp = sample.after_lp
|
||||
assert after_lp is not None
|
||||
assert after_lp.variables_old is not None
|
||||
assert after_lp.variables is not None
|
||||
assert _round(after_lp.variables.values) == (1.0, 0.923077, 1.0, 0.0, 67.0)
|
||||
assert after_lp.lp_solve is not None
|
||||
assert after_lp.variables_old["x[0]"].value is not None
|
||||
assert after_lp.variables_old["x[1]"].value is not None
|
||||
assert after_lp.variables_old["x[2]"].value is not None
|
||||
assert after_lp.variables_old["x[3]"].value is not None
|
||||
assert round(after_lp.variables_old["x[0]"].value, 3) == 1.000
|
||||
assert round(after_lp.variables_old["x[1]"].value, 3) == 0.923
|
||||
assert round(after_lp.variables_old["x[2]"].value, 3) == 1.000
|
||||
assert round(after_lp.variables_old["x[3]"].value, 3) == 0.000
|
||||
assert after_lp.lp_solve.lp_value is not None
|
||||
assert round(after_lp.lp_solve.lp_value, 3) == 1287.923
|
||||
assert after_lp.lp_solve.lp_log is not None
|
||||
|
||||
Reference in New Issue
Block a user