From: Serhiy Storchaka <storchaka@gmail.com>
Date: Sat, 26 Jan 2013 09:53:17 +0000 (+0200)
Subject: Optimize the test for issue #13454.
X-Git-Tag: v3.3.1rc1~289
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=1c7181d764c3aa031de690865ed9f49296b86029;p=python

Optimize the test for issue #13454.
Now it requires almost 4x less memory and is almost 2x faster.
---

1c7181d764c3aa031de690865ed9f49296b86029
diff --cc Lib/test/test_itertools.py
index efd50bae10,dfa371e2b8..53926a9df5
--- a/Lib/test/test_itertools.py
+++ b/Lib/test/test_itertools.py
@@@ -1206,73 -930,10 +1206,71 @@@ class TestBasicOps(unittest.TestCase)
          del a
          self.assertRaises(ReferenceError, getattr, p, '__class__')
  
 +        ans = list('abc')
 +        long_ans = list(range(10000))
 +
 +        # check copy
 +        a, b = tee('abc')
 +        self.assertEqual(list(copy.copy(a)), ans)
 +        self.assertEqual(list(copy.copy(b)), ans)
 +        a, b = tee(list(range(10000)))
 +        self.assertEqual(list(copy.copy(a)), long_ans)
 +        self.assertEqual(list(copy.copy(b)), long_ans)
 +
 +        # check partially consumed copy
 +        a, b = tee('abc')
 +        take(2, a)
 +        take(1, b)
 +        self.assertEqual(list(copy.copy(a)), ans[2:])
 +        self.assertEqual(list(copy.copy(b)), ans[1:])
 +        self.assertEqual(list(a), ans[2:])
 +        self.assertEqual(list(b), ans[1:])
 +        a, b = tee(range(10000))
 +        take(100, a)
 +        take(60, b)
 +        self.assertEqual(list(copy.copy(a)), long_ans[100:])
 +        self.assertEqual(list(copy.copy(b)), long_ans[60:])
 +        self.assertEqual(list(a), long_ans[100:])
 +        self.assertEqual(list(b), long_ans[60:])
 +
 +        # check deepcopy
 +        a, b = tee('abc')
 +        self.assertEqual(list(copy.deepcopy(a)), ans)
 +        self.assertEqual(list(copy.deepcopy(b)), ans)
 +        self.assertEqual(list(a), ans)
 +        self.assertEqual(list(b), ans)
 +        a, b = tee(range(10000))
 +        self.assertEqual(list(copy.deepcopy(a)), long_ans)
 +        self.assertEqual(list(copy.deepcopy(b)), long_ans)
 +        self.assertEqual(list(a), long_ans)
 +        self.assertEqual(list(b), long_ans)
 +
 +        # check partially consumed deepcopy
 +        a, b = tee('abc')
 +        take(2, a)
 +        take(1, b)
 +        self.assertEqual(list(copy.deepcopy(a)), ans[2:])
 +        self.assertEqual(list(copy.deepcopy(b)), ans[1:])
 +        self.assertEqual(list(a), ans[2:])
 +        self.assertEqual(list(b), ans[1:])
 +        a, b = tee(range(10000))
 +        take(100, a)
 +        take(60, b)
 +        self.assertEqual(list(copy.deepcopy(a)), long_ans[100:])
 +        self.assertEqual(list(copy.deepcopy(b)), long_ans[60:])
 +        self.assertEqual(list(a), long_ans[100:])
 +        self.assertEqual(list(b), long_ans[60:])
 +
 +        # check pickle
 +        self.pickletest(iter(tee('abc')))
 +        a, b = tee('abc')
 +        self.pickletest(a, compare=ans)
 +        self.pickletest(b, compare=ans)
 +
      # Issue 13454: Crash when deleting backward iterator from tee()
      def test_tee_del_backward(self):
-         forward, backward = tee(range(20000000))
-         for i in forward:
-             pass
- 
+         forward, backward = tee(repeat(None, 20000000))
+         any(forward)  # exhaust the iterator
          del backward
  
      def test_StopIteration(self):