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 test

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-02-24 10:11:19 -06:00
parent 9acffadabe
commit 30e7d7313c
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
miplearn/components/tests/test_primal.py
View File

@@ -29,10 +29,8 @@ def test_predict():
comp.fit(instances)
solution = comp.predict(instances[0], models[0])
assert models[0].x in solution.keys()
assert solution[models[0].x][0] == 1
assert solution[models[0].x][1] == 1
assert solution[models[0].x][2] == 1
assert solution[models[0].x][3] == 1
for idx in range(4):
assert idx in solution[models[0].x].keys()
# def test_warm_start_save_load():
# state_file = tempfile.NamedTemporaryFile(mode="r")