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

Implement SetPackPerturber and SetCoverPerturber

Browse Source
This commit is contained in:
Alinson S. Xavier
2025-12-08 13:47:33 -06:00
parent 427bd1d806
commit 4137378bb8
4 changed files with 213 additions and 106 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
tests/problems/test_setcover.py
View File

@@ -23,51 +23,16 @@ def test_set_cover_generator() -> None:
n_elements=randint(low=3, high=4),
n_sets=randint(low=5, high=6),
costs=uniform(loc=0.0, scale=100.0),
costs_jitter=uniform(loc=0.95, scale=0.10),
density=uniform(loc=0.5, scale=0),
K=uniform(loc=25, scale=0),
fix_sets=False,
)
data = gen.generate(2)
data = gen.generate(1)
assert data[0].costs.round(1).tolist() == [136.8, 86.2, 25.7, 27.3, 102.5]
assert data[0].incidence_matrix.tolist() == [
[1, 0, 1, 0, 1],
[1, 1, 0, 0, 0],
[1, 0, 0, 1, 1],
]
assert data[1].costs.round(1).tolist() == [63.5, 76.6, 48.1, 74.1, 93.3]
assert data[1].incidence_matrix.tolist() == [
[1, 1, 0, 1, 1],
[0, 1, 0, 1, 0],
[0, 1, 1, 0, 0],
]
def test_set_cover_generator_with_fixed_sets() -> None:
np.random.seed(42)
gen = SetCoverGenerator(
n_elements=randint(low=3, high=4),
n_sets=randint(low=5, high=6),
costs=uniform(loc=0.0, scale=100.0),
costs_jitter=uniform(loc=0.95, scale=0.10),
density=uniform(loc=0.5, scale=0.00),
fix_sets=True,
)
data = gen.generate(3)
assert data[0].costs.tolist() == [136.75, 86.17, 25.71, 27.31, 102.48]
assert data[1].costs.tolist() == [135.38, 82.26, 26.92, 26.58, 98.28]
assert data[2].costs.tolist() == [138.37, 85.15, 26.95, 27.22, 106.17]
print(data[0].incidence_matrix)
for i in range(3):
assert data[i].incidence_matrix.tolist() == [
[1, 0, 1, 0, 1],
[1, 1, 0, 0, 0],
[1, 0, 0, 1, 1],
]
def test_set_cover() -> None: