3dc324d3f1
This adds the Python files in the tests/ directory to be formatted with ./tools/codeformat.py. The basics/ subdirectory is excluded for now so we aren't changing too much at once. In a few places `# fmt: off`/`# fmt: on` was used where the code had special formatting for readability or where the test was actually testing the specific formatting.
44 lines
890 B
Python
44 lines
890 B
Python
# test that we can generate a traceback without allocating
|
|
|
|
import micropython
|
|
import sys
|
|
|
|
try:
|
|
import uio
|
|
except ImportError:
|
|
print("SKIP")
|
|
raise SystemExit
|
|
|
|
# preallocate exception instance with some room for a traceback
|
|
global_exc = StopIteration()
|
|
try:
|
|
raise global_exc
|
|
except:
|
|
pass
|
|
|
|
|
|
def test():
|
|
micropython.heap_lock()
|
|
global global_exc
|
|
global_exc.__traceback__ = None
|
|
try:
|
|
raise global_exc
|
|
except StopIteration:
|
|
print("StopIteration")
|
|
micropython.heap_unlock()
|
|
|
|
|
|
# call test() with heap allocation disabled
|
|
test()
|
|
|
|
# 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)
|