summaryrefslogtreecommitdiff
path: root/numpy/random/mtrand/mtrand_py_helper.h
blob: 266847cbe9fc301c8b9c532a9db9e75a31f340ba (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
#ifndef _MTRAND_PY_HELPER_H_
#define _MTRAND_PY_HELPER_H_

#include <Python.h>

static PyObject *empty_py_bytes(npy_intp length, void **bytes)
{
    PyObject *b;
#if PY_MAJOR_VERSION >= 3
    b = PyBytes_FromStringAndSize(NULL, length);
    if (b) {
        *bytes = PyBytes_AS_STRING(b);
    }
#else
    b = PyString_FromStringAndSize(NULL, length);
    if (b) {
        *bytes = PyString_AS_STRING(b);
    }
#endif
    return b;
}

#endif /* _MTRAND_PY_HELPER_H_ */