7081ea4119
When printing exceptions from files sent to a target by pyboard.py the filename in the exception is <stdin>, which differs to when running the script on the PC. So we strip out the filename to make the outputs the same on all targets (see also misc/print_exception.py test).
37 lines
811 B
Python
37 lines
811 B
Python
# test that we can generate a traceback without allocating
|
|
|
|
import micropython
|
|
import sys
|
|
import uio
|
|
|
|
# preallocate exception instance with some room for a traceback
|
|
global_exc = StopIteration()
|
|
try:
|
|
raise global_exc
|
|
except:
|
|
pass
|
|
|
|
def test():
|
|
global global_exc
|
|
global_exc.__traceback__ = None
|
|
try:
|
|
raise global_exc
|
|
except StopIteration:
|
|
print('StopIteration')
|
|
|
|
# call test() with heap allocation disabled
|
|
micropython.heap_lock()
|
|
test()
|
|
micropython.heap_unlock()
|
|
|
|
# print the exception that was raised
|
|
buf = uio.StringIO()
|
|
sys.print_exception(global_exc, buf)
|
|
for l in buf.getvalue().split("\n"):
|
|
# uPy on pyboard prints <stdin> as file, so remove filename.
|
|
if l.startswith(" File "):
|
|
l = l.split('"')
|
|
print(l[0], l[2])
|
|
else:
|
|
print(l)
|