tests/micropython/heapalloc_iter: Improve skippability.

1 files changed, 7 insertions, 2 deletions

diff --git a/tests/micropython/heapalloc_iter.py b/tests/micropython/heapalloc_iter.py
index e1ed3daa3b..45d3519e45 100644
--- a/tests/micropython/heapalloc_iter.py
+++ b/tests/micropython/heapalloc_iter.py
@@ -1,10 +1,15 @@
 # test that iterating doesn't use the heap
+try:
+    import array
+except ImportError:
+    import sys
+    print("SKIP")
+    sys.exit()
 
 try:
     from micropython import heap_lock, heap_unlock
 except (ImportError, AttributeError):
     heap_lock = heap_unlock = lambda:0
-import array
 
 def do_iter(l):
     for i in l:
@@ -20,7 +25,7 @@ ar = array.array('H', (123, 456))
 t = (1, 2, 3)
 l = [1, 2]
 d = {1:2}
-s = {1}
+s = set((1,))
 fs = frozenset((1,))
 g1 = (100 + x for x in range(2))
 g2 = gen_func()