对pytest中的关键字参数使用parametrize

问题描述

我正在学习pytest，并尝试使用pytest.mark.parametrize作为关键字参数。

这是不带pytest.mark.parametrize的简单示例：

G = 10
H = 2

def f(g=G, h=H):
    return 5 * g + h

def test_f1():
    assert f(h=4) == 54
    assert f(g=20) == 102

这是我使用pytest.mark.parametrize进行的一次不成功的尝试。它不起作用，但它有助于理解我想要实现的目标：

import pytest

G = 10
H = 2

def f(g=G, h=H):
    return 5 * g + h  

@pytest.mark.parametrize("arg, expected", [
    ("h=4", 54),
    ("g=20", 102),
    ])

def test_f2(arg, expected):
    assert f(eval(arg)) == expected

---

解决方案

函数f接受关键字参数，因此您需要将测试参数分配给关键字。幸运的是，Python提供了一种非常方便的方式将关键字参数传递给函数.字典：

d = {'h': 4}
f(**d)

d之前的**前缀将"解压缩"字典，将每个键/值对作为关键字参数传递给函数。

对于您的pytest示例，这意味着您只需要将字符串参数替换为字典，将test_f2中的eval替换为关键字UnPacker：

@pytest.mark.parametrize("arg,expected", [
    ({'h':4}, 54),
    ({'g':20}, 102),
    ])

def test_f2(arg, expected):
    assert f(**arg) == expected