导航

转载:Python自动单元测试实例

Posted on 2014-03-03 23:34  littledrop  阅读(180)  评论(0编辑  收藏  举报

 

转自:http://blog.csdn.net/ithomer/article/details/5969842

Python自动单元测试实例

分类: Linux/Shell Script 3886人阅读 评论(0) 收藏 举报

Python unittest模块的相关用法

 

待测源码类:widget.py

  1. #!/usr/bin/env python  
  2.   
  3. class Widget:  
  4.     def __init__(self, size = (4040)):  
  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  

测试代码类:widget_unittest.py

  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(), (4040))  
  16.       
  17.     def testReSize(self):  
  18.         self.widget.resize(20,30)  
  19.         self.assertEqual(self.widget.getSize(), (2030))  
  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)  

运行结果: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")