1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495 |
- import os
- import pytest
- import shutil
- import subprocess
- import sys
- import warnings
- import numpy as np
- try:
- import cffi
- except ImportError:
- cffi = None
- if sys.flags.optimize > 1:
-
-
- cffi = None
- try:
- with warnings.catch_warnings(record=True) as w:
-
- warnings.filterwarnings('always', '', DeprecationWarning)
- import numba
- except ImportError:
- numba = None
- try:
- import cython
- from Cython.Compiler.Version import version as cython_version
- except ImportError:
- cython = None
- else:
- from distutils.version import LooseVersion
-
-
-
-
- required_version = LooseVersion('0.29.21')
- if LooseVersion(cython_version) < required_version:
-
- cython = None
- @pytest.mark.skipif(cython is None, reason="requires cython")
- @pytest.mark.slow
- def test_cython(tmp_path):
- srcdir = os.path.join(os.path.dirname(__file__), '..')
- shutil.copytree(srcdir, tmp_path / 'random')
-
- build_dir = tmp_path / 'random' / '_examples' / 'cython'
- subprocess.check_call([sys.executable, 'setup.py', 'build', 'install',
- '--prefix', str(tmp_path / 'installdir'),
- '--single-version-externally-managed',
- '--record', str(tmp_path/ 'tmp_install_log.txt'),
- ],
- cwd=str(build_dir),
- )
-
-
- with open(build_dir / 'extending.c') as fid:
- txt_to_find = 'NumPy API declarations from "numpy/__init__.pxd"'
- for i, line in enumerate(fid):
- if txt_to_find in line:
- break
- else:
- assert False, ("Could not find '{}' in C file, "
- "wrong pxd used".format(txt_to_find))
-
- so1 = so2 = None
- with open(tmp_path /'tmp_install_log.txt') as fid:
- for line in fid:
- if 'extending.' in line:
- so1 = line.strip()
- if 'extending_distributions' in line:
- so2 = line.strip()
- assert so1 is not None
- assert so2 is not None
-
- from importlib.machinery import ExtensionFileLoader
- extending = ExtensionFileLoader('extending', so1).load_module()
- extending_distributions = ExtensionFileLoader('extending_distributions', so2).load_module()
-
- from numpy.random import PCG64
- values = extending_distributions.uniforms_ex(PCG64(0), 10, 'd')
- assert values.shape == (10,)
- assert values.dtype == np.float64
- @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
|