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

FileInstance.save: create file when it does not already exist

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-07-27 11:22:40 -05:00
parent 4f14b99a75
commit d30c3232e6
3 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/instance/test_file.py
View File

@@ -15,17 +15,17 @@ def test_usage() -> None:
original = GurobiSolver().build_test_instance_knapsack()
# Save instance to disk
file = tempfile.NamedTemporaryFile()
FileInstance.save(original, file.name)
sample = Hdf5Sample(file.name)
filename = tempfile.mktemp()
FileInstance.save(original, filename)
sample = Hdf5Sample(filename)
assert len(sample.get_bytes("pickled")) > 0
# Solve instance from disk
solver = LearningSolver(solver=GurobiSolver())
solver.solve(FileInstance(file.name))
solver.solve(FileInstance(filename))
# Assert HDF5 contains training data
sample = FileInstance(file.name).get_samples()[0]
sample = FileInstance(filename).get_samples()[0]
assert sample.get_scalar("mip_lower_bound") == 1183.0
assert sample.get_scalar("mip_upper_bound") == 1183.0
assert len(sample.get_vector("lp_var_values")) == 5