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

Make MaxWeightStableSetGenerator return data class

Browse Source
This commit is contained in:
Alinson S. Xavier
2022-02-22 09:16:37 -06:00
parent 1811492557
commit b0d63a0a2d
3 changed files with 27 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/problems/test_stab.py
View File

@@ -29,8 +29,8 @@ def test_stab_generator_fixed_graph() -> None:
p=uniform(loc=0.05, scale=0.0),
fix_graph=True,
)
instances = gen.generate(1_000)
weights = np.array([instance.weights for instance in instances])
data = gen.generate(1_000)
weights = np.array([d.weights for d in data])
weights_avg_actual = np.round(np.average(weights, axis=0))
weights_avg_expected = [55.0] * 10
assert list(weights_avg_actual) == weights_avg_expected
@@ -46,8 +46,8 @@ def test_stab_generator_random_graph() -> None:
p=uniform(loc=0.5, scale=0.0),
fix_graph=False,
)
instances = gen.generate(1_000)
n_nodes = [instance.graph.number_of_nodes() for instance in instances]
n_edges = [instance.graph.number_of_edges() for instance in instances]
data = gen.generate(1_000)
n_nodes = [d.graph.number_of_nodes() for d in data]
n_edges = [d.graph.number_of_edges() for d in data]
assert np.round(np.mean(n_nodes)) == 35.0
assert np.round(np.mean(n_edges), -1) == 300.0