def test_requirements(self):
""" Creates zoo and cages: """
zoo = Zoo()
cage_1 = Cage()
cage_2 = Cage()
""" Creation of animals: """
mouse = Mouse()
wildcat = WildCat()
lion = Lion()
""" Attempt to add objects that are not cages to a zoo """
zoo.add_cages([mouse])
self.assertFalse(zoo.cages, [mouse])
""" Tests zoo with empty cages """
self.assertEqual(cage_1.n_of_animals(), 0)
zoo.add_cages([cage_1, cage_2])
self.assertEqual(zoo.cages, [cage_1, cage_2])
""" Multiple addition of animals among repeated ones """
cage_1.add_animals([mouse, mouse, lion, lion, mouse])
# Set up to test more than one duplicated animal and different orders
self.assertEqual(cage_1.animals_list, [mouse, lion])
""" Tests zoo attributes """
zoo.cages = []
zoo.add_cages([cage_1, cage_2])
self.assertEqual(zoo.cages, [cage_1, cage_2])
self.assertEqual(zoo.n_of_cages(), 2)
self.assertEqual(zoo.n_of_animals(), 2)
""" Tests competition between prey and predator """
cage_2.add_animals([mouse, wildcat, lion])
self.assertEqual(cage_2.animals_list, [lion])
""" Tests if animal already in another cage """
# ----- Should I do it?
""" Attempt to add objects that are not animals to a cage """
cage_1.animals_list = []
cage_1.add_animals([zoo])
self.assertFalse(cage_1.animals_list, [zoo])
