pytest 参数化进阶-Toy模板网

这篇具有很好参考价值的文章主要介绍了pytest 参数化进阶。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

前言：

语法

参数化误区

实践

简要回顾

前言：

pytest是一个功能强大的Python测试框架，它提供了参数化功能，可以帮助简化测试用例的编写和管理。

语法

本文就赶紧聊一聊 pytest 的参数化是怎么玩的。

@pytest.mark.parametrize

@user1ize("test_input,expected", [("3+5", 8), ("2+4", 6), ("6*9", 42)])
def test_eval(test_input, expected):
    assert eval(test_input) == expected

可以自定义变量，test_input 对应的值是"3+5" "2+4" "6*9"，expected 对应的值是 8 6 42，多个变量用 tuple，多个 tuple 用 list
参数化的变量是引用而非复制，意味着如果值是 list 或 dict，改变值会影响后续的 test
重叠产生笛卡尔积

import pytest


@user2ize("x", [0, 1])
@user3ize("y", [2, 3])
def test_foo(x, y):
    pass

@pytest.fixture()

@pytest.fixture(scope="module", params=["smtp.gmail.com", "mail.python.org"])
def smtp_connection(request):
    smtp_connection = smtplib.SMTP(request.param, 587, timeout=5)

只能使用 request.param 来引用
参数化生成的 test 带有 ID，可以使用-k来筛选执行。默认是根据函数名[参数名]来的，可以使用 ids 来定义

// list
@pytest.fixture(params=[0, 1], ids=["spam", "ham"])
// function
@pytest.fixture(params=[0, 1], ids=idfn)

使用--collect-only命令行参数可以看到生成的 IDs。

参数添加 marker

我们知道了参数化后会生成多个 tests，如果有些 test 需要 marker，可以用 pytest.param 来添加

marker 方式

# content of test_expectation.py
import pytest


@user7ize(
    "test_input,expected",
    [("3+5", 8), ("2+4", 6), pytest.param("6*9", 42, marks=pytest.mark.xfail)],
)
def test_eval(test_input, expected):
    assert eval(test_input) == expected

fixture 方式

# content of test_fixture_marks.py
import pytest


@pytest.fixture(params=[0, 1, pytest.param(2, marks=pytest.mark.skip)])
def data_set(request):
    return request.param
def test_data(data_set):
    pass

pytest_generate_tests

用来自定义参数化方案。使用到了 hook，hook 的知识我会写在《pytest hook》中，欢迎关注公众号 dongfanger 获取最新文章。文章来源地址https://www.toymoban.com/news/detail-595043.html