class TestAnimals(unittest.TestCase):
def setUp(self):
self.animal = Animal("tiger", 30, "Kumcho", "male", 60)
def test_init(self):
animal = Animal("Tiger", 20, "Dingo", "male", 90)
self.assertEqual(animal.species, "Tiger")
self.assertEqual(animal.age, 20)
self.assertEqual(animal.name, "Dingo")
self.assertEqual(animal.gender, "male")
self.assertEqual(animal.weight, 90)
def test_load_json_invalid_file(self):
with self.assertRaises(FileNotFoundError):
self.animal.load_json(str(uuid4()))
def test_load_json_success(self):
self.animal.load_json("database.json")
self.assertEqual(self.animal.life_expectancy, 20.0)
self.assertEqual(self.animal.food_type, "meat")
def test_load_json_invalid_species(self):
test_animal = Animal(str(uuid4()), 10, "TEST", "male", 15)
with self.assertRaises(ValueError):
test_animal.load_json("database.json")
def test_get_current_animal_year(self):
self.assertEqual(self.animal.get_current_animal_year(), 3)
def test_get_chance_of_dying_valid_life_expectancy(self):
self.animal.load_json("database.json")
self.assertEqual(self.animal.get_chance_of_dying(), 0.15)
def test_grow(self):
self.animal.load_json("database.json")
self.animal.grow()
self.assertEqual(self.animal.age, 31)
self.assertEqual(self.animal.weight, 60.8)
def test_eat(self):
self.animal.load_json("database.json")
self.animal.eat()
self.assertEqual(self.animal.weight, 60.6)
def test_is_alive(self):
self.animal.load_json("database.json")
dead_or_alive = set()
for i in range(100):
status = self.animal.is_alive()
dead_or_alive.add(status)
self.assertEqual(len(dead_or_alive), 2)
def test_remove_dead_animals(self):
bear2 = Animal("bear", 12, "Pesho2", "male", 60)
self.zoo.accommodate_an_animal(bear2)
bear2.is_alive = False
self.zoo.remove_dead_animals()
self.assertEqual(self.zoo.animals, [self.bear, self.panda])
