python自动单元测试实例

Python unittest模块的相关用法

 

待测源码类：widget.py

[python] view plain copy print ?

1. #!/usr/bin/env python  
2.   
3. class Widget:  
4.     def __init__(self, size = (40, 40)):  
5.         self._size = size  
6.   
7.     def getSize(self):  
8.         return self._size  
9.   
10.     def resize(self, width, height):  
11.         if width == 0  or height < 0:  
12.             raise ValueError, "illegal size"  
13.         self._size = (width, height)  
14.   
15.     def dispose(self):  
16.         pass  

#!/usr/bin/env python class Widget: def __init__(self, size = (40, 40)): self._size = size def getSize(self): return self._size def resize(self, width, height): if width == 0 or height < 0: raise ValueError, "illegal size" self._size = (width, height) def dispose(self): pass

测试代码类：widget_unittest.py

[python] view plain copy print ?

1. #!/usr/bin/env python  
2.   
3. import sys  
4. import unittest  
5. from widget import Widget  
6.   
7. class WidgetTestCase(unittest.TestCase):  
8.     def setUp(self):  
9.         self.widget = Widget()          
10.       
11.     def tearDown(self):  
12.         self.widget = None  
13.       
14.     def testSize(self):  
15.         self.assertEqual(self.widget.getSize(), (40, 40))  
16.       
17.     def testReSize(self):  
18.         self.widget.resize(20,30)  
19.         self.assertEqual(self.widget.getSize(), (20, 30))  
20.   
21. def suite():  
22.     suite = unittest.TestSuite()  
23.     suite.addTest(WidgetTestCase("testSize"))  
24.     suite.addTest(WidgetTestCase("testReSize"))  
25.     return suite  
26.   
27. if __name__ == "__main__":      
28.     unittest.TextTestRunner().run(suite())      
29.     # If you use IDE to execute test script in an external console,  
30.     # uncomment following code line.  
31.     # while True  
32.     #     if raw_input("Type y or Y to close!") in ("y","Y"): sys.exit(0)  

#!/usr/bin/env python import sys import unittest from widget import Widget class WidgetTestCase(unittest.TestCase): def setUp(self): self.widget = Widget() def tearDown(self): self.widget = None def testSize(self): self.assertEqual(self.widget.getSize(), (40, 40)) def testReSize(self): self.widget.resize(20,30) self.assertEqual(self.widget.getSize(), (20, 30)) def suite(): suite = unittest.TestSuite() suite.addTest(WidgetTestCase("testSize")) suite.addTest(WidgetTestCase("testReSize")) return suite if __name__ == "__main__": unittest.TextTestRunner().run(suite()) # If you use IDE to execute test script in an external console, # uncomment following code line. # while True # if raw_input("Type y or Y to close!") in ("y","Y"): sys.exit(0)

运行结果：python widget_unittest.py

[work@db-testing-com06-vm3.db01.baidu.com test]$ python widget_unittest.py
..
----------------------------------------------------------------------
Ran 2 tests in 0.000s

OK

 

动态方法不再覆盖runTest()方法，而是为测试类编写多个测试方法（按习惯这些方法通常以test开头），在创建TestCase子类的实例时必须给出测试方法的名称，来为PyUnit测试框架指明运行该测试用例时究竟应该调用测试类中的哪个方法.

如果用于测试的类中所有的测试方法都以test开，Python程序员甚至可以用PyUnit模块提供的makeSuite()方法来构造一个TestSuite：

def suite():
    return unittest.makeSuite(WidgetTestCase, "test")