tests/micropython: Get heapalloc_traceback test running on baremetal.

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).

1 files changed, 14 insertions, 2 deletions

diff --git a/tests/micropython/heapalloc_traceback.py b/tests/micropython/heapalloc_traceback.py
index 9f43da259..808df0225 100644
--- a/tests/micropython/heapalloc_traceback.py
+++ b/tests/micropython/heapalloc_traceback.py
@@ -2,6 +2,7 @@
 
 import micropython
 import sys
+import uio
 
 # preallocate exception instance with some room for a traceback
 global_exc = StopIteration()
@@ -15,10 +16,21 @@ def test():
     global_exc.__traceback__ = None
     try:
         raise global_exc
-    except StopIteration as e:
-        sys.print_exception(e)
+    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)