tests/micropython/heapalloc_iter: Improve skippability.

pfalcon · pfalcon · commit 806c07c8989e · 2017-04-03T00:27:01.000+03:00
diff --git 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 @@ def gen_func():
 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()
