zoo-project/test_zoo.py at master · phxcnk/zoo-project · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
import unittest
from zoo import Zoo

class TestZoo(unittest.TestCase):
    def setUp(self):
        self.zoo = Zoo()

    def test_ticket_price_child(self):
        self.assertEqual(self.zoo.get_ticket_price(10), 50)

    def test_ticket_price_teen(self):
        self.assertEqual(self.zoo.get_ticket_price(15), 100)

    def test_ticket_price_adult(self):
        self.assertEqual(self.zoo.get_ticket_price(30), 150)

    def test_ticket_price_senior(self):
        self.assertEqual(self.zoo.get_ticket_price(70), 100)

    def test_invalid_age(self):
        with self.assertRaises(ValueError):
            self.zoo.get_ticket_price(-1)

if __name__ == '__main__':
    unittest.main()