防止自定义断言显示在 python unittest 的回溯中答案

【问题标题】：Prevent custom assert from showing in traceback of python unittest防止自定义断言显示在 python unittest 的回溯中
【发布时间】：2018-04-19 18:29:44
【问题描述】：

我想向 TestCase 添加一些自定义断言方法。作为一个简单的例子，我只是在下面的测试类中放了一个。它按预期工作，但是当生成输出时，回溯会在输出中包含自定义断言。

要让它表现得像 assertEqual() 的必要步骤是什么？ assertEqual 的代码在 TestCase 中，但引发断言的实际行没有出现在回溯中。我需要做什么才能使 test_something2 的输出看起来更像 test_something1 的？

import unittest
class TestCustomAssert(unittest.TestCase):
    def assertSomething(self, s):
        self.assertEqual(s, 'something')
    def test_something1(self):
        self.assertEqual('foo', 'something')
    def test_something2(self):
        self.assertSomething('foo')

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

输出

python3 custom_assert.py 
FF
======================================================================
FAIL: test_something1 (__main__.TestCustomAssert)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "custom_assert.py", line 8, in test_something1
    self.assertEqual('foo', 'something')
AssertionError: 'foo' != 'something'
- foo                                                                                                                    
+ something                                                                                                              


======================================================================                                                   
FAIL: test_something2 (__main__.TestCustomAssert)                                                                        
----------------------------------------------------------------------                                                   
Traceback (most recent call last):                                                                                       
  File "custom_assert.py", line 10, in test_something2                                                                   
    self.assertSomething('foo')                                                                                          
  File "custom_assert.py", line 6, in assertSomething                                                                    
    self.assertEqual(s, 'something')                                                                                     
AssertionError: 'foo' != 'something'                                                                                     
- foo                                                                                                                    
+ something                                                                                                              


----------------------------------------------------------------------                                                   
Ran 2 tests in 0.000s                                                                                                    

FAILED (failures=2)

【问题讨论】：

标签： python-3.x python-unittest

【解决方案1】：

unittest 不会从 globals 中包含 __unittest=True 的帧打印回溯。

来自unittest.result：

def _is_relevant_tb_level(self, tb):
    return '__unittest' in tb.tb_frame.f_globals

所以，如果你制作了一个辅助模块，你可以模仿那个行为：

helper.py：

__unittest = True
def assert_stuff(s):
    assert s == 'something', "%s is not something" % s

现在你可以从你的测试用例中调用这个助手了。

我想可以制作一个非常简洁的装饰器，让这种魔法在没有辅助模块的情况下更加自动化，但恕我直言，无论如何你都不应该努力减少回溯。

【讨论】：