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

Rename features.variables to variables_old; update FeatureExtractor

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-15 06:54:27 -05:00
parent 08f0bedbe0
commit fec0113722
9 changed files with 150 additions and 173 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/components/test_primal.py
View File

@@ -28,7 +28,7 @@ def sample() -> Sample:
sample = Sample(
after_load=Features(
instance=InstanceFeatures(),
variables={
variables_old={
"x[0]": Variable(category="default"),
"x[1]": Variable(category=None),
"x[2]": Variable(category="default"),
@@ -36,7 +36,7 @@ def sample() -> Sample:
},
),
after_lp=Features(
variables={
variables_old={
"x[0]": Variable(),
"x[1]": Variable(),
"x[2]": Variable(),
@@ -44,7 +44,7 @@ def sample() -> Sample:
},
),
after_mip=Features(
variables={
variables_old={
"x[0]": Variable(value=0.0),
"x[1]": Variable(value=1.0),
"x[2]": Variable(value=1.0),
@@ -53,13 +53,13 @@ def sample() -> Sample:
),
)
sample.after_load.instance.to_list = Mock(return_value=[5.0]) # type: ignore
sample.after_lp.variables["x[0]"].to_list = Mock( # type: ignore
sample.after_lp.variables_old["x[0]"].to_list = Mock( # type: ignore
return_value=[0.0, 0.0]
)
sample.after_lp.variables["x[2]"].to_list = Mock( # type: ignore
sample.after_lp.variables_old["x[2]"].to_list = Mock( # type: ignore
return_value=[1.0, 0.0]
)
sample.after_lp.variables["x[3]"].to_list = Mock( # type: ignore
sample.after_lp.variables_old["x[3]"].to_list = Mock( # type: ignore
return_value=[1.0, 1.0]
)
return sample