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])])