Is numpy already mocked when setup.py is running?

JohnVinyard · Sep 7, 2017 · b679dd4 · b679dd4
1 parent fa2374c
commit b679dd4
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -22,10 +22,14 @@
     import numpy as np
     from distutils.extension import Extension
 
+    include = np.get_include()
+    if not isinstance(include, str):
+        raise ImportError('numpy is mocked')
+
     countbits = Extension(
         name='countbits',
         sources=['zounds/nputil/countbits.pyx'],
-        include_dirs=[np.get_include()],
+        include_dirs=[include],
         extra_compile_args=[
             '-shared',
             '-pthread',
@@ -36,7 +40,7 @@
             '-fno-strict-aliasing'
         ])
     extension_modules = [countbits]
-except:
+except ImportError:
     extension_modules = []