commit 2e2909f58482f5c780c58549f04f7cf325b94e02
author Raymond Hettinger <python@rcn.com> Thu Feb 19 02:15:14 2009 +0000
committer Raymond Hettinger <python@rcn.com> Thu Feb 19 02:15:14 2009 +0000
tree 4fe3fb9b2305efe850f5f54a2fe5b9858b9ea3e2
parent 08259e8f254a0595f8ed1be29b2d135f9b0d2d50

Add keyword arg support to itertools.compress().

Lib/test/test_itertools.py

diff --git a/Lib/test/test_itertools.py b/Lib/test/test_itertools.py
index a33939e..2cdb1d7 100644
--- a/Lib/test/test_itertools.py
+++ b/Lib/test/test_itertools.py
@@ -306,6 +306,7 @@
                 self.assertEqual(comb, sorted(set(cwr) & set(perm)))            # comb: both a cwr and a perm
 
     def test_compress(self):
+        self.assertEqual(list(compress(data='ABCDEF', selectors=[1,0,1,0,1,1])), list('ACEF'))
         self.assertEqual(list(compress('ABCDEF', [1,0,1,0,1,1])), list('ACEF'))
         self.assertEqual(list(compress('ABCDEF', [0,0,0,0,0,0])), list(''))
         self.assertEqual(list(compress('ABCDEF', [1,1,1,1,1,1])), list('ABCDEF'))