def test_zoo_tend_animal_no_budget(self):
z = Zoo("Zoo", 250, 2, 2)
z.add_animal(Lion("John", "m", 2), 100)
z.add_animal(Tiger("Bill", "f", 4), 100)
res = z.tend_animals()
self.assertEqual(
res, "You have no budget to tend the animals. They are unhappy.")
def test_zoo_tend_animal_success(self):
z = Zoo("Zoo", 500, 2, 2)
z.add_animal(Lion("John", "m", 2), 100)
z.add_animal(Tiger("Bill", "f", 4), 100)
res = z.tend_animals()
self.assertEqual(z._Zoo__budget, 205)
self.assertEqual(
res,
"You tended all the animals. They are happy. Budget left: 205")
Caretaker("Bill", 21, 68),
Caretaker("Marie", 32, 105),
Caretaker("Stacy", 35, 140),
Vet("Peter", 40, 300),
Vet("Kasey", 37, 280),
Vet("Sam", 29, 220)
]
# Adding all animals
for i in range(len(animals)):
animal = animals[i]
price = prices[i]
print(zoo.add_animal(animal, price))
# Adding all workers
for worker in workers:
print(zoo.hire_worker(worker))
# Tending animals
print(zoo.tend_animals())
# Paying keepers
print(zoo.pay_workers())
# Fireing worker
print(zoo.fire_worker("Adam"))
# # Printing statuses
print(zoo.animals_status())
print(zoo.workers_status())
