unittest指南——不拼花哨，只拼实用

news2024/12/24 11:41:46
  • 📢专注于分享软件测试干货内容，欢迎点赞 👍 收藏 ⭐留言 📝 如有错误敬请指正！
  • 📢交流讨论：欢迎加入我们一起学习！
  • 📢资源分享：耗时200+小时精选的「软件测试」资料包
  • 📢 最困难的时候，也就是我们离成功不远的时候！

“ Python为开发者提供了内置的单元测试框架 unittest，它是一种强大的工具，能够有效地编写和执行单元测试。unittest 提供了完整的测试结构，支持自动化测试的执行，能够对测试用例进行组织，并且提供了丰富的断言方法。最终，unittest 会生成详细的测试报告，这个框架非常简单且易于使用。”

unittest核心概念

在 unittest 中，有四个核心概念：

  • TestCase（测试用例）：每个测试用例实例用于封装一个或多个测试函数。

  • TestSuite（测试套件）：这是多个测试用例的集合，用于组织和执行多个测试用例。

  • TestLoader（测试加载器）：这是一个用于将测试用例加载到测试套件中的工具。

  • TextTestRunner（测试运行器）：这是用于执行测试用例的运行器，负责运行测试并生成结果报告。

  • Fixture（环境管理机制）：这是测试用例的环境搭建和销毁部分，包括前置条件和后置条件。

在这里插入图片描述

unittest的工作流程

  1. 编写继承自 unittest.TestCase 的测试用例类，其中每个测试函数都是一个独立的测试用例。
  2. 使用 TestLoader 加载测试用例，并将它们组织成 TestSuite 对象。
  3. 使用 TestRunner 运行 TestSuite 中的测试用例，并输出测试结果。

使用unittest初级指南

  1. 导入 unittest 模块以及被测试的文件或类。
  2. 创建一个测试类，并继承 unittest.TestCase，所有自定义的单元测试类都要继承它，作为基类。
  3. 重写 setUp 和 tearDown 方法，用于初始化和清理测试环境（如果有必要）。
  4. 定义测试函数，函数名以 test_ 开头，这样才能被识别并执行。
  5. 在测试函数中使用断言来判断测试结果是否符合预期。
  6. 调用 unittest.main() 方法运行测试用例，按照函数名的排序执行测试。

以下是一个简单的例子：

import unittest
 
def login(username, password):
    if username == 'kira' and password == '123':
        res = {"code": 200, "msg": "登录成功"}
        return res
    return {"code": 400, "msg": "登录失败"}
 
class TestLogin(unittest.TestCase):
 
    def test_login_success(self):
        """测试登录成功"""
        test_data = {"username": "kira", "password": "test"}
        expect_data = {"code": 200, "msg": "登录成功"}
        res = login(**test_data)
        self.assertEqual(res, expect_data)
 
    def test_login_error_with_error_password(self):
        """账号正确，密码错误，登录失败"""
        test_data = {"username": "kira", "password": "12345"}
        expect_data = {"code": 400, "msg": "登录失败"}
        res = login(**test_data)
        self.assertEqual(res, expect_data)
 
    # 更多测试函数类似...
 
if __name__ == '__main__':
    unittest.main()

以上是一个简单的测试用例，包含了两个测试函数。运行脚本将输出测试结果。

unittest核心概念

测试脚手架

测试脚手架 是测试用例的前置条件和后置条件，确保测试环境的初始化和清理，从而保证测试的准确性和可靠性。

import unittest
 
class MyTestCase(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        # 类级别的前置条件设置，整个类运行最先只执行一次
        print("setUpClass")
 
    @classmethod
    def tearDownClass(cls):
        # 类级别的后置条件清理，整个类运行最后结束执行一次
        print("tearDownClass")
 
    def setUp(self):
        # 测试方法级别的前置条件设置，所有测试方法运行前都执行一次
        print("setUp")
 
    def tearDown(self):
        # 测试方法级别的后置条件清理，所有测试方法运行结束都执行一次
        print("tearDown")
 
    def test_example(self):
        # 测试用例
        print("test_example")
 
if __name__ == "__main__":
    unittest.main()
  1. setUp()：每个测试方法运行前执行，用于测试前置的初始化工作。
  2. tearDown()：每个测试方法结束后执行，用于测试后的清理工作。
  3. setUpClass()：所有的测试方法运行前执行，用于单元测试类运行前的准备工作。使用 @classmethod 装饰器装饰，整个测试类运行过程中只会执行一次。
  4. tearDownClass()：所有的测试方法结束后执行，用于单元测试类运行后的清理工作。使用 @classmethod 装饰器装饰，整个测试类运行过程中只会执行一次。

测试用例

测试用例 是最小的测试单元，用于检测特定的输入集合的特定的返回值。unittest 提供了 TestCase 基类，所有的测试类都需要继承该基类，而在该类下的函数如果以 test_ 开头，则被标识为测试函数：

class MyTestCase(unittest.TestCase
 
):
 
    def test_addition(self):
        result = 2 + 3
        self.assertEqual(result, 5)  # 使用断言方法验证结果是否相等
 
    def test_subtraction(self):
        result = 5 - 3
        self.assertTrue(result == 2)  # 使用断言方法验证结果是否为True
 
    # 更多测试用例函数...

断言方法

以下是常用的断言方法：

  • assertEqual(a, b, msg=None)：验证 a 等于 b。

  • assertNotEqual(a, b)：验证 a 不等于 b。

  • assertTrue(x)：验证 x 是否为 True。

  • assertFalse(x)：验证 x 是否为 False。

  • assertIs(a, b)：验证 a 是否是 b。

  • assertIsNot(a, b)：验证 a 是否不是 b。

  • assertIsNone(x)：验证 x 是否为 None。

  • assertIsNotNone(x)：验证 x 是否不为 None。

  • assertIn(a, b)：验证 a 是否在 b 中。

  • assertNotIn(a, b)：验证 a 是否不在 b 中。

  • assertIsInstance(a, b)：验证 a 是否是 b 类型的实例。

  • assertNotIsInstance(a, b)：验证 a 是否不是 b 类型的实例。

可以使用这些方法进行断言，也可以直接使用原生的assert来断言，如果断言失败，测试用例会被定义为执行失败。

忽略特定测试方法

unittest 提供了一些方法来跳过特定的测试用例：

  • @unittest.skip(reason)：强制跳过，reason 是跳过的原因。

  • @unittest.skipIf(condition, reason)：当 condition 为 True 时跳过。

  • @unittest.skipUnless(condition, reason)：当 condition 为 False 时跳过。

  • @unittest.expectedFailure：如果测试失败，这个测试用例不会计入失败的统计。

  • 使用实例方法：self.skipTest() 使用和上述类似。

import sys
import unittest
 
class Test1(unittest.TestCase):
    @unittest.expectedFailure  # 即使失败也会被计为成功的用例
    def test_1(self):
        assert 1 + 1 == 3
 
    @unittest.skip('无条件跳过')  # 不管什么情况都会进行跳过
    def test_2(self):
        print("2+2...", 4)
 
    @unittest.skipIf(sys.platform == "win32", "跳过")  # 如果系统平台为 Windows 则跳过
    def test_3(self):
        print("3+3...", 6)
 
    @unittest.skipUnless(sys.platform == "win32", "跳过")  # 除非系统平台为 Windows，否则跳过
    def test_4(self):
        print("4+4...", 8)
 
    def test_5(self):
        self.skipTest("跳过")
        print("5+5...", 10)
 
if __name__ == "__main__":
    unittest.main(verbosity=2)

测试套件

测试套件用于收集和组织多个测试用例，便于集中执行。

  1. 通过 unittest.main() 方法直接加载单元测试的测试模块，这是一种简单的加载方式。所有测试用例的执行顺序按照方法名的字符串表示的 ASCII 码升序排序，通过命名时使用 test_01_xxx 来指定执行顺序。

  2. 将所有的单元测试用例 TestCase 加载到测试套件 Test Suite 集合中，然后一次性加载所有测试对象。

通过 TestSuite 对象收集

此方式适用于需要自定义组合特定测试用例的情况。

import unittest
 
class MyTestCase(unittest.TestCase):
    def test_addition(self):
        result = 2 + 3
        self.assertEqual(result, 5)
 
def suite():
    suite = unittest.TestSuite()
    suite.addTest(MyTestCase('test_addition'))
    return suite
 
if __name__ == '__main__':
    runner = unittest.TextTestRunner()
    runner.run(suite())

通过 TestLoader 对象收集

TestLoader 是 unittest 框架提供的加载测试用例的类。

import unittest
 
if __name__ == '__main__':
    loader = unittest.defaultTestLoader
    
    # 自动加载当前模块中所有以 'test_' 开头的测试用例函数
    suite = loader.loadTestsFromModule(__name__)
 
    runner = unittest.TextTestRunner()
    runner.run(suite)
 

import unittest
 
class MyTestCase(unittest.TestCase):
    def test_addition(self):
        result = 2 + 3
        self.assertEqual(result, 5)
 
if __name__ == '__main__':
    loader = unittest.defaultTestLoader
    
    # 自动加载 MyTestCase 类中的所有测试用例
    suite = loader.loadTestsFromTestCase(MyTestCase)
 
    runner = unittest.TextTestRunner()
    runner.run(suite)
 
 

import unittest
 
if __name__ == '__main__':
    loader = unittest.defaultTestLoader
    
    # 自动加载指定名称的测试用例
    suite = loader.loadTestsFromName('module.MyTestCase.test_addition')
 
    runner = unittest.TextTestRunner()
    runner.run(suite)
 

import unittest
 
if __name__ == '__main__':
    loader = unittest.defaultTestLoader
    
    # 自动发现并加载指定目录中的测试用例模块
    suite = loader.discover(start_dir='test_directory', pattern='test_*.py', top_level_dir=None)
 
    runner = unittest.TextTestRunner()
    runner.run(suite)

测试运行器

测试运行器是用于执行和输出测试结果的组件。常用的运行器有：

  • unittest.TextTestRunner：这是 unittest 框架中默认的测试运行器，会在命令行输出测试结果。通过调用 run() 方法运行测试套件，并将测试结果打印到控制台。
import unittest
 
if __name__ == '__main__':
    loader = unittest.defaultTestLoader
    suite = loader.discover(start_dir='tests', pattern='test_*.py')
    
    runner = unittest.TextTestRunner()
    result = runner.run(suite)
  • HTMLTestRunner：这是一个第三方库，能够生成漂亮的 HTML 测试报告，需要进行安装。你可以通过搜索获取相关文件进行安装。
import unittest
from HTMLTestRunner import HTMLTestRunner
 
if __name__ == '__main__':
    loader = unittest.defaultTestLoader
    suite = loader.discover(start_dir='tests', pattern='test_*.py')
    
    with open('test_report.html', 'wb') as report_file:
        runner = HTMLTestRunner(stream=report_file, title='Test Report', description='Test Results')
        result = runner.run(suite)
  • XMLTestRunner：这是另一个第三方库，用于生成 XML 格式的测试报告。
import unittest
from xmlrunner import XMLTestRunner
 
if __name__ == '__main__':
    loader = unittest.defaultTestLoader
    suite = loader.discover(start_dir='tests', pattern='test_*.py')
    
    with open('test_report.xml', 'wb') as report_file:
        runner = XMLTestRunner(output=report_file)
        result = runner.run(suite)

你也可以自定义测试运行器。继承 unittest.TestRunner 类并实现 run() 方法，以创建自己的测试运行器。

import unittest
 
class MyTestRunner(unittest.TextTestRunner):
    def run(self, test):
        print("Running tests with MyTestRunner")
        result = super().run(test)
        return result
 
if __name__ == '__main__':
    loader = unittest.defaultTestLoader
    suite = loader.discover(start_dir='tests', pattern='test_*.py')
    
    runner = MyTestRunner()
    result = runner.run(suite)

通常使用 HTMLTestRunner 即可满足需求，它非常易用。

实战一个测试案例

假设有一个测试函数 login：

# login.py
def login(username, password):
    """模拟登录校验"""
    if username == 'kira' and password == '123456':
        return {"code": 0, "msg": "登录成功"}
    else:
        return {"code": 1, "msg": "账号或密码不正确"}

设计用例

根据函数的参数和逻辑，设计如下用例：

编写测试用例并运行

import unittest
from login import login
 
class TestLogin(unittest.TestCase):
    def test_login_correct(self):
        """测试账号密码正确"""
        test_data = {"username": "kira", "password": "123456"}
        expect_data = {"code": 0, "msg": "登录成功"}
        res = login(**test_data)
        self.assertEqual(res, expect_data)
 
    def test_login_wrong_password(self):
        """测试账号正确密码不正确"""
        test_data = {"username": "kira", "password": "123"}
        expect_data = {"code": 1, "msg": "账号或密码不正确"}
        res = login(**test_data)
        self.assertEqual(res, expect_data)
 
    def test_login_wrong_username(self):
        """测试账号错误密码正确"""
        test_data = {"username": "kir", "password": "123456"}
        expect_data = {"code": 1, "msg": "账号或密码不正确"}
        res = login(**test_data)
        self.assertEqual(res, expect_data)
 
if __name__ == '__main__':
    unittest.main()

这是一个简单的测试用例，包含了三个测试函数。运行测试用例后，会输出测试结果，看完是否觉得unittest非常简单易用。ner.run(suite)

最后的分享

如果你想学习自动化测试，那么下面这套视频应该会帮到你很多

如何逼自己1个月学完自动化测试，学完即就业，小白也能信手拈来，拿走不谢，允许白嫖....

最后我这里给你们分享一下我所积累和整理的一些文档和学习资料，有需要直接领取就可以了！


以上内容，对于软件测试的朋友来说应该是最全面最完整的备战仓库了，为了更好地整理每个模块，我也参考了很多网上的优质博文和项目，力求不漏掉每一个知识点，很多朋友靠着这些内容进行复习，拿到了BATJ等大厂的offer，这个仓库也已经帮助了很多的软件测试的学习者，希望也能帮助到你。

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.coloradmin.cn/o/1247865.html

如若内容造成侵权/违法违规/事实不符，请联系多彩编程网进行投诉反馈，一经查实，立即删除！

相关文章

BGP联邦及路由反射器配置

BGP联邦及路由反射器配置

需求 1 AS1存在两个环回，一个地址为192.168.1.0/24，该地址不能再任何协议中宣告 AS3存在两个环回，一个地址为192.168.2.0/24，该地址不能再任何协议中宣告 AS1还有一个环回地址为10.1.1.0/24，AS3另一个环回地址是11.1.1…
阅读更多...
树形 DP：树的直径

树形 DP：树的直径

leetCode 104.二叉树的最大深度104. 二叉树的最大深度 - 力扣（LeetCode） class Solution { public:int maxDepth(TreeNode* root) {if(root nullptr) return 0;int lDepth maxDepth(root->left);int rDepth maxDepth(root->right);return max(l…
阅读更多...
HashMap的实现原理，HashMap方法详解，hash()计算的原理，扩容机制

HashMap的实现原理，HashMap方法详解，hash()计算的原理，扩容机制

文章目录 说一下HashMap的实现原理(非常重要)①HashMap的工作原理HashMap存储结构常用的变量HashMap 构造函数tableSizeFor() put()方法详解hash()计算原理resize() 扩容机制get()方法为什么HashMap链表会形成死循环 说一下HashMap的实现原理(非常重要) HashMap概述&#xff1…
阅读更多...
冷空气来袭，关注身体状况，手表这几个功能速get

冷空气来袭，关注身体状况，手表这几个功能速get

进入小雪节气后，冷空气活动更加频繁，气温会越来越低，或进入感冒发烧的高发期。大家在感觉到身体不适时，要多关注一下自己的体温、血氧饱和度、心率等指标，通过手表就可以掌握这些身体讯号，速来了解一下&…
阅读更多...
SQL 通配符：用于模糊搜索和匹配的 SQL 关键技巧

SQL 通配符：用于模糊搜索和匹配的 SQL 关键技巧

SQL通配符字符 通配符字符用于替代字符串中的一个或多个字符。通配符字符与LIKE运算符一起使用。LIKE运算符用于在WHERE子句中搜索列中的指定模式。 示例 返回所有以字母 ‘a’ 开头的客户： SELECT * FROM Customers WHERE CustomerName LIKE a%;通配符字符 符…
阅读更多...
智能WIFI 电子标签

智能WIFI 电子标签

7.5/10.2/13.3寸大屏电子墨水屏高清显示 无需部署专用基站，基于标准WIFI网络通信实时在线 无需接线，极简安装 超低功耗：充一次电续航一年~ 系统联动，信息统一，WiFi控制，批量快速刷新 随改随变化&#…
阅读更多...
【数据分享】我国12.5米分辨率的DEM地形数据（免费获取/地理坐标系）

【数据分享】我国12.5米分辨率的DEM地形数据（免费获取/地理坐标系）

DEM地形数据是我们在各种研究和设计中经常使用的数据！之前我们分享过500米分辨率的DEM地形数据、90米分辨率的DEM地形数据、30米分辨率的DEM地形数据（均可查看之前的文章获悉详情）。 本次我们为大家带来的是分辨率为12.5m的DEM地形数据&#…
阅读更多...
Linux后台运行Python的py文件，如何使ssh工具退出后仍能运行

Linux后台运行Python的py文件，如何使ssh工具退出后仍能运行

常规运行 python3 mysqlbak.py ssh工具退出后，或ctrlc中断后，程序将不在运行 后台运行 nohup python3 mysqlbak.py > mysqlbak.log & > mysqlbak.log为可选项，输出日志到指定文件，如果不写，输出日志到nohup…
阅读更多...
CENTOS8.2下的内核启动参数cmdline更新

CENTOS8.2下的内核启动参数cmdline更新

error: environment block too small. 删掉grubenv重新生成。 查看是什么方式启动 [rootlocalhost boot]# [ -d /sys/firmware/efi ] && echo UEFI || echo BIOS UEFI 先改etc/default/grub GRUB_CMDLINE_LINUX"consolettyS0,115200n8 crashkernelauto ignore_…
阅读更多...
大语言模型概述（一）：基于亚马逊云科技的研究分析与实践

大语言模型概述（一）：基于亚马逊云科技的研究分析与实践

大型语言模型指的是具有数十亿参数（B）的预训练语言模型（例如：GPT-3, Bloom, LLaMA)。这种模型可以用于各种自然语言处理任务，如文本生成、机器翻译和自然语言理解等。 大语言模型的这些参数是在大量文本数据上训练的。…
阅读更多...
LeeCode前端算法基础100题（2）- 最多水的容器

LeeCode前端算法基础100题（2）- 最多水的容器

一、问题详情： 给定一个长度为 n 的整数数组 height 。有 n 条垂线，第 i 条线的两个端点是 (i, 0) 和 (i, height[i]) 。 找出其中的两条线，使得它们与 x 轴共同构成的容器可以容纳最多的水。 返回容器可以储存的最大水量。 说明：…
阅读更多...
C++ 之win32多线程应用总结

C++ 之win32多线程应用总结

InterlockedIncrement 函数的作用： 在多线程同时对一个变量访问时，保证一个线程访问变量时其他线程不能访问 事件是很常用的多线程同步互斥机制 HANDLE CreateEvent(LPSECURITY_ATTRIBUTES lpEventAttributes,　// SECURITY_ATTRIBUTES结构指针&…
阅读更多...
长文讲清荧光定量PCR（最新版）q-pcr rt-pcr

长文讲清荧光定量PCR（最新版）q-pcr rt-pcr

为讲透qPCR，我会持续更新本文，点关注追踪查看。 1.初阶认识 这个阶段，我们要明白一些概念和术语，避免自己在师兄面前错误的瞎哔哔，比如： 问：RT-PCR、qPCR、Real-time PCR、real-time RT-PCR有…
阅读更多...
青云科技容器平台与星辰天合存储产品完成兼容性互认证

青云科技容器平台与星辰天合存储产品完成兼容性互认证

近日， 北京青云科技股份有限公司（以下简称：青云科技）的 KubeSphere 企业版容器平台成功完成了与 XSKY星辰天合的企业级分布式统一数据平台 V6（简称：XEDP）以及天合翔宇分布式存储系统 V6&#xf…
阅读更多...
python——第十二天

python——第十二天

内置模块或者其他模块学习方式： dir help os模块负责程序与操作系统的交互，提供了访问操作系统底层的接口；即os模块提供了非常丰富的方法用来处理文件和目录。 os： os.path 遍历C盘代码 import os from os import path def …
阅读更多...
0003Java程序设计-ssm基于微信小程序的家教信息管理系统

0003Java程序设计-ssm基于微信小程序的家教信息管理系统

文章目录 摘要目 录系统实现开发环境 编程技术交流、源码分享、模板分享、网课分享 企鹅🐧裙：776871563 摘要 本文讲述了基于微信小程序的家教信息管理系统的设计与实现。结合线上管理的特点，分析了家教信息管理系统的现状，给出…
阅读更多...
leetcode：环形链表的入环点

leetcode：环形链表的入环点

题目描述 题目链接:力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台 题目分析 我们假设起点到环的入口点的距离是L，入口点到相遇点的距离是X，环的长度是C 那么画图我们可以得知： 从开始到相遇时slow走的距离是LX从…
阅读更多...
qRT-PCR相对定量计算详解qPCR相对定量计算方式——2^-(∆∆Ct) deta t

qRT-PCR相对定量计算详解qPCR相对定量计算方式——2^-(∆∆Ct) deta t

做完转录组分析之后，一般都要求做qRT-PCR来验证二代测序得到的转录本表达是否可靠。荧光定量PCR是一种相对表达定量的方法，他的计算方法有很多，常用的相对定量数据分析方法有双标曲线法，ΔCt法，2^-ΔΔCt法(Livak法)&a…
阅读更多...
STM32入门笔记15_PWR电源管理模块

STM32入门笔记15_PWR电源管理模块

PWR和低功耗模式 PWR简介 PWR(Power Control) 电源控制PWR负责管理STM32内部的电源供电部分，可以实现可编程电压检测器和低功耗模式的功能可编程电压检测器(PVD) 可以监控VDD电源电压，当VDD下降到PVD阈值以下或上升到PVD阈值之上时，PVD会触…
阅读更多...
通过线性回归进行房价预测

通过线性回归进行房价预测

房价预测一直是房地产行业和投资者关注的重要问题。线性回归是一种常用的回归算法，可以建立输入变量和连续输出变量之间的关系。在本文中，我们将探讨如何使用线性回归算法来进行房价预测，并介绍该方法的步骤和实践技巧。 一、线性回归算法简…
阅读更多...
推荐文章
最新文章

Copyright @ 2022~2023