commit 15a4950da1a80b334ee4985c62e66a1e1071a951
author Raymond Hettinger <python@rcn.com> Thu Feb 19 02:17:09 2009 +0000
committer Raymond Hettinger <python@rcn.com> Thu Feb 19 02:17:09 2009 +0000
tree f4038ec0edc4fa562dd334b4a9b1b244dc24dd21
parent bc670849405d9bda3bbf6f1d28a252abe778bdd9

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 c34b915..94159b5 100644
--- a/Lib/test/test_itertools.py
+++ b/Lib/test/test_itertools.py
@@ -310,6 +310,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'))