[Python-checkins] r66060 - python/trunk/Lib/test/crashers/iter.py

[armin.rigo]

Author: armin.rigo
Date: Fri Aug 29 23:21:52 2008
New Revision: 66060

Log:
A collection of crashers, all variants of the idea
of issue #3720.


Added:
   python/trunk/Lib/test/crashers/iter.py   (contents, props changed)

Added: python/trunk/Lib/test/crashers/iter.py
==============================================================================
--- (empty file)
+++ python/trunk/Lib/test/crashers/iter.py	Fri Aug 29 23:21:52 2008
@@ -0,0 +1,53 @@
+# Calls to PyIter_Next, or direct calls to tp_iternext, on an object
+# which might no longer be an iterable because its 'next' method was
+# removed.  These are all variants of Issue3720.
+
+"""
+Run this script with an argument between 1 and <N> to test for
+different crashes.
+"""
+N = 8
+
+import sys
+
+class Foo(object):
+    def __iter__(self):
+        return self
+    def next(self):
+        del Foo.next
+        return (1, 2)
+
+def case1():
+    list(enumerate(Foo()))
+
+def case2():
+    x, y = Foo()
+
+def case3():
+    filter(None, Foo())
+
+def case4():
+    map(None, Foo(), Foo())
+
+def case5():
+    max(Foo())
+
+def case6():
+    sum(Foo(), ())
+
+def case7():
+    dict(Foo())
+
+def case8():
+    sys.stdout.writelines(Foo())
+
+# etc...
+
+
+if __name__ == '__main__':
+    if len(sys.argv) < 2:
+        print __doc__.replace('<N>', str(N))
+    else:
+        n = int(sys.argv[1])
+        func = globals()['case%d' % n]
+        func()