1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
from contextlib import contextmanager
import sys
import traceback
# ######### Testing tools
@contextmanager
def store_exceptions_on_all_threads():
"""Context manager that captures all exceptions, even those coming from
the UI thread. On exit, the first exception is raised (if any).
"""
exceptions = []
def excepthook(type, value, tb):
exceptions.append(value)
message = 'Uncaught exception:\n'
message += ''.join(traceback.format_exception(type, value, tb))
sys.stderr.write(message)
try:
sys.excepthook = excepthook
yield
finally:
if len(exceptions) > 0:
raise exceptions[0]
sys.excepthook = sys.__excepthook__
@contextmanager
def assert_raises(ExceptionClass):
try:
yield
except ExceptionClass:
pass
else:
msg = 'Test should have failed with {}.'
raise Exception(msg.format(ExceptionClass.__name__))
|
