def test_zoo_add_animal_success(self): z = Zoo("My Zoo", 1500, 6, 10) res = z.add_animal(Lion("Neo", "Male", 2), 1000) self.assertEqual(res, "Neo the Lion added to the zoo") self.assertEqual(len(z.animals), 1) self.assertEqual(z._Zoo__budget, 500)
def test_zoo_add_animal_no_budget(self): z = Zoo("My Zoo", 500, 6, 10) res = z.add_animal(Lion("Neo", "Male", 2), 1000) self.assertEqual(res, "Not enough budget") self.assertEqual(len(z.animals), 0) self.assertEqual(z._Zoo__budget, 500)
def test_zoo_fire_worker_unsuccessful(self): z = Zoo("Zoo", 1500, 1, 1) res = z.fire_worker("K") self.assertEqual(res, "There is no K in the zoo") self.assertEqual(z.workers, [])
def test_zoo_profit(self): z = Zoo("Mine", 250, 2, 2) z.profit(250) self.assertEqual(z._Zoo__budget, 500)
def test_zoo_fire_worker_success(self): z = Zoo("Zoo", 1500, 1, 1) z.hire_worker(Keeper("K", 45, 100)) res = z.fire_worker("K") self.assertEqual(res, "K fired successfully") self.assertEqual(z.workers, [])
def test_zoo_hire_worker_no_space(self): z = Zoo("Some Zoo", 1500, 1, 0) res = z.hire_worker(Vet("I am Vet", 20, 500)) self.assertEqual(res, "Not enough space for worker") self.assertEqual(len(z.workers), 0) self.assertEqual(z._Zoo__workers_capacity, 0)
def test_zoo_hire_worker_success(self): z = Zoo("Some Zoo", 1500, 1, 1) res = z.hire_worker(Vet("I am Vet", 20, 500)) self.assertEqual(res, "I am Vet the Vet hired successfully") self.assertEqual(len(z.workers), 1) self.assertEqual(z._Zoo__workers_capacity, 1)
def test_zoo_add_animal_no_space(self): z = Zoo("My Zoo", 1500, 0, 10) res = z.add_animal(Lion("Neo", "Male", 2), 1000) self.assertEqual(res, "Not enough space for animal") self.assertEqual(len(z.animals), 0) self.assertEqual(z._Zoo__budget, 1500)
from project.lion import Lion from project.tiger import Tiger from project.cheetah import Cheetah from project.keeper import Keeper from project.caretaker import Caretaker from project.vet import Vet from project.zoo import Zoo zoo = Zoo("Zootopia", 3000, 5, 8) # Animals creation animals = [ Cheetah("Cheeto", "Male", 2), Cheetah("Cheetia", "Female", 1), Lion("Simba", "Male", 4), Tiger("Zuba", "Male", 3), Tiger("Tigeria", "Female", 1), Lion("Nala", "Female", 4) ] # Animal prices prices = [200, 190, 204, 156, 211, 140] # Workers creation workers = [ Keeper("John", 26, 100), Keeper("Adam", 29, 80), Keeper("Anna", 31, 95), Caretaker("Bill", 21, 68), Caretaker("Marie", 32, 105), Caretaker("Stacy", 35, 140),