编写一个简单的单元测试用例

开发一个简单的计算器，用于计算两个数的加减乘除，示例：

 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

module

使用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)

test