python - python (2.7) unittest 的测试描述如何修改

Question

看来unittest模块在Python 2.7中改了不少

我有一个测试用例:

class DemoTest(unittest.TestCase):
  def test_foo(self):
      """Test foo"""
      pass

控制台输出为:

测试 foo ... ok

升级到 Python 2.7 后，控制台输出现在是:

test_foo (testcase.demotest.DemoTest)

测试 foo ... ok

第一行描述没用。我想隐藏它，但不知道该怎么做。

Answer 1

鉴于您已经为测试编写了文档字符串，额外的输出看起来有点多余。下面是一种可以抑制它的方法；您需要将其添加到测试文件的顶部:

from unittest.runner import TextTestResult
TextTestResult.getDescription = lambda _, test: test.shortDescription()