circuitpython/tests/micropython/heapalloc_traceback.py
2017-03-09 10:26:31 +01:00

42 lines
888 B
Python

# test that we can generate a traceback without allocating
import micropython
import sys
try:
import uio
except ImportError:
import sys
print("SKIP")
sys.exit()
# 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)