3K: random: make mtrand to import

1 files changed, 23 insertions, 0 deletions

diff --git a/numpy/random/mtrand/mtrand_py_helper.h b/numpy/random/mtrand/mtrand_py_helper.h
new file mode 100644
index 000000000..2e7e4b7dc
--- /dev/null
+++ b/numpy/random/mtrand/mtrand_py_helper.h
@@ -0,0 +1,23 @@
+#ifndef _MTRAND_PY_HELPER_H_
+#define _MTRAND_PY_HELPER_H_
+
+#include <Python.h>
+
+static PyObject *empty_py_bytes(unsigned long 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_ */