一、概述
本文介绍python的单元测试框架unittest，unittest原名为PyUnit，是由java的JUnit衍生而来，这是Python自带的标准模块unittest。unittest是基于java中的流行单元测试框架junit设计的，其功能强大且灵活，对于熟悉junit的人来说掌握unittest很简单。 unittest涉及的知识点较多，但核心的就那一些，本文只介绍最核心和基础的内容。
类似junit，使用unittest编写python的单元测试代码，包括如下几个步骤：
1）编写一个python类，继承unittest模块中的TestCase类，这就是一个测试类；
2）在上面编写的测试类中定义测试方法（这个就是指的测试用例），每个方法的方法名要求以 test 打头，没有额外的参数。 在该测试方法中调用被测试代码，校验测试结果，TestCase类中提供了很多标准的校验方法，如最常见的assertEqual；
3）执行 unittest.main() ，该函数会负责运行测试，它会实例化所有TestCase的子类，并运行其中所有以test打头的方法。
二、案例
我们下面看一个例子，编写如下的python文件，为了简单，我们将被测函数与测试代码放在一个文件中了：
# coding:utf-8 import unittest #被测试方法 def cal(a,b): return a+b class singletest(unittest.TestCase): def setUp(self): print "setUp" pass def tearDown(self): print "tearDown" pass def test_01(self): print ("test_01") expected = 6 result = cal(2, 4) self.assertEqual(expected, result) def test_02(self): print ("test_02") expected = 0 result = cal(2, 1) self.assertEqual(expected, result) #执行测试代码 if __name__ == '__main__': unittest.main()
运行结果如下：
1 .F 2 ====================================================================== 3 FAIL: test_02 (__main__.singletest) 4 ---------------------------------------------------------------------- 5 Traceback (most recent call last): 6 File "C:/Users/zhangxiaoding/PycharmProjects/Mytest/singletest/testcase.py", line 32, in test_02 7 setUp 8 self.assertEqual(expected, result) 9 AssertionError: 0 != 3 10 11 ---------------------------------------------------------------------- 12 Ran 2 tests in 0.001s 13 test_01 14 15 tearDown 16 FAILED (failures=1) 17 setUp 18 test_02 19 tearDown 20 21 Process finished with exit code 1
上面显示出了执行出错的用例，以及出错信息。最后两句给出了整个测试执行所花的时间，以及测试失败的用例数。
如果我们将test_02修改正确，运行结果输出如下：
1 .. 2 ---------------------------------------------------------------------- 3 Ran 2 tests in 0.001s 4 5 OK 6 setUp 7 test_01 8 tearDown 9 setUp 10 test_02 11 tearDown 12 13 Process finished with exit code 0
全部运行通过！