U
|
P ±dqT ã @ s\ d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl Z ddl
|
m Z d dl m Z eddZdtd
|
d Ze jd d dd Ze jd d dd Ze jd d dd Ze jd d dd Zdd Zdd Zdd Ze jjd d Zd d Z d d! Z d"d# Z d$d% Z d&d' Z d(d) Z!d*d+ Z"d,d- Z#d.d/ Z$d0d1 Z%d2d3 Z&d4d5 Z'd6d7 Z(e jjd8d9d:d; Z)d<d= Z*d>d? Z+d@dA Z,dBdC Z-dDdE Z.dFdG Z/dHdI Z0dJdK Z1dLdM Z2e jjd8d9dNdO Z3dPdQ Z4dRdS Z5dTdU Z6dVdW Z7dXdY Z8dZd[ Z9d\d] Z:d^d_ Z;d`da Z<dbdc Z=ddde Z>dfdg Z?dhdi Z@djdk ZAdldm ZBdndo ZCdpdq ZDdrds ZEdS )ué N)ÚPath)Ú
|
namedtupleé )Úutil)ÚmainÚPPathsz(finp, f90inp, pyf, wrap77, wrap90, cmodfÚuntitledc
|
C sR t | }t| d¡| d¡| d¡| | d¡| | d¡| | d¡dS )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Úwrap77Zwrap90Úcmodf)r r Ú with_suffixÚ with_name)Z fname_inpÚmnameÚbpath© r úSd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/f2py/tests/test_f2py2e.pyÚ get_io_paths s úr Úsession)Zscopec C s2 t dddd¡ ¡ }| ¡ d }|j|dd |S )ú'Generates a single f90 file for testingÚtestsÚsrcÚcliz hiworld.f90z hello.f90Úascii©Úencoding©r ZgetpathÚ read_textZ getbasetempÚ
|
write_text©Útmpdir_factoryÚfdatÚfnr r r Úhello_world_f908 s r& c C s2 t dddd¡ ¡ }| ¡ d }|j|dd |S )ú'Generates a single f77 file for testingr r r zhi77.fzhello.fr r r r" r r r Úhello_world_f77A s r( c C s2 t dddd¡ ¡ }| ¡ d }|j|dd |S )r' r r Z return_realzfoo77.fzfoo.fr r r r" r r r Ú retreal_f77J s r) c C s` t dddd¡ ¡ }t dddd¡ ¡ }| ¡ d }| ¡ d }|j|dd |j|dd |S )
|
r r r Úf2cmapzisoFortranEnvMap.f90z .f2py_f2cmapz
|
f2cmap.f90Zmapfiler r r )r# r$ r* r% Zfmapr r r Ú
|
f2cmap_f90R s r+ c C s t |}t |jd }| tdd| d| ¡ ¡ t |j¡6 t | ¡ \}}d|ks`t
|
t | ¡ srt
|
W 5 Q R X dS )zIEnsures that a signature file is generated via the CLI
|
CLI :: -h
|
r Úargvzf2py -h ú úSaving signatures to fileN) r ÚstemÚsetattrÚsysÚsplitr Ú switchdirÚparentÚf2pycliÚ
|
readouterrÚAssertionErrorÚexists)Úcapfdr& Ú monkeypatchÚipathZopathÚoutÚ_r r r Ú test_gen_pyf^ s r> c C sf t |}| tdd| ¡ ¡ t |j¡0 t | ¡ \}}d|ksLt d|ksXt W 5 Q R X dS )zHEnsures that a signature file can be dumped to stdout
|
CLI :: -h
|
r, zf2py -h stdout r. zfunction hi() ! in N©
|
r r0 r1 r2 r r3 r4 r5 r6 r7 ©r9 r& r: r; r< r= r r r Útest_gen_pyf_stdoutm s rA c
|
C s t |}| tdd| ¡ ¡ t |j¡L t djddd t t
|
¡$ t | ¡ \}}d|ksjt W 5 Q R X W 5 Q R X 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 r0 r1 r2 r r3 r4 r! ÚpytestÚraisesÚ
|
SystemExitr5 r6 r7 )r9 r& r: r; r= Úerrr r r Útest_gen_pyf_no_overwritez s rI c C sª t |dd}|j}d}d}| tdd| d| ¡ ¡ t |j¡\ t | ¡ \}}| ¡ D ]} d| d |ks`t
|
q`| ¡ D ]}
|
d
|
|
|
d |kst
|
qW 5 Q R X d S ) z9Tests that functions can be skipped
|
CLI :: skip:
|
Útest©r út0 t4 t8 sd s8 s4útd s0r, úf2py z -m test skip: z=buildmodule: Could not found the body of interfaced routine "ú ". Skipping.ú Constructing wrapper function "ú"N© r r r0 r1 r2 r r3 r4 r5 r6 r7 ) r9 r) r: Úfoutlr; ÚtoskipÚ remainingr< rH ÚskeyÚrkeyr r r Útest_f2py_skip s& þ
|
ÿÿ rX c C sª t |dd}|j}d}d}| tdd| d| ¡ ¡ t |j¡\ t | ¡ \}}| ¡ D ]} d| d |ks`t
|
q`| ¡ D ]}
|
d
|
|
|
d |kst
|
qW 5 Q R X d S ) z>Test that functions can be kept by only:
|
CLI :: only:
|
rJ rK rL rM r, rN z -m test only: ú<buildmodule: Could not find the body of interfaced routine "rO rP rQ NrR ) r9 r) r: rS r; rT Útokeepr< rH rV rW r r r Útest_f2py_only¡ s& þ
|
ÿÿ r[ c C s t |dd}|j}d}t|}d}d} | tdd| d| d | d
|
| ¡ ¡ t |j¡\ t |
|
¡ \}
|
} | ¡ D ]} d | d | ksxt qx| ¡ D ]} d | d|
|
kst qW 5 Q R X dS )zfTests that it is possible to return to file processing mode
|
CLI :: :
|
BUG: numpy-gh #20520
|
rJ rK rL ztd s0 hiÚblahr, rN ú -m z only: z : rY rO rP rQ N) r r r r0 r1 r2 r r3 r4 r5 r6 r7 )r9 r& r) r: rS r; rT Zipath2rZ r r< rH rV rW r r r Útest_file_processing_switch¸ s* ý
|
ÿÿ r^ c C sv d}t ||d}|j}| tdd| d| ¡ ¡ t |j¡ t W 5 Q R X t
|
|j ¡sbt t
|
|j ¡srt dS )zHChecks the generation of files based on a module name
|
CLI :: -m
|
ÚhirK r, rN r] N)r r r0 r1 r2 r r3 r4 r5 r r8 r r7 r )r9 r& r: ZMNAMErS r; r r r Útest_mod_gen_f77Ö s r` c C sú t |dd}|j}t d¡}t d¡}| tdd| d ¡ ¡ t |j ¡< t
|
| ¡ \}}| |¡dk srt | |¡dkst W 5 Q R X | tdd| d ¡ ¡ t |j ¡< t
|
| ¡ \}}| |¡dksÚt | |¡dk sìt W 5 Q R X dS )
|
zHLowers cases by flag or when -h is present
|
|
CLI :: --[no-]lower
|
rJ rK zHI\(\)zhi\(\)r, rN z -m test --lowerNz -m test --no-lower)r r ÚreÚcompiler0 r1 r2 r r3 r4 r5 r6 Úsearchr7 © r9 r( r: rS r; ZcapshiZcapslor< r= r r r Útest_lower_cmodç s$
|
|
ÿ re c C s
|
t |dd}|j}t d¡}t d¡}| tdd| d|j d ¡ ¡ t |j
|
¡< t | ¡ \}}| |¡d k szt| |¡d kstW 5 Q R X | tdd| d|j d
|
¡ ¡ t |j
|
¡< t | ¡ \}}| |¡d ksêt| |¡d k sütW 5 Q R X d S ) z^Lowers cases in signature files by flag or when -h is present
|
|
CLI :: --[no-]lower -h
|
rJ rK z Block: HIz Block: hir, rN z -h z -m test --overwrite-signatureNz) -m test --overwrite-signature --no-lower)r r ra rb r0 r1 r r2 r r3 r4 r5 r6 rc r7 rd r r r Útest_lower_sig s0
|
|
ý ý rf c
|
C sv t |}d}d}| tdd| d| d| ¡ ¡ t |j¡, t | ¡ \}}d| d|ksht W 5 Q R X d S )
|
zNEnsures that the build directory can be specified
|
|
CLI :: --build-dir
|
r\ Ztttmpr, úf2py -m r- z --build-dir zWrote C/API module "rQ Nr? )r9 r& r: r; r Zodirr< r= r r r Útest_build_dir( s ÿ rh c C sn t |}| tdd| d ¡ ¡ t |j¡6 t djddd t | ¡ \}}d|ks`t
|
W 5 Q R X d S )
|
zXEnsures that the build directory can be specified
|
|
CLI :: --overwrite-signature
|
r, rB z --overwrite-signaturerC rD r r r. N) r r0 r1 r2 r r3 r4 r! r5 r6 r7 r@ r r r Útest_overwrite9 s þ ri c
|
C s t |}d}| tdd| d| d ¡ ¡ t |j¡R t | ¡ \}}d|ksXt t | d
|
¡ }d| ¡ ks|t W 5 Q R X W 5 Q R X d S )
|
zJEnsures that TeX documentation is written out
|
|
CLI :: --latex-doc
|
r\ r, rg r- z --latex-docú Documentation is saved to fileú
|
module.texú\documentclassN© r r0 r1 r2 r r3 r4 r5 r6 r7 ÚopenÚread©r9 r& r: r; r r< r= Zotexr r r Ú test_latexdocJ s ÿ rq c C sf t |}d}| tdd| d| d ¡ ¡ t |j¡$ t | ¡ \}}d|ksXt W 5 Q R X dS )zMEnsures that TeX documentation is written out
|
|
CLI :: --no-latex-doc
|
r\ r, rg r- z --no-latex-docrj Nr? ©r9 r& r: r; r r< r= r r r Útest_nolatexdoc\ s ÿ rs c
|
C s t |}d}| tdd| d| d ¡ ¡ t |j¡T t | ¡ \}}d|ksXt t d| d
|
¡ }d | ¡ ks~t W 5 Q R X W 5 Q R X d
|
S ) zEnsures that truncated documentation is written out
|
|
TODO: Test to ensure this has no effect without --latex-doc
|
CLI :: --latex-doc --short-latex
|
r\ r, rg r- z --latex-doc --short-latexrj ú./rk rl Nrm rp r r r Útest_shortlatexl s ý ru c
|
C s t |}d}| tdd| d| d ¡ ¡ t |j¡T t | ¡ \}}d|ksXt t d| d
|
¡ }d | ¡ ks~t W 5 Q R X W 5 Q R X d
|
S ) zIEnsures that RsT documentation is written out
|
|
CLI :: --rest-doc
|
r\ r, rg r- z --rest-docú#ReST Documentation is saved to filert z module.restz.. -*- rest -*-Nrm )r9 r& r: r; r r< r= Zorstr r r Ú test_restdoc s ÿ rw c C sf t |}d}| tdd| d| d ¡ ¡ t |j¡$ t | ¡ \}}d|ksXt W 5 Q R X dS )zLEnsures that TeX documentation is written out
|
|
CLI :: --no-rest-doc
|
r\ r, rg r- z --no-rest-docrv Nr? rr r r r Útest_norestexdoc s ÿ rx c
|
C s~ t |}d}| tdd| d| d ¡ ¡ t |j¡< t t d| d ¡ }d| ¡ ksft
|
W 5 Q R X W 5 Q R X d S )
|
zIEnsures that debugging wrappers are written
|
|
CLI :: --debug-capi
|
r\ r, rg r- z --debug-capirt r
|
z#define DEBUGCFUNCSN© r r0 r1 r2 r r3 r4 r5 rn ro r7 )r9 r& r: r; r Úocmodr r r Útest_debugcapi¤ s ÿr{ zConsistently fails on CI.)Úreasonc C s t | }d}| tdd| d| d ¡ ¡ t |j¡P t t d¡}t j
|
|ddd }d
|
}t d ¡}|j |ksvt|j|kstW 5 Q R X d S ) zEEnsures that debugging wrappers work
|
|
CLI :: --debug-capi -c
|
r\ r, rg r- 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 r0 r1 r2 r r3 r4 r5 ÚshlexÚ
|
subprocessÚrunÚtextwrapÚdedentÚstdoutr7 Ústderr)r& r: r; r Úcmd_runÚroutÚeoutZeerrr r r Útest_debugcapi_bld´ s ÿ
|
|
|
r c C sº t |}d}| tdd| d| ¡ ¡ t |j¡ t W 5 Q R X | ¡ \}}d|ks`t | tdd| d| d ¡ ¡ t |j¡$ t | ¡ \}}d|ks¬t W 5 Q R X dS )zoEnsures that fortran subroutine wrappers for F77 are included by default
|
|
CLI :: --[no]-wrap-functions
|
r\ r, rg r- ú Fortran 77 wrappers are saved toz --wrap-functionsNr? rr r r r Útest_wrapfunc_defÒ s ÿ r c C sf t |}d}| tdd| d| d ¡ ¡ t |j¡$ t | ¡ \}}d|ksXt W 5 Q R X dS )zeEnsures that fortran subroutine wrappers for F77 can be disabled
|
|
CLI :: --no-wrap-functions
|
r\ r, rg r- z --no-wrap-functionsr Nr? rr r r r Útest_nowrapfuncë s ÿ r c
|
C s t |}d}| tdd| d| d ¡ ¡ t |j¡L t t d| d ¡ &}| ¡ }d|ksjt
|
d |ksvt
|
W 5 Q R X W 5 Q R X d
|
S ) zcAdd to the include directories
|
|
CLI :: -include
|
TODO: Document this in the help string
|
r\ r, rg r- z' -include<stdbool.h> -include<stdio.h> rt r
|
z#include <stdbool.h>z#include <stdio.h>Nry )r9 r& r: r; r rz Zocmrr r r Útest_inclheaderû s ý r c C s dS )z?Add to the include directories
|
|
CLI :: --include-paths
|
Nr r r r r Ú test_inclpath s r c C s dS )z;Add to the include directories
|
|
CLI :: --help-link
|
Nr r r r r Ú
|
test_hlink s r c C s t |}| tdd| d ¡ ¡ t |j¡T t | ¡ \}}d|ksNt d|ksZt d|ksft d|ksrt d|ks~t W 5 Q R X d S )
|
zOCheck that Fortran-to-Python KIND specs can be passed
|
|
CLI :: --f2cmap
|
r, ú f2py -m blah 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 changesNr? )r9 r+ r: r; r< r= r r r Ú test_f2cmap$ s r c C s` t |}| tdd| d ¡ ¡ t |j¡( t | ¡ \}}t |dksRt
|
W 5 Q R X dS )z)Reduce verbosity
|
|
CLI :: --quiet
|
r, r z --quietr N) r r0 r1 r2 r r3 r4 r5 r6 Úlenr7 r@ r r r Ú
|
test_quiet6 s r c C s\ t |}| tdd| d ¡ ¡ t |j¡$ t | ¡ \}}d|ksNt W 5 Q R X dS )z-Increase verbosity
|
|
CLI :: --verbose
|
r, r z
|
--verboseZ analyzelineNr? r@ r r r Ú test_verboseD s r c C sX | tdd ¡ ¡ t t¡2 t | ¡ \}}ddl}|j |
|
¡ ksJt W 5 Q R X dS )z"Ensure version
|
|
CLI :: -v
|
r, zf2py -vr N) r0 r1 r2 rE rF rG r5 r6 ÚnumpyÚ __version__Ústripr7 )r9 r: r< r= Únpr r r Ú test_versionR s r c C sp t | }| tdd| d ¡ ¡ t |j¡8 t t d¡}t j
|
|ddd}d}|j |ksbt W 5 Q R X d S )
|
z
|
CLI :: -c
|
r, r z -cz"python -c "import blah; blah.hi()"Tr} r~ r N) r r0 r1 r2 r r3 r4 r5 r r r r r7 )r& r: r; r r r r r r Ú test_npdistop` s
|
r c C s dS )z
|
CLI :: -c --fcompiler
|
Nr r r r r Útest_npd_fcompilert s r c C s dS )z
|
CLI :: -c --compiler
|
Nr r r r r Útest_npd_compiler| s r c C s dS )z$
|
CLI :: -c --help-fcompiler
|
Nr r r r r Útest_npd_help_fcompiler s r c C s dS )z
|
CLI :: -c --f77exec
|
Nr r r r r Útest_npd_f77exec s r c C s dS )z
|
CLI :: -c --f90exec
|
Nr r r r r Útest_npd_f90exec s r¡ c C s dS )z
|
CLI :: -c --f77flags
|
Nr r r r r Útest_npd_f77flags s r¢ c C s dS )z
|
CLI :: -c --f90flags
|
Nr r r r r Útest_npd_f90flags¤ s r£ c C s dS )z
|
CLI :: -c --opt
|
Nr r r r r Ú test_npd_opt¬ s r¤ c C s dS )z
|
CLI :: -c --arch
|
Nr r r r r Ú test_npd_arch´ s r¥ c C s dS )z
|
CLI :: -c --noopt
|
Nr r r r r Útest_npd_noopt¼ s r¦ c C s dS )z
|
CLI :: -c --noarch
|
Nr r r r r Útest_npd_noarchÄ s r§ c C s dS )z
|
CLI :: -c --debug
|
Nr r r r r Útest_npd_debugÌ s r¨ c C s dS )z%
|
CLI :: -c --link-<resource>
|
Nr r r r r Útest_npd_link_autoÔ s r© c C s dS )z/
|
CLI :: -c -L/path/to/lib/ -l<libname>
|
Nr r r r r Ú test_npd_libÜ s rª c C s dS )z
|
CLI :: -D<define>
|
Nr r r r r Útest_npd_defineä s r« c C s dS )z
|
CLI :: -U<name>
|
Nr r r r r Útest_npd_undefineì s r¬ c C s dS )z$
|
CLI :: -I/path/to/include/
|
Nr r r r r Ú test_npd_inclô s r c C s dS )z8
|
CLI :: <filename>.o <filename>.so <filename>.a
|
Nr r r r r Útest_npd_linkerü s r® )r )Fr ra r1 r r Úpathlibr Ú collectionsr rE Ú r Znumpy.f2py.f2py2er r5 r r Zfixturer& r( r) r+ r> rA rI ÚmarkZxfailrX r[ r^ r` re rf rh ri rq rs ru rw rx 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 (
|
|
'
|
|
|
|
|
|
|
|
|
'
|
|
|
