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

Make TravelingSalesmanGenerator return data class

Browse Source
This commit is contained in:
Alinson S. Xavier
2022-02-22 09:23:55 -06:00
parent 03e5acb11a
commit 87bba1b38e
3 changed files with 16 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/problems/test_tsp.py
View File

@@ -14,17 +14,17 @@ from miplearn.solvers.tests import assert_equals
def test_generator() -> None:
instances = TravelingSalesmanGenerator(
data = TravelingSalesmanGenerator(
x=uniform(loc=0.0, scale=1000.0),
y=uniform(loc=0.0, scale=1000.0),
n=randint(low=100, high=101),
gamma=uniform(loc=0.95, scale=0.1),
fix_cities=True,
).generate(100)
assert len(instances) == 100
assert instances[0].n_cities == 100
assert norm(instances[0].distances - instances[0].distances.T) < 1e-6
d = [instance.distances[0, 1] for instance in instances]
assert len(data) == 100
assert data[0].n_cities == 100
assert norm(data[0].distances - data[0].distances.T) < 1e-6
d = [d.distances[0, 1] for d in data]
assert np.std(d) > 0