3
���h�% � @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z m Z ddl
mZ ddlmZ ddlmZ dadadd � Zd
d � Zd d
� Zdd� Zeg g g dfdd��Zeg g g ddfdd��Zdadd� Z dd� Z dd� Z dd� Z ed d � �Z G d d � d �Z!dS ) z�
Utility functions for
- building and importing modules on test time, using a temporary location
- detecting if compilers are present
� N)�asbytes�asstr)�temppath)�
import_module)�md5i c C s^ t d k rZytjjt � W n tk
r, Y nX ytjt � W n ttfk
rT Y nX d a d S )N) � _module_dir�sys�path�remove�
ValueError�shutil�rmtree�IOError�OSError� r r �6/tmp/pip-build-5_djhm0z/numpy/numpy/f2py/tests/util.py�_cleanup s r c C s6 t d kr2tj� a tjt� t tjkr2tjjdt � t S )Nr ) r �tempfile�mkdtemp�atexit�registerr r r �insertr r r r �get_module_dir. 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)�d�namer r r �get_temp_module_name8 s
r c s i �� �fdd�}� j |_ |S )Nc sn t | |f�}|�krTy� | |��|< W n, tk
rR } z|�|< � W Y d d }~X nX �| }t|t�rj|�|S )N)�repr� Exception�
isinstance)�a�kw�key�e�ret)�func�memor r �wrapperG s
z_memoize.<locals>.wrapper)�__name__)r( r* r )r( r) r �_memoizeD s r, c C sr dt tj� }t� }g }g }xr| D ]j} tjj| �s>td| ��tjj|tjj| ��}
t j
| |
� |j |
� tjj |
�\} } | dkr"|j |
� q"W |dkr�t
� }dd |g| | }
|r�|
d
g| 7 }
|r�|
d g| 7 }
tj� }zftj|� tjd|g|
}tj|tjtjd �}|j� \}}|jd
k�r@td|dd� t|�f ��W dtj|� x|D ]} tj| � �qTW X t|�S )zH
Compile and import a f2py module, built from the given files.
zEimport sys; sys.path = %s; import numpy.f2py as f2py2e; f2py2e.main()z%s is not a file�.f90�.f�.c�.pyfNz-cz-mzskip:zonly:)�stdout�stderrr zRunning f2py failed: %s
%s� )r- r. r/ r0 )r r r r �os�isfiler �join�basenamer �copyfile�append�splitextr �getcwd�chdir�
executable�
subprocess�Popen�PIPE�STDOUT� communicate�
returncoder �unlinkr )� source_files�options�skip�only� module_name�coder � dst_sourcesZ f2py_sources�fn�dst�base�extZ f2py_opts�cwd�cmd�p�out�errr r r � build_module[ sF
rU c C sV |dkr d}t |d��6}t|d��}|j| � W dQ R X t|g||||d�S Q R X dS )z6
Compile and import Fortran code using f2py.
Nz.f)�suffix�w)rF rG rH rI )r �open�writerU )Z source_coderF rG rH rV rI r �fr r r �
build_code� s r[ c C s� t d k r t S d a tjd�} | tttj�d� } tj� }z`t jj
|d�}t |d��}|j | � W d Q R X tj
ddg}tj|tjtj|d�}|j� \}}W d tj|� X tjd|�}|r�tt|jd ���tt|jd
���tt|jd ���fa t S )
NFa� import os
import sys
sys.path = %(syspath)s
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)
)�syspathzsetup.pyrW �config)r1 r2 rP s COMPILERS:(\d+),(\d+),(\d+)r � � )FFF)�_compiler_status�textwrap�dedent�dictr r r r r r4 r6 rX rY r= r> r? r@ rA rB r r
�re�search�bool�int�group) rJ Ztmpdir�scriptrZ rQ rR rS rT �mr r r �_get_compiler_status� s,
rk c C s
t � d S )Nr )rk r r r r �has_c_compiler� s rl c C s
t � d S )Nr )rk r r r r �has_f77_compiler� s rm c C s
t � d S )Nr^ )rk r r r r �has_f90_compiler� s rn c K s� ddl m} ddlm} t� }g }xP| D ]H}tjj|�sDtd| ��tjj |tjj
|��} t j || � |j
| � q(W tj|�jdd�}tjd�t|ttj�d� }
tjj |t� d �} |j
| � t| d
��} | jt|
�� W d Q R X tj� }
zdtj|� tj| d d
g}tj |tj tj d�}|j � \}}|j dk�rNtd|dd � t!|�f ��W d tj|
� x|D ]}tj"|� �qbW X t#|� tj$| S )z6
Build a module via distutils and import it.
r )�
Configuration)�setupz%s is not a file�
z
a� import os
import sys
sys.path = %(syspath)s
def configuration(parent_name='',top_path=None):
from numpy.distutils.misc_util import Configuration
config = Configuration('', parent_name, top_path)
%(config_code)s
return config
if __name__ == "__main__":
from numpy.distutils.core import setup
setup(configuration=configuration)
)� config_coder\ z.py�wbN� build_extz-i)r1 r2 z%Running distutils build failed: %s
%sr3 )%Znumpy.distutils.misc_utilro Znumpy.distutils.corerp r r4 r r5 r r6 r7 r r8 r9 ra rb �replacerc r r r rX rY r r; r< r= r> r? r@ rA rB rC r rD �
__import__r )rE rr rI r$ ro rp r rK rL rM rJ ri rZ rP rQ rR rS rT r r r �build_module_distutils� s@
rw c @ s4 e Zd ZdZdZg Zg Zg ZdZdZ dZ
dd� Z dS )�F2PyTestNz.fc C s t jdkrtjd� | jd k r"d S t� s2tjd� g }| jrH|j| j� | jd k r^|j | j
� d}d}x*|D ]"}|j d�r�d}ql|j d�rld}qlW |r�t � r�tjd� |r�t
� r�tjd � | jd k r�t| j| j| j| j| j
| jd
�| _| jd k �rt| j| j| j| j| jd �| _d S ) N�win32z)Fails with MinGW64 Gfortran (Issue #9673)zNo C compiler availableFz.fTz.f90z No Fortran 77 compiler availablez No Fortran 90 compiler available)rF rG rH rV rI )rF rG rH rI )r �platform�pytestrG �modulerl �sources�extendrJ r9 rV �endswithrm rn r[ rF rH rI rU )�self�codesZ needs_f77Z needs_f90rL r r r rp B s>
zF2PyTest.setup) r+ �
__module__� __qualname__rJ r} rF rG rH rV r| rI rp r r r r rx 8 s rx )"�__doc__r4 r r> r r r ra rd r{ Z numpy.compatr r Z
numpy.testingr � importlibr �hashlibr r r r r r r, rU r[ r` rk rl rm rn rw rx r r r r �<module> s>
87 G |