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

Switch tests to simpler Knapsack encoding; remove outdated test

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-02-04 11:34:03 -06:00
parent c82de560f4
commit 6685f4ff23
4 changed files with 32 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
miplearn/tests/test_solver.py
View File

@@ -3,17 +3,17 @@
# Written by Alinson S. Xavier <axavier@anl.gov>
from miplearn import LearningSolver
from miplearn.problems.knapsack import MultiKnapsackInstance
from miplearn.problems.knapsack import KnapsackInstance
from miplearn.branching import BranchPriorityComponent
from miplearn.warmstart import WarmStartComponent
import numpy as np
def _get_instance():
return MultiKnapsackInstance(
weights=np.array([[23., 26., 20., 18.]]),
prices=np.array([505., 352., 458., 220.]),
capacities=np.array([67.])
return KnapsackInstance(
weights=[23., 26., 20., 18.],
prices=[505., 352., 458., 220.],
capacity=67.,
)
def test_solver():
@@ -50,8 +50,8 @@ def test_parallel_solve():
solver = LearningSolver()
results = solver.parallel_solve(instances, n_jobs=3)
assert len(results) == 10
assert len(solver.components["warm-start"].x_train[0]) == 10
assert len(solver.components["warm-start"].y_train[0]) == 10
assert len(solver.components["warm-start"].x_train["default"]) == 40
assert len(solver.components["warm-start"].y_train["default"]) == 40
def test_solver_random_branch_priority():
instance = _get_instance()