Python单元测试浅析

python单元测试浅析

作者：CC-ROYAL

1.单元测试作用

单元测试是用来对一个模块，一个函数或者一个类进行正确性检测的测试工作。

比如对函数 abs()，我们可以编写出以下几个测试用例：

正数输入：比如 1，1.2，0.8等测试用例，期待返回值与输入值相同

负数输入：比如-1，-1.2，-0.8等测试用例，期待返回值与输入值相反

0输入：期待输入0

非数值类型：比如 None，[ ]，{ }等，期待输出错误 TypeError

把上面的例子输入到一个模块中，就是一个简单的、完整的单元测试用例。

单元测试的作用在于：单元测试模块编写好，并且测试通过后，如果对abs()进行修改，则可直接复用原有的单元测试模块，对修改后的函数进行测试。如果测试不通过，则需要考虑修改测试用例，或者对函数修改进行正确性检测。

2.python单元测试

python中单元测试的模块有很多种，例如 unittest、等

以 unittest 模块单元测试为例，在 python 中编写单元测试，需要引入 python 自带的 unittest 模块。

以 my.py 和 my_unittest.py 为例来简单解释 abs() 函数的python的单元测试：

my.py代码如下

#!/usr/bin/env python
#encoding: utf-8

class myclass:
def __init__(self):
pass

def sum(self, x, y):
return x+y

def sub(self, x, y):
return x-y

my_unittest.py代码如下

#!/usr/bin/env python
#encoding: utf-8

import unittest
import myclass

class mytest(unittest.TestCase):

##初始化工作
def setUp(self):
self.tclass = myclass.myclass()   ##实例化了被测试模块中的类

#退出清理工作
def tearDown(self):
pass

#具体的测试用例，一定要以test开头
def testsum(self):
self.assertEqual(self.tclass.sum(1, 2), 3)

if __name__ =='__main__':
unittest.main()

my.py 中定义了一个 myclass 类，以及类的初始化， sum 和 sub 方法。my_unittest.py 则是对 myclass 的单元测试模块，定义了一个单元测试用例 testsum 以及 setUp 方法（每一个单元测试用例执行前运行）、tearDown 方法（每一个单元测试执行完成时运行）。unittest.main() 则是单元测试模块运行时的入口。应注意以下要点。

要执行的单元测试用例必须以 test 开头。

单元测试用例同一的初始化和析构方法可以在 setUp 和 tearDown 方法中定义。

单元测试模块运行：python3 -m unittest my_unittest.py

执行结果如下

----------------------------------------------------------------------
Ran 1 test in 0.000s

OK

执行结果中，说明了执行测试用例的数量、时间和结果。如果测试用例没通过，则提示相应的错误。