Pytest中print的坑

在Python中，print语句用于输出信息到控制台。然而，在Pytest测试框架中，print语句可能会引发一些问题，因为Pytest重定向了标准输出（stdout）到自己的日志记录系统。这意味着当你在测试中使用print语句时，输出不会被显示在控制台上，而是被记录在Pytest的日志中。这可能会导致一些困惑，特别是当你期望看到某些输出以便验证代码行为时。
要解决这个问题，你可以使用Pytest的fixture功能来捕获print语句的输出并将其存储在变量中。这样，你就可以在测试中进行断言以验证输出是否符合预期。
下面是一个示例代码，演示了如何在Pytest中使用fixture捕获print语句的输出：

import pytest
@pytest.fixture
def capture_output():
import sys
from io import StringIO
old_stdout = sys.stdout
new_stdout = StringIO()
sys.stdout = new_stdout
yield new_stdout
sys.stdout = old_stdout

然后，在你的测试函数中使用这个fixture，如下所示：

def test_my_function(capture_output):
print('This is a test output')
output = capture_output.getvalue()
assert output == 'This is a test output
'

在这个示例中，我们定义了一个名为capture_output的fixture，它使用StringIO模块创建一个新的字符串输出流。在测试函数中，我们使用这个fixture并调用getvalue()方法来获取print语句的输出。然后，我们可以使用断言来验证输出是否符合预期。
请注意，这种方法只适用于捕获print语句的输出。如果你需要在测试中执行其他类型的输出操作（如文件写入），可能需要使用其他方法来捕获这些输出。另外，使用这种方法会覆盖标准输出流（stdout），因此在测试完成后需要恢复原始的标准输出流。这可以通过在fixture的yield语句之后将标准输出流设置回原始值来实现。
总之，虽然print语句在Pytest中可能会导致一些问题，但通过使用fixture和其他技巧，你可以成功捕获print语句的输出并进行断言验证。通过遵循这些方法，你可以避免潜在的问题并确保你的Pytest测试能够准确地验证代码的行为。