如何用pytest测试Python无返回值函数并实现100%代码覆盖率？

我刚开始用pytest做Python单元测试，有返回值的函数可以用assert断言预期值和返回值，但遇到一个无返回值、只在结尾执行打印的函数，比如：

from random import randint

def function() -> None:
    number = randint(0, 4)

    if (number == 0):
       print("Number 0")

    elif (number == 1):
       print("Number 1")

    elif (number == 2):
       print("Number 2")

    elif (number == 3):
       print("Number 3")

    elif (number == 4):
       print("Number 4")

我想实现100%代码覆盖率，想到过修改函数让它返回值再打印，但想知道能不能不修改函数，直接测试打印语句？

解决方案 #

不用修改原函数，你可以通过捕获标准输出+Mock随机数生成的方式覆盖所有分支，实现100%代码覆盖率，具体步骤如下：

1. 用capsys捕获打印输出 #

pytest提供的capsys fixture可以捕获函数执行时的标准输出（stdout）和标准错误（stderr）。调用目标函数后，通过capsys.readouterr()就能获取到打印的内容。

2. Mockrandint控制分支走向 #

原函数里的randint是随机的，无法稳定触发每个分支，所以需要用unittest.mock.patch替换randint的返回值，手动指定它返回0-4中的每个数字，从而覆盖所有if/elif分支。

3. 编写参数化测试用例 #

用@pytest.mark.parametrize批量生成测试用例，避免重复代码，高效覆盖所有场景。

完整测试代码示例 #

import pytest
from unittest.mock import patch
from your_module import function  # 替换成你的函数所在模块名

@pytest.mark.parametrize("test_input, expected_output", [
    (0, "Number 0\n"),
    (1, "Number 1\n"),
    (2, "Number 2\n"),
    (3, "Number 3\n"),
    (4, "Number 4\n"),
])
def test_function(test_input, expected_output, capsys):
    # Mock randint，让它返回指定的test_input
    with patch('random.randint') as mock_randint:
        mock_randint.return_value = test_input
        function()
        # 捕获打印的输出
        captured = capsys.readouterr()
        # 断言输出是否符合预期
        assert captured.out == expected_output

注意事项 #

• patch的路径要匹配函数中randint的导入方式：如果函数里用from random import randint，则需要patchyour_module.randint（替换为你的模块名）；如果是直接用random.randint，则patchrandom.randint即可。
• 打印内容默认带换行符\n，断言时要对应上，也可以用strip()去除后再比较，比如assert captured.out.strip() == expected_output.strip()。

执行这个测试用例后，每个分支都会被触发一次，代码覆盖率就能达到100%，且完全不需要修改原函数代码。

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