def test_removes_exception_after_printing(self, sio):
"""
After pretty printing `exception` is removed from the event_dict.
"""
epp = ExceptionPrettyPrinter(sio)
try:
raise ValueError
except ValueError:
ed = format_exc_info(None, None, {"exc_info": True})
assert "exception" in ed
new_ed = epp(None, None, ed)
assert "exception" not in new_ed
def test_prints_exception(self, sio):
"""
If there's an `exception` key in the event_dict, just print it out.
This happens if `format_exc_info` was run before us in the chain.
"""
epp = ExceptionPrettyPrinter(file=sio)
try:
raise ValueError
except ValueError:
ed = format_exc_info(None, None, {'exc_info': True})
epp(None, None, ed)
out = sio.getvalue()
assert 'test_prints_exception' in out
assert 'raise ValueError' in out
def test_stdout_by_default(self):
"""
If no file is supplied, use stdout.
"""
epp = ExceptionPrettyPrinter()
assert sys.stdout is epp._file
