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

Better encapsulate solvers

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-02-22 20:22:33 -06:00
parent 141c8f0fdf
commit f8e8aeb973
3 changed files with 109 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
miplearn/tests/test_solver.py
View File

@@ -16,17 +16,21 @@ def _get_instance():
def test_solver():
instance = _get_instance()
solver = LearningSolver()
solver.solve(instance)
assert instance.solution["x"][0] == 1.0
assert instance.solution["x"][1] == 0.0
assert instance.solution["x"][2] == 1.0
assert instance.solution["x"][3] == 1.0
assert instance.lower_bound == 1183.0
assert instance.upper_bound == 1183.0
solver.fit()
solver.solve(instance)
for internal_solver in ["cplex", "gurobi"]:
solver = LearningSolver(time_limit=300,
gap_tolerance=1e-3,
threads=1,
solver=internal_solver,
)
results = solver.solve(instance)
assert instance.solution["x"][0] == 1.0
assert instance.solution["x"][1] == 0.0
assert instance.solution["x"][2] == 1.0
assert instance.solution["x"][3] == 1.0
assert instance.lower_bound == 1183.0
assert instance.upper_bound == 1183.0
solver.fit()
solver.solve(instance)
def test_solve_save_load_state():