Pytest数据驱动怎么玩？实战教程来了！

原创

于 2025-01-09 17:07:04 发布 · 718 阅读

20 ·

CC 4.0 BY-SA版权

文章标签：

#pytest #微信小程序 #postman #jmeter #功能测试 #测试工具 #软件测试

Pytest是一种用于Python的测试框架，它可以帮助我们编写简洁、可维护的测试代码。数据驱动是一种测试方法，它能够通过不同的数据来驱动测试用例的执行，从而提高测试覆盖率和测试效率。本文将介绍如何在Pytest中使用数据驱动进行测试，包括如何使用pytest.mark.parametrize装饰器和如何使用测试数据文件。

首先，我们需要安装Pytest。可以通过以下命令在命令行中安装：

pip install pytest

安装完成后，我们就可以开始使用Pytest进行数据驱动测试了。

使用pytest.mark.parametrize装饰器

pytest.mark.parametrize是一个装饰器，它可以将多个参数组合成不同的测试用例，并为每个参数组合执行一次测试。

下面是一个示例，假设我们要测试一个计算器类的加法方法：

class Calculator:
    def add(self, a, b):
        return a + b

def test_add():
    calculator = Calculator()
    assert calculator.add(2, 3) == 5

上面的测试用例只测试了一个参数组合（2和3），如果我们要测试更多的参数组合，就需要编写多个测试用例。使用pytest.mark.parametrize可以简化这个过程。

import pytest

@pytest.mark.parametrize("a, b, expected", [
    (2, 3, 5),
    (0, 0, 0),
    (-1, 1, 0),
])
def test_add(a, b, expected):
    calculator = Calculator()
    assert calculator.add(a, b) == expected