1 files changed, 17 insertions, 9 deletions
diff --git a/tests/micropython/heapalloc.py b/tests/micropython/heapalloc.py
index b4b27d19b7..2dc7fa5e7e 100644
--- a/tests/micropython/heapalloc.py
+++ b/tests/micropython/heapalloc.py
@@ -2,23 +2,31 @@
 
 import gc
 
-def f(a):
+def f1(a):
     print(a)
 
-def g(a, b=2):
+def f2(a, b=2):
     print(a, b)
 
+def f3(a, b, c, d):
+    x1 = x2 = a
+    x3 = x4 = b
+    x5 = x6 = c
+    x7 = x8 = d
+    print(x1, x3, x5, x7, x2 + x4 + x6 + x8)
+
 global_var = 1
 
-def h():
+def test():
     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
+        f1(i)           # function call
+        f1(i * 2 + 1)   # binary operation with small ints
+        f1(a=i)         # keyword arguments
+        f2(i)           # default arg (second one)
+        f2(i, i)        # 2 args
+    f3(1, 2, 3, 4)  # function with lots of local state
 
 # call h with heap allocation disabled and all memory used up
 gc.disable()
@@ -27,5 +35,5 @@ try:
         'a'.lower # allocates 1 cell for boundmeth
 except MemoryError:
     pass
-h()
+test()
 gc.enable()