tools/testrunner: minor code beautifications

This commit is contained in:
Hauke Petersen 2017-11-30 10:02:58 +01:00
parent b62ebf5e44
commit 2ead5ef3a0

View File

@ -26,9 +26,7 @@ def find_exc_origin(exc_info):
pos = list_until(extract_tb(exc_info), pos = list_until(extract_tb(exc_info),
lambda frame: frame[0].startswith(PEXPECT_PATH) lambda frame: frame[0].startswith(PEXPECT_PATH)
)[-1] )[-1]
return pos[3], \ return (pos[3], os.path.relpath(os.path.abspath(pos[0]), RIOTBASE), pos[1])
os.path.relpath(os.path.abspath(pos[0]), RIOTBASE), \
pos[1]
def run(testfunc, timeout=10, echo=True, traceback=False): def run(testfunc, timeout=10, echo=True, traceback=False):
env = os.environ.copy() env = os.environ.copy()
@ -49,16 +47,14 @@ def run(testfunc, timeout=10, echo=True, traceback=False):
try: try:
testfunc(child) testfunc(child)
except pexpect.TIMEOUT: except pexpect.TIMEOUT:
line, filename, lineno = find_exc_origin(sys.exc_info()[2]) trace = find_exc_origin(sys.exc_info()[2])
print("Timeout in expect script at \"%s\" (%s:%d)" % print("Timeout in expect script at \"%s\" (%s:%d)" % trace)
(line, filename, lineno))
if traceback: if traceback:
print_tb(sys.exc_info()[2]) print_tb(sys.exc_info()[2])
return 1 return 1
except pexpect.EOF: except pexpect.EOF:
line, filename, lineno = find_exc_origin(sys.exc_info()[2]) trace = find_exc_origin(sys.exc_info()[2])
print("Unexpected end of file in expect script at \"%s\" (%s:%d)" % print("Unexpected end of file in expect script at \"%s\" (%s:%d)" % trace)
(line, filename, lineno))
if traceback: if traceback:
print_tb(sys.exc_info()[2]) print_tb(sys.exc_info()[2])
return 1 return 1