TST. API: test using distributions.h via cffi

1 files changed, 10 insertions, 2 deletions

diff --git a/numpy/random/tests/test_extending.py b/numpy/random/tests/test_extending.py
index 6f0f7a462..607853663 100644
--- a/numpy/random/tests/test_extending.py
+++ b/numpy/random/tests/test_extending.py
@@ -3,11 +3,15 @@ import pytest
 import warnings
 
 try:
+    import cffi
+except ImportError:
+    cffi = None
+
+try:
     with warnings.catch_warnings(record=True) as w:
         # numba issue gh-4733
         warnings.filterwarnings('always', '', DeprecationWarning)
         import numba
-    import cffi
 except ImportError:
     numba = None
 
@@ -26,7 +30,11 @@ def test_cython():
         sys.argv = argv
         os.chdir(curdir)
 
-@pytest.mark.skipif(numba is None, reason="requires numba")
+@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