分享unittest单元测试框架中几种常用的用例加载方法
作者:小公瑾 发布时间:2023-04-25 06:53:34
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)
来源:https://www.cnblogs.com/xiaogongjin/p/11665911.html
![](https://www.aspxhome.com/images/zang.png)
![](https://www.aspxhome.com/images/jiucuo.png)
猜你喜欢
- 导言在前面的教程我们看到了如何使用两个页面(一个主页,用于列出供应商; 一个明细页,用于显示选定供应商提供的产品)创建主/从报表 . 这种两
- 服务器:#server.py#!/usr/bin/env python#-*-coding:utf-8-*-import sysimport
- 在为一个项目添加权限时,遇到一个问题,就是为项目所有的url设置权限,但是一个一个手动输入太麻烦了,所以考虑用代码获取到一个项目所有的url
- 练手项目,先上图先实现一个简单的串口工具,为之后的上位机做准备代码如下:github 下载地址pyserial_demo.pyimport
- 阻塞定义当来自应用程序的第一个连接控制锁而第二个连接需要相冲突的锁类型时,将发生阻塞。其结果是强制第二个连接等待,而在第一个连接上阻塞。不管
- 如下所示:import ospath="/home/test/" #待读取的文件夹path_list=os.listdi
- 用了smarty,tp过后,也想了解了解其模板技术是怎么实现,于是写一个简单的模板类,大致就是读取模板文件->替换模板文件的内容-&g
- 研究网页编码很长时间了,因为最近要设计一个友情链接检测的VBS脚本,而与你链接的人的页面很可能是各种编码,以前采取的方法是:如果用GB231
- 第一种情况是返回的游标是某个具体的表或视图的数据,如:SQL-Code:CREATE OR REPLACE P
- networkx是Python的一个包,用于构建和操作复杂的图结构,提供分析图的算法。图是由顶点、边和可选的属性构成的数据结构,顶点表示数据
-   本文介绍基于Python中ArcPy模块,实现基于栅格图像批量裁剪栅格图像,同时对齐各个栅格图像的空
- <!--这是一个主页文件--><html><head><meta http-equiv="
- 1. 首先导入一些python画图的包,读取txt文件,假设我现在有两个模型训练结果的records.txt文件import numpy a
- Ubuntu16.04自带python2.7与python3.5,某个项目编译却要求python版本大于等于3.7,遂考虑在原系统基础上再安
- python具体强大的库文件,很多功能都有相应的库文件,所以很有必要进行学习一下,其中有一个ftp相应的库文件ftplib,我们只需要其中的
- 提到sa弱口令,我们首先就会想到,许多数据库都有1个类似的超级管理员账号,比如:Oracle是"system"和&quo
- 目标:爬取自己账号中购买的课程视频。一、实现登录账号这里采用的是手动输入验证码的方式,有能力的盆友也可以通过图像识别的方式自动填写验证码。登
- 本文实例讲述了Python使用cx_Freeze库生成msi格式安装文件的方法。分享给大家供大家参考,具体如下:①.需要在目录下面创建一个文
- 如何做一个分页程序? 这在ASP中确实容易实现,但需要技巧,看看下面的分页代码和说明: <angu
- 新搞了台linux云主机,瞎折腾折腾,先装个Python3。Linux环境下有其他软件需要Python2,如YUM,所以安装的Python3