def test_regrowth_to_maximum(self, example_pop_herb):
"""
Asserts that amount of fodder is equal to f_max after each year.
"""
landscape = Landscape(example_pop_herb)
landscape.regrowth()
assert landscape.fodder_amount == landscape.params["f_max"]
def test_have_all_herbivores_been_fed(self):
"""
Tests feed_all_herbivores method.
Tests that all herbivores have gained weight in a situation where there
is plenty of food available.
"""
test_population_feed = [
{
"species": "Herbivore",
"age": 3,
"weight": 20.0
},
{
"species": "Herbivore",
"age": 3,
"weight": 20.0
},
{
"species": "Herbivore",
"age": 3,
"weight": 20.0
},
]
landscape = Landscape(test_population_feed)
landscape.feed_all_herbivores()
assert landscape.pop_herb[0].weight > test_population_feed[0]["weight"]
assert landscape.pop_herb[1].weight > test_population_feed[1]["weight"]
assert landscape.pop_herb[2].weight > test_population_feed[2]["weight"]
def teardown_feeding_test(self):
"""
Sets up parameters for test_fittest_animal_eats_first method.
After the test is run, we reset the parameters to their initial values.
"""
yield None
Landscape.reset_params()
def test_sort_several_herbivores_by_fitness(self, example_pop_herb):
"""
Tests that sort_herb_population_by_fitness works on a list of several
herbivores.
"""
landscape = Landscape(example_pop_herb)
landscape.sort_herb_population_by_fitness()
assert landscape.pop_herb[0].fitness > landscape.pop_herb[1].fitness
assert landscape.pop_herb[1].fitness > landscape.pop_herb[2].fitness
def test_sort_several_carnivores_by_fitness(self, example_pop_carn):
"""
Tests that sort_carn_population_by_fitness works on a list of several
carnivores.
"""
landscape = Landscape(example_pop_carn)
landscape.sort_carn_population_by_fitness()
assert landscape.pop_carn[0].fitness > landscape.pop_carn[1].fitness
assert landscape.pop_carn[1].fitness > landscape.pop_carn[2].fitness
def test_have_all_animals_lost_weight(self, example_pop_herb):
"""
Test for make_all_animals_lose_weight method.
Test that all animals in the population has lost some weight.
"""
landscape = Landscape(example_pop_herb)
landscape.make_all_animals_lose_weight()
assert landscape.pop_herb[0].weight < example_pop_herb[0]["weight"]
assert landscape.pop_herb[1].weight < example_pop_herb[1]["weight"]
assert landscape.pop_herb[2].weight < example_pop_herb[2]["weight"]
def test_correct_fodder_amount_for_carn(self, example_pop_herb,
example_pop_carn):
"""
Tests available_fodder_carnivore method.
Asserts that the fodder amount available for a carnivore is calculated
correctly.
"""
total_pop = example_pop_herb + example_pop_carn
landscape = Landscape(total_pop)
assert landscape.available_fodder_carnivore() == 80
def test_have_all_animals_aged(self, example_pop_herb):
"""
Tests make_all_animals_older method.
Test that all animals in the population has aged by one year.
"""
landscape = Landscape(example_pop_herb)
landscape.make_all_animals_older()
assert landscape.pop_herb[0].age == example_pop_herb[0]["age"] + 1
assert landscape.pop_herb[1].age == example_pop_herb[1]["age"] + 1
assert landscape.pop_herb[2].age == example_pop_herb[2]["age"] + 1
def test_amount_returned_when_no_fodder_available_to_herb(
self, example_pop_herb):
"""
Tests available_fodder_herbivore method.
Asserts that no fodder is provided to the herbivore when there is no
fodder available.
"""
landscape = Landscape(example_pop_herb)
landscape.fodder_amount = 0
assert landscape.available_fodder_herbivore() == 0
assert landscape.fodder_amount == 0
def test_eaten_herbs_have_been_removed(self, example_pop_herb):
"""
Tests remove_all_eaten_herbivores method.
Asserts that eaten herbivores are removed from pop_herb in landscape
cell.
:param example_pop_herb: list
List of herbivores
"""
landscape = Landscape(example_pop_herb)
landscape.remove_all_eaten_herbivores(landscape.pop_herb)
assert len(landscape.pop_herb) == 0
def test_sort_single_herbivore_by_fitness(self):
"""
Tests that sort_herb_population_by_fitness
works for a single-element list.
"""
landscape = Landscape([{
"species": "Herbivore",
"age": 1,
"weight": 10.0
}])
landscape.sort_herb_population_by_fitness()
assert landscape.pop_herb[0].fitness == approx(0.49979521641750696)
def test_fittest_animal_eats_first(self, teardown_feeding_test,
example_pop_herb):
"""
Tests feed_all_herbivores.
Asserts that the strongest animal has eaten first, in a situation
where there is a limited supply of food.
"""
landscape = Landscape(example_pop_herb)
landscape.params["f_max"] = Herbivore.params["F"]
landscape.feed_all_herbivores()
assert landscape.pop_herb[0].weight > example_pop_herb[1]["weight"]
assert landscape.pop_herb[1].weight == example_pop_herb[2]["weight"]
assert landscape.pop_herb[2].weight == example_pop_herb[0]["weight"]
def test_has_dead_animal_been_removed(self, example_pop_herb):
"""
Test for remove_all_dead_animals method.
Tests that a dead animal has been removed from population list because
it has fitness equal to zero, due to no weight.
"""
numpy.random.seed(1)
landscape = Landscape(example_pop_herb)
landscape.pop_herb[2].weight = 0
for animal in landscape.pop_herb:
animal.find_fitness()
landscape.remove_all_dead_animals()
assert len(landscape.pop_herb) == 2
def test_have_carnivore_been_fed(self, example_pop_herb, example_pop_carn,
mocker):
"""
Tests feed_all_carnivores method.
Assert that a carnivore gains weight after eating, and that the eaten
herbivore is removed from population.
"""
mocker.patch('numpy.random.random', return_value=0.00001)
# Adds one strong carnivore and one weak herbivore to population.
landscape = Landscape([example_pop_herb[0]] + [example_pop_carn[1]])
old_weight = landscape.pop_carn[0].weight
landscape.feed_all_carnivores()
assert old_weight < landscape.pop_carn[0].weight
assert len(landscape.pop_herb) == 0
def test_amount_returned_when_restricted_fodder_is_available_to_herb(
self, example_pop_herb):
"""
Tests available_fodder_herbivore method.
Asserts that all of the fodder in the cell is provided to a herbivore
when it's demand is larger than the amount of fodder available.
Also asserts that the amount of fodder left now is zero.
"""
landscape = Landscape(example_pop_herb)
landscape.fodder_amount = Herbivore.params["F"] / 2
old_fodder_amount = landscape.fodder_amount
available_fodder_to_herb = landscape.available_fodder_herbivore()
assert available_fodder_to_herb == Herbivore.params["F"] / 2
assert landscape.fodder_amount == old_fodder_amount - \
available_fodder_to_herb
def test_amount_returned_when_enough_fodder_is_available_to_herb(
self, example_pop_herb):
"""
Tests available_fodder_herbivore method.
Asserts that a herbivore is provided with the amount it desires, when
enough fodder is available.
Also asserts that the amount of fodder
provided is subtracted from the initial fodder amount.
"""
landscape = Landscape(example_pop_herb)
landscape.regrowth()
old_fodder_amount = landscape.fodder_amount
available_fodder_to_herb = landscape.available_fodder_herbivore()
assert available_fodder_to_herb == Herbivore.params["F"]
assert landscape.fodder_amount == old_fodder_amount - \
available_fodder_to_herb
def test_constructor_landscape(self, example_pop_herb):
"""
Asserts that the Landscape class enables creation of class
instances and it's population attribute has correct length.
"""
landscape = Landscape(example_pop_herb)
assert isinstance(landscape, Landscape)
assert len(landscape.pop_herb) == len(example_pop_herb)
def test_newborn_animals_have_been_created(self):
"""
Tests add_newborn_animals method.
Asserts that in a situation when all animals will give birth,
the population list has been updated with the right number of animals.
"""
numpy.random.seed(1)
test_population_birth = [{
"species": "Herbivore",
"age": 2,
"weight": 70.0
}, {
"species": "Herbivore",
"age": 3,
"weight": 90.0
}, {
"species": "Herbivore",
"age": 3,
"weight": 70.0
}, {
"species": "Herbivore",
"age": 3,
"weight": 80.0
}, {
"species": "Herbivore",
"age": 3,
"weight": 60.0
}, {
"species": "Herbivore",
"age": 5,
"weight": 90.0
}]
landscape = Landscape(test_population_birth)
for animal in landscape.pop_herb:
animal.find_fitness()
landscape.add_newborn_animals()
assert len(landscape.pop_herb) == 2 * len(test_population_birth)