pytest中@pytest.mark.parametrize和自定义装饰器如何一起使用?

767 篇文章 2 订阅
575 篇文章 11 订阅

在Pytest中,@pytest.mark.parametrize装饰器用于将测试函数参数化。它允许我们通过传递多个参数和对应的值来运行相同的测试函数多次。另一方面,自定义装饰器可以用于添加额外的功能或重用代码。在本文中,我们将详细介绍如何在Pytest中同时使用@pytest.mark.parametrize和自定义装饰器。

首先,我们需要了解如何用@pytest.mark.parametrize装饰器来参数化测试函数。假设我们有一个简单的函数,用于计算两个数的和:

def add_numbers(a, b):
    return a + b

现在,我们希望测试这个函数的不同输入组合。我们可以使用@pytest.mark.parametrize来实现:

import pytest

@pytest.mark.parametrize("a, b, expected", [
    (1, 2, 3),
    (10, 20, 30),
    (-1, 1, 0),
])
def test_add_numbers(a, b, expected):
    result = add_numbers(a, b)
    assert result == expected

在这个示例中,我们在测试函数上使用了@pytest.mark.parametrize装饰器。我们传递了三个参数:a,b和expected。然后,我们提供了一个参数值列表,其中每个元组都代表一个测试用例。这些元组的顺序与参数列表的顺序相对应。在每次运行测试时,pytest会为a,b和expected分别传递列表中的值,并执行测试函数。

现在,让我们看看如何与自定义装饰器一起使用@pytest.mark.parametrize。假设我们有一个自定义装饰器,用于记录测试函数的运行时间:

import time

def record_time(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        execution_time = end_time - start_time
        print(f"Execution time: {execution_time} seconds")
        return result
    return wrapper
现在我也找了很多测试的朋友,做了一个分享技术的交流群,共享了很多我们收集的技术文档和视频教程。
如果你不想再体验自学时找不到资源,没人解答问题,坚持几天便放弃的感受
可以加入我们一起交流。而且还有很多在自动化,性能,安全,测试开发等等方面有一定建树的技术大牛
分享他们的经验,还会分享很多直播讲座和技术沙龙
可以免费学习!划重点!开源的!!!
qq群号:691998057【暗号:csdn999】

在这个示例中,record_time装饰器将计算测试函数的运行时间并打印出来。现在,我们将这个装饰器应用于我们的测试函数:

@pytest.mark.parametrize("a, b, expected", [
    (1, 2, 3),
    (10, 20, 30),
    (-1, 1, 0),
])
@record_time
def test_add_numbers(a, b, expected):
    result = add_numbers(a, b)
    assert result == expected

现在,每次运行test_add_numbers函数时,record_time装饰器都会被触发,计算并打印出运行时间。同时,@pytest.mark.parametrize装饰器将根据提供的参数值运行测试函数多次。

需要注意的是,装饰器的顺序很重要。在这个示例中,我们首先使用@pytest.mark.parametrize装饰器,然后使用record_time装饰器。如果装饰器的顺序相反,pytest就会报错。

总的来说,使用@pytest.mark.parametrize和自定义装饰器可以为我们提供更灵活和重用代码的方式。

我们可以使用@pytest.mark.parametrize来参数化测试函数,用不同的输入执行多次测试。同时,自定义装饰器可以添加额外的功能,例如记录运行时间。通过正确的装饰器顺序,我们可以同时使用这两种装饰器来优化和扩展我们的测试代码。

下面是配套资料,对于做【软件测试】的朋友来说应该是最全面最完整的备战仓库,这个仓库也陪伴我走过了最艰难的路程,希望也能帮助到你!

史上最全软件测试自学资料大全(学习路线+视频教程+文档资料+PDF面试文档)作为一个软件测试的过来人,小濠想尽自己最大的努力,帮助每一个伙伴都能顺利找到满意的工作。我花费了很多时间细心钻研,在收集了近 200G 干货、 1500份资料中精心筛选出来了这份「软件测试资料包」。你只需要花很少的时间就能快速地icon-default.png?t=N7T8https://mp.weixin.qq.com/s?__biz=Mzk0OTI0ODM2Mg==&mid=2247489248&idx=1&sn=3d7dbe28d58e5147c409c7e5f059d8ea&chksm=c35a1ee5f42d97f397fd420ce40f72b60195d4ffb42c34f61300463711347815f2e5656defa4&token=1425287504&lang=zh_CN#rd最后: 可以在公众号:程序员小濠 ! 免费领取一份216页软件测试工程师面试宝典文档资料。以及相对应的视频学习教程免费分享!,其中包括了有基础知识、Linux必备、Shell、互联网程序原理、Mysql数据库、抓包工具专题、接口测试工具、测试进阶-Python编程、Web自动化测试、APP自动化测试、接口自动化测试、测试高级持续集成、测试架构开发测试框架、性能测试、安全测试等。

如果我的博客对你有帮助、如果你喜欢我的博客内容,请 “点赞” “评论” “收藏” 一键三连哦!

  • 8
    点赞
  • 15
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值