summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_itertools.py
diff options
context:
space:
mode:
authorRaymond Hettinger <python@rcn.com>2003-08-29 23:09:58 (GMT)
committerRaymond Hettinger <python@rcn.com>2003-08-29 23:09:58 (GMT)
commita56f6b6600ea8906562cfaa31859b72a53a8c0d7 (patch)
tree5379c5da992031907cc63202247dd8fa185cb9df /Lib/test/test_itertools.py
parentb738041c5db7a04b71ed4c3c8128dabc935ad60b (diff)
downloadcpython-a56f6b6600ea8906562cfaa31859b72a53a8c0d7.zip
cpython-a56f6b6600ea8906562cfaa31859b72a53a8c0d7.tar.gz
cpython-a56f6b6600ea8906562cfaa31859b72a53a8c0d7.tar.bz2
SF bug #793826: using itertools.izip to mutate tuples
Avoid Armin Rigo's dastardly exercise in re-entrancy.
Diffstat (limited to 'Lib/test/test_itertools.py')
-rw-r--r--Lib/test/test_itertools.py33
1 files changed, 32 insertions, 1 deletions
diff --git a/Lib/test/test_itertools.py b/Lib/test/test_itertools.py
index f96ccd5..057b576 100644
--- a/Lib/test/test_itertools.py
+++ b/Lib/test/test_itertools.py
@@ -427,6 +427,36 @@ class TestVariousIteratorArgs(unittest.TestCase):
self.assertRaises(TypeError, list, dropwhile(isOdd, N(s)))
self.assertRaises(ZeroDivisionError, list, dropwhile(isOdd, E(s)))
+class RegressionTests(unittest.TestCase):
+
+ def test_sf_793826(self):
+ # Fix Armin Rigo's successful efforts to wreak havoc
+
+ def mutatingtuple(tuple1, f, tuple2):
+ # this builds a tuple t which is a copy of tuple1,
+ # then calls f(t), then mutates t to be equal to tuple2
+ # (needs len(tuple1) == len(tuple2)).
+ def g(value, first=[1]):
+ if first:
+ del first[:]
+ f(z.next())
+ return value
+ items = list(tuple2)
+ items[1:1] = list(tuple1)
+ gen = imap(g, items)
+ z = izip(*[gen]*len(tuple1))
+ z.next()
+
+ def f(t):
+ global T
+ T = t
+ first[:] = list(T)
+
+ first = []
+ mutatingtuple((1,2,3), f, (4,5,6))
+ second = list(T)
+ self.assertEqual(first, second)
+
libreftest = """ Doctest for examples in the library reference: libitertools.tex
@@ -568,7 +598,8 @@ False
__test__ = {'libreftest' : libreftest}
def test_main(verbose=None):
- test_classes = (TestBasicOps, TestVariousIteratorArgs, TestGC)
+ test_classes = (TestBasicOps, TestVariousIteratorArgs, TestGC,
+ RegressionTests)
test_support.run_unittest(*test_classes)
# verify reference counting