tests/micropython: Improve skipping of tests using micropython module.

Signed-off-by: Damien George <damien@micropython.org>

tests/micropython/emg_exc.py

@@ -1,10 +1,9 @@
 # test that emergency exceptions work
 
-import micropython
 import sys
 
 try:
-    import io
+    import io, micropython
 except ImportError:
     print("SKIP")
     raise SystemExit

tests/micropython/emg_exc.py.exp

@@ -1,4 +1,4 @@
 Traceback (most recent call last):
-, line 22, in f
+, line 21, in f
 ValueError: 1
 

tests/micropython/extreme_exc.py

@@ -1,6 +1,10 @@
 # test some extreme cases of allocating exceptions and tracebacks
 
-import micropython
+try:
+    import micropython
+except ImportError:
+    print("SKIP")
+    raise SystemExit
 
 # Check for stackless build, which can't call functions without
 # allocating a frame on the heap.

tests/micropython/heap_lock.py

@@ -1,6 +1,10 @@
 # check that heap_lock/heap_unlock work as expected
 
-import micropython
+try:
+    import micropython
+except ImportError:
+    print("SKIP")
+    raise SystemExit
 
 l = []
 l2 = list(range(100))

tests/micropython/heap_locked.py

@@ -1,8 +1,10 @@
 # test micropython.heap_locked()
 
-import micropython
+try:
+    import micropython
 
-if not hasattr(micropython, "heap_locked"):
+    micropython.heap_locked
+except (AttributeError, ImportError):
     print("SKIP")
     raise SystemExit
 

tests/micropython/heapalloc.py

@@ -1,6 +1,10 @@
 # check that we can do certain things without allocating heap memory
 
-import micropython
+try:
+    import micropython
+except ImportError:
+    print("SKIP")
+    raise SystemExit
 
 # Check for stackless build, which can't call functions without
 # allocating a frame on heap.

tests/micropython/heapalloc_exc_compressed.py

@@ -1,4 +1,10 @@
-import micropython
+try:
+    import micropython
+
+    micropython.heap_lock
+except (ImportError, AttributeError):
+    print("SKIP")
+    raise SystemExit
 
 # Tests both code paths for built-in exception raising.
 # mp_obj_new_exception_msg_varg (exception requires decompression at raise-time to format)

tests/micropython/heapalloc_exc_compressed_emg_exc.py

@@ -1,4 +1,10 @@
-import micropython
+try:
+    import micropython
+
+    micropython.heap_lock
+except (ImportError, AttributeError):
+    print("SKIP")
+    raise SystemExit
 
 # Does the full test from heapalloc_exc_compressed.py but while the heap is
 # locked (this can only work when the emergency exception buf is enabled).

tests/micropython/heapalloc_exc_raise.py

@@ -1,6 +1,12 @@
 # Test that we can raise and catch (preallocated) exception
 # without memory allocation.
-import micropython
+try:
+    import micropython
+
+    micropython.heap_lock
+except (ImportError, AttributeError):
+    print("SKIP")
+    raise SystemExit
 
 e = ValueError("error")
 

tests/micropython/heapalloc_fail_bytearray.py

@@ -1,6 +1,12 @@
 # test handling of failed heap allocation with bytearray
 
-import micropython
+try:
+    import micropython
+
+    micropython.heap_lock
+except (ImportError, AttributeError):
+    print("SKIP")
+    raise SystemExit
 
 
 class GetSlice:

tests/micropython/heapalloc_fail_dict.py

@@ -1,6 +1,12 @@
 # test handling of failed heap allocation with dict
 
-import micropython
+try:
+    import micropython
+
+    micropython.heap_lock
+except (ImportError, AttributeError):
+    print("SKIP")
+    raise SystemExit
 
 # create dict
 x = 1

tests/micropython/heapalloc_fail_list.py

@@ -1,6 +1,12 @@
 # test handling of failed heap allocation with list
 
-import micropython
+try:
+    import micropython
+
+    micropython.heap_lock
+except (ImportError, AttributeError):
+    print("SKIP")
+    raise SystemExit
 
 
 class GetSlice:

tests/micropython/heapalloc_fail_memoryview.py

@@ -1,6 +1,12 @@
 # test handling of failed heap allocation with memoryview
 
-import micropython
+try:
+    import micropython
+
+    micropython.heap_lock
+except (ImportError, AttributeError):
+    print("SKIP")
+    raise SystemExit
 
 
 class GetSlice:

tests/micropython/heapalloc_fail_set.py

@@ -1,6 +1,12 @@
 # test handling of failed heap allocation with set
 
-import micropython
+try:
+    import micropython
+
+    micropython.heap_lock
+except (ImportError, AttributeError):
+    print("SKIP")
+    raise SystemExit
 
 # create set
 x = 1

tests/micropython/heapalloc_fail_tuple.py

@@ -1,6 +1,12 @@
 # test handling of failed heap allocation with tuple
 
-import micropython
+try:
+    import micropython
+
+    micropython.heap_lock
+except (ImportError, AttributeError):
+    print("SKIP")
+    raise SystemExit
 
 # create tuple
 x = 1

tests/micropython/heapalloc_inst_call.py

@@ -1,6 +1,13 @@
 # Test that calling clazz.__call__() with up to at least 3 arguments
 # doesn't require heap allocation.
-import micropython
+
+try:
+    import micropython
+
+    micropython.heap_lock
+except (ImportError, AttributeError):
+    print("SKIP")
+    raise SystemExit
 
 
 class Foo0:

tests/micropython/heapalloc_int_from_bytes.py

@@ -1,6 +1,13 @@
 # Test that int.from_bytes() for small number of bytes generates
 # small int.
-import micropython
+
+try:
+    import micropython
+
+    micropython.heap_lock
+except (ImportError, AttributeError):
+    print("SKIP")
+    raise SystemExit
 
 micropython.heap_lock()
 print(int.from_bytes(b"1", "little"))

tests/micropython/heapalloc_str.py

@@ -1,5 +1,12 @@
 # String operations which don't require allocation
-import micropython
+
+try:
+    import micropython
+
+    micropython.heap_lock
+except (ImportError, AttributeError):
+    print("SKIP")
+    raise SystemExit
 
 micropython.heap_lock()
 

tests/micropython/heapalloc_super.py

@@ -1,5 +1,12 @@
 # test super() operations which don't require allocation
-import micropython
+
+try:
+    import micropython
+
+    micropython.heap_lock
+except (ImportError, AttributeError):
+    print("SKIP")
+    raise SystemExit
 
 # Check for stackless build, which can't call functions without
 # allocating a frame on heap.

tests/micropython/heapalloc_traceback.py

@@ -1,10 +1,9 @@
 # test that we can generate a traceback without allocating
 
-import micropython
 import sys
 
 try:
-    import io
+    import io, micropython
 except ImportError:
     print("SKIP")
     raise SystemExit

tests/micropython/heapalloc_traceback.py.exp

@@ -1,5 +1,5 @@
 StopIteration
 Traceback (most recent call last):
-  File  , line 25, in test
+  File  , line 24, in test
 StopIteration: 
 

tests/micropython/heapalloc_yield_from.py

@@ -1,6 +1,12 @@
 # Check that yield-from can work without heap allocation
 
-import micropython
+try:
+    import micropython
+
+    micropython.heap_lock
+except (ImportError, AttributeError):
+    print("SKIP")
+    raise SystemExit
 
 
 # Yielding from a function generator

tests/micropython/kbd_intr.py

@@ -1,10 +1,10 @@
 # test the micropython.kbd_intr() function
 
-import micropython
-
 try:
+    import micropython
+
     micropython.kbd_intr
-except AttributeError:
+except (ImportError, AttributeError):
     print("SKIP")
     raise SystemExit
 

tests/micropython/meminfo.py

@@ -1,9 +1,10 @@
 # tests meminfo functions in micropython module
 
-import micropython
+try:
+    import micropython
 
-# these functions are not always available
+    micropython.mem_info
-if not hasattr(micropython, "mem_info"):
+except (ImportError, AttributeError):
     print("SKIP")
     raise SystemExit
 

tests/micropython/memstats.py

@@ -1,9 +1,10 @@
 # tests meminfo functions in micropython module
 
-import micropython
+try:
+    import micropython
 
-# these functions are not always available
+    micropython.mem_total
-if not hasattr(micropython, "mem_total"):
+except (ImportError, AttributeError):
     print("SKIP")
     raise SystemExit
 

tests/micropython/opt_level.py

@@ -1,4 +1,10 @@
-import micropython as micropython
+# test micropython.opt_level()
+
+try:
+    import micropython
+except ImportError:
+    print("SKIP")
+    raise SystemExit
 
 # check we can get and set the level
 micropython.opt_level(0)

tests/micropython/opt_level_lineno.py

@@ -1,4 +1,10 @@
-import micropython as micropython
+# test micropython.opt_level() and line numbers
+
+try:
+    import micropython
+except ImportError:
+    print("SKIP")
+    raise SystemExit
 
 # check that level 3 doesn't store line numbers
 # the expected output is that any line is printed as "line 1"

tests/micropython/ringio.py

@@ -1,10 +1,10 @@
 # Check that micropython.RingIO works correctly.
 
-import micropython
-
 try:
+    import micropython
+
     micropython.RingIO
-except AttributeError:
+except (ImportError, AttributeError):
     print("SKIP")
     raise SystemExit
 

tests/micropython/ringio_async.py

@@ -1,9 +1,7 @@
 # Check that micropython.RingIO works correctly with asyncio.Stream.
 
-import micropython
-
 try:
-    import asyncio
+    import asyncio, micropython
 
     asyncio.StreamWriter
     micropython.RingIO

tests/micropython/ringio_big.py

@@ -1,10 +1,10 @@
 # Check that micropython.RingIO works correctly.
 
-import micropython
-
 try:
+    import micropython
+
     micropython.RingIO
-except AttributeError:
+except (ImportError, AttributeError):
     print("SKIP")
     raise SystemExit
 

tests/micropython/schedule.py

@@ -1,10 +1,10 @@
 # test micropython.schedule() function
 
-import micropython
-
 try:
+    import micropython
+
     micropython.schedule
-except AttributeError:
+except (ImportError, AttributeError):
     print("SKIP")
     raise SystemExit
 

tests/micropython/stack_use.py

@@ -1,7 +1,10 @@
 # tests stack_use function in micropython module
-import micropython
 
-if not hasattr(micropython, "stack_use"):
+try:
+    import micropython
+
+    micropython.stack_use
+except (ImportError, AttributeError):
     print("SKIP")
     raise SystemExit