def test_all_strategy(self):
want = [
ensemble.Candidate("all", [self.fake_builder_1, self.fake_builder_2],
[])
]
got = ensemble.AllStrategy().generate_ensemble_candidates(
[self.fake_builder_1, self.fake_builder_2], None)
self.assertEqual(want, got)
def test_all_strategy_with_previous_ensemble_subnetwork_builders(self):
want = [
ensemble.Candidate("all", [self.fake_builder_1, self.fake_builder_2],
[self.fake_builder_3, self.fake_builder_4])
]
got = ensemble.AllStrategy().generate_ensemble_candidates(
[self.fake_builder_1, self.fake_builder_2],
[self.fake_builder_3, self.fake_builder_4])
self.assertEqual(want, got)
def test_all_strategy(self):
self.assertEqual(
ensemble.AllStrategy().generate_ensemble_candidates(
[self.fake_builder_1, self.fake_builder_2]),
[ensemble.Candidate([self.fake_builder_1, self.fake_builder_2])])