tests: Add feature test for when heap allocation is disabled.

1 files changed, 26 insertions, 0 deletions

diff --git a/tests/micropython/heapalloc.py b/tests/micropython/heapalloc.py
new file mode 100644
index 000000000..c62428a08
--- /dev/null
+++ b/tests/micropython/heapalloc.py
@@ -0,0 +1,26 @@
+# check that we can do certain things without allocating heap memory
+
+import gc
+
+def f(a):
+    print(a)
+
+def g(a, b=2):
+    print(a, b)
+
+global_var = 1
+
+def h():
+    global global_var
+    global_var = 2      # set an existing global variable
+    for i in range(2):  # for loop
+        f(i)            # function call
+        f(i * 2 + 1)    # binary operation with small ints
+        f(a=i)          # keyword arguments
+        g(i)            # default arg (second one)
+        g(i, i)         # 2 args
+
+# call h with heap allocation disabled
+gc.disable()
+h()
+gc.enable()