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

Fix failing tests

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-01-07 09:41:55 -06:00
parent 4a26de5ff1
commit 144ee668e9
1 changed files with 10 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
miplearn/components/tests/test_relaxation.py
View File

@@ -61,19 +61,15 @@ def _setup():
return solver, internal, instance, classifiers return solver, internal, instance, classifiers
def test_usage(): def test_drop_redundant():
solver, internal, instance, classifiers = _setup() solver, internal, instance, classifiers = _setup()
component = RelaxationComponent() component = DropRedundantInequalitiesStep()
drop_ineqs_step = component.steps[1] component.classifiers = classifiers
drop_ineqs_step.classifiers = classifiers
# LearningSolver calls before_solve # LearningSolver calls before_solve
component.before_solve(solver, instance, None) component.before_solve(solver, instance, None)
# Should relax integrality of the problem
internal.relax.assert_called_once()
# Should query list of constraints # Should query list of constraints
internal.get_constraint_ids.assert_called_once() internal.get_constraint_ids.assert_called_once()
@@ -99,10 +95,10 @@ def test_usage():
) )
# Should ask ML to predict whether constraint should be removed # Should ask ML to predict whether constraint should be removed
drop_ineqs_step.classifiers["type-a"].predict_proba.assert_called_once_with( component.classifiers["type-a"].predict_proba.assert_called_once_with(
[[1.0, 0.0], [0.5, 0.5]] [[1.0, 0.0], [0.5, 0.5]]
) )
drop_ineqs_step.classifiers["type-b"].predict_proba.assert_called_once_with([[1.0]]) component.classifiers["type-b"].predict_proba.assert_called_once_with([[1.0]])
# Should ask internal solver to remove constraints predicted as redundant # Should ask internal solver to remove constraints predicted as redundant
assert internal.extract_constraint.call_count == 2 assert internal.extract_constraint.call_count == 2
@@ -129,12 +125,13 @@ def test_usage():
} }
def test_usage_with_check_dropped(): def test_drop_redundant_with_check_dropped():
solver, internal, instance, classifiers = _setup() solver, internal, instance, classifiers = _setup()
component = RelaxationComponent(check_dropped=True, violation_tolerance=1e-3) component = DropRedundantInequalitiesStep(
drop_ineqs_step = component.steps[1] check_dropped=True, violation_tolerance=1e-3
drop_ineqs_step.classifiers = classifiers )
component.classifiers = classifiers
# LearningSolver call before_solve # LearningSolver call before_solve
component.before_solve(solver, instance, None) component.before_solve(solver, instance, None)