diff options
| -rw-r--r-- | Modules/mathmodule.c | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/Modules/mathmodule.c b/Modules/mathmodule.c index f835005..92f5f42 100644 --- a/Modules/mathmodule.c +++ b/Modules/mathmodule.c @@ -1137,15 +1137,22 @@ math_factorial(PyObject *self, PyObject *arg) PyObject *result, *iobj, *newresult; if (PyFloat_Check(arg)) { + PyObject *lx; double dx = PyFloat_AS_DOUBLE((PyFloatObject *)arg); - if (dx != floor(dx)) { + if (!(Py_IS_FINITE(dx) && dx == floor(dx))) { PyErr_SetString(PyExc_ValueError, "factorial() only accepts integral values"); return NULL; } + lx = PyLong_FromDouble(dx); + if (lx == NULL) + return NULL; + x = PyLong_AsLong(lx); + Py_DECREF(lx); } + else + x = PyLong_AsLong(arg); - x = PyLong_AsLong(arg); if (x == -1 && PyErr_Occurred()) return NULL; if (x < 0) { |