Python：在单元测试期间尝试捕获输出时出现问题 - Unittest 库答案

【问题标题】：Python: Problem trying catching an output during a unit test - Unittest libraryPython：在单元测试期间尝试捕获输出时出现问题 - Unittest 库
【发布时间】：2019-05-28 18:31:14
【问题描述】：

我正在尝试捕获要测试的输出，我正在使用 unittest 和 unittest.mock 库。我查看了许多其他相关问题（例如：How to assert output with nosetest/unittest in python? 等），他们提出的所有问题都类似于我的代码中包含的内容，但我得到的只是一个空字符串，并且找不到我在哪里搞错了。

我的函数做这样的事情：

def my_func():
  a = int(input("put a 2 digit number"))
  b = [1,2,3,4,5,6,7,8,9,0]
  if a in b:
    print("wrong number")

这是相关的测试：

import io
import sys
import unittest
from unittest.mock import patch
from io import StringIO
from somemodule import my_func

def test_entering_wrong_value(self):

        user_input = ['2']

        with patch('builtins.input', side_effect=user_input):
            self.my_func()
            capturedOutput = io.StringIO()
            sys.stdout = capturedOutput
            output = capturedOutput.getvalue().strip()
            self.assertEqual('wrong number', output)

我不明白这应该捕获输出的字符串并与测试值进行比较，但无法捕获字符串，我只是得到一个空字符串和这个错误：

AssertionError: 'wrong number' != ''

我在哪里搞砸了？

【问题讨论】：

标签： python-3.x unit-testing io python-unittest

【解决方案1】：

在打印之前，您必须重定向 stdout。您无法在之后重定向并期望获得未被重定向的prints：

import io
import sys
import unittest
from unittest.mock import patch
from somemodule import my_func

def test_entering_wrong_value(self):

        user_input = ['2']

        with patch('builtins.input', side_effect=user_input):
            capturedOutput = io.StringIO()
            sys.stdout = capturedOutput
            self.my_func()
            output = capturedOutput.getvalue().strip()
            sys.stdout = sys.__stdout__ # Restore stdout so prints work normally
            self.assertEqual('wrong number', output)

【讨论】：