如何将pytest.main的输出信息写入字符串对象？

嘿，这个问题我之前也碰到过！pytest.main()确实只返回代表错误码的整数，官方文档里没明确提捕获输出的方法，但我们可以通过重定向系统标准输出和错误流来实现，直接上可行方案：

方法：用StringIO临时重定向输出流 #

这个思路是临时把sys.stdout和sys.stderr替换成内存中的字符串流，让pytest的所有输出都写到这个流里，之后再把原始流恢复，最后读取流中的内容即可。

import sys
from io import StringIO
import pytest

def capture_pytest_test_output():
    # 先保存原始的标准输出和错误流，避免后续程序输出异常
    original_stdout = sys.stdout
    original_stderr = sys.stderr
    
    try:
        # 创建内存字符串流，用来捕获输出
        captured_stream = StringIO()
        sys.stdout = captured_stream
        sys.stderr = captured_stream
        
        # 运行你的pytest命令，这里替换成你需要的参数
        exit_code = pytest.main(["-v", "path/to/your/tests/"])
        
        # 读取捕获到的所有输出内容
        test_output = captured_stream.getvalue()
        return exit_code, test_output
    finally:
        # 不管pytest运行成功还是失败，都要恢复原始流
        sys.stdout = original_stdout
        sys.stderr = original_stderr

# 调用示例
exit_code, output_str = capture_pytest_test_output()
print("=== 捕获到的pytest输出 ===")
print(output_str)

关键细节说明 #

• 为什么要恢复原始流？ 如果不恢复，后续程序的print()或者其他输出操作会继续写到我们创建的StringIO里，导致你看不到控制台输出，所以一定要在finally块里恢复，确保万无一失。
• 捕获完整输出： pytest的部分日志可能会写到stderr，所以我们同时重定向了stdout和stderr，这样能捕获到所有测试相关的输出信息。
• 自定义pytest参数： 你可以像在命令行一样给pytest.main()传参数，比如["-x", "--tb=short", "test_module.py"]，完全和命令行调用的参数一致。

如果是在pytest测试用例内部捕获单个测试的输出，那可以用pytest自带的capsys或capfd fixture，但这个场景下我们是外部调用pytest.main()，上面的方法是最直接有效的。

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