Python 3.10结构模式匹配的高效测试方法问询

我正在用Python 3.10构建命令行界面（CLI），其中save命令会根据传入的选项调用不同函数，可选选项为"A", "B", "C", "D", "E"。我用Python 3.10的结构模式匹配写了如下代码：

# __main__.py

from my_module import func_A, func_B, func_C, func_D, func_E, get_from_cache

@cli.command()
def save(option):
    match option:
        case "A":
            to_save = func_A()
        case "B":
            to_save = func_B()
        case "C":
            to_save = func_C()
        case "D":
            to_save = func_D()
        case "E":
            to_save = func_E()
        case None:
            to_save = get_from_cache()
        case _:
            print("Not an Option")
            return

    # rest of save command

我已经单独测试了每个函数，但Pytest Coverage报告显示__main__.py中的模式匹配语句未被覆盖。请问有没有Pythonic的高效方法测试传入指定选项时是否调用了正确的函数？我目前想到的唯一方案是为每个函数打桩并为A-E选项分别编写测试，但当选项增加时，需要新增桩函数和测试用例。

高效解决方案：参数化测试+函数映射优化 #

第一步：优化原代码（可选但推荐） #

把模式匹配改成选项-函数映射字典，既符合DRY原则，也让后续测试更简洁，新增选项只需维护字典而非修改分支逻辑：

# __main__.py
from my_module import func_A, func_B, func_C, func_D, func_E, get_from_cache

# 建立选项到对应函数的映射
OPTION_TO_FUNC = {
    "A": func_A,
    "B": func_B,
    "C": func_C,
    "D": func_D,
    "E": func_E,
    None: get_from_cache
}

@cli.command()
def save(option):
    try:
        target_func = OPTION_TO_FUNC[option]
        to_save = target_func()
    except KeyError:
        print("Not an Option")
        return

    # rest of save command

第二步：用Pytest参数化测试覆盖所有场景 #

利用Pytest的@pytest.mark.parametrize一次性定义所有测试用例，不用为每个选项单独写测试函数，新增选项只需在测试用例列表中加一行即可：

# test_cli.py
import pytest
from unittest.mock import patch
from __main__ import save  # 导入你的save命令函数

# 测试用例：(选项值, 预期调用的函数名, 是否为有效选项)
TEST_SCENARIOS = [
    ("A", "func_A", True),
    ("B", "func_B", True),
    ("C", "func_C", True),
    ("D", "func_D", True),
    ("E", "func_E", True),
    (None, "get_from_cache", True),
    ("X", None, False),  # 测试无效选项
    ("invalid", None, False),
]

@pytest.mark.parametrize("option, expected_func, is_valid", TEST_SCENARIOS)
def test_save_command(option, expected_func, is_valid):
    # 一次性打桩所有目标函数
    with patch("__main__.func_A") as mock_a, \
         patch("__main__.func_B") as mock_b, \
         patch("__main__.func_C") as mock_c, \
         patch("__main__.func_D") as mock_d, \
         patch("__main__.func_E") as mock_e, \
         patch("__main__.get_from_cache") as mock_cache, \
         patch("builtins.print") as mock_print:

        # 执行save命令
        save(option)

        if is_valid:
            # 验证对应函数被调用一次，其余函数未被调用
            target_mock = locals()[f"mock_{expected_func.lower()}"]
            target_mock.assert_called_once()
            
            # 检查其他函数未触发调用
            all_mocks = [mock_a, mock_b, mock_c, mock_d, mock_e, mock_cache]
            for mock in all_mocks:
                if mock != target_mock:
                    mock.assert_not_called()
        else:
            # 无效选项时，验证提示信息打印，且无任何业务函数被调用
            mock_print.assert_called_once_with("Not an Option")
            for mock in [mock_a, mock_b, mock_c, mock_d, mock_e, mock_cache]:
                mock.assert_not_called()

方案优势 #

• 扩展性强：新增选项只需在OPTION_TO_FUNC字典和TEST_SCENARIOS列表中各加一行，无需修改核心逻辑和测试结构
• 覆盖全面：一次测试覆盖所有有效/无效选项场景，确保Coverage报告拉满
• 符合Pythonic风格：用字典映射替代冗余的case分支，代码更简洁易维护

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