def update_zoos(self): zoos = [] all_zoos = self.get_zoos() for zoo in all_zoos: zoo = Zoo(zoo["name"], zoo["capacity"], zoo["budget"]) for animal in self.fetch_animals_from_zoo(zoo.get_name()): animal = Animal(animal["species"], animal["age"], animal["name"], animal["gender"], animal["weight"]) zoo.add_animal(animal) zoos.append(zoo) return zoos
class ZooTests(unittest.TestCase): def setUp(self): self.zoo = Zoo("Alexandria Zoo", 2, 1000) self.tiger_vitaly = Animal("tiger", 1, "Vitaly", "male", 180) self.lion_alex = Animal("lion", 1, "Alex", "male", 210) self.hippo_gloria = Animal("hippo", 2, "Gloria", "female", 1600) def test_get_zoo_name(self): self.assertEqual("Alexandria Zoo", self.zoo.get_name()) def test_get_zoo_max_capacity(self): self.assertEqual(2, self.zoo.get_max_capacity()) def test_get_zoo_budget(self): self.assertEqual(1000, self.zoo.get_budget()) def test_get_animals(self): self.assertEqual("", self.zoo.get_animals()) def test_add_animal(self): self.assertTrue(self.zoo.add_animal(self.tiger_vitaly)) self.assertEqual("<Vitaly>: <tiger>, <1>, <180>", self.zoo.get_animals()) def test_add_animal_when_no_space(self): self.assertTrue(self.zoo.add_animal(self.tiger_vitaly)) self.assertEqual("<Vitaly>: <tiger>, <1>, <180>", self.zoo.get_animals()) self.assertTrue(self.zoo.add_animal(self.lion_alex)) self.assertEqual("<Vitaly>: <tiger>, <1>, <180>\n<Alex>: <lion>, <1>, <210>", self.zoo.get_animals()) self.assertFalse(self.zoo.add_animal(self.hippo_gloria)) self.assertEqual("<Vitaly>: <tiger>, <1>, <180>\n<Alex>: <lion>, <1>, <210>", self.zoo.get_animals()) def test_remove_animal(self): self.assertTrue(self.zoo.add_animal(self.tiger_vitaly)) self.assertEqual("<Vitaly>: <tiger>, <1>, <180>", self.zoo.get_animals()) self.assertTrue(self.zoo.remove_animal(0)) self.assertEqual("", self.zoo.get_animals()) def test_is_there_space(self): self.assertTrue(self.zoo.is_there_space()) def test_is_there_space_when_no_space(self): self.assertTrue(self.zoo.add_animal("Tiger")) self.assertTrue(self.zoo.add_animal("Lion")) self.assertFalse(self.zoo.is_there_space()) def test_get_daily_expenses_when_empty_zoo(self): self.assertEqual(0, self.zoo.get_daily_expenses()) def test_get_daily_expenses(self): self.assertTrue(self.zoo.add_animal(self.tiger_vitaly)) expected = 4 * (self.tiger_vitaly.get_food_weight_ratio() * self.tiger_vitaly.weight) self.assertEqual(expected, self.zoo.get_daily_expenses())
class ZooTests(unittest.TestCase): def setUp(self): self.tiger = Animal('tiger', 2, 'Pesho', 'male', 100) self.lion = Animal('lion', 5, 'Gosho', 'female', 120) self.goat = Animal('goat', 3, 'Kolio', 'male', 80) self.zoo = Zoo([self.tiger, self.lion, self.goat], 1000, 100000, 'Zoo') def test_zoo_get_animals(self): self.assertEqual([self.tiger, self.lion, self.goat], self.zoo.get_animals()) def test_zoo_get_capacity(self): self.assertEqual(1000, self.zoo.get_capacity()) def test_zoo_get_budget(self): self.assertEqual(100000, self.zoo.get_budget()) def test_zoo_get_name(self): self.assertEqual('Zoo', self.zoo.get_name()) @unittest.skip("Skip this.") def test_zoo_save(self): self.zoo.save() def test_zoo_income(self): self.assertEqual(180, self.zoo.daily_income()) def test_accomodate(self): self.zoo.accomodate(self.tiger) self.zoo.accomodate(self.lion) self.zoo.accomodate(self.goat) def test_daily_outcome(self): self.assertEqual(101, self.zoo.daily_outcome()) def test_simulate_period_of_time(self): self.zoo.simulate_time('months', 10) self.assertEqual(123700, self.zoo.get_budget())
class TestZoo(unittest.TestCase): """docstring for ZooTest""" def setUp(self): self.zoo = Zoo("Sofia", 2, 3000) def test_atributes(self): self.assertEqual("Sofia", self.zoo.get_name()) self.assertEqual(2, self.zoo.get_capacity()) self.assertEqual([], self.zoo.get_animals()) def test_accomodate_animal(self): self.zoo.accommodate_animal("tiger", 18, "Zyblyo", "male", 19) self.assertEqual(1, len(self.zoo.get_animals())) self.assertEqual(True, self.zoo.accommodate_animal("tiger", 18, "Anastasij", "male", 19)) def test_see_animals(self): self.zoo.accommodate_animal("tiger", 18, "Zyblyo", "male", 19) expected = "Zyblyo\t - tiger, age: 18 years, weight: 19 kg" result = self.zoo.see_animals() self.assertEqual(expected, result) def test_remove_animal(self): self.zoo.accommodate_animal("tiger", 18, "Zyblyo", "male", 19) self.zoo.remove_animal("tiger", "Zyblyo") self.assertEqual(0, len(self.zoo.get_animals())) def test_move_animal(self): self.zoo.accommodate_animal("tiger", 18, "Spiridon", "male", 19) self.zoo.move_animal("tiger", "Spiridon") self.assertEqual(0, len(self.zoo.get_animals())) def test_daily_incomes(self): self.zoo.accommodate_animal("tiger", 18, "Tsveta", "female", 19) self.zoo.accommodate_animal("tiger", 18, "Svetla", "female", 19) self.assertEqual(120, self.zoo.daily_incomes()) def test_daily_expenses(self): self.zoo.accommodate_animal("tiger", 18, "Zyblyo", "male", 19) self.assertEqual(19 * 0.06 * 4, self.zoo.daily_expenses()) def test_born_animal(self): self.zoo.accommodate_animal("tiger", 18, "Snejan", "male", 19) self.zoo.accommodate_animal("tiger", 18, "Spiridonka", "female", 19) self.zoo.born_animal("tiger", "Spiridonka") self.assertEqual(3, len(self.zoo.get_animals())) def test_ready_to_give_birth(self): self.zoo.accommodate_animal("tiger", 18, "Snejan", "male", 19) self.zoo.accommodate_animal("tiger", 18, "Spiridonka", "female", 19) self.zoo.born_animal("tiger", "Spiridonka") self.assertFalse(self.zoo.born_animal("tiger", "Spiridonka")) def test_update_animals_from_database(self): self.assertEqual(0, len(self.zoo.get_animals())) self.db = self.zoo.get_database() animal1 = Animal("lion", 24, "Svetla", "female", 150) self.db.insert_animal(animal1) self.zoo.__animals = self.zoo.update_animals_from_database() self.assertEqual(1, len(self.zoo.get_animals())) first_animal = self.zoo.get_animals()[0] self.assertTrue(isinstance(first_animal, Animal)) def tearDown(self): call("rm Sofia.db", shell=True)