diff options
| author | Christian Heimes <christian@cheimes.de> | 2008-03-16 00:07:10 (GMT) | 
|---|---|---|
| committer | Christian Heimes <christian@cheimes.de> | 2008-03-16 00:07:10 (GMT) | 
| commit | dd15f6c315f20c1a9a540dd757cd63e27dbe9f3c (patch) | |
| tree | 29466152518fc877722f97df2be69df9559aec29 /Lib/test/test_itertools.py | |
| parent | 462187540764e3bcc2324fb0d51315f5c28ed088 (diff) | |
| download | cpython-dd15f6c315f20c1a9a540dd757cd63e27dbe9f3c.zip cpython-dd15f6c315f20c1a9a540dd757cd63e27dbe9f3c.tar.gz cpython-dd15f6c315f20c1a9a540dd757cd63e27dbe9f3c.tar.bz2  | |
Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from
svn+ssh://pythondev@svn.python.org/python/trunk
........
  r61239 | andrew.kuchling | 2008-03-05 01:44:41 +0100 (Wed, 05 Mar 2008) | 1 line
  Add more items; add fragmentary notes
........
  r61240 | amaury.forgeotdarc | 2008-03-05 02:50:33 +0100 (Wed, 05 Mar 2008) | 13 lines
  Issue#2238: some syntax errors from *args or **kwargs expressions
  would give bogus error messages, because of untested exceptions::
      >>> f(**g(1=2))
      XXX undetected error
      Traceback (most recent call last):
        File "<stdin>", line 1, in <module>
      TypeError: 'int' object is not iterable
  instead of the expected SyntaxError: keyword can't be an expression
  Will backport.
........
  r61241 | neal.norwitz | 2008-03-05 06:10:48 +0100 (Wed, 05 Mar 2008) | 3 lines
  Remove the files/dirs after closing the DB so the tests work on Windows.
  Patch from Trent Nelson.  Also simplified removing a file by using test_support.
........
  r61242 | neal.norwitz | 2008-03-05 06:14:18 +0100 (Wed, 05 Mar 2008) | 3 lines
  Get this test to pass even when there is no sound card in the system.
  Patch from Trent Nelson.  (I can't test this.)
........
  r61243 | neal.norwitz | 2008-03-05 06:20:44 +0100 (Wed, 05 Mar 2008) | 3 lines
  Catch OSError when trying to remove a file in case removal fails. This
  should prevent a failure in tearDown masking any real test failure.
........
  r61244 | neal.norwitz | 2008-03-05 06:38:06 +0100 (Wed, 05 Mar 2008) | 5 lines
  Make the timeout longer to give slow machines a chance to pass the test
  before timing out.  This doesn't change the duration of the test under
  normal circumstances.  This is targetted at fixing the spurious failures
  on the FreeBSD buildbot primarily.
........
  r61245 | neal.norwitz | 2008-03-05 06:49:03 +0100 (Wed, 05 Mar 2008) | 1 line
  Tabs -> spaces
........
  r61246 | neal.norwitz | 2008-03-05 06:50:20 +0100 (Wed, 05 Mar 2008) | 1 line
  Use -u urlfetch to run more tests
........
  r61247 | neal.norwitz | 2008-03-05 06:51:20 +0100 (Wed, 05 Mar 2008) | 1 line
  test_smtplib sometimes reports leaks too, suppress it
........
  r61248 | jeffrey.yasskin | 2008-03-05 07:19:56 +0100 (Wed, 05 Mar 2008) | 5 lines
  Fix test_socketserver on Windows after r61099 added several signal.alarm()
  calls (which don't exist on non-Unix platforms).
  Thanks to Trent Nelson for the report and patch.
........
  r61249 | georg.brandl | 2008-03-05 08:10:35 +0100 (Wed, 05 Mar 2008) | 2 lines
  Fix some rst.
........
  r61252 | thomas.heller | 2008-03-05 15:53:39 +0100 (Wed, 05 Mar 2008) | 2 lines
  News entry for yesterdays commit.
........
  r61253 | thomas.heller | 2008-03-05 16:34:29 +0100 (Wed, 05 Mar 2008) | 3 lines
  Issue 1872: Changed the struct module typecode from 't' to '?', for
  compatibility with PEP3118.
........
  r61254 | skip.montanaro | 2008-03-05 17:41:09 +0100 (Wed, 05 Mar 2008) | 4 lines
  Elaborate on the role of the altinstall target when installing multiple
  versions.
........
  r61255 | georg.brandl | 2008-03-05 20:31:44 +0100 (Wed, 05 Mar 2008) | 2 lines
  #2239: PYTHONPATH delimiter is os.pathsep.
........
  r61256 | raymond.hettinger | 2008-03-05 21:59:58 +0100 (Wed, 05 Mar 2008) | 1 line
  C implementation of itertools.permutations().
........
  r61257 | raymond.hettinger | 2008-03-05 22:04:32 +0100 (Wed, 05 Mar 2008) | 1 line
  Small code cleanup.
........
  r61260 | martin.v.loewis | 2008-03-05 23:24:31 +0100 (Wed, 05 Mar 2008) | 2 lines
  cd PCbuild only after deleting all pyc files.
........
  r61261 | raymond.hettinger | 2008-03-06 02:15:52 +0100 (Thu, 06 Mar 2008) | 1 line
  Add examples.
........
  r61262 | andrew.kuchling | 2008-03-06 02:36:27 +0100 (Thu, 06 Mar 2008) | 1 line
  Add two items
........
  r61263 | georg.brandl | 2008-03-06 07:47:18 +0100 (Thu, 06 Mar 2008) | 2 lines
  #1725737: ignore other VC directories other than CVS and SVN's too.
........
  r61264 | martin.v.loewis | 2008-03-06 07:55:22 +0100 (Thu, 06 Mar 2008) | 4 lines
  Patch #2232: os.tmpfile might fail on Windows if the user has no
  permission to create files in the root directory.
  Will backport to 2.5.
........
  r61269 | georg.brandl | 2008-03-06 08:19:15 +0100 (Thu, 06 Mar 2008) | 2 lines
  Expand on re.split behavior with captured expressions.
........
  r61270 | georg.brandl | 2008-03-06 08:22:09 +0100 (Thu, 06 Mar 2008) | 2 lines
  Little clarification of assignments.
........
  r61271 | georg.brandl | 2008-03-06 08:31:34 +0100 (Thu, 06 Mar 2008) | 2 lines
  Add isinstance/issubclass to tutorial.
........
  r61272 | georg.brandl | 2008-03-06 08:34:52 +0100 (Thu, 06 Mar 2008) | 2 lines
  Add missing NEWS entry for r61263.
........
  r61273 | georg.brandl | 2008-03-06 08:41:16 +0100 (Thu, 06 Mar 2008) | 2 lines
  #2225: return nonzero status code from py_compile if not all files could be compiled.
........
  r61274 | georg.brandl | 2008-03-06 08:43:02 +0100 (Thu, 06 Mar 2008) | 2 lines
  #2220: handle matching failure more gracefully.
........
  r61275 | georg.brandl | 2008-03-06 08:45:52 +0100 (Thu, 06 Mar 2008) | 2 lines
  Bug #2220: handle rlcompleter attribute match failure more gracefully.
........
  r61278 | martin.v.loewis | 2008-03-06 14:49:47 +0100 (Thu, 06 Mar 2008) | 1 line
  Rely on x64 platform configuration when building _bsddb on AMD64.
........
  r61279 | martin.v.loewis | 2008-03-06 14:50:28 +0100 (Thu, 06 Mar 2008) | 1 line
  Update db-4.4.20 build procedure.
........
  r61285 | raymond.hettinger | 2008-03-06 21:52:01 +0100 (Thu, 06 Mar 2008) | 1 line
  More tests.
........
  r61286 | raymond.hettinger | 2008-03-06 23:51:36 +0100 (Thu, 06 Mar 2008) | 1 line
  Issue 2246:  itertools grouper object did not participate in GC (should be backported).
........
  r61288 | raymond.hettinger | 2008-03-07 02:33:20 +0100 (Fri, 07 Mar 2008) | 1 line
  Tweak recipes and tests
........
  r61289 | jeffrey.yasskin | 2008-03-07 07:22:15 +0100 (Fri, 07 Mar 2008) | 5 lines
  Progress on issue #1193577 by adding a polling .shutdown() method to
  SocketServers. The core of the patch was written by Pedro Werneck, but any bugs
  are mine. I've also rearranged the code for timeouts in order to avoid
  interfering with the shutdown poll.
........
  r61290 | nick.coghlan | 2008-03-07 15:13:28 +0100 (Fri, 07 Mar 2008) | 1 line
  Speed up with statements by storing the __exit__ method on the stack instead of in a temp variable (bumps the magic number for pyc files)
........
  r61298 | andrew.kuchling | 2008-03-07 22:09:23 +0100 (Fri, 07 Mar 2008) | 1 line
  Grammar fix
........
  r61303 | georg.brandl | 2008-03-08 10:54:06 +0100 (Sat, 08 Mar 2008) | 2 lines
  #2253: fix continue vs. finally docs.
........
  r61304 | marc-andre.lemburg | 2008-03-08 11:01:43 +0100 (Sat, 08 Mar 2008) | 3 lines
  Add new name for Mandrake: Mandriva.
........
  r61305 | georg.brandl | 2008-03-08 11:05:24 +0100 (Sat, 08 Mar 2008) | 2 lines
  #1533486: fix types in refcount intro.
........
  r61312 | facundo.batista | 2008-03-08 17:50:27 +0100 (Sat, 08 Mar 2008) | 5 lines
  Issue 1106316. post_mortem()'s parameter, traceback, is now
  optional: it defaults to the traceback of the exception that is currently
  being handled.
........
  r61313 | jeffrey.yasskin | 2008-03-08 19:26:54 +0100 (Sat, 08 Mar 2008) | 2 lines
  Add tests for with and finally performance to pybench.
........
  r61314 | jeffrey.yasskin | 2008-03-08 21:08:21 +0100 (Sat, 08 Mar 2008) | 2 lines
  Fix pybench for pythons < 2.6, tested back to 2.3.
........
  r61317 | jeffrey.yasskin | 2008-03-08 22:35:15 +0100 (Sat, 08 Mar 2008) | 3 lines
  Well that was dumb. platform.python_implementation returns a function, not a
  string.
........
  r61329 | georg.brandl | 2008-03-09 16:11:39 +0100 (Sun, 09 Mar 2008) | 2 lines
  #2249: document assertTrue and assertFalse.
........
  r61332 | neal.norwitz | 2008-03-09 20:03:42 +0100 (Sun, 09 Mar 2008) | 4 lines
  Introduce a lock to fix a race condition which caused an exception in the test.
  Some buildbots were consistently failing (e.g., amd64).
  Also remove a couple of semi-colons.
........
  r61344 | raymond.hettinger | 2008-03-11 01:19:07 +0100 (Tue, 11 Mar 2008) | 1 line
  Add recipe to docs.
........
  r61350 | guido.van.rossum | 2008-03-11 22:18:06 +0100 (Tue, 11 Mar 2008) | 3 lines
  Fix the overflows in expandtabs().  "This time for sure!"
  (Exploit at request.)
........
  r61351 | raymond.hettinger | 2008-03-11 22:37:46 +0100 (Tue, 11 Mar 2008) | 1 line
  Improve docs for itemgetter().  Show that it works with slices.
........
  r61363 | georg.brandl | 2008-03-13 08:15:56 +0100 (Thu, 13 Mar 2008) | 2 lines
  #2265: fix example.
........
  r61364 | georg.brandl | 2008-03-13 08:17:14 +0100 (Thu, 13 Mar 2008) | 2 lines
  #2270: fix typo.
........
  r61365 | georg.brandl | 2008-03-13 08:21:41 +0100 (Thu, 13 Mar 2008) | 2 lines
  #1720705: add docs about import/threading interaction, wording by Nick.
........
  r61366 | andrew.kuchling | 2008-03-13 12:07:35 +0100 (Thu, 13 Mar 2008) | 1 line
  Add class decorators
........
  r61367 | raymond.hettinger | 2008-03-13 17:43:17 +0100 (Thu, 13 Mar 2008) | 1 line
  Add 2-to-3 support for the itertools moved to builtins or renamed.
........
  r61368 | raymond.hettinger | 2008-03-13 17:43:59 +0100 (Thu, 13 Mar 2008) | 1 line
  Consistent tense.
........
  r61369 | raymond.hettinger | 2008-03-13 20:03:51 +0100 (Thu, 13 Mar 2008) | 1 line
  Issue 2274:  Add heapq.heappushpop().
........
  r61370 | raymond.hettinger | 2008-03-13 20:33:34 +0100 (Thu, 13 Mar 2008) | 1 line
  Simplify the nlargest() code using heappushpop().
........
  r61371 | brett.cannon | 2008-03-13 21:27:00 +0100 (Thu, 13 Mar 2008) | 4 lines
  Move test_thread over to unittest. Commits GHOP 237.
  Thanks Benjamin Peterson for the patch.
........
  r61372 | brett.cannon | 2008-03-13 21:33:10 +0100 (Thu, 13 Mar 2008) | 4 lines
  Move test_tokenize to doctest.
  Done as GHOP 238 by Josip Dzolonga.
........
  r61373 | brett.cannon | 2008-03-13 21:47:41 +0100 (Thu, 13 Mar 2008) | 4 lines
  Convert test_contains, test_crypt, and test_select to unittest.
  Patch from GHOP 294 by David Marek.
........
  r61374 | brett.cannon | 2008-03-13 22:02:16 +0100 (Thu, 13 Mar 2008) | 4 lines
  Move test_gdbm to use unittest.
  Closes issue #1960. Thanks Giampaolo Rodola.
........
  r61375 | brett.cannon | 2008-03-13 22:09:28 +0100 (Thu, 13 Mar 2008) | 4 lines
  Convert test_fcntl to unittest.
  Closes issue #2055. Thanks Giampaolo Rodola.
........
  r61376 | raymond.hettinger | 2008-03-14 06:03:44 +0100 (Fri, 14 Mar 2008) | 1 line
  Leave heapreplace() unchanged.
........
  r61378 | martin.v.loewis | 2008-03-14 14:56:09 +0100 (Fri, 14 Mar 2008) | 2 lines
  Patch #2284: add -x64 option to rt.bat.
........
  r61379 | martin.v.loewis | 2008-03-14 14:57:59 +0100 (Fri, 14 Mar 2008) | 2 lines
  Use -x64 flag.
........
  r61382 | brett.cannon | 2008-03-14 15:03:10 +0100 (Fri, 14 Mar 2008) | 2 lines
  Remove a bad test.
........
  r61383 | mark.dickinson | 2008-03-14 15:23:37 +0100 (Fri, 14 Mar 2008) | 9 lines
  Issue 705836: Fix struct.pack(">f", 1e40) to behave consistently
  across platforms:  it should now raise OverflowError on all
  platforms.  (Previously it raised OverflowError only on
  non IEEE 754 platforms.)
  Also fix the (already existing) test for this behaviour
  so that it actually raises TestFailed instead of just
  referencing it.
........
  r61387 | thomas.heller | 2008-03-14 22:06:21 +0100 (Fri, 14 Mar 2008) | 1 line
  Remove unneeded initializer.
........
  r61388 | martin.v.loewis | 2008-03-14 22:19:28 +0100 (Fri, 14 Mar 2008) | 2 lines
  Run debug version, cd to PCbuild.
........
  r61392 | georg.brandl | 2008-03-15 00:10:34 +0100 (Sat, 15 Mar 2008) | 2 lines
  Remove obsolete paragraph. #2288.
........
  r61395 | georg.brandl | 2008-03-15 01:20:19 +0100 (Sat, 15 Mar 2008) | 2 lines
  Fix lots of broken links in the docs, found by Sphinx' external link checker.
........
  r61396 | skip.montanaro | 2008-03-15 03:32:49 +0100 (Sat, 15 Mar 2008) | 1 line
  note that fork and forkpty raise OSError on failure
........
  r61402 | skip.montanaro | 2008-03-15 17:04:45 +0100 (Sat, 15 Mar 2008) | 1 line
  add %f format to datetime - issue 1158
........
  r61403 | skip.montanaro | 2008-03-15 17:07:11 +0100 (Sat, 15 Mar 2008) | 2 lines
  .
........
Diffstat (limited to 'Lib/test/test_itertools.py')
| -rw-r--r-- | Lib/test/test_itertools.py | 224 | 
1 files changed, 204 insertions, 20 deletions
diff --git a/Lib/test/test_itertools.py b/Lib/test/test_itertools.py index d44235b..335e47d 100644 --- a/Lib/test/test_itertools.py +++ b/Lib/test/test_itertools.py @@ -51,22 +51,21 @@ def fact(n):      'Factorial'      return prod(range(1, n+1)) -def permutations(iterable, r=None): -    # XXX use this until real permutations code is added -    pool = tuple(iterable) -    n = len(pool) -    r = n if r is None else r -    for indices in product(range(n), repeat=r): -        if len(set(indices)) == r: -            yield tuple(pool[i] for i in indices) -  class TestBasicOps(unittest.TestCase):      def test_chain(self): -        self.assertEqual(list(chain('abc', 'def')), list('abcdef')) -        self.assertEqual(list(chain('abc')), list('abc')) -        self.assertEqual(list(chain('')), []) -        self.assertEqual(take(4, chain('abc', 'def')), list('abcd')) -        self.assertRaises(TypeError, list,chain(2, 3)) + +        def chain2(*iterables): +            'Pure python version in the docs' +            for it in iterables: +                for element in it: +                    yield element + +        for c in (chain, chain2): +            self.assertEqual(list(c('abc', 'def')), list('abcdef')) +            self.assertEqual(list(c('abc')), list('abc')) +            self.assertEqual(list(c('')), []) +            self.assertEqual(take(4, c('abc', 'def')), list('abcd')) +            self.assertRaises(TypeError, list,c(2, 3))      def test_chain_from_iterable(self):          self.assertEqual(list(chain.from_iterable(['abc', 'def'])), list('abcdef')) @@ -121,6 +120,8 @@ class TestBasicOps(unittest.TestCase):                      self.assertEqual(len(set(c)), r)                    # no duplicate elements                      self.assertEqual(list(c), sorted(c))                # keep original ordering                      self.assert_(all(e in values for e in c))           # elements taken from input iterable +                    self.assertEqual(list(c), +                                     [e for e in values if e in c])      # comb is a subsequence of the input iterable                  self.assertEqual(result, list(combinations1(values, r))) # matches first pure python version                  self.assertEqual(result, list(combinations2(values, r))) # matches first pure python version @@ -131,9 +132,10 @@ class TestBasicOps(unittest.TestCase):      def test_permutations(self):          self.assertRaises(TypeError, permutations)              # too few arguments          self.assertRaises(TypeError, permutations, 'abc', 2, 1) # too many arguments -##        self.assertRaises(TypeError, permutations, None)        # pool is not iterable -##        self.assertRaises(ValueError, permutations, 'abc', -2)  # r is negative -##        self.assertRaises(ValueError, permutations, 'abc', 32)  # r is too big +        self.assertRaises(TypeError, permutations, None)        # pool is not iterable +        self.assertRaises(ValueError, permutations, 'abc', -2)  # r is negative +        self.assertRaises(ValueError, permutations, 'abc', 32)  # r is too big +        self.assertRaises(TypeError, permutations, 'abc', 's')  # r is not an int or None          self.assertEqual(list(permutations(range(3), 2)),                                             [(0,1), (0,2), (1,0), (1,2), (2,0), (2,1)]) @@ -186,7 +188,7 @@ class TestBasicOps(unittest.TestCase):                      self.assertEqual(result, list(permutations(values)))       # test default r          # Test implementation detail:  tuple re-use -##        self.assertEqual(len(set(map(id, permutations('abcde', 3)))), 1) +        self.assertEqual(len(set(map(id, permutations('abcde', 3)))), 1)          self.assertNotEqual(len(set(map(id, list(permutations('abcde', 3))))), 1)      def test_count(self): @@ -416,12 +418,46 @@ class TestBasicOps(unittest.TestCase):                                   list(product(*args, **dict(repeat=r))))          self.assertEqual(len(list(product(*[range(7)]*6))), 7**6)          self.assertRaises(TypeError, product, range(6), None) + +        def product1(*args, **kwds): +            pools = list(map(tuple, args)) * kwds.get('repeat', 1) +            n = len(pools) +            if n == 0: +                yield () +                return +            if any(len(pool) == 0 for pool in pools): +                return +            indices = [0] * n +            yield tuple(pool[i] for pool, i in zip(pools, indices)) +            while 1: +                for i in reversed(range(n)):  # right to left +                    if indices[i] == len(pools[i]) - 1: +                        continue +                    indices[i] += 1 +                    for j in range(i+1, n): +                        indices[j] = 0 +                    yield tuple(pool[i] for pool, i in zip(pools, indices)) +                    break +                else: +                    return + +        def product2(*args, **kwds): +            'Pure python version used in docs' +            pools = list(map(tuple, args)) * kwds.get('repeat', 1) +            result = [[]] +            for pool in pools: +                result = [x+[y] for x in result for y in pool] +            for prod in result: +                yield tuple(prod) +          argtypes = ['', 'abc', '', range(0), range(4), dict(a=1, b=2, c=3),                      set('abcdefg'), range(11), tuple(range(13))]          for i in range(100):              args = [random.choice(argtypes) for j in range(random.randrange(5))]              expected_len = prod(map(len, args))              self.assertEqual(len(list(product(*args))), expected_len) +            self.assertEqual(list(product(*args)), list(product1(*args))) +            self.assertEqual(list(product(*args)), list(product2(*args)))              args = map(iter, args)              self.assertEqual(len(list(product(*args))), expected_len) @@ -661,6 +697,81 @@ class TestBasicOps(unittest.TestCase):              self.assertRaises(StopIteration, next, f(lambda x:x, []))              self.assertRaises(StopIteration, next, f(lambda x:x, StopNow())) +class TestExamples(unittest.TestCase): + +    def test_chain(self): +        self.assertEqual(''.join(chain('ABC', 'DEF')), 'ABCDEF') + +    def test_chain_from_iterable(self): +        self.assertEqual(''.join(chain.from_iterable(['ABC', 'DEF'])), 'ABCDEF') + +    def test_combinations(self): +        self.assertEqual(list(combinations('ABCD', 2)), +                         [('A','B'), ('A','C'), ('A','D'), ('B','C'), ('B','D'), ('C','D')]) +        self.assertEqual(list(combinations(range(4), 3)), +                         [(0,1,2), (0,1,3), (0,2,3), (1,2,3)]) + +    def test_count(self): +        self.assertEqual(list(islice(count(10), 5)), [10, 11, 12, 13, 14]) + +    def test_cycle(self): +        self.assertEqual(list(islice(cycle('ABCD'), 12)), list('ABCDABCDABCD')) + +    def test_dropwhile(self): +        self.assertEqual(list(dropwhile(lambda x: x<5, [1,4,6,4,1])), [6,4,1]) + +    def test_groupby(self): +        self.assertEqual([k for k, g in groupby('AAAABBBCCDAABBB')], +                         list('ABCDAB')) +        self.assertEqual([(list(g)) for k, g in groupby('AAAABBBCCD')], +                         [list('AAAA'), list('BBB'), list('CC'), list('D')]) + +    def test_filter(self): +        self.assertEqual(list(filter(lambda x: x%2, range(10))), [1,3,5,7,9]) + +    def test_filterfalse(self): +        self.assertEqual(list(filterfalse(lambda x: x%2, range(10))), [0,2,4,6,8]) + +    def test_map(self): +        self.assertEqual(list(map(pow, (2,3,10), (5,2,3))), [32, 9, 1000]) + +    def test_islice(self): +        self.assertEqual(list(islice('ABCDEFG', 2)), list('AB')) +        self.assertEqual(list(islice('ABCDEFG', 2, 4)), list('CD')) +        self.assertEqual(list(islice('ABCDEFG', 2, None)), list('CDEFG')) +        self.assertEqual(list(islice('ABCDEFG', 0, None, 2)), list('ACEG')) + +    def test_zip(self): +        self.assertEqual(list(zip('ABCD', 'xy')), [('A', 'x'), ('B', 'y')]) + +    def test_zip_longest(self): +        self.assertEqual(list(zip_longest('ABCD', 'xy', fillvalue='-')), +                         [('A', 'x'), ('B', 'y'), ('C', '-'), ('D', '-')]) + +    def test_permutations(self): +        self.assertEqual(list(permutations('ABCD', 2)), +                         list(map(tuple, 'AB AC AD BA BC BD CA CB CD DA DB DC'.split()))) +        self.assertEqual(list(permutations(range(3))), +                         [(0,1,2), (0,2,1), (1,0,2), (1,2,0), (2,0,1), (2,1,0)]) + +    def test_product(self): +        self.assertEqual(list(product('ABCD', 'xy')), +                         list(map(tuple, 'Ax Ay Bx By Cx Cy Dx Dy'.split()))) +        self.assertEqual(list(product(range(2), repeat=3)), +                        [(0,0,0), (0,0,1), (0,1,0), (0,1,1), +                         (1,0,0), (1,0,1), (1,1,0), (1,1,1)]) + +    def test_repeat(self): +        self.assertEqual(list(repeat(10, 3)), [10, 10, 10]) + +    def test_stapmap(self): +        self.assertEqual(list(starmap(pow, [(2,5), (3,2), (10,3)])), +                         [32, 9, 1000]) + +    def test_takewhile(self): +        self.assertEqual(list(takewhile(lambda x: x<5, [1,4,6,4,1])), [1,4]) + +  class TestGC(unittest.TestCase):      def makecycle(self, iterator, container): @@ -672,6 +783,14 @@ class TestGC(unittest.TestCase):          a = []          self.makecycle(chain(a), a) +    def test_chain_from_iterable(self): +        a = [] +        self.makecycle(chain.from_iterable([a]), a) + +    def test_combinations(self): +        a = [] +        self.makecycle(combinations([1,2,a,3], 3), a) +      def test_cycle(self):          a = []          self.makecycle(cycle([a]*2), a) @@ -684,6 +803,13 @@ class TestGC(unittest.TestCase):          a = []          self.makecycle(groupby([a]*2, lambda x:x), a) +    def test_issue2246(self): +        # Issue 2246 -- the _grouper iterator was not included in GC +        n = 10 +        keyfunc = lambda x: x +        for i, j in groupby(range(n), key=keyfunc): +            keyfunc.__dict__.setdefault('x',[]).append(j) +      def test_filter(self):          a = []          self.makecycle(filter(lambda x:True, [a]*2), a) @@ -696,6 +822,12 @@ class TestGC(unittest.TestCase):          a = []          self.makecycle(zip([a]*2, [a]*3), a) +    def test_zip_longest(self): +        a = [] +        self.makecycle(zip_longest([a]*2, [a]*3), a) +        b = [a, None] +        self.makecycle(zip_longest([a]*2, [a]*3, fillvalue=b), a) +      def test_map(self):          a = []          self.makecycle(map(lambda x:x, [a]*2), a) @@ -704,6 +836,14 @@ class TestGC(unittest.TestCase):          a = []          self.makecycle(islice([a]*2, None), a) +    def test_permutations(self): +        a = [] +        self.makecycle(permutations([1,2,a,3], 3), a) + +    def test_product(self): +        a = [] +        self.makecycle(product([1,2,a,3], repeat=3), a) +      def test_repeat(self):          a = []          self.makecycle(repeat(a), a) @@ -1115,7 +1255,7 @@ Samuele  ...     return sum(map(operator.mul, vec1, vec2))  >>> def flatten(listOfLists): -...     return list(chain(*listOfLists)) +...     return list(chain.from_iterable(listOfLists))  >>> def repeatfunc(func, times=None, *args):  ...     "Repeat calls to func with specified arguments." @@ -1134,6 +1274,38 @@ Samuele  ...         pass  ...     return zip(a, b) +>>> def grouper(n, iterable, fillvalue=None): +...     "grouper(3, 'abcdefg', 'x') --> ('a','b','c'), ('d','e','f'), ('g','x','x')" +...     args = [iter(iterable)] * n +...     kwds = dict(fillvalue=fillvalue) +...     return zip_longest(*args, **kwds) + +>>> def roundrobin(*iterables): +...     "roundrobin('abc', 'd', 'ef') --> 'a', 'd', 'e', 'b', 'f', 'c'" +...     # Recipe credited to George Sakkis +...     pending = len(iterables) +...     nexts = cycle(iter(it).__next__ for it in iterables) +...     while pending: +...         try: +...             for next in nexts: +...                 yield next() +...         except StopIteration: +...             pending -= 1 +...             nexts = cycle(islice(nexts, pending)) + +>>> def powerset(iterable): +...     "powerset('ab') --> set([]), set(['a']), set(['b']), set(['a', 'b'])" +...     # Recipe credited to Eric Raymond +...     pairs = [(2**i, x) for i, x in enumerate(iterable)] +...     for n in range(2**len(pairs)): +...         yield set(x for m, x in pairs if m&n) + +>>> def compress(data, selectors): +...     "compress('abcdef', [1,0,1,0,1,1]) --> a c e f" +...     for d, s in zip(data, selectors): +...         if s: +...             yield d +  This is not part of the examples but it tests to make sure the definitions  perform as purported. @@ -1199,6 +1371,18 @@ False  >>> dotproduct([1,2,3], [4,5,6])  32 +>>> list(grouper(3, 'abcdefg', 'x')) +[('a', 'b', 'c'), ('d', 'e', 'f'), ('g', 'x', 'x')] + +>>> list(roundrobin('abc', 'd', 'ef')) +['a', 'd', 'e', 'b', 'f', 'c'] + +>>> list(map(sorted, powerset('ab'))) +[[], ['a'], ['b'], ['a', 'b']] + +>>> list(compress('abcdef', [1,0,1,0,1,1])) +['a', 'c', 'e', 'f'] +  """  __test__ = {'libreftest' : libreftest} @@ -1206,7 +1390,7 @@ __test__ = {'libreftest' : libreftest}  def test_main(verbose=None):      test_classes = (TestBasicOps, TestVariousIteratorArgs, TestGC,                      RegressionTests, LengthTransparency, -                    SubclassWithKwargsTest) +                    SubclassWithKwargsTest, TestExamples)      test_support.run_unittest(*test_classes)      # verify reference counting  | 
