禁止单元测试中的打印输出

Question

编辑：请注意我使用的是 Python 2.6（已标记）

假设我有以下内容：

class Foo:
    def bar(self):
        print 'bar'
        return 7

假设我有以下单元测试：

import unittest
class ut_Foo(unittest.TestCase):
    def test_bar(self):
        obj = Foo()
        res = obj.bar()
        self.assertEqual(res, 7)

所以如果我跑：

unittest.main()

我明白了：

bar # <-- I don't want this, but I *do* want the rest
.
----------------------------------------------------------------------
Ran 1 test in 0.002s

OK
Exit code:  False

我的问题是：有没有办法抑制正在测试的对象的输出，同时仍然获得 unittest 框架的输出？

编辑 这个问题不是 flagged question 的重复问题，flagged question 询问关于在普通 python 脚本中使特定函数的标准输出静音。

而这个问题是关于在运行它的单元测试时隐藏 python 脚本的正常标准输出。我仍然希望显示 unittest 标准输出，并且我不想禁用我测试脚本的标准输出。

Answer 1

使用选项“-b”调用您的单元测试 - 缓冲区标准输出和标准错误

Foo.py

class Foo:
    def bar(self):
        print "bar"
        return 7

test.py

import unittest
from Foo import Foo

class test_Foo(unittest.TestCase):
    def test_bar(self):
        obj = Foo()
        res = obj.bar()
        self.assertEqual(res, 7)

if __name__ == "__main__":
    unittest.main()

使用 -b 选项运行它

$ python test.py -b
.
----------------------------------------------------------------------
Ran 1 test in 0.000s

OK

替代方案：使用nose

$ pip install nose

什么安装命令nosetests

注意，我已经修改了测试套件，使其具有以test 为前缀的类和方法，以满足nose 默认测试发现规则。

nosetests 默认不显示输出

$ nosetests
.
----------------------------------------------------------------------
Ran 1 test in 0.002s

OK

如果要查看输出，请使用-s 开关：

$ nosetests -s
bar
.
----------------------------------------------------------------------
Ran 1 test in 0.002s

OK

Answer 2

您可以通过禁用 sys.stdout 并在测试完成后启用它来抑制输出：

import sys
import io
import unittest

class ut_Foo(unittest.TestCase):
    def test_bar(self):

        #You suppress here:
        suppress_text = io.StringIO()
        sys.stdout = suppress_text 
        
        obj = Foo()
        res = obj.bar()
        self.assertEqual(res, 7)
        
        #You release here:
        sys.stdout = sys.__stdout__

所有这些都来自：

https://codingdose.info/2018/03/22/supress-print-output-in-python/