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

Add types to remaining files; activate mypy's disallow_untyped_defs

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-07 21:25:30 -05:00
parent f5606efb72
commit e9cd6d1715
21 changed files with 102 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/problems/test_stab.py
View File

@@ -10,16 +10,16 @@ from miplearn.problems.stab import MaxWeightStableSetInstance
from miplearn.solvers.learning import LearningSolver
def test_stab():
def test_stab() -> None:
graph = nx.cycle_graph(5)
weights = [1.0, 1.0, 1.0, 1.0, 1.0]
weights = np.array([1.0, 1.0, 1.0, 1.0, 1.0])
instance = MaxWeightStableSetInstance(graph, weights)
solver = LearningSolver()
stats = solver.solve(instance)
assert stats["Lower bound"] == 2.0
def test_stab_generator_fixed_graph():
def test_stab_generator_fixed_graph() -> None:
np.random.seed(42)
from miplearn.problems.stab import MaxWeightStableSetGenerator
@@ -36,7 +36,7 @@ def test_stab_generator_fixed_graph():
assert list(weights_avg_actual) == weights_avg_expected
def test_stab_generator_random_graph():
def test_stab_generator_random_graph() -> None:
np.random.seed(42)
from miplearn.problems.stab import MaxWeightStableSetGenerator