def test_sort_all_animals_by_fitness(self): """Tests if the sort all animals by fitness method sorts a sample of animals placed on a location by fitness. """ i = Island() loc = (2, 7) for _ in range(10): Herbivore(i, loc) Carnivore(i, loc) local_herb_list = i.get_herb_list_on_loc(loc) local_carn_list = i.get_carn_list_on_loc(loc) local_herb_list.sort(key=lambda herb: herb.fitness, reverse=True) local_carn_list.sort(key=lambda carn: carn.fitness, reverse=True) i.sort_all_animals_by_fitness() assert i.get_herb_list_on_loc(loc) == local_herb_list assert i.get_carn_list_on_loc(loc) == local_carn_list
def test_get_carn_list_on_loc(self): """Tests if get carn list on loc method retrieves carn list attribute of cell on locatrion. """ i = Island() loc = (1, 8) for _ in range(3): Carnivore(i, loc) carn_list_manually = i.island_dict[loc].carn_pop_list assert i.get_carn_list_on_loc(loc) == carn_list_manually