開發一個簡單的計算器，用於計算兩個數的加減乘除，示例：

 1 class Calculator():
 2     ‘‘‘實現簡單的加減乘除‘‘‘
 3     def __init__(self,a,b):
 4         self.a = int(a)
 5         self.b = int(b)
 6 
 7     def add(self):
 8         return self.a + self.b
 9 
10     def sub(self):
11         return self.a - self.b
12 
13     def mul(self):
 
14         return self.a * self.b
15 
16     def div(self):
17         return self.a / self.b

使用unittest單元測試框架對Calculator累的方法進行測試。

 1 from module import Calculator
 2 import unittest
 3 
 4 class ModuleTest(unittest.TestCase):
 5 
 6     def setUp(self):
 7         self.cal = Calculator(8,4)
 
 8 
 9     def tearDown(self):
10         pass
11 
12     def test_add(self):
13         result = self.cal.add()
14         self.assertEqual(result,12)
15 
16     def test_sub(self):
17         result = self.cal.sub()
18         self.assertEqual(result,4)
19 
20     def test_mul(self):
21         result = self.cal.mul()
 
22         self.assertEqual(result,32)
23 
24     def test_div(self):
25         result = self.cal.div()
26         self.assertEqual(result,2)
27 
28 if __name__ == "__main__":
29     #構造測試集
30     suite = unittest.TestSuite()
31     suite.addTest(ModuleTest("test_add"))
32     suite.addTest(ModuleTest("test_sub"))
33     suite.addTest(ModuleTest("test_mul"))
34     suite.addTest(ModuleTest("test_div"))
35     #執行測試
36     runer = unittest.TextTestRunner()
37     runer.run(suite)

編寫一個簡單的單元測試用例