如何在Pytest中为@patch设置多返回值，实现类似@parametrize的测试？

完全可行，你可以通过两种方式实现多轮测试不同mock返回值的需求，以下是具体方案：

方案一：结合装饰器实现 #

将@pytest.mark.parametrize放在@patch装饰器上方，通过测试函数参数传递不同的返回值，并在函数内修改mock对象的返回值：

import pytest
from unittest.mock import patch
from your_module import class_to_patch

@pytest.mark.parametrize("mock_return, expected", [("x", 1), ("y", 1)])
@patch("class_to_patch.method_to_patch", autospec=True)
def test_method(mock_method, mock_return, expected):
    # 为当前用例设置mock返回值
    mock_method.return_value = mock_return
    # 执行被测逻辑（根据实际场景调整，示例中假设创建实例后触发逻辑）
    instance = class_to_patch()
    # 这里根据实际被测逻辑调整断言，比如如果是调用方法后得到结果：
    # result = instance.method_to_patch()
    # assert result == mock_return
    assert instance == expected

关键说明： #

• 装饰器顺序必须是@pytest.mark.parametrize在上，@patch在下，因为Python装饰器是从下往上执行的，确保参数能正确传入测试函数。
• autospec=True会保留原方法的签名，避免因mock与原方法签名不匹配导致的隐藏问题。

方案二：使用上下文管理器（更灵活） #

如果觉得装饰器顺序容易混淆，可以在测试函数内部用patch上下文管理器直接绑定对应返回值：

import pytest
from unittest.mock import patch
from your_module import class_to_patch

@pytest.mark.parametrize("mock_return, expected", [("x", 1), ("y", 1)])
def test_method(mock_return, expected):
    # 每个用例单独创建mock，自动处理patch的启停
    with patch("class_to_patch.method_to_patch", autospec=True, return_value=mock_return):
        instance = class_to_patch()
        # 调整为实际被测逻辑的断言
        assert instance == expected

额外注意事项： #

• 确保patch的路径正确：必须是被测代码中导入class_to_patch的路径，而非class_to_patch的定义路径。例如如果被测模块my_app.service中导入了class_to_patch，则patch路径应为my_app.service.class_to_patch.method_to_patch，否则mock不会生效。
• 示例中的断言逻辑是基于你提供的代码编写的，实际测试中请根据被测函数的真实行为调整断言内容。

内容的提问来源于stack exchange，提问作者rick_grimes