tests/micropython: Add tests for try and with blocks under native/viper.

1 files changed, 39 insertions, 0 deletions

diff --git a/tests/micropython/native_try.py b/tests/micropython/native_try.py
new file mode 100644
index 000000000..2e41bf2ea
--- /dev/null
+++ b/tests/micropython/native_try.py
@@ -0,0 +1,39 @@
+# test native try handling
+
+# basic try-finally
+@micropython.native
+def f():
+    try:
+        fail
+    finally:
+        print('finally')
+try:
+    f()
+except NameError:
+    print('NameError')
+
+# nested try-except with try-finally
+@micropython.native
+def f():
+    try:
+        try:
+            fail
+        finally:
+            print('finally')
+    except NameError:
+        print('NameError')
+f()
+
+# check that locals written to in try blocks keep their values
+@micropython.native
+def f():
+    a = 100
+    try:
+        print(a)
+        a = 200
+        fail
+    except NameError:
+        print(a)
+        a = 300
+    print(a)
+f()