You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32 lines
1.2 KiB
32 lines
1.2 KiB
# MIPLearn: Extensible Framework for Learning-Enhanced Mixed-Integer Optimization
|
|
# Copyright (C) 2020, UChicago Argonne, LLC. All rights reserved.
|
|
# Released under the modified BSD license. See COPYING.md for more details.
|
|
|
|
from miplearn import GurobiSolver
|
|
from miplearn.features import ModelFeaturesExtractor
|
|
from tests.fixtures.knapsack import get_knapsack_instance
|
|
|
|
|
|
def test_knapsack() -> None:
|
|
for solver_factory in [GurobiSolver]:
|
|
# Initialize model, instance and internal solver
|
|
solver = solver_factory()
|
|
instance = get_knapsack_instance(solver)
|
|
model = instance.to_model()
|
|
solver.set_instance(instance, model)
|
|
|
|
# Extract all model features
|
|
extractor = ModelFeaturesExtractor(solver)
|
|
features = extractor.extract()
|
|
|
|
# Test constraint features
|
|
print(solver, features)
|
|
assert features["constraints"]["eq_capacity"]["lhs"] == {
|
|
"x[0]": 23.0,
|
|
"x[1]": 26.0,
|
|
"x[2]": 20.0,
|
|
"x[3]": 18.0,
|
|
}
|
|
assert features["constraints"]["eq_capacity"]["sense"] == "<"
|
|
assert features["constraints"]["eq_capacity"]["rhs"] == 67.0
|