关于unittest单元测试框架中常用的几种用例加载方法

　　unittest模块是python自带的一个单元测试模块，我们可以用来做单元测试。unittest模块包含了如下几个子模块：

• 测试用例：testcase
• 测试集：testsuite
• 加载用例：testloader
• 执行用例：texttestrunner

　　首先编写一个简单的加减乘除数学方法类：

class mathcalculate:
    '''
    加减乘除的计算类
    '''
    def __init__(self, first_num, second_num):
        self.first_num = first_num
        self.second_num = second_num

    def add(self):
        '''
        加法
        :return:
        '''
        return self.first_num + self.second_num

    def subtract(self):
        '''
        减法
        :return:
        '''
        return self.first_num - self.second_num

    def multiply(self):
        '''
        乘法
        :return:
        '''
        return self.first_num * self.second_num

    def divide(self):
        '''
        除法
        :return:
        '''
        try:
            return round(self.first_num / self.second_num, 2)
        except zerodivisionerror:
            return '∞'

　　接下来我们利用unittest中的testcase类来编写测试用例，编写之前，要先知道testcase类以及类方法的用法：

1. 要引用testcase类，需要引入unittest模块
2. 引用testcase是编写一个测试类，所以要新建一个类
3. 测试类里面的方法名都是以test_开头，这样unittest才能识别到用例并去执行

　　新建一个模块，编写一个简单的测试用例类：

import unittest
from python_for_unittest_testsuite.mathcalculate import mathcalculate as mc


class testmathcalculate(unittest.testcase):
    '''
    测试加减乘除类
    '''

    def test_add(self):
        actual = mc(1, 1).add()
        self.assertequal(2, actual, msg='两数相加测试失败')

    def test_subtract(self):
        actual = mc(1, 1).subtract()
        self.assertequal(0, actual, msg='两数相减测试失败')

    def test_multiply(self):
        actual = mc(1, 1).multiply()
        self.assertequal(1, actual, msg='两数相乘测试失败')

    def test_divide(self):
        actual = mc(1, 1).divide()
        self.assertequal(1, actual, msg='两数相除测试失败')


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

　　我们运行代码之后，得到如下结果：

 　　但是这样远远达不到我们的要求，还需要继续优化，接下来进入正题......

　　上面我们看到的是利用unittest.main()执行所有的用例，但是有的时候我们不需要执行所有的测试用例，并且可以按照自己的需求去添加用例，这时我们就要使用testsuite来帮助我们更好地去实现单元测试

　　方法一：加载测试用例，直接使用addtest()方法添加测试用例，代码示范如下：

import unittest
from python_for_unittest_testsuite.test_math_calculate import testmathcalculate

suite = unittest.testsuite()
# 增加一条用例  addtest()方法传入：类名(测试方法名)
suite.addtest(testmathcalculate('test_add'))
suite.addtest(testmathcalculate('test_subtract'))
suite.addtest(testmathcalculate('test_multiply'))
suite.addtest(testmathcalculate('test_divide'))

# 上下文管理器，生成测试报告
with open('test_math.txt', mode='w+', encoding='utf-8') as f:
    runner = unittest.texttestrunner(f, 'text', 2)
    runner.run(suite)

 　　方法二：加载测试类，通过testloader里面的loadtestsfromtestcase来加载测试类，代码示范如下：

import unittest
from python_for_unittest_testsuite.test_math_calculate import testmathcalculate

suite = unittest.testsuite()
loader = unittest.testloader()
# 加载测试类
suite.addtest(loader.loadtestsfromtestcase(testmathcalculate))


# 上下文管理器，生成测试报告
with open('test_math.txt', mode='w+', encoding='utf-8') as f:
    runner = unittest.texttestrunner(f, 'text', 2)
    runner.run(suite)

　　方法三：加载测试模块，通过testloader里面的loadtestsfrommodule来加载测试类，代码示范如下：

import unittest
from python_for_unittest_testsuite import test_math_calculate


suite = unittest.testsuite()
loader = unittest.testloader()
# 加载测试模块
suite.addtest(loader.loadtestsfrommodule(test_math_calculate))


# 上下文管理器，生成测试报告
with open('test_math.txt', mode='w+', encoding='utf-8') as f:
    runner = unittest.texttestrunner(f, 'text', 2)
    runner.run(suite)

　　方法四：自动搜寻测试用例，根据传入的测试模块的所属目录路径自动去查找测试用例，代码示范如下：

import unittest

# 根据传入的测试模块的所属目录路径自动去查找
suite = unittest.defaulttestloader.discover("g:\pycharm\pycharmprojects\python_class\python_for_unittest_testsuite")

# 上下文管理器，生成测试报告
with open('test_math.txt', mode='w+', encoding='utf-8') as f:
    runner = unittest.texttestrunner(f, 'text', 2)
    runner.run(suite)