blob: e9ff9e540135cb521aa6f0d91168085e8fbe3718 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
from numpy.core.numerictypes import obj2arrtype
__all__ = ['issubclass_', 'get_numpy_include', 'issubarrtype']
def issubclass_(arg1, arg2):
try:
return issubclass(arg1, arg2)
except TypeError:
return False
def issubarrtype(arg1, arg2):
return issubclass(obj2arrtype(arg1), obj2arrtype(arg2))
def get_numpy_include():
"""Return the directory in the package that contains the numpy/*.h header
files.
Extension modules that need to compile against numpy should use this
function to locate the appropriate include directory. Using distutils:
import numpy
Extension('extension_name', ...
include_dirs=[numpy.get_numpy_include()])
"""
from numpy.distutils.misc_util import get_numpy_include_dirs
include_dirs = get_numpy_include_dirs()
assert len(include_dirs)==1,`include_dirs`
return include_dirs[0]
|
