def setUp(self): self.ES = ExampleSelectorGA()
class TestExampleSelectorGA(unittest.TestCase): def setUp(self): self.ES = ExampleSelectorGA() def test_set_data_from_path(self): self.ES.set_data_from_path("train_0", "test_0") assert all(x[0] == "1" for x in self.ES.train_actives) assert all(x[0] == "0" for x in self.ES.train_decoys) def test_set_up_deap(self): self.ES.set_data_from_path("train_0", "test_0") self.ES.setup_deap() def test_individual(self): self.ES.set_data_from_path("train_0", "test_0") self.ES.setup_deap() ind1 = self.ES.toolbox.individual() ind2 = self.ES.toolbox.individual() assert len(ind1) == len(ind2) self.ES.set_delimiting_indices() sp_ind1 = self.ES.single_ppl_ind() sp_ind2 = self.ES.single_ppl_ind() assert len(sp_ind1) == len(sp_ind2) assert sum(sp_ind1) == sum(sp_ind2) def test_translate(self): self.ES.set_data_from_path("train_0", "test_0") self.ES.setup_deap() individual = self.ES.toolbox.individual() self.ES.translate(individual) def test_mutate(self): self.ES.set_data_from_path("train_0", "test_0") self.ES.setup_deap() self.ES.set_delimiting_indices() individual = self.ES.toolbox.individual() self.ES.mutate(individual, 1.0) sp_individual = self.ES.single_ppl_ind() self.ES.sp_mutate(sp_individual, 1.0) assert sum(sp_individual) == len(self.ES.delimiting_indices) + 1 def test_crossover(self): self.ES.set_data_from_path("train_0", "test_0") self.ES.setup_deap() self.ES.set_delimiting_indices() individual = self.ES.toolbox.individual() other = self.ES.toolbox.individual() self.ES.uniform_crossover(individual, other, 1.0) sp_individual = self.ES.single_ppl_ind() sp_other = self.ES.single_ppl_ind() self.ES.sp_crossover(sp_individual, sp_other, 1.0) assert sum(sp_individual) == sum(sp_other) assert sum(sp_individual) == len(self.ES.delimiting_indices) + 1