1 files changed, 51 insertions, 0 deletions
diff --git a/numpy/random/tests/test_extending.py b/numpy/random/tests/test_extending.py
new file mode 100644
index 000000000..807de1a25
--- /dev/null
+++ b/numpy/random/tests/test_extending.py
@@ -0,0 +1,51 @@
+import os, sys
+import pytest
+import warnings
+
+try:
+    import cffi
+except ImportError:
+    cffi = None
+
+if sys.flags.optimize > 1:
+    # no docstrings present to inspect when PYTHONOPTIMIZE/Py_OptimizeFlag > 1
+    # cffi cannot succeed
+    cffi = None
+
+try:
+    with warnings.catch_warnings(record=True) as w:
+        # numba issue gh-4733
+        warnings.filterwarnings('always', '', DeprecationWarning)
+        import numba
+except ImportError:
+    numba = None
+
+try:
+    import cython
+except ImportError:
+    cython = None
+
+@pytest.mark.skipif(cython is None, reason="requires cython")
+def test_cython():
+    curdir = os.getcwd()
+    argv = sys.argv
+    examples = (os.path.dirname(__file__), '..', '_examples')
+    try:
+        os.chdir(os.path.join(*examples))
+        sys.argv = argv[:1] + ['build']
+        with warnings.catch_warnings(record=True) as w:
+            # setuptools issue gh-1885
+            warnings.filterwarnings('always', '', DeprecationWarning)
+            from numpy.random._examples.cython import setup
+    finally:
+        sys.argv = argv
+        os.chdir(curdir)
+
+@pytest.mark.skipif(numba is None or cffi is None,
+                    reason="requires numba and cffi")
+def test_numba():
+        from numpy.random._examples.numba import extending
+
+@pytest.mark.skipif(cffi is None, reason="requires cffi")
+def test_cffi():
+        from numpy.random._examples.cffi import extending