Вот декоратор, который должен делать снимок экрана после каждого исключения и неудачного утверждения, которое, я считаю, то, что вы хотите. Попробуйте что-то вроде следующего?
def decorator_screenshot(func):
def wrapper(func, *args, **kwargs):
try:
return func(*args, **kwargs)
except:
return get_screenshot()
return wrapper
@decorator_screenshot
def test_something():
Assert.fail("failed test")