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

Implement iteration_cb for LearningSolver; reactivate TSP

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-09-23 17:46:18 -05:00
parent 5390a5b656
commit e527e75481
8 changed files with 91 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
miplearn/components/tests/test_lazy_static.py
View File

@@ -95,8 +95,9 @@ def test_usage_with_solver():
])
internal.add_constraint.reset_mock()
# LearningSolver calls callback (first time)
component.on_callback(solver, instance, None)
# LearningSolver calls after_iteration (first time)
should_repeat = component.after_iteration(solver, instance, None)
assert should_repeat
# Should ask internal solver to verify if constraints in the pool are
# satisfied and add the ones that are not
@@ -105,8 +106,9 @@ def test_usage_with_solver():
internal.add_constraint.assert_called_once_with("<c2>")
internal.add_constraint.reset_mock()
# LearningSolver calls callback (second time)
component.on_callback(solver, instance, None)
# LearningSolver calls after_iteration (second time)
should_repeat = component.after_iteration(solver, instance, None)
assert not should_repeat
# The lazy constraint pool should be empty by now, so no calls should be made
internal.is_constraint_satisfied.assert_not_called()