Speed-up math.dist() by 30% (GH-9628)
diff --git a/Modules/mathmodule.c b/Modules/mathmodule.c
index e872e47..e956314 100644
--- a/Modules/mathmodule.c
+++ b/Modules/mathmodule.c
@@ -2101,8 +2101,8 @@
/*[clinic input]
math.dist
- p: object(subclass_of='&PyTuple_Type')
- q: object(subclass_of='&PyTuple_Type')
+ p: object
+ q: object
/
Return the Euclidean distance between two points p and q.
@@ -2116,7 +2116,7 @@
static PyObject *
math_dist_impl(PyObject *module, PyObject *p, PyObject *q)
-/*[clinic end generated code: output=56bd9538d06bbcfe input=937122eaa5f19272]*/
+/*[clinic end generated code: output=56bd9538d06bbcfe input=8c83c07c7a524664]*/
{
PyObject *item;
double max = 0.0;
@@ -2126,6 +2126,11 @@
double diffs_on_stack[NUM_STACK_ELEMS];
double *diffs = diffs_on_stack;
+ if (!PyTuple_Check(p) || !PyTuple_Check(q)) {
+ PyErr_SetString(PyExc_TypeError, "dist argument must be a tuple");
+ return NULL;
+ }
+
m = PyTuple_GET_SIZE(p);
n = PyTuple_GET_SIZE(q);
if (m != n) {