예제 #1
0
파일: test_zoo.py 프로젝트: mimipaskova/Zoo
class testZoo(unittest.TestCase):

    def setUp(self):
        self.testAnim = Animal("zvqr", 10, 'gosh', 'male', 100)
        self.testZoo = Zoo(15, 12000)

    def test_get_animal_no_animals(self):
        self.assertEqual(0, self.testZoo.get_animal_count())

    def test_add_animal_with_count(self):
        self.testZoo.add_animal(self.testAnim)
        self.assertEqual(1, self.testZoo.get_animal_count())

    def test_remove_animal(self):
        self.testZoo.add_animal(self.testAnim)
        self.testZoo.remove_animal(self.testAnim)
        self.assertEqual(0, self.testZoo.get_animal_count())

    def test_income(self):
        self.testZoo.add_animal(self.testAnim)
        # Add some animals
        self.testAnimTosh = Animal("zver", 10, 'tosh', 'female', 120)
        self.testAnimTesi = Animal("cver", 10, 'tesi', 'male', 20)

        self.testZoo.add_animal(self.testAnimTesi)
        self.testZoo.add_animal(self.testAnimTosh)

        self.assertEqual(180, self.testZoo.income())
예제 #2
0
파일: test_zoo.py 프로젝트: danailbd/Zoo
class testZoo(unittest.TestCase):

    def setUp(self):
        self.testZoo = Zoo(15, 12000)

    def test_get_animal_no_animals(self):
        self.assertEqual(0, self.testZoo.get_animal_count())