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

Add pre argument to sample_xy

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-13 19:19:49 -05:00
parent a01c179341
commit bec7dae6d9
8 changed files with 64 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/components/test_dynamic_lazy.py
View File

@@ -87,7 +87,6 @@ def training_instances() -> List[Instance]:
def test_sample_xy(training_instances: List[Instance]) -> None:
comp = DynamicLazyConstraintsComponent()
comp.dynamic.known_cids = ["c1", "c2", "c3", "c4"]
x_expected = {
"type-a": [[5.0, 1.0, 2.0, 3.0], [5.0, 4.0, 5.0, 6.0]],
"type-b": [[5.0, 1.0, 2.0], [5.0, 3.0, 4.0]],
@@ -99,6 +98,7 @@ def test_sample_xy(training_instances: List[Instance]) -> None:
x_actual, y_actual = comp.sample_xy(
training_instances[0],
training_instances[0].samples[0],
pre=[{"c1", "c2", "c3", "c4"}],
)
assert_equals(x_actual, x_expected)
assert_equals(y_actual, y_expected)