分享unittest单元测试框架中几种常用的用例加载方法
程序员文章站
2022-03-05 23:26:49
unittest模块是python自带的一个单元测试模块,我们可以用来做单元测试。unittest模块包含了如下几个子模块: 测试用例:testcase 测试集:testsuite 加载用例:t...
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类以及类方法的用法:
- 要引用testcase类,需要引入unittest模块
- 引用testcase是编写一个测试类,所以要新建一个类
- 测试类里面的方法名都是以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)
以上就是unittest单元测试框架中常用的几种用例加载方法的详细内容,更多关于unittest单元测试框架 用例加载的资料请关注其它相关文章!