tests: Add test for recursive iternext stack overflow.

micropython · Jun 3, 2015 · 80f638f · 80f638f
1 parent 953c23b
commit 80f638f
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 0 deletions.
diff --git a/tests/misc/recursive_iternext.py b/tests/misc/recursive_iternext.py
@@ -0,0 +1,33 @@
+# This tests that recursion with iternext doesn't lead to segfault.
+
+try:
+    x = (1, 2)
+    for i in range(1000):
+        x = enumerate(x)
+    tuple(x)
+except RuntimeError:
+    print("RuntimeError")
+
+try:
+    x = (1, 2)
+    for i in range(1000):
+        x = filter(None, x)
+    tuple(x)
+except RuntimeError:
+    print("RuntimeError")
+
+try:
+    x = (1, 2)
+    for i in range(1000):
+        x = map(max, x, ())
+    tuple(x)
+except RuntimeError:
+    print("RuntimeError")
+
+try:
+    x = (1, 2)
+    for i in range(1000):
+        x = zip(x)
+    tuple(x)
+except RuntimeError:
+    print("RuntimeError")
diff --git a/tests/misc/recursive_iternext.py.exp b/tests/misc/recursive_iternext.py.exp
@@ -0,0 +1,4 @@
+RuntimeError
+RuntimeError
+RuntimeError
+RuntimeError