U
|
P ±dp* ã @ sB d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
|
Z
|
ddl Z ddl m Z ddlmZmZ ddlmZmZ ddlmZ dadadd Zd
|
d Zd d Zdd Zeg g g dfddZ eg g g ddfddZ da dd Z dd Z dd Z!dd Z"ed d Z#G d d d Z$d d! Z%e
|
j&d"d# Z'dS )$z¥
|
Utility functions for
|
|
- building and importing modules on test time, using a temporary location
|
- detecting if compilers are present
|
- determining paths to tests
|
|
é N)ÚPath)ÚasbytesÚasstr)ÚtemppathÚIS_WASM)Ú import_modulei c C sZ t d k rVztj t ¡ W n tk
|
r, Y nX zt t ¡ W n tk
|
rP Y nX d a d S ©N)Ú _module_dirÚsysÚpathÚremoveÚ
|
ValueErrorÚshutilÚrmtreeÚOSError© r r úLd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/f2py/tests/util.pyÚ_cleanup" s r c C s6 t d kr2t ¡ a t t¡ t tjkr2tj dt ¡ t S ©Nr ) r ÚtempfileÚmkdtempÚatexitÚregisterr r
|
r Úinsertr r r r Úget_module_dir0 s
|
|
r c C s, t dt } td7 a| tjkr(td| S )Nz_test_ext_module_%dé z%Temporary module name already in use.)r Ú _module_numr
|
ÚmodulesÚ RuntimeError)Únamer r r Úget_temp_module_name: s
|
r c s i fdd} j |_ |S )Nc
|
sn t | |f}|krTz | ||< W n, tk
|
rR } z||< W 5 d }~X Y nX | }t|trj||S r )ÚreprÚ ExceptionÚ
|
isinstance)ÚaÚkwÚkeyÚeÚret©ÚfuncÚmemor r ÚwrapperI s
|
z_memoize.<locals>.wrapper)Ú__name__)r* r, r r) r Ú_memoizeF s r. c
|
C sr dt jd}t }g }g }| D ]j} tj | ¡s<td| tj |tj | ¡¡}
|
t | |
|
¡ |
|
|
|
¡ tj |
|
¡\} } | dkr |
|
|
|
¡ q |st |dkr¢t }dd|g| | } |rÆ| dg| 7 } |rØ| d g| 7 } t ¡ }zft |¡ t jd|g| }tj|tjtjd
|
}| ¡ \}}|jd krDtd |d d
t|f W 5 t |¡ |D ]} t | ¡ qVX t|S )zH
|
Compile and import a f2py module, built from the given files.
|
|
zimport sys; sys.path = z&; import numpy.f2py; numpy.f2py.main()ú%s is not a file)ú.f90ú.fz.cú.pyfNz-cz-mzskip:zonly:©ÚstdoutÚstderrr zRunning f2py failed: %s
|
%sé )r
|
r r ÚosÚisfiler ÚjoinÚbasenamer ÚcopyfileÚappendÚsplitextÚAssertionErrorr ÚgetcwdÚchdirÚunlinkÚ
|
executableÚ
|
subprocessÚPopenÚPIPEÚSTDOUTÚ communicateÚ
|
returncoder r )Ú source_filesÚoptionsÚskipÚonlyÚ module_nameÚcodeÚdÚ dst_sourcesZ f2py_sourcesÚfnÚdstÚbaseÚextZ f2py_optsÚcwdÚcmdÚpÚoutÚerrr r r Ú build_module_ sL
|
|
þ ÿ
|
|
rZ c
|
C sb |dkr d}t |dB}t|d}| | ¡ W 5 Q R X t|g||||dW 5 Q R £ S Q R X dS )z6
|
Compile and import Fortran code using f2py.
|
|
Nr1 )ÚsuffixÚw©rJ rK rL rM )r ÚopenÚwriterZ )Z source_coderJ rK rL r[ rM r Úfr r r Ú
|
build_code s üra c C s t d k r t S da trt S t dttj d¡} | tttjd } t ¡ }z`t j |d¡}t|d}| | ¡ W 5 Q R X tjddg}tj|tjtj|d}| ¡ \}}W 5 t
|
|¡ X t d |¡}|rþtt| d
|
¡tt| d ¡tt| d ¡fa t S ) N)FFFz8 import os
|
import sys
|
sys.path = a{
|
|
def configuration(parent_name='',top_path=None):
|
global config
|
from numpy.distutils.misc_util import Configuration
|
config = Configuration('', parent_name, top_path)
|
return config
|
|
from numpy.distutils.core import setup
|
setup(configuration=configuration)
|
|
config_cmd = config.get_config_cmd()
|
have_c = config_cmd.try_compile('void foo() {}')
|
print('COMPILERS:%%d,%%d,%%d' %% (have_c,
|
config.have_f77c(),
|
config.have_f90c()))
|
sys.exit(99)
|
)Zsyspathzsetup.pyr\ Úconfig)r4 r5 rU s COMPILERS:(\d+),(\d+),(\d+)r é é )Ú_compiler_statusr ÚtextwrapÚdedentr! r
|
r Údictr r r r r7 r9 r^ r_ rB rC rD rE rF rG ÚreÚsearchÚboolÚintÚgroup) rN ZtmpdirÚscriptr` rV rW rX rY Úmr r r Ú_get_compiler_status¸ s: ý
|
ý ýrp c C s
|
t d S r ©rp r r r r Úhas_c_compilerö s rr c C s
|
t d S )Nr rq r r r r Úhas_f77_compilerú s rs c C s
|
t d S )Nrc rq r r r r Úhas_f90_compilerþ s rt c
|
K sf t }g }| D ]H}tj |¡s*td| tj |tj |¡¡}t ||¡ | |¡ qt
|
|¡ dd¡}dt tj d| d}tj |t d ¡} | | ¡ t| d}
|
|
|
t|¡ W 5 Q R X t ¡ } zdt |¡ tj| d d
|
g} tj| tjtjd } | ¡ \}}| j d kr.td | dd
t |f W 5 t | ¡ |D ]}t |¡ q@X t | tj | S )z6
|
Build a module via distutils and import it.
|
|
r/ Ú
|
z
|
z!
|
import os
|
import sys
|
sys.path = z¥
|
|
def configuration(parent_name='',top_path=None):
|
from numpy.distutils.misc_util import Configuration
|
config = Configuration('', parent_name, top_path)
|
z
|
return config
|
|
if __name__ == "__main__":
|
from numpy.distutils.core import setup
|
setup(configuration=configuration)
|
z.pyÚwbÚ build_extz-ir3 r z%Running distutils build failed: %s
|
%sr6 N) r r7 r r8 r r9 r: r r; r< rf rg Úreplacer! r
|
r r^ r_ r r? r@ rA rB rC rD rE rF rG rH r Ú
|
__import__r )rI Z config_coderM r% rO rP rQ rR rN rn r` rU rV rW rX rY r r r Úbuild_module_distutils sH ýø
|
|
þ ÿ
|
|
rz c @ s< e Zd ZdZdZg Zg Zg ZdZdZ e
|
dd Z dd Z dS )ÚF2PyTestNr1 c C s* t | }d|j dd¡d d|j dS )NÚ_Ú.r éÿÿÿÿZ _ext_module)ÚtypeÚ
|
__module__Úrsplitr- )ÚselfÚclsr r r rM U s zF2PyTest.module_namec C sL t jdkrt d¡ | jd k r"d S t s2t d¡ g }| jrH| | j¡ | jd k r^| | j
|
¡ d}d}d}|D ]>}t | d¡rd}qnt | d¡rd}qnt | d¡rnd}qn|rÂt sÂt d ¡ |rÖt sÖt d
|
¡ |rðt sðt sðt d ¡ | jd k r t| j| j| j| j| j
|
| jd | _| jd k rHt| j| j| j| j| jd | _d S )NÚwin32z)Fails with MinGW64 Gfortran (Issue #9673)zNo C compiler availableFr1 Tr0 r2 z No Fortran 77 compiler availablez No Fortran 90 compiler availablez No Fortran compiler available)rJ rK rL r[ rM r] )r
|
ÚplatformÚpytestrK Úmodulerr ÚsourcesÚextendrN r< r[ ÚstrÚendswithrs rt ra rJ rL rM rZ )r ÚcodesZ needs_f77Z needs_f90Z needs_pyfrQ r r r Ú setup_methodZ sX
|
|
|
|
|
|
|
|
|
|
ú ûzF2PyTest.setup_method) r- r Ú __qualname__rN r rJ rK rL r[ r ÚpropertyrM r r r r r r{ L s
|
r{ c G s t tjjj ¡ }|j| S r )r ÚnumpyZf2pyÚ__file__ÚparentÚresolveÚjoinpath)r$ rO r r r Úgetpath s r c c s. t ¡ }t | ¡ z
|
d V W 5 t |¡ X d S r )r rU r7 r@ )r Zcurpathr r r Ú switchdir s
|
|
|
r )(Ú__doc__r7 r
|
rC r r r rf ri r Ú
|
contextlibr Úpathlibr Z numpy.compatr r Z numpy.testingr r Ú importlibr r r r r r r. rZ ra re rp rr rs rt rz r{ r Úcontextmanagerr r r r r Ú<module> sR
|
:û >
|
DJ
|
