使用unittest.mock.patch捕获stdout时无法获取print输出的问题咨询

我来帮你分析下这个问题哈～

你遇到的情况并不是没有捕获到输出，而是你在测试函数里的打印操作也被patch给拦截了，导致这些内容没有显示到控制台而已。

问题原因 #

当你用@patch('sys.stdout', new_callable=io.StringIO)装饰测试函数时，整个测试函数执行期间，sys.stdout都被替换成了mock_stdout这个StringIO对象：

• greet_user("Alice")里的print确实把内容写入了mock_stdout
• 但你之后调用的print("OUTPUT WITH PATCH:")和print(mock_stdout.getvalue())，同样会把内容写入mock_stdout，而不是真实的控制台

所以这些内容都存在mock_stdout里，但没机会显示到屏幕上，让你误以为没捕获到输出。

解决方法 #

根据你的需求，有两种常见的处理方式：

方式1：验证输出是否正确（推荐的测试方式）

单元测试的核心是验证结果，而不是手动打印查看。你可以直接断言捕获的输出是否符合预期：

@patch('sys.stdout', new_callable=io.StringIO)
def test_with_patch(self, mock_stdout):
    greet_user("Alice")
    # 断言捕获的输出是否正确（strip()去掉换行符的影响）
    self.assertEqual(mock_stdout.getvalue().strip(), "Hello, Alice!")

这样测试会自动验证输出，运行测试时如果结果符合预期就会通过，不符合则报错。

方式2：将捕获的内容打印到真实控制台

如果你确实需要把捕获的内容显示到控制台，可以直接调用系统原本的stdout来写入：

@patch('sys.stdout', new_callable=io.StringIO)
def test_with_patch(self, mock_stdout):
    greet_user("Alice")
    captured_output = mock_stdout.getvalue()
    # 使用sys.__stdout__访问真实的标准输出
    import sys
    sys.__stdout__.write("OUTPUT WITH PATCH:\n")
    sys.__stdout__.write(captured_output)

这样就能在控制台看到捕获到的输出内容了。

补充说明 #

你看到的测试输出里，.OUTPUT WITHOUT PATCH:的顺序是因为unittest会先输出测试进度的.，然后test_without_patch里的print直接写入真实控制台，所以会显示对应的内容；而test_with_patch里的打印都被拦截了，所以没显示出来。

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