blob: e64f5d978dbbdac3a57cb6b5370c4264bf76e1ff (
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 obj2sctype
__all__ = ['issubclass_', 'get_numpy_include', 'issubsctype']
def issubclass_(arg1, arg2):
try:
return issubclass(arg1, arg2)
except TypeError:
return False
def issubsctype(arg1, arg2):
return issubclass(obj2sctype(arg1), obj2sctype(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]
|
