如何正确测试Python中input()调用时的stdout输出？

首先，你的问题核心出在**patch装饰器的参数传递顺序上！Python的unittest.mock.patch装饰器是按从下往上**的顺序把mock对象传递给测试方法的，你把参数顺序搞反了，才导致拿到的mock_stdout其实是input的MagicMock对象，自然断言会失败。

错误原因分析 #

你写的修复代码里，装饰器顺序是：

@patch('sys.stdout', new_callable=StringIO)
@patch('builtins.input', return_value='0')

但测试方法的参数是(self, mock_stdout, input)——这就错了！因为@patch('builtins.input')是更靠近测试方法的装饰器（也就是代码里更靠下的那个），它对应的mock应该是方法的第一个参数，而@patch('sys.stdout')的mock是第二个参数。

正确的测试代码 #

调整参数顺序，同时给mock对象起更清晰的名字避免混淆：

from unittest import TestCase, patch
from io import StringIO
from your_module import GameDisplay  # 替换成实际的模块名

class TestGameDisplay(TestCase):
    def setUp(self):
        self.game_display = GameDisplay()  # 初始化被测试对象

    @patch('builtins.input', return_value='0')
    @patch('sys.stdout', new_callable=StringIO)
    def test_prompt_output(self, mock_stdout, mock_input):
        # 调用被测试方法
        self.game_display.prompt('Choose 0: ')
        # 断言stdout的输出正确
        self.assertEqual(mock_stdout.getvalue(), 'Choose 0: ')

额外优化说明 #

• 参数顺序规则：记住patch装饰器的参数传递是逆序的——最靠近测试方法的装饰器对应的mock是方法的第一个参数，依次往上推。
• 为什么键盘模拟不规范：用pynput模拟键盘输入依赖系统级事件，不仅不稳定，还会干扰用户操作，完全不如直接mockbuiltins.input可靠。
• 验证输入返回值：如果需要验证prompt方法的返回值是否正确，还可以加一行断言：

  result = self.game_display.prompt('Choose 0: ')
  self.assertEqual(result, '0')
  

这样修改后，测试就能正确捕获stdout的输出，断言也会顺利通过啦。

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