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.
20 lines
691 B
20 lines
691 B
# MIPLearn: Extensible Framework for Learning-Enhanced Mixed-Integer Optimization
|
|
# Copyright (C) 2020-2022, UChicago Argonne, LLC. All rights reserved.
|
|
# Released under the modified BSD license. See COPYING.md for more details.
|
|
|
|
from typing import List
|
|
|
|
from miplearn.extractors.dummy import DummyExtractor
|
|
from miplearn.h5 import H5File
|
|
|
|
|
|
def test_dummy(multiknapsack_h5: List[str]) -> None:
|
|
ext = DummyExtractor()
|
|
with H5File(multiknapsack_h5[0], "r") as h5:
|
|
x = ext.get_instance_features(h5)
|
|
assert x.shape == (1,)
|
|
x = ext.get_var_features(h5)
|
|
assert x.shape == (100, 1)
|
|
x = ext.get_constr_features(h5)
|
|
assert x.shape == (4, 1)
|