commit 473170908e11e347aca4adf23738a82162b981e3
author Raymond Hettinger <python@rcn.com> Thu Jan 17 03:02:14 2008 +0000
committer Raymond Hettinger <python@rcn.com> Thu Jan 17 03:02:14 2008 +0000
tree 2fa25a5ccd4abfc71e6a0f5193838a726ce4c68a
parent 3ad2acc8575e1977cece844b17c572550503a615

Make starmap() match its pure python definition and accept any itertable input (not just tuples).

Lib/test/test_itertools.py

diff --git a/Lib/test/test_itertools.py b/Lib/test/test_itertools.py
index 1a3064c..9d19228 100644
--- a/Lib/test/test_itertools.py
+++ b/Lib/test/test_itertools.py
@@ -292,7 +292,8 @@
         self.assertEqual(take(3, starmap(operator.pow, izip(count(), count(1)))),
                          [0**1, 1**2, 2**3])
         self.assertEqual(list(starmap(operator.pow, [])), [])
-        self.assertRaises(TypeError, list, starmap(operator.pow, [[4,5]]))
+        self.assertEqual(list(starmap(operator.pow, [iter([4,5])])), [4**5])
+        self.assertRaises(TypeError, list, starmap(operator.pow, [None]))
         self.assertRaises(TypeError, starmap)
         self.assertRaises(TypeError, starmap, operator.pow, [(4,5)], 'extra')
         self.assertRaises(TypeError, starmap(10, [(4,5)]).next)