Fast case for count from zero

rhettinger · Jun 25, 2024 · 34bf546 · 34bf546
1 parent ace2045
commit 34bf546
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/Modules/_collectionsmodule.c b/Modules/_collectionsmodule.c
@@ -2575,7 +2575,11 @@ _collections__count_elements_impl(PyObject *module, PyObject *mapping,
             oldval = PyObject_CallFunctionObjArgs(bound_get, key, zero, NULL);
             if (oldval == NULL)
                 break;
-            newval = PyNumber_Add(oldval, one);
+            if (oldval == zero) {
+                newval = Py_NewRef(one);
+            } else {
+                newval = PyNumber_Add(oldval, one);
+            }
             Py_DECREF(oldval);
             if (newval == NULL)
                 break;