def test_all_imports_pyx():
""" Tests: test_all_imports_pyx: for rebuild, syntax correctness and internal imports
"""
print('::: TEST: test_all_imports_pyx()')
remove_files = []
remove_dirs = []
all_modules_path = []
for root, dirnames, filenames in walk(ROOT_PACKAGE_PATH):
all_modules_path.extend(glob(root + '/*.pyx'))
for pyx_module_file_path in all_modules_path:
module_filename = path_basename(pyx_module_file_path)
module_filename_no_ext = path_splitext(module_filename)[0]
cython_extension_module_path, cython_module_c_file_path, cython_build_dir_path = build_cython_extension(
pyx_module_file_path,
cython_force_rebuild=True
)
so_loader = ExtensionFileLoader(module_filename_no_ext, cython_extension_module_path)
so_loader.load_module(module_filename_no_ext)
# add for cleanup
remove_files.append(cython_module_c_file_path)
remove_dirs.append(cython_build_dir_path)
# Cleanup
try:
for file_ in remove_files:
if path_exists(file_):
os_remove(file_)
for dir_ in remove_dirs:
if path_exists(dir_):
rmtree(dir_)
except Exception as err:
raise Exception('test_all_imports_pyx', 'Could not cython_clean_up: Exception: <{}>'.format(err))
def import_module(self, module_dir=None, silent=True):
"""Build an import the module
Keyword Args:
module_dir(str): The location to store all the files of the module (source, temporary objects,
shared object). Default to a temporary location.
silent(bool): Silent compilation. Default True
Returns:
The loaded C extension
Raises:
ImportError: if the C++ code could not be compiled or the module could not be loaded
"""
# Build module
cpp_code = self.get_cpp_code()
extension_kwargs = dict()
if self._enable_numpy:
import numpy as np
extension_kwargs['include_dirs'] = [np.get_include()]
module_filename = build_install_module(cpp_code, self._name, extension_kwargs=extension_kwargs,
module_dir=module_dir, silent=silent)
if module_filename is None:
raise ImportError('Module %s could not be load' % self._name)
# Load module
file_loader = ExtensionFileLoader(self._name, module_filename)
imported_module = file_loader.load_module(self._name)
return imported_module
def _import_path(path, module_name=MODULE_NAME):
"""
Function which imports ``path`` and returns it as a module. This is
meant to import pyd files produced by :meth:`Distribution._build` in
a Python 2/3 agnostic fashion.
:param str path:
The path to the file to import
:keyword str module_name:
Optional name of the module being imported. By default
this will use ``_pywincffi`` if no value is provided.
:raises ResourceNotFoundError:
Raised if ``path`` does not exist.
"""
logger.debug("_import_path(%r, module_name=%r)", path, module_name)
if not isfile(path):
raise ResourceNotFoundError("Module path %r does not exist" % path)
elif ExtensionFileLoader is not None:
loader = ExtensionFileLoader(module_name, path)
return loader.load_module(module_name)
elif imp is not None: # pragma: no cover
return imp.load_dynamic(module_name, path)
else: # pragma: no cover
raise NotImplementedError(
"Neither `imp` or `ExtensionFileLoader` were imported")
def _import_path(path, module_name=MODULE_NAME):
"""
Function which imports ``path`` and returns it as a module. This is
meant to import pyd files produced by :meth:`Distribution._build` in
a Python 2/3 agnostic fashion.
:param str path:
The path to the file to import
:keyword str module_name:
Optional name of the module being imported. By default
this will use ``_pywincffi`` if no value is provided.
:raises ResourceNotFoundError:
Raised if ``path`` does not exist.
"""
if not isfile(path):
raise ResourceNotFoundError("Module path %r does not exist" % path)
elif ExtensionFileLoader is not None:
loader = ExtensionFileLoader(module_name, path)
return loader.load_module(module_name)
elif imp is not None: # pragma: no cover
return imp.load_dynamic(module_name, path)
else: # pragma: no cover
raise NotImplementedError(
"Neither `imp` or `ExtensionFileLoader` were imported")
def load_dynamic_ext(name, path):
''' Load compiled shared object and return as python module. '''
loader = ExtensionFileLoader(name, path)
return loader.load_module()
def load_dynamic_ext(name, path):
''' Load compiled shared object and return as python module. '''
loader = ExtensionFileLoader(name, path)
return loader.load_module()