def test_formatting(self):
with pytest.raises(Buzz) as err_info:
raise Buzz('fail message with no formatting')
assert 'fail message with no formatting' in str(err_info.value)
with pytest.raises(Buzz) as err_info:
raise Buzz('fail message with {}', 'arg formatting')
assert 'fail message with arg formatting' in str(err_info.value)
with pytest.raises(Buzz) as err_info:
raise Buzz('fail message with {fmt}', fmt='kwarg formatting')
assert 'fail message with kwarg formatting' in str(err_info.value)
def test_get_traceback(self):
try:
raise Buzz("Original Error")
except Exception as err:
trace = err.get_traceback()
last_frame = traceback.format_tb(trace)[-1]
assert 'test_base.py' in last_frame
assert 'test_get_traceback' in last_frame
assert 'Buzz("Original Error")' in last_frame
