From 806c07c8989e44e1be4832fd37688e2b5828b3af Mon Sep 17 00:00:00 2001
From: Paul Sokolovsky <pfalcon@users.sourceforge.net>
Date: Mon, 3 Apr 2017 00:27:01 +0300
Subject: [PATCH] tests/micropython/heapalloc_iter: Improve skippability.

---
 tests/micropython/heapalloc_iter.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/tests/micropython/heapalloc_iter.py b/tests/micropython/heapalloc_iter.py
index e1ed3daa3..45d3519e4 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()
-- 
GitLab