def test_cross_over_generation(self):
ga = GASolver(max_generations, population_size, antibody_cnt, nc_cnt,
c_cnt, markers, cell_cnt, mu_list, sigma_list)
ga.create_random_population()
ga.survive_n_fittest(0, population_size / 2)
start_ind = population_size / 2
new_start_ind = ga.cross_over_generation(1, start_ind)
self.assertEqual(ga.total_population, population_size * 5 / 4)
self.assertEqual(new_start_ind, start_ind * 3 / 2)
# half is zero
self.assertTrue([0, 0, 0, 0] not in ga.population[0].tolist())
x = [el for el in ga.population[1] if el.tolist() == [0, 0, 0, 0]]
self.assertEqual(len(x), population_size / 4)
def test_survive_n_fittest(self):
ga = GASolver(max_generations, population_size, antibody_cnt, nc_cnt,
c_cnt, markers, cell_cnt, mu_list, sigma_list)
ga.create_random_population()
fit_cnt = ga.survive_n_fittest(0, 2)
self.assertNotEqual(ga.population[1][0].tolist(), [0, 0, 0, 0])
self.assertNotEqual(ga.population[1][1].tolist(), [0, 0, 0, 0])
self.assertEqual(ga.population[1][fit_cnt + 1].tolist(), [0, 0, 0, 0])
self.assertEqual(ga.population[1][fit_cnt + 2].tolist(), [0, 0, 0, 0])