Python单元测试框架unittest之单用例管理（一）

一、概述

本文介绍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

全部运行通过！