File: //usr/local/lib/python3.10/dist-packages/numpy/f2py/tests/__pycache__/util.cpython-310.pyc
o
���g�+ � @ sx d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
mZ ddlmZ ddlmZ ddlmZmZ ddlmZ dadaejd kriee�j j j Z e e �d
��Z d d � Z d
d� Z!dd� Z"dd� Z#e#g g g dfdd��Z$e#g g g ddfdd��Z%da&dd� Z'dd� Z(dd � Z)d d � Z*e#d 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)�asstr)� asunicode)�temppath�IS_WASM)�
import_modulei �cygwinz**/*.dllc C sZ t d ur+ztj�t � W n ty Y nw zt�t � W n ty& Y nw d a d S d S �N)� _module_dir�sys�path�remove�
ValueError�shutil�rmtree�OSError� r r �@/usr/local/lib/python3.10/dist-packages/numpy/f2py/tests/util.py�_cleanup( s � ��r c C s6 t d u rt�� a t�t� t tjvrtj�dt � t S �Nr ) r
�tempfile�mkdtemp�atexit�registerr r r �insertr r r r �get_module_dir6 s
r c C s, t � dt } td7 a| tjv rtd��| 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
sh t | |f�}|�vr'z � | i |���|< W n ty& } z|�|< � d }~ww �| }t|t�r2|�|S r )�repr� Exception�
isinstance)�a�kw�key�e�ret��func�memor r �wrapperO s ��
z_memoize.<locals>.wrapper)�__name__)r+ r- r r* r �_memoizeL s
r/ c
C s� dt j�d�}t� }g }g }| D ]5} tj�| �s td| ��tj�|tj�| ��}
t� | |
� |�
|
� tj� |
�\} } | dv rE|�
|
� q|sJJ �|du rQt � }dd|g| | }
|rc|
dg| 7 }
|rl|
d g| 7 }
t�
� }zAt�|� t jd|g|
}tj|tjtjd
�}|�� \}}|jd kr�td |d
d� t|�f ��W t�|� |D ]} t�| � q�nt�|� |D ]} t�| � q�w t jdkr�t�t�tj�|d� |���� t� g d�t � 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 z{:s}*)z/usr/bin/rebasez
--databasez --obliviousz --verbose) r r r �os�isfiler �join�basenamer �copyfile�append�splitextr! �getcwd�chdir�
executable�
subprocess�Popen�PIPE�STDOUT� communicate�
returncoder �unlink�platform� _module_list�extend�glob�format�
check_callr )� source_files�options�skip�only� module_name�code�d� dst_sources� f2py_sources�fn�dst�base�ext� f2py_opts�cwd�cmd�p�out�errr r r � build_modulee sl
�
�
��
�
� �
���rb c C s� |du rd}t |d��,}t|d��
}|�| � W d � n1 s!w Y t|g||||d�W d � S 1 s:w Y dS )z6
Compile and import Fortran code using f2py.
Nr2 )�suffix�w�rP rQ rR rS )r �open�writerb )� source_coderP rQ rR rc rS r �fr r r �
build_code� s ��$�rj c C s t d urt S da tr t S t�dttj�� d��} | tttj�d� } t� � }z?t
j� |d�}t |d��
}|�
| � W d � n1 sCw Y tjddg}tj|tjtj|d�}|�� \}}W t�|� nt�|� w 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)
)�syspathzsetup.pyrd �config)r5 r6 r] s COMPILERS:(\d+),(\d+),(\d+)r � � )�_compiler_statusr �textwrap�dedentr" r r �dictr r r8 r: rf rg rA rB rC rD rE rF r r �re�search�bool�int�group) rT �tmpdir�scriptri r^ r_ r` ra �mr r r �_get_compiler_status� s<
� � � �r{ c C �
t � d S r �r{ r r r r �has_c_compiler �
r~ c C r| )Nr r} r r r r �has_f77_compiler r r� c C r| )Nrm r} r r r r �has_f90_compiler r r� c
K s� t � }g }| D ]$}tj�|�std| ��tj�|tj�|��}t�||� |� |� qt
� |�� dd�}dt
tj�� d|� d�}tj�|t� d �} |� | � t| d��}
|
�|�d �� W d
� n1 shw Y t�� } z@t�|� tj| d d g} tj| tjtjd
�}
|
�� \}}|
jdkr�td| dd
� t |�f ��W t�| � |D ]}t� |� q�nt�| � |D ]}t� |� q�w t |� tj | S )z6
Build a module via distutils and import it.
r0 �
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�latin1N� build_extz-ir4 r z%Running distutils build failed: %s
%sr7 ) r r8 r r9 r r: r; r r< r= rp rq �replacer" r r! rf rg �encoder? r@ rA rB rC rD rE rF rG r rH �
__import__r )rO � config_coderS r& rU rV rX rY rT ry ri r] r^ r_ r` ra r r r �build_module_distutils sV ��
�
�
��
�
� �
r� c @ s< e Zd ZdZdZg Zg Zg ZdZdZ e
dd� �Z dd� Z dS )�F2PyTestNr2 c C s* t | �}d|j�dd�d � d|j� d�S )N�_�.r ���� _ext_module)�type�
__module__�rsplitr. )�self�clsr r r rS j s "zF2PyTest.module_namec C sL t jdkr
t�d� | jd urd S t� st�d� g }| jr$|�| j� | jd ur/|� | j
� d}d}d}|D ] }t |�� d�rCd}q7t |�� d�rMd}q7t |�� d�rVd}q7|rat
� sat�d � |rkt� skt�d
� |rxt� sxt
� sxt�d � | jd ur�t| j| j| j| j| j
| jd �| _| jd ur�t| j| j| j| j| jd
�| _d S d S )N�win32z)Fails with MinGW64 Gfortran (Issue #9673)zNo C compiler availableFr2 Tr1 r3 z No Fortran 77 compiler availablez No Fortran 90 compiler availablez No Fortran compiler available)rP rQ rR rc rS re )r rI �pytestrQ �moduler~ �sourcesrK rT r= rc �str�endswithr� r� rj rP rR rS rb )r� �codes� needs_f77� needs_f90� needs_pyfrX r r r � setup_methodo s\
�
�
��zF2PyTest.setup_method)
r. r� � __qualname__rT r� rP rQ rR rc r� �propertyrS r� r r r r r� a s
r� c G s t tjj�j�� }|j| � S r )r �numpy�f2py�__file__�parent�resolve�joinpath)r% rU r r r �getpath� s
r� c c s8 � t �� }t�| � z d V W t�|� d S t�|� w r )r r] r8 r@ )r �curpathr r r � switchdir� s �
r� )0�__doc__rL r8 r rB r r r rp rs r� �
contextlibr� �pathlibr � numpy.compatr � numpy._utilsr �
numpy.testingr r � importlibr r
r rI r� r� �NUMPY_INSTALL_ROOT�listrJ r r r! r/ rb rj ro r{ r~ r� r� r� r� r� �contextmanagerr� r r r r �<module> s^
I �>
DJ