[pypy-svn] pypy default: Allow object to be passed as a kwarg to itertools.repeat.

alex_gaynor commits-noreply at bitbucket.org
Thu Jan 20 21:59:25 CET 2011


Author: Alex Gaynor <alex.gaynor at gmail.com>
Branch: 
Changeset: r41094:99eee51786c6
Date: 2011-01-20 14:58 -0600
http://bitbucket.org/pypy/pypy/changeset/99eee51786c6/

Log:	Allow object to be passed as a kwarg to itertools.repeat.

diff --git a/pypy/module/itertools/test/test_itertools.py b/pypy/module/itertools/test/test_itertools.py
--- a/pypy/module/itertools/test/test_itertools.py
+++ b/pypy/module/itertools/test/test_itertools.py
@@ -81,6 +81,8 @@
         raises(StopIteration, it.next)
         raises(StopIteration, it.next)
 
+        assert list(itertools.repeat(object='a', times=3)) == ['a', 'a', 'a']
+
     def test_repeat_overflow(self):
         import itertools
         import sys

diff --git a/pypy/module/itertools/interp_itertools.py b/pypy/module/itertools/interp_itertools.py
--- a/pypy/module/itertools/interp_itertools.py
+++ b/pypy/module/itertools/interp_itertools.py
@@ -91,8 +91,8 @@
             s = 'repeat(%s)' % (objrepr,)
         return self.space.wrap(s)
 
-def W_Repeat___new__(space, w_subtype, w_obj, w_times=None):
-    return space.wrap(W_Repeat(space, w_obj, w_times))
+def W_Repeat___new__(space, w_subtype, w_object, w_times=None):
+    return space.wrap(W_Repeat(space, w_object, w_times))
 
 W_Repeat.typedef = TypeDef(
         'repeat',


More information about the Pypy-commit mailing list