Python 单元测试

本篇为Python的单元测试的方法及示例

概念

结果

示例

概念

用来对一个函数、一个类或者一个模块来进行正确性校验工作

结果

1.单元测试通过，说明我们测试的函数功能正常

2.单元测试不通过，说明函数功能有bug，或者测试条件输入有误

示例

对函数进行测试

创建函数文件

首先编写函数文件，在当前目录下创建单元测试函数文件

示例如下：

python 复制代码

# 两数相加
def mySum(num1, num2):
    return num1 + num2

# 两数相减
def mySub(num1, num2):
    return num1 - num2

创建测试文件

接着在创建myTest文件进行测试，在当前目录下创建myTest文件。

示例如下：

python 复制代码

import unittest
from 单元测试函数文件 import mySum
from 单元测试函数文件 import mySub


class Test(unittest.TestCase):
    def setUp(self):
        print('开始测试时自动调用')

    def tearDown(self):
        print('结束测试时自动调用')

    def test_mySum(self):
        self.assertEqual(mySum(1, 2), 3, '两数相加结果有误')

    def test_mySub(self):
        self.assertEqual(mySub(3, 1), 2, '两数相减结果有误')


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

测试结果

运行测试文件，查看测试结果。

成功

失败

修改函数结果为一个错误返回后，提示如下：

对类进行测试

创建待测试类

创建person类，并设置类方法

示例如下：

python 复制代码

class Person(object):
    def __init__(self, name, age):
        self.name = name
        self.age = age

    def get_age(self):
        return self.age

创建测试文件

在当前目录文件下，创建myTest文件对Person类中的函数进行测试。

示例如下：

python 复制代码

import unittest
from Person import Person

class Test(unittest.TestCase):
    def test_init(self):
        per = Person('hello', 20)
        self.assertEqual(per.name, 'hello', '属性赋值有误')


    def test_get_age(self):
        per = Person('hai', 25)
        self.assertEqual(per.get_age(), per.age, 'get_age函数有误')


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

测试结果与对函数进行单元测试相似，就不在截图示例了。

文档测试

创建函数

引入doctest库，创建两数相加函数。

并对函数增加注释，最后通过文档进行测试结果是否符合。

示例如下：

python 复制代码

import doctest

# doctest模块可以提取注释中的代码执行
# doctest严格按照python交互模式的输入提取

def mySum(x, y):
    '''
    得到x与y两数之和
    :param x: 第一个数
    :param y: 第二个数
    :return: sum

    example:
    >>> print(mySum(1, 2))
    3

    '''

    return x + y

注意：注释代码前需要有一个空格

进行测试

示例如下：

python 复制代码

if __name__=='__main__':
    doctest.testmod(verbose=True)

效果如下：

总结

测试是一种必要的保障项目质量的有效方法。