在失败的断言中触发一个函数

Question

如果断言失败，是否有触发函数的方法？

我现在正在考虑unittest，但如果需要的话可以更改测试框架。

我想当断言失败

Answer 1

该采取截图：

try: 
    assert something == something_else 
except AssertionError: 
    take_screenshot() 

是你可以做的一种方式。

在某些情况下，你可能想使用一个装饰：

>>> def assertd(function): 
     def _f(*args): 
      try: 
       return function(*args) 
      except AssertionError: 
       takescreenshot() 
     return _f 

>>> @assertd 
def f(x, y): 
    assert x > 0 
    return x + y 

>>> f(1, 2) 
3 
>>> f(5, 6) 
11 
>>> f(-7, 8) 
<Screenshot taken>