一、概述
本文介绍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
importunittest#被测试方法
defcal(a,b):return a+bclasssingletest(unittest.TestCase):defsetUp(self):print "setUp"
pass
deftearDown(self):print "tearDown"
pass
deftest_01(self):print ("test_01")
expected= 6result= cal(2, 4)
self.assertEqual(expected, result)deftest_02(self):print ("test_02")
expected=0
result= cal(2, 1)
self.assertEqual(expected, result)#执行测试代码
if __name__ == '__main__':
unittest.main()
运行结果如下:
1 .F2 ======================================================================
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, intest_027 setUp8 self.assertEqual(expected, result)9 AssertionError: 0 != 3
10
11 ----------------------------------------------------------------------
12 Ran 2 tests in0.001s13 test_0114
15 tearDown16 FAILED (failures=1)17 setUp18 test_0219 tearDown20
21 Process finished with exit code 1
上面显示出了执行出错的用例,以及出错信息。最后两句给出了整个测试执行所花的时间,以及测试失败的用例数。
如果我们将test_02修改正确,运行结果输出如下:
1 ..2 ----------------------------------------------------------------------
3 Ran 2 tests in0.001s4
5 OK6 setUp7 test_018 tearDown9 setUp10 test_0211 tearDown12
13 Process finished with exit code 0
全部运行通过!