简单使用Python Unittest写单元测试用例

简单使用Python Unittest写单元测试用例

作为开发人员，对写的代码进行单元测试是最基本的

对python语言来讲，其单元测试的利器是pyUnit， 这个从Python2.1之后的版本中，就已经是自带的了。

下面通过一个基本的例子来讲述如何来使用unittest.

首先是一个简单的功能。

area.py

1. class Area:  
2.     def __init__(self,width=100,length=100):  
3.         self._width = width  
4.         self._length = length  
5.       
6.     def getWidth(self):  
7.         return self._width  
8.   
9.     def getLength(self):  
10.         return self._length  
11.   
12.     def getArea(self):  
13.         return self._width*self._length  
14.   
15.     def setWidth(self,width):  
16.         if width <= 0:  
17.             raise ValueError,"Illeage Width value"  
18.         self._width = width  
19.   
20.     def setLenth(self,length):  
21.         if length <= 0:  
22.             raise ValueError,"Illeage Length value"  
23.         self._length = length   

ut_area.py

1. from area import Area  
2. import unittest  
3.   
4. class AreaTestCase(unittest.TestCase):  
5.     def setUp(self):  
6.         self.area = Area()  
7.       
8.     def tearDown(self):  
9.         self.area = None  
10.       
11.     def testArea(self):  
12.         self.assertEqual(self.area.getArea(),10000)  
13.       
14.     def testWidth(self):  
15.         self.area.setWidth(10)  
16.         self.assertEqual(self.area.getWidth(),10)  
17.     def testLength(self):  
18.         self.area.setLenth(10)  
19.         self.assertEqual(self.area.getLength(),10)  
20.   
21. if __name__ == "__main__":  
22.     unittest.main()  

我们来看一下，单元测试的结果。

 

单元测试也可以以如下的方法来实现 

首先我们的测试类要继承于unittest.TestCase. 如果采用动态测试的方法可以为每个需要测试的方法编写测试方法，使用assertEqual( , ). 然后把我们的测试都放到unittest.TestSuite()容器中，最后使用 unittest.TextTestRunner().run(suite)方法自动测试。

1. suite = unittest.TestSuite()  
2.     suite.addTest(AreaTestCase("testArea"))  
3.     suite.addTest(AreaTestCase("testWidth"))  
4.     suite.addTest(AreaTestCase("testLength"))  
5.   
6.   
7.     runner = unittest.TextTestRunner()  
8.     runner.run(suite)  
9.     print "#"*80  
10.     unittest.main()  

我们把两种方法进行了对比，结果是一样的，

 

对TestCase,有以下的一些函数可以来使用

 

官方资料：http://docs.python.org/library/unittest.html