unittest是python单元测试框架,又叫做PyUnit。

之所以称为框架是它代替开发人员完成了一些调用、IO等与单元测试无直接关系的支撑代码,让开发人员可以专注与测试用例的编写,简化单元测试工作。

 

单元测试是一种基本的,由开发人员(而不是测试人员)完成的测试,保证一个程序基本单元的正确性。单元的概念我个人理解就类似电子设备中的元器件,一个个元器件(单元)组成了整个电子设备(程序)。而元器件的功能是单一的、确定的,可以在电子设备未完全成型(程序未完全实现)之前对其正确性进行测试(即单元测试)。同时,这些元器件是独立的,可以方便的更换(高内聚,低耦合,可以任意重构)。所以单元测试是测试驱动开发、极限编程、敏捷中的重要概念。

 

一个简单的例子:

#FileName: ut_target.py
class EqualToZero(Exception): pass
class SplitZero(object):
    def splitzero(self, num):
        if num > 0:
            return "num is bigger than zero"
        elif num < 0:
            return "num is smaller than zero"
        else:
            raise EqualToZero

 

 

#FileName: utest.py
from ut_target import SplitZero, EqualToZero
import unittest
 
class SzTestCase(unittest.TestCase):
    def setUp(self):
        print "test start"
    def tearDown(self):
        print "test stop"
    def testSzBig(self):
        num = 10
        sz = SplitZero()
        self.assertEqual(sz.splitzero(num),
                         "num is bigger than zero")
    def testSzSmall(self):
        num = -10
        sz = SplitZero()
        self.assertEqual(sz.splitzero(num),
                         "num is smaller than zero")
    def testSzEqual(self):
        num = 0
        sz = SplitZero()
        self.assertRaises(EqualToZero, sz.splitzero, num)
 
if __name__ == "__main__":
    unittest.main()

 

执行python utest.py -v

testSzBig (__main__.SzTestCase) ... test start
test stop
ok
testSzEqual (__main__.SzTestCase) ... test start
test stop
ok
testSzSmall (__main__.SzTestCase) ... test start
test stop
ok
 
----------------------------------------------------------------------
Ran 3 tests in 0.000s
 
OK

 

unittest基本使用方法

1.import unittest

2.定义一个继承自unittest.TestCase的测试用例类

3.定义setUp和tearDown,在每个测试用例前后做一些辅助工作。

4.定义测试用例,名字以test开头。

5.一个测试用例应该只测试一个方面,测试目的和测试内容应很明确。主要是调用assertEqual、assertRaises等断言方法判断程序执行结果和预期值是否相符。

6.调用unittest.main()启动测试

7.如果测试未通过,会输出相应的错误提示。如果测试全部通过则不显示任何东西,这时可以添加-v参数显示详细信息。

 

参考

[1]http://docs.python.org/release/2.6.8/library/unittest.html

[2]http://www.ibm.com/developerworks/cn/linux/l-pyunit/index.html