BUG: PyErr_NoMemory when PyArray_Zeros fails to initialize

grahamc · juliantaylor · commit c3888e73fff4 · 2015-01-25T16:57:39.000+01:00
diff --git a/numpy/fft/fftpack_litemodule.c b/numpy/fft/fftpack_litemodule.c
@@ -151,7 +151,7 @@ fftpack_rfftf(PyObject *NPY_UNUSED(self), PyObject *args)
     PyObject *op1, *op2;
     PyArrayObject *data, *ret;
     PyArray_Descr *descr;
-    double *wsave, *dptr, *rptr;
+    double *wsave = NULL, *dptr, *rptr;
     npy_intp nsave;
     int npts, nrepeats, i, rstep;
 
@@ -168,6 +168,9 @@ fftpack_rfftf(PyObject *NPY_UNUSED(self), PyObject *args)
     PyArray_DIMS(data)[PyArray_NDIM(data) - 1] = npts/2 + 1;
     ret = (PyArrayObject *)PyArray_Zeros(PyArray_NDIM(data),
             PyArray_DIMS(data), PyArray_DescrFromType(NPY_CDOUBLE), 0);
+    if (ret == NULL) {
+        goto fail;
+    }
     PyArray_DIMS(data)[PyArray_NDIM(data) - 1] = npts;
     rstep = PyArray_DIM(ret, PyArray_NDIM(ret) - 1)*2;
 
