如何通过pytest自动让所有写入标准输出/标准错误的测试用例失败

刚好碰到过一模一样的需求！之前维护老库的时候，总是有人不小心把调试print语句提交上去，pytest默认又不会管这些输出，真的头疼。下面给你分享两种实用的解决方案，完全不用在每个测试里重复写检查逻辑：

一、全局自动检查所有测试的stdout/stderr #

我们可以用pytest的自动生效fixture（设置autouse=True），让它在所有测试执行前后自动捕获并检查输出。只需要在项目根目录的conftest.py里写这段代码：

import pytest

@pytest.fixture(autouse=True)
def check_no_stray_output(capsys: pytest.CaptureFixture[str]):
    # 先让测试正常跑起来
    yield
    # 测试跑完后捕获输出并做断言
    out, err = capsys.readouterr()
    assert out == "", f"测试意外输出到stdout: {repr(out)}"
    assert err == "", f"测试意外输出到stderr: {repr(err)}"

原理很简单：yield之前的代码会在测试开始前执行，yield之后的代码是测试结束后执行的清理/检查逻辑。这样所有测试都会自动被检查，只要有任何stdout或stderr输出，测试就会直接失败，还会告诉你具体输出了什么内容。

如果有个别测试确实需要输出（比如测试日志功能的用例），可以给它加个标记跳过检查：

1. 先修改上面的fixture，加上标记判断：

import pytest

@pytest.fixture(autouse=True)
def check_no_stray_output(capsys: pytest.CaptureFixture[str], request):
    # 如果测试带有no_check_output标记，就跳过检查
    if request.node.get_closest_marker("no_check_output"):
        yield
        return
    yield
    out, err = capsys.readouterr()
    assert out == "", f"测试意外输出到stdout: {repr(out)}"
    assert err == "", f"测试意外输出到stderr: {repr(err)}"

2. 然后在需要跳过的测试上加上标记：

@pytest.mark.no_check_output
def test_legitimate_output():
    print("这是合法的输出，不会触发断言失败")

二、禁用stdin读取（可选） #

你提到还想让读取stdin的操作抛出异常，这个也可以用fixture实现，同样放在conftest.py里：

import sys
from io import StringIO

@pytest.fixture(autouse=True)
def disable_stdin():
    # 先保存原来的stdin，测试结束后要恢复
    original_stdin = sys.stdin
    # 自定义一个会抛出异常的stdin替代类
    class ForbiddenStdin(StringIO):
        def read(self, *args, **kwargs):
            raise RuntimeError("禁止读取标准输入！")
        def readline(self, *args, **kwargs):
            raise RuntimeError("禁止读取标准输入！")
        # 可以根据需要覆盖其他读取方法，比如readlines、__iter__等
    sys.stdin = ForbiddenStdin()
    yield
    # 测试结束后恢复原来的stdin，避免影响其他测试
    sys.stdin = original_stdin

这样不管测试里是用input()还是直接调用sys.stdin.read()，都会立刻抛出RuntimeError，让测试失败，从根源上杜绝非法的stdin读取。

把这两个fixture都放在conftest.py里，pytest会自动加载它们，你的整个测试套件就会自动执行这些检查了，再也不用手动一个个加代码啦！

备注：内容来源于stack exchange，提问作者zwol