Index: Lib/test/test_functools.py =================================================================== --- Lib/test/test_functools.py (revision 83938) +++ Lib/test/test_functools.py (working copy) @@ -321,10 +321,11 @@ self.sofar.append(n*n) n += 1 return self.sofar[i] - - self.assertEqual(self.func(lambda x, y: x+y, ['a', 'b', 'c'], ''), 'abc') + def add(x, y): + return x + y + self.assertEqual(self.func(add, ['a', 'b', 'c'], ''), 'abc') self.assertEqual( - self.func(lambda x, y: x+y, [['a', 'c'], [], ['d', 'w']], []), + self.func(add, [['a', 'c'], [], ['d', 'w']], []), ['a','c','d','w'] ) self.assertEqual(self.func(lambda x, y: x*y, range(2,8), 1), 5040) @@ -332,16 +333,28 @@ self.func(lambda x, y: x*y, range(2,21), 1), 2432902008176640000 ) - self.assertEqual(self.func(lambda x, y: x+y, Squares(10)), 285) - self.assertEqual(self.func(lambda x, y: x+y, Squares(10), 0), 285) - self.assertEqual(self.func(lambda x, y: x+y, Squares(0), 0), 0) + self.assertEqual(self.func(add, Squares(10)), 285) + self.assertEqual(self.func(add, Squares(10), 0), 285) + self.assertEqual(self.func(add, Squares(0), 0), 0) self.assertRaises(TypeError, self.func) self.assertRaises(TypeError, self.func, 42, 42) self.assertRaises(TypeError, self.func, 42, 42, 42) self.assertEqual(self.func(42, "1"), "1") # func is never called with one item self.assertEqual(self.func(42, "", "1"), "1") # func is never called with one item self.assertRaises(TypeError, self.func, 42, (42, 42)) + self.assertRaises(TypeError, self.func, add, []) # arg 2 must not be empty sequence with no initial value + self.assertRaises(TypeError, self.func, add, "") + self.assertRaises(TypeError, self.func, add, ()) + self.assertRaises(TypeError, self.func, add, object()) + class TestFailingIter: + def __iter__(self): + raise RuntimeError + self.assertRaises(RuntimeError, self.func, add, TestFailingIter()) + + self.assertEqual(self.func(add, [], None), None) + self.assertEqual(self.func(add, [], 42), 42) + class BadSeq: def __getitem__(self, index): raise ValueError Index: Lib/test/test_builtin.py =================================================================== --- Lib/test/test_builtin.py (revision 83938) +++ Lib/test/test_builtin.py (working copy) @@ -1234,6 +1234,7 @@ class G: pass self.assertRaises(TypeError, zip, a, G()) + self.assertRaises(RuntimeError, zip, a, TestFailingIter()) # Make sure zip doesn't try to allocate a billion elements for the # result list when one of its arguments doesn't say how long it is. Index: Modules/_functoolsmodule.c =================================================================== --- Modules/_functoolsmodule.c (revision 83938) +++ Modules/_functoolsmodule.c (working copy) @@ -302,8 +302,9 @@ it = PyObject_GetIter(seq); if (it == NULL) { - PyErr_SetString(PyExc_TypeError, - "reduce() arg 2 must support iteration"); + if (PyErr_ExceptionMatches(PyExc_TypeError)) + PyErr_SetString(PyExc_TypeError, + "reduce() arg 2 must support iteration"); Py_XDECREF(result); return NULL; }