File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1111
1212from unittest .mock import patch
1313
14+
1415class TestPetShop (unittest .TestCase ):
1516
1617 def test_dog_pet_shop_shall_show_dog_instance (self ):
@@ -31,6 +32,33 @@ def test_cat_pet_shop_shall_show_cat_instance(self):
3132 self .assertEqual (mock_f_get_pet .call_count , 1 )
3233 self .assertEqual (mock_f_get_food .call_count , 1 )
3334
35+
36+ class TestCat (unittest .TestCase ):
37+
38+ @classmethod
39+ def setUpClass (cls ):
40+ cls .c = Cat ()
41+
42+ def test_cat_shall_meow (cls ):
43+ cls .assertEqual (cls .c .speak (), 'meow' )
44+
45+ def test_cat_shall_be_printable (cls ):
46+ cls .assertEqual (str (cls .c ), 'Cat' )
47+
48+
49+ class TestDog (unittest .TestCase ):
50+
51+ @classmethod
52+ def setUpClass (cls ):
53+ cls .d = Dog ()
54+
55+ def test_dog_shall_woof (cls ):
56+ cls .assertEqual (cls .d .speak (), 'woof' )
57+
58+ def test_dog_shall_be_printable (cls ):
59+ cls .assertEqual (str (cls .d ), 'Dog' )
60+
61+
3462if __name__ == "__main__" :
3563 unittest .main ()
3664
You can’t perform that action at this time.
0 commit comments