TST: Test building third party C extensions with Py_LIMITED_API

lpsinger · lpsinger · commit 52b2da37e0b1 · 2022-01-13T13:26:14.000-05:00
Fixes #13784.
diff --git a/numpy/core/tests/examples/example_limited_api.c b/numpy/core/tests/examples/example_limited_api.c
@@ -0,0 +1,22 @@
+/*
+ * Test that third-party extensions that use the Numpy C API can be built with
+ * the limited Python C API (see https://docs.python.org/3/c-api/stable.html).
+ */
+
+#define Py_LIMITED_API 0x03060000
+
+#include <Python.h>
+#include <numpy/arrayobject.h>
+#include <numpy/ufuncobject.h>
+
+static PyModuleDef moduledef = {
+    .m_base = PyModuleDef_HEAD_INIT,
+    .m_name = "example_limited_api"
+};
+
+PyMODINIT_FUNC PyInit_example_limited_api(void)
+{
+    import_array();
+    import_umath();
+    return PyModule_Create(&moduledef);
+}
diff --git a/numpy/core/tests/examples/setup.py b/numpy/core/tests/examples/setup.py
@@ -9,16 +9,24 @@
 from setuptools.extension import Extension
 import os
 
+include_dirs = [np.get_include()]
 macros = [("NPY_NO_DEPRECATED_API", 0)]
 
 checks = Extension(
     "checks",
     sources=[os.path.join('.', "checks.pyx")],
-    include_dirs=[np.get_include()],
+    include_dirs=include_dirs,
     define_macros=macros,
 )
 
-extensions = [checks]
+example_limited_api = Extension(
+    "example_limited_api",
+    sources=[os.path.join('.', "example_limited_api.c")],
+    include_dirs=include_dirs,
+    define_macros=macros,
+)
+
+extensions = [checks, example_limited_api]
 
 setup(
     ext_modules=cythonize(extensions)
