Replace instance.samples by instance.get/push_sample

This commit is contained in:
2021-06-29 16:49:24 -05:00
parent a5092cc2b9
commit 80281df8d8
9 changed files with 48 additions and 34 deletions

View File

@@ -70,7 +70,7 @@ def sample() -> Sample:
@pytest.fixture
def instance(sample: Sample) -> Instance:
instance = Mock(spec=Instance)
instance.samples = [sample]
instance.get_samples = Mock(return_value=[sample]) # type: ignore
instance.has_static_lazy_constraints = Mock(return_value=True)
return instance
@@ -111,7 +111,7 @@ def test_usage_with_solver(instance: Instance) -> None:
)
stats: LearningSolveStats = {}
sample = instance.samples[0]
sample = instance.get_samples()[0]
assert sample.after_load is not None
assert sample.after_mip is not None
assert sample.after_mip.extra is not None