File: //usr/local/lib/python3.10/dist-packages/numpy/f2py/tests/__pycache__/test_f2py2e.cpython-310.pyc
o
���gOc � @ s d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl Z d dl
Z
ddl m Z d dl
mZ edd�Zd�d
d �Ze
jd d
�dd� �Ze
jd d
�dd� �Ze
jd d
�dd� �Ze
jd d
�dd� �Ze
jd d
�dd� �Ze
jd d
�dd� �Zdd� Zd d � Zd d � Zd d!� Z d"d#� Z d$d%� Z e
j j e �!� d&kp�ej"d'kd(d)�d*d+� �Z#e
j j e �!� d&kp�ej"d'kd(d)�d,d-� �Z$e
j j%d.d/� �Z&d0d1� Z'd2d3� Z(d4d5� Z)d6d7� Z*d8d9� Z+d:d;� Z,d<d=� Z-d>d?� Z.d@dA� Z/dBdC� Z0dDdE� Z1dFdG� Z2dHdI� Z3dJdK� Z4e
j j%dLd)�dMdN� �Z5dOdP� Z6dQdR� Z7dSdT� Z8dUdV� Z9dWdX� Z:dYdZ� Z;d[d\� Z<d]d^� Z=d_d`� Z>e
j j%dLd)�dadb� �Z?dcdd� Z@dedf� ZAdgdh� ZBdidj� ZCdkdl� ZDdmdn� ZEdodp� ZFdqdr� ZGdsdt� ZHdudv� ZIdwdx� ZJdydz� ZKd{d|� ZLd}d~� ZMdd�� ZNd�d�� ZOd�d�� ZPd�d�� ZQdS )�� N)�Path)�
namedtuple� )�util)�main�PPathsz(finp, f90inp, pyf, wrap77, wrap90, cmodf�untitledc
C sR t | �}t|�d�|�d�|�d�|�|� d��|�|� d��|�|� d��d�S )ax Takes in a temporary file for testing and returns the expected output and input paths
Here expected output is essentially one of any of the possible generated
files.
..note::
Since this does not actually run f2py, none of these are guaranteed to
exist, and module names are typically incorrect
Parameters
----------
fname_inp : str
The input filename
mname : str, optional
The name of the module, untitled by default
Returns
-------
genp : NamedTuple PPaths
The possible paths which are generated, not all of which exist
z.fz.f90�.pyfz-f2pywrappers.fz-f2pywrappers2.f90�module.c)�finp�f90inp�pyf�wrap77�wrap90�cmodf)r r � with_suffix� with_name)� fname_inp�mname�bpath� r �G/usr/local/lib/python3.10/dist-packages/numpy/f2py/tests/test_f2py2e.py� get_io_paths s �r �session)�scopec C �2 t �dddd��� }| �� d }|j|dd� |S )�'Generates a single f90 file for testing�tests�src�cliz hiworld.f90z hello.f90�ascii��encoding�r �getpath� read_text� getbasetemp�
write_text��tmpdir_factory�fdat�fnr r r �hello_world_f909 � r, c C �2 t �dddd��� }| �� d }|j|dd� |S )z(F90 file for testing warnings in gh23598r r � crackfortranzgh23598Warn.f90r r! r# r( r r r � gh23598_warnB r- r0 c C r. )z8F90 file for testing disallowed CLI arguments in ghff819r r r z gh_22819.pyfr r! r# r( r r r � gh22819_cliK r- r1 c C r )�'Generates a single f77 file for testingr r r zhi77.fzhello.fr r! r# r( r r r �hello_world_f77T r- r3 c C r )r2 r r � return_realzfoo77.fzfoo.fr r! r# r( r r r � retreal_f77] r- r5 c C s` t �dddd��� }t �dddd��� }| �� d }| �� d }|j|dd � |j|dd � |S )
r r r �f2cmapzisoFortranEnvMap.f90z .f2py_f2cmapz
f2cmap.f90�mapfiler r! r# )r) r* r6 r+ �fmapr r r �
f2cmap_f90e s r9 c C s� t |�}|�tdd|� ��� � t�|j��0 t� dd� |j�d�D �}d|vs+J �d|vs1J �d|v s7J �d |v s=J � W d � d S 1 sIw Y d S ) z�Check that module names are handled correctly
gh-22819
Essentially, the -m name cannot be used to import the module, so the module
named in the .pyf needs to be used instead
CLI :: -m and a .pyf file
�argv�
f2py -m blah c S s g | ] }|� � r|j�qS r )�is_file�name)�.0�itemr r r �
<listcomp>} s z$test_gh22819_cli.<locals>.<listcomp>�*z blahmodule.czblah-f2pywrappers.fztest_22819-f2pywrappers.fztest_22819module.cz
Ignoring blahN) r �setattr�sys�splitr � switchdir�parent�f2pycli�rglob)�capfdr1 � monkeypatch�ipath� gen_pathsr r r �test_gh22819_cliq s "�rM c C s� t |�}|�tdd|� d��� � t�|j��+ tjt dd�� t
� W d � n1 s-w Y W d � dS W d � dS 1 sEw Y dS )zBOnly one .pyf file allowed
gh-22819
CLI :: .pyf files
r: r; z
hello.pyfzOnly one .pyf file per call)�matchN) r rB rC rD r rE rF �pytest�raises�
ValueErrorrG )rI r1 rJ rK r r r �test_gh22819_many_pyf� s ��"�rR c C sz t |dd�}|j}|�tdd|� d��� � t�|j�� t� |j �
� }d|vs+J �W d � d S 1 s6w Y d S )N�test�r r: �f2py z -m testzintproductf2pywrap, intpr) r r rB rC rD r rE rF rG r r% )rI r0 rJ �foutlrK �wrapperr r r �test_gh23598_warn� s �
"�rX c C s� t |�}t |�jd }|�tdd|� d|� ��� � t�|j��! t� | � � \}}d|v s0J �t |� ��
� s9J �W d � dS 1 sDw Y dS )zIEnsures that a signature file is generated via the CLI
CLI :: -h
r r: zf2py -h � �Saving signatures to fileN) r �stemrB rC rD r rE rF rG �
readouterr�exists)rI r, rJ rK �opath�out�_r r r � test_gen_pyf� s "�ra c C s| t |�}|�tdd|� ��� � t�|j�� t� | �� \}}d|v s&J �d|v s,J �W d � dS 1 s7w Y dS )zHEnsures that a signature file can be dumped to stdout
CLI :: -h
r: zf2py -h stdout rZ zfunction hi() ! in N� r rB rC rD r rE rF rG r\ �rI r, rJ rK r_ r` r r r �test_gen_pyf_stdout� s "�rd c C s� t |�}|�tdd|� ��� � t�|j��> t d�jddd� t� t
�� t � | � � \}}d|v s5J �W d � n1 s?w Y W d � dS W d � dS 1 sWw Y dS ) zjEnsures that the CLI refuses to overwrite signature files
CLI :: -h without --overwrite-signature
r: �f2py -h faker.pyf � faker.pyf� Fake newsr r! z&Use --overwrite-signature to overwriteN)
r rB rC rD r rE rF r' rO rP �
SystemExitrG r\ )rI r, rJ rK r` �errr r r �test_gen_pyf_no_overwrite� s ��"�rj �Linux)� � zCompiler and 3.12 required)�reasonc C sp t |�}|�tdd|� ��� � t�|j�� t� | �� \}}d|v s&J �W d � dS 1 s1w Y dS )z@Check that modules are named correctly
CLI :: defaults
r: zf2py --backend meson -c zuntitledmodule.cNrb rc r r r �test_untitled_cli� s "�ro c C sN d}t ||d�}|j}|�tdd|� d|� ��� � t�|j�� t� | � � \}}d|v s0J �W d � n1 s:w Y |�tdd�� � t�|j�� t� | � � \}}d |v s^J �W d � n1 shw Y d
}|�tdd|� d |� d ��� � t�|j�� t� | � � \}}d
|v s�J �W d � dS 1 s�w Y dS )zqCheck that no distutils imports are performed on 3.12
CLI :: --fcompiler --help-link --backend distutils
�hirT r: rU z -c --fcompiler=gfortran -m z%--fcompiler cannot be used with mesonNzf2py --help-linkzUse --dep for meson builds�hi2z -c -m z --backend distutilsz.Cannot use distutils backend with Python>=3.12)
r r rB rC rD r rE rF rG r\ )rI r, rJ �MNAMErV rK r_ r` r r r �!test_no_py312_distutils_fcompiler� s8 � �
� �� "�rs c C s� t |dd�}|j}d}d}|�tdd|� d|� ��� � t�|j��4 t� | � � \}}|�� D ] } d| � d �|v s<J �q0|�� D ] }
d
|
� d �|v sMJ �qAW d � d S 1 sYw Y d S )
z9Tests that functions can be skipped
CLI :: skip:
rS rT �t0 t4 t8 sd s8 s4�td s0r: rU z -m test skip: z=buildmodule: Could not found the body of interfaced routine "� ". Skipping.� Constructing wrapper function "�"N�
r r rB rC rD r rE rF rG r\ ) rI r5 rJ rV rK �toskip� remainingr_ ri �skey�rkeyr r r �test_f2py_skip� s& �
� �"�r~ c C s� t |dd�}|j}d}d}|�tdd|� d|� ��� � t�|j��4 t� | � � \}}|�� D ] } d| � d �|v s<J �q0|�� D ] }
d
|
� d �|v sMJ �qAW d � d S 1 sYw Y d S )
z>Test that functions can be kept by only:
CLI :: only:
rS rT rt ru r: rU z -m test only: �<buildmodule: Could not find the body of interfaced routine "rv rw rx Nry ) rI r5 rJ rV rK rz �tokeepr_ ri r| r} r r r �test_f2py_only s& �
� �"�r� c C s� t |dd�}|j}d}t|�}d}d} |�tdd|� d| � d |� d
|� ��� � t�|j��4 t � | �
� \}
} |�� D ] } d | � d �| v sHJ �q<|�� D ] }
d
|
� d�|
v sYJ �qMW d � dS 1 sew Y dS )zfTests that it is possible to return to file processing mode
CLI :: :
BUG: numpy-gh #20520
rS rT rt ztd s0 hi�blahr: rU � -m z only: z : r rv rw rx N) r r r rB rC rD r rE rF rG r\ )rI r, r5 rJ rV rK rz �ipath2r� r r_ ri r| r} r r r �test_file_processing_switch' s, �
� �"�r� c C s� d}t ||d�}|j}|�tdd|� d|� ��� � t�|j�� t� W d � n1 s.w Y t �
|j �s;J �t �
|j �sCJ �dS )zHChecks the generation of files based on a module name
CLI :: -m
rp rT r: rU r� N)
r r rB rC rD r rE rF rG r r] r r )rI r, rJ rr rV rK r r r �test_mod_gen_f77E s �r� c C s� d}t ||d�}|j}|�tdd|� d|� d��� � t�|j��4 t� t d��
� �}|� � }d|v s6J �W d � n1 s@w Y W d � d S W d � d S 1 sXw Y d S )
z�Check that pyf files are correctly generated with module structure
CLI :: -m <name> -h pyf_file
BUG: numpy-gh #20520
rp rT r: rU r� z
-h hi.pyfzhi.pyfzpython module hiN) r r rB rC rD r rE rF rG r �open�read)rI r3 rJ rr rV rK �hipyf�pyfdatr r r �test_mod_gen_gh25263V s ��"�r� c C s$ t |dd�}|j}t�d�}t�d�}|�tdd|� d��� � t�|j ��# t
� | � � \}}|� |�dus9J �|� |�du sBJ �W d � n1 sLw Y |�tdd|� d ��� � t�|j ��$ t
� | � � \}}|� |�du swJ �|� |�dus�J �W d � dS 1 s�w Y dS )
zHLowers cases by flag or when -h is present
CLI :: --[no-]lower
rS rT zHI\(\)zhi\(\)r: rU z -m test --lowerNz -m test --no-lower)
r r �re�compilerB rC rD r rE rF rG r\ �search� rI r3 rJ rV rK �capshi�capslor_ r` r r r �test_lower_cmodf s(
�� "�r� c C s4 t |dd�}|j}t�d�}t�d�}|�tdd|� d|j� d��� � t� |j
��# t � | � � \}}|�
|�d us=J �|�
|�d u sFJ �W d � n1 sPw Y |�tdd|� d|j� d
��� � t� |j
��$ t � | � � \}}|�
|�d u sJ �|�
|�d us�J �W d � d S 1 s�w Y d S ) z^Lowers cases in signature files by flag or when -h is present
CLI :: --[no-]lower -h
rS rT z Block: HIz Block: hir: rU z -h z -m test --overwrite-signatureNz) -m test --overwrite-signature --no-lower)r r r� r� rB rC r
rD r rE rF rG r\ r� r� r r r �test_lower_sig� s6
� �� "�r� c
C s� t |�}d}d}|�tdd|� d|� d|� ��� � t�|j�� t� | �� \}}d|� d�|v s4J �W d � d S 1 s?w Y d S )
zNEnsures that the build directory can be specified
CLI :: --build-dir
r� �tttmpr: �f2py -m rY z
--build-dir zWrote C/API module "rx Nrb )rI r, rJ rK r �odirr_ r` r r r �test_build_dir� s � "�r� c C s� t |�}|�tdd|� d��� � t�|j��! t d�jddd� t� | � � \}}d|v s0J �W d � d S 1 s;w Y d S )
zXEnsures that the build directory can be specified
CLI :: --overwrite-signature
r: re z --overwrite-signaturerf rg r r! rZ N)
r rB rC rD r rE rF r' rG r\ rc r r r �test_overwrite� s � "�r� c C s� t |�}d}|�tdd|� d|� d��� � t�|j��A t� | �� \}}d|v s,J �t |� d��� � �}d|�
� v s>J �W d � n1 sHw Y W d � d S W d � d S 1 s`w Y d S )
zJEnsures that TeX documentation is written out
CLI :: --latex-doc
r� r: r� rY z --latex-doc� Documentation is saved to file�
module.tex�\documentclassN� r rB rC rD r rE rF rG r\ r� r� �rI r, rJ rK r r_ r` �otexr r r �
test_latexdoc� s � ��"�r� c C �| t |�}d}|�tdd|� d|� d��� � t�|j�� t� | �� \}}d|vs,J �W d � dS 1 s7w Y dS )zMEnsures that TeX documentation is written out
CLI :: --no-latex-doc
r� r: r� rY z --no-latex-docr� Nrb �rI r, rJ rK r r_ r` r r r �test_nolatexdoc� � � "�r� c C s� t |�}d}|�tdd|� d|� d��� � t�|j��B t� | �� \}}d|v s,J �t d|� d��� � �}d |�
� vs?J �W d
� n1 sIw Y W d
� d
S W d
� d
S 1 saw Y d
S ) z�Ensures that truncated documentation is written out
TODO: Test to ensure this has no effect without --latex-doc
CLI :: --latex-doc --short-latex
r� r: r� rY z --latex-doc --short-latexr� �./r� r� Nr� r� r r r �test_shortlatex� s" � ��"�r� c C s� t |�}d}|�tdd|� d|� d��� � t�|j��B t� | �� \}}d|v s,J �t d|� d��� � �}d |�
� v s?J �W d
� n1 sIw Y W d
� d
S W d
� d
S 1 saw Y d
S ) zIEnsures that RsT documentation is written out
CLI :: --rest-doc
r� r: r� rY z --rest-doc�#ReST Documentation is saved to filer� z module.restz.. -*- rest -*-Nr� )rI r, rJ rK r r_ r` �orstr r r � test_restdoc s � ��"�r� c C r� )zLEnsures that TeX documentation is written out
CLI :: --no-rest-doc
r� r: r� rY z --no-rest-docr� Nrb r� r r r �test_norestexdoc r� r� c C s� t |�}d}|�tdd|� d|� d��� � t�|j��6 t� t d|� d���� �}d|� � v s3J �W d � n1 s=w Y W d � d S W d � d S 1 sUw Y d S )
zIEnsures that debugging wrappers are written
CLI :: --debug-capi
r� r: r� rY z
--debug-capir� r
z#define DEBUGCFUNCSN�
r rB rC rD r rE rF rG r� r� )rI r, rJ rK r �ocmodr r r �test_debugcapi# s � ��"�r� zConsistently fails on CI.c C s� t | �}d}|�tdd|� d|� d��� � t�|j��. t� t�d�}t j
|ddd �}d
}t � d �}|j
|ks;J �|j|ksBJ �W d � d S 1 sMw Y d S )
zEEnsures that debugging wrappers work
CLI :: --debug-capi -c
r� r: r� rY z -c --debug-capiz#python3 -c "import blah; blah.hi()"T�UTF-8��capture_outputr" �
Hello World
a debug-capi:Python C/API function blah.hi()
debug-capi:float hi=:output,hidden,scalar
debug-capi:hi=0
debug-capi:Fortran subroutine `f2pywraphi(&hi)'
debug-capi:hi=0
debug-capi:Building return value.
debug-capi:Python C/API function blah.hi: successful.
debug-capi:Freeing memory.
N)r rB rC rD r rE rF rG �shlex�
subprocess�run�textwrap�dedent�stdout�stderr)r, rJ rK r �cmd_run�rout�eout�eerrr r r �test_debugcapi_bld3 s �
"�r� c C s� t |�}d}|�tdd|� d|� ��� � t�|j�� t� W d � n1 s)w Y | �� \}}d|v s:J �|�tdd|� d|� d��� � t�|j�� t� | �� \}}d|v s`J �W d � dS 1 skw Y dS )zoEnsures that fortran subroutine wrappers for F77 are included by default
CLI :: --[no]-wrap-functions
r� r: r� rY N� Fortran 77 wrappers are saved toz --wrap-functionsrb r� r r r �test_wrapfunc_defQ s � � "�r� c C r� )zeEnsures that fortran subroutine wrappers for F77 can be disabled
CLI :: --no-wrap-functions
r� r: r� rY z --no-wrap-functionsr� Nrb r� r r r �test_nowrapfuncj r� r� c C s� t |�}d}|�tdd|� d|� d��� � t�|j��> t� t d|� d���� �}|� � }d|v s5J �d |v s;J �W d
� n1 sEw Y W d
� d
S W d
� d
S 1 s]w Y d
S ) zcAdd to the include directories
CLI :: -include
TODO: Document this in the help string
r� r: r� rY z' -include<stdbool.h> -include<stdio.h> r� r
z#include <stdbool.h>z#include <stdio.h>Nr� )rI r, rJ rK r r� �ocmrr r r �test_inclheaderz s$ � ��"�r� c C � dS )z?Add to the include directories
CLI :: --include-paths
Nr r r r r �
test_inclpath� � r� c C r� )z;Add to the include directories
CLI :: --help-link
Nr r r r r �
test_hlink� r� r� c C s� t |�}|�tdd|� d��� � t�|j��0 t� | �� \}}d|v s'J �d|v s-J �d|v s3J �d|v s9J �d|v s?J �W d � d S 1 sJw Y d S )
zOCheck that Fortran-to-Python KIND specs can be passed
CLI :: --f2cmap
r: r; z --f2cmap mapfilez!Reading f2cmap from 'mapfile' ...z&Mapping "real(kind=real32)" to "float"z'Mapping "real(kind=real64)" to "double"z,Mapping "integer(kind=int64)" to "long_long"z0Successfully applied user defined f2cmap changesNrb )rI r9 rJ rK r_ r` r r r � test_f2cmap� s "�r� c C sv t |�}|�tdd|� d��� � t�|j�� t� | �� \}}t |�dks)J �W d � dS 1 s4w Y dS )z)Reduce verbosity
CLI :: --quiet
r: r; z --quietr N)
r rB rC rD r rE rF rG r\ �lenrc r r r �
test_quiet� s "�r� c C sr t |�}|�tdd|� d��� � t�|j�� t� | �� \}}d|v s'J �W d � dS 1 s2w Y dS )z-Increase verbosity
CLI :: --verbose
r: r; z
--verbose� analyzelineNrb rc r r r � test_verbose� s "�r� c C sn |� tdd�� � t�t�� t� | �� \}}ddl}|j |�
� ks%J �W d � dS 1 s0w Y dS )z"Ensure version
CLI :: -v
r: zf2py -vr N) rB rC rD rO rP rh rG r\ �numpy� __version__�strip)rI rJ r_ r` �npr r r � test_version� s "�r� c C s� t | �}|�tdd|� d��� � t�|j��" t� t�d�}t j
|ddd�}d}|j |ks1J �W d � d S 1 s<w Y d S )
z
CLI :: -c
r: r; z -cz"python -c "import blah; blah.hi()"Tr� r� r� N) r rB rC rD r rE rF rG r� r� r� r� )r, rJ rK r� r� r� r r r �
test_npdistop� s
"�r� c C r� )z
CLI :: -c --fcompiler
Nr r r r r �test_npd_fcompiler� � r� c C r� )z
CLI :: -c --compiler
Nr r r r r �test_npd_compiler� r� r� c C r� )z$
CLI :: -c --help-fcompiler
Nr r r r r �test_npd_help_fcompiler r� r� c C r� )z
CLI :: -c --f77exec
Nr r r r r �test_npd_f77exec r� r� c C r� )z
CLI :: -c --f90exec
Nr r r r r �test_npd_f90exec r� r� c C r� )z
CLI :: -c --f77flags
Nr r r r r �test_npd_f77flags r� r� c C r� )z
CLI :: -c --f90flags
Nr r r r r �test_npd_f90flags# r� r� c C r� )z
CLI :: -c --opt
Nr r r r r � test_npd_opt+ r� r� c C r� )z
CLI :: -c --arch
Nr r r r r �
test_npd_arch3 r� r� c C r� )z
CLI :: -c --noopt
Nr r r r r �test_npd_noopt; r� r� c C r� )z
CLI :: -c --noarch
Nr r r r r �test_npd_noarchC r� r� c C r� )z
CLI :: -c --debug
Nr r r r r �test_npd_debugK r� r� c C r� )z%
CLI :: -c --link-<resource>
Nr r r r r �test_npd_link_autoS r� r� c C r� )z/
CLI :: -c -L/path/to/lib/ -l<libname>
Nr r r r r � test_npd_lib[ r� r� c C r� )z
CLI :: -D<define>
Nr r r r r �test_npd_definec r� r� c C r� )z
CLI :: -U<name>
Nr r r r r �test_npd_undefinek r� r� c C r� )z$
CLI :: -I/path/to/include/
Nr r r r r �
test_npd_incls r� r� c C r� )z8
CLI :: <filename>.o <filename>.so <filename>.a
Nr r r r r �test_npd_linker{ r� r� )r )Rr� r� rC r� r� �pathlibr � collectionsr �platformrO � r �numpy.f2py.f2py2er rG r r �fixturer, r0 r1 r3 r5 r9 rM rR rX ra rd rj �mark�skipif�system� version_inforo rs �xfailr~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r �<module> s� (
'
�
'