testing

Author: eendebakpt

Objects/rangeobject.c

@@ -216,6 +216,12 @@ static long compute_range_length_long(PyObject *start,
     }
 }
 
+static void print_str(PyObject *o)
+{
+    PyObject_Print(o, stdout, Py_PRINT_RAW);
+}
+
+
 /* Return number of items in range (lo, hi, step) as a PyLong object,
  * when arguments are PyLong objects.  Arguments MUST return 1 with
  * PyLong_Check().  Return NULL when there is an error.
@@ -240,6 +246,17 @@ compute_range_length(PyObject *start, PyObject *stop, PyObject *step)
     assert(PyLong_Check(stop));
     assert(PyLong_Check(step));
 
+    /* fast path for one argument case of range */
+    if (start == zero && step == one) {
+        if (_PyLong_IsPositive( (const PyLongObject *)stop) ) {
+                Py_INCREF(stop);
+                return stop;
+        }
+        else {
+                return zero;
+        }
+    }
+
     /* fast path when all arguments fit into a long integer */
     long len = compute_range_length_long(start, stop, step);
     if (len >= 0) {
@@ -293,6 +310,7 @@ compute_range_length(PyObject *start, PyObject *stop, PyObject *step)
     Py_DECREF(diff);
     Py_DECREF(step);
     Py_DECREF(tmp1);
+
     return result;
 
   Fail:
@@ -1135,13 +1153,19 @@ range_iter(PyObject *seq)
 
     /* If all three fields and the length convert to long, use the int
      * version */
-    lstart = PyLong_AsLong(r->start);
-    if (lstart == -1 && PyErr_Occurred()) {
+    lstop = PyLong_AsLong(r->stop);
+    if (lstop == -1 && PyErr_Occurred()) {
         PyErr_Clear();
         goto long_range;
     }
-    lstop = PyLong_AsLong(r->stop);
-    if (lstop == -1 && PyErr_Occurred()) {
+
+    if (r->start == (PyLongObject *)_PyLong_GetZero() && r->step == _PyLong_GetOne() ) {
+            /* fast path for one argument range */
+            return fast_range_iter(0, lstop, 1, Py_MAX(0, (long)lstop));
+    }
+
+    lstart = PyLong_AsLong(r->start);
+    if (lstart == -1 && PyErr_Occurred()) {
         PyErr_Clear();
         goto long_range;
     }