From: Ezio Melotti <ezio.melotti@gmail.com>
Date: Sun, 18 Nov 2012 21:14:42 +0000 (+0200)
Subject: #7782: add a test for test_iter.
X-Git-Tag: v2.7.4rc1~361
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=6451497df166adf4e57689f571d840d3cca3f12d;p=thirdparty%2FPython%2Fcpython.git

#7782: add a test for test_iter.
---

diff --git a/Lib/test/test_iter.py b/Lib/test/test_iter.py
index 64984cc6e6a5..8dc39d2e43fa 100644
--- a/Lib/test/test_iter.py
+++ b/Lib/test/test_iter.py
@@ -908,6 +908,21 @@ class TestCase(unittest.TestCase):
         except TypeError:
             pass
 
+    def test_extending_list_with_iterator_does_not_segfault(self):
+        # The code to extend a list with an iterator has a fair
+        # amount of nontrivial logic in terms of guessing how
+        # much memory to allocate in advance, "stealing" refs,
+        # and then shrinking at the end.  This is a basic smoke
+        # test for that scenario.
+        def gen():
+            for i in range(500):
+                yield i
+        lst = [0] * 500
+        for i in range(240):
+            lst.pop(0)
+        lst.extend(gen())
+        self.assertEqual(len(lst), 760)
+
 
 def test_main():
     run_unittest(TestCase)