From 9b4197ba53af85195cd722c4f85effee671c3474 Mon Sep 17 00:00:00 2001 From: Raymond Hettinger Date: Mon, 30 Nov 2009 21:13:52 +0000 Subject: [PATCH] Handle step values other than one. --- Modules/itertoolsmodule.c | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/Modules/itertoolsmodule.c b/Modules/itertoolsmodule.c index 6b5c249f00..633d0a7e0d 100644 --- a/Modules/itertoolsmodule.c +++ b/Modules/itertoolsmodule.c @@ -3379,7 +3379,7 @@ static PyObject * count_reduce(countobject *lz) { if (lz->cnt == PY_SSIZE_T_MAX) - return Py_BuildValue("O(O)", Py_TYPE(lz), lz->long_cnt); + return Py_BuildValue("O(OO)", Py_TYPE(lz), lz->long_cnt, lz->long_step); return Py_BuildValue("O(n)", Py_TYPE(lz), lz->cnt); } @@ -3388,6 +3388,7 @@ PyDoc_STRVAR(count_reduce_doc, "Return state information for pickling."); static PyMethodDef count_methods[] = { {"__reduce__", (PyCFunction)count_reduce, METH_NOARGS, count_reduce_doc}, + {NULL, NULL} /* sentinel */ }; PyDoc_STRVAR(count_doc, -- 2.50.1