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

Implement TravelingSalesmanPerturber

Browse Source
This commit is contained in:
Alinson S. Xavier
2025-12-08 15:10:24 -06:00
parent 4137378bb8
commit 1d44980a7b
23 changed files with 128 additions and 90 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/components/lazy/test_mem.py
View File

@@ -22,7 +22,7 @@ def test_mem_component(
for h5 in [tsp_gp_h5, tsp_pyo_h5]:
clf = Mock(wraps=DummyClassifier())
comp = MemorizingLazyComponent(clf=clf, extractor=default_extractor)
comp.fit(tsp_gp_h5)
comp.fit(h5)
# Should call fit method with correct arguments
clf.fit.assert_called()
@@ -43,7 +43,7 @@ def test_mem_component(
# Call before-mip
stats: Dict[str, Any] = {}
model = Mock()
comp.before_mip(tsp_gp_h5[0], model, stats)
comp.before_mip(h5[0], model, stats)
# Should call predict with correct args
clf.predict.assert_called()