def test_death_probability(mocker):
"""Test that the death function returns true if the mocked value is lower than the probability
of death. Also the opposite if the mocked value is higher than the probability of death."""
mocker.patch("numpy.random.random", return_value=0)
herb = Herbivore(5, 10)
carn = Carnivore(6, 15)
assert herb.death() is True
assert carn.death() is True
mocker.patch("numpy.random.random", return_value=1)
herb = Herbivore(5, 10)
carn = Carnivore(6, 15)
assert herb.death() is False
assert carn.death() is False
def test_death_weight():
"""Testing that the animal dies if the weight is 0."""
herb = Herbivore(3, 0)
carn = Carnivore(5, 0)
assert herb.death() is True
assert carn.death() is True
