Issue 2582: Fix pickling of xrange objects.
diff --git a/Lib/test/test_xrange.py b/Lib/test/test_xrange.py
index eafb7fa..ab68c36 100644
--- a/Lib/test/test_xrange.py
+++ b/Lib/test/test_xrange.py
@@ -57,6 +57,16 @@
self.assertEqual(len(r), sys.maxint)
self.assertRaises(OverflowError, xrange, -sys.maxint-1, sys.maxint, 2)
+ def test_getnewargs(self):
+ def test(*args):
+ r = xrange(*args)
+ return list(r) == list(xrange(*r.__getnewargs__()))
+ tests = [(13,), (0, 11), (-22, 10), (20, 3, -1),
+ (13, 21, 3), (-2, 2, 2)]
+ for t in tests:
+ self.assert_(test(*t),
+ "xrange.__getnewargs__() failed with %r" % (t,))
+
def test_main():
test.test_support.run_unittest(XrangeTest)
diff --git a/Objects/rangeobject.c b/Objects/rangeobject.c
index da4356b..bfc038e 100644
--- a/Objects/rangeobject.c
+++ b/Objects/rangeobject.c
@@ -129,6 +129,16 @@
return rtn;
}
+/* Pickling support */
+static PyObject *
+range_getnewargs(rangeobject *r)
+{
+ return Py_BuildValue("(iii)",
+ r->start,
+ r->start + r->len * r->step,
+ r->step);
+}
+
static PySequenceMethods range_as_sequence = {
(lenfunc)range_length, /* sq_length */
0, /* sq_concat */
@@ -145,6 +155,7 @@
static PyMethodDef range_methods[] = {
{"__reversed__", (PyCFunction)range_reverse, METH_NOARGS, reverse_doc},
+ {"__getnewargs__", (PyCFunction)range_getnewargs, METH_NOARGS},
{NULL, NULL} /* sentinel */
};