o p>eBT@sxddlZddlZddlZddlZddlZddlmZddlmZddl Z ddl m Z ddl m ZeddZdxd d Ze jd d ddZe jd d ddZe jd d ddZe jd d ddZe jd d ddZddZddZddZddZe jjd d!Zd"d#Zd$d%Zd&d'Z d(d)Z!d*d+Z"d,d-Z#d.d/Z$d0d1Z%d2d3Z&d4d5Z'd6d7Z(d8d9Z)d:d;Z*e jjdd?Z+d@dAZ,dBdCZ-dDdEZ.dFdGZ/dHdIZ0dJdKZ1dLdMZ2dNdOZ3dPdQZ4e jjddfdgZ?dhdiZ@djdkZAdldmZBdndoZCdpdqZDdrdsZEdtduZFdvdwZGdS)yN)Path) namedtuple)util)mainPPathsz(finp, f90inp, pyf, wrap77, wrap90, cmodfuntitledc CsRt|}t|d|d|d||d||d||ddS)axTakes 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.f90module.c)finpf90inppyfwrap77wrap90cmodf)rr with_suffix with_name)Z fname_inpmnamebpathr]/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/numpy/f2py/tests/test_f2py2e.py get_io_pathssrsession)scopecC2tdddd}|d}|j|dd|S)'Generates a single f90 file for testingtestssrccliz hiworld.f90z hello.f90asciiencodingrZgetpath read_textZ getbasetemp write_texttmpdir_factoryfdatfnrrrhello_world_f908 r)cCs2tdddd}|d}|j|dd|S)z(F90 file for testing warnings in gh23598rrZ crackfortranzgh23598Warn.f90rr r"r%rrr gh23598_warnAr*r+cCr)'Generates a single f77 file for testingrrrzhi77.fzhello.frr r"r%rrrhello_world_f77Jr*r-cCr)r,rrZ return_realzfoo77.fzfoo.frr r"r%rrr retreal_f77Sr*r.cCs`tdddd}tdddd}|d}|d}|j|dd |j|dd |S) rrrf2cmapzisoFortranEnvMap.f90z .f2py_f2cmapz f2cmap.f90Zmapfilerr r")r&r'r/r(Zfmaprrr f2cmap_f90[s  r0cCszt|dd}|j}|tdd|dt|jt|j }d|vs+JWddS1s6wYdS)Ntestrargvf2py z -m testzintproductf2pywrap, intpr) rr setattrsyssplitr switchdirparentf2pyclirr#)capfdr+ monkeypatchfoutlipathwrapperrrrtest_gh23598_warngs  "r@cCst|}t|jd}|tdd|d|t|j!t| \}}d|vs0Jt| s9JWddS1sDwYdS)zIEnsures that a signature file is generated via the CLI CLI :: -h r r3zf2py -h  Saving signatures to fileN) rstemr5r6r7rr8r9r: readouterrexists)r;r)r<r>Zopathout_rrr test_gen_pyfts  "rHcCs|t|}|tdd|t|jt|\}}d|vs&Jd|vs,JWddS1s7wYdS)zHEnsures that a signature file can be dumped to stdout CLI :: -h r3zf2py -h stdout rBzfunction hi() ! in N rr5r6r7rr8r9r:rDr;r)r<r>rFrGrrrtest_gen_pyf_stdouts  "rKc Cst|}|tdd|t|j>tdjdddt t t | \}}d|vs5JWdn1s?wYWddSWddS1sWwYdS) zjEnsures that the CLI refuses to overwrite signature files CLI :: -h without --overwrite-signature r3f2py -h faker.pyf faker.pyf Fake newsrr z&Use --overwrite-signature to overwriteN) rr5r6r7rr8r9r$pytestraises SystemExitr:rD)r;r)r<r>rGerrrrrtest_gen_pyf_no_overwrites  "rSc Cst|dd}|j}d}d}|tdd|d|t|j4t| \}}|D] } d| d |vstoskip remainingrFrRskeyrkeyrrrtest_f2py_skips(     "r^c Cst|dd}|j}d}d}|tdd|d|t|j4t| \}}|D] } d| d |vsTest that functions can be kept by only: CLI :: only: r1r2rTrUr3r4z -m test only: rZtokeeprFrRr\r]rrrtest_f2py_onlys(     "rac Cst|dd}|j}d}t|}d}d} |tdd|d| d |d |t|j4t | \} } |D] } d | d | vsHJq<|D] } d | d| vsYJqMWddS1sewYdS)zfTests that it is possible to return to file processing mode CLI :: : BUG: numpy-gh #20520 r1r2rTztd s0 hiblahr3r4 -m z only: z : r_rVrWrXN) rr rr5r6r7rr8r9r:rD)r;r)r.r<r=r>rZZipath2r`rrFrRr\r]rrrtest_file_processing_switchs.     "rdcCsd}t||d}|j}|tdd|d|t|j tWdn1s.wYt |j s;Jt |j sCJdS)zHChecks the generation of files based on a module name CLI :: -m hir2r3r4rcN) rr r5r6r7rr8r9r:rrErr)r;r)r<ZMNAMEr=r>rrrtest_mod_gen_f77s rfc Cs$t|dd}|j}td}td}|tdd|dt|j #t | \}}| |dus9J| |dusBJWdn1sLwY|tdd|d t|j $t | \}}| |duswJ| |dusJWddS1swYdS) zHLowers cases by flag or when -h is present CLI :: --[no-]lower r1r2zHI\(\)zhi\(\)r3r4z -m test --lowerNz -m test --no-lower) rr recompiler5r6r7rr8r9r:rDsearch r;r-r<r=r>ZcapshiZcapslorFrGrrrtest_lower_cmods(     "rkc Cs4t|dd}|j}td}td}|tdd|d|jdt |j #t | \}}| |d us=J| |d usFJWd n1sPwY|tdd|d|jd t |j $t | \}}| |d usJ| |d usJWd d S1swYd S) z^Lowers cases in signature files by flag or when -h is present CLI :: --[no-]lower -h r1r2z Block: HIz Block: hir3r4z -h z -m test --overwrite-signatureNz) -m test --overwrite-signature --no-lower)rr rgrhr5r6r r7rr8r9r:rDrirjrrrtest_lower_sigs6     "rlc Cst|}d}d}|tdd|d|d|t|jt|\}}d|d|vs4JWd d S1s?wYd S) zNEnsures that the build directory can be specified CLI :: --build-dir rbZtttmpr3f2py -m rAz --build-dir zWrote C/API module "rXNrI)r;r)r<r>rZodirrFrGrrrtest_build_dir>s "rncCst|}|tdd|dt|j!tdjdddt| \}}d|vs0JWd d S1s;wYd S) zXEnsures that the build directory can be specified CLI :: --overwrite-signature r3rLz --overwrite-signaturerMrNrr rBN) rr5r6r7rr8r9r$r:rDrJrrrtest_overwriteOs "roc Cst|}d}|tdd|d|dt|jAt|\}}d|vs,Jt|d }d| vs>JWd n1sHwYWd d SWd d S1s`wYd S) zJEnsures that TeX documentation is written out CLI :: --latex-doc rbr3rmrAz --latex-docDocumentation is saved to file module.tex\documentclassN rr5r6r7rr8r9r:rDopenreadr;r)r<r>rrFrGZotexrrr test_latexdoc`s  "rwc C|t|}d}|tdd|d|dt|jt|\}}d|vs,JWddS1s7wYdS)zMEnsures that TeX documentation is written out CLI :: --no-latex-doc rbr3rmrAz --no-latex-docrpNrIr;r)r<r>rrFrGrrrtest_nolatexdocr "rzc Cst|}d}|tdd|d|dt|jBt|\}}d|vs,Jtd|d }d | vs?JWd n1sIwYWd d SWd d S1sawYd S) zEnsures that truncated documentation is written out TODO: Test to ensure this has no effect without --latex-doc CLI :: --latex-doc --short-latex rbr3rmrAz --latex-doc --short-latexrp./rqrrNrsrvrrrtest_shortlatexs"  "r}c Cst|}d}|tdd|d|dt|jBt|\}}d|vs,Jtd|d }d | vs?JWd n1sIwYWd d SWd d S1sawYd S) zIEnsures that RsT documentation is written out CLI :: --rest-doc rbr3rmrAz --rest-doc#ReST Documentation is saved to filer|z module.restz.. -*- rest -*-Nrs)r;r)r<r>rrFrGZorstrrr test_restdocs  "rc Crx)zLEnsures that TeX documentation is written out CLI :: --no-rest-doc rbr3rmrAz --no-rest-docr~NrIryrrrtest_norestexdocr{rc Cst|}d}|tdd|d|dt|j6ttd|d}d| vs3JWd n1s=wYWd d SWd d S1sUwYd S) zIEnsures that debugging wrappers are written CLI :: --debug-capi rbr3rmrAz --debug-capir|r z#define DEBUGCFUNCSN rr5r6r7rr8r9r:rtru)r;r)r<r>rocmodrrrtest_debugcapis"rzConsistently fails on CI.)reasonc Cst|}d}|tdd|d|dt|j.ttd}t j |ddd }d }t d }|j |ks;J|j|ksBJWd d S1sMwYd S) zEEnsures that debugging wrappers work CLI :: --debug-capi -c rbr3rmrAz -c --debug-capiz#python3 -c "import blah; blah.hi()"TUTF-8capture_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)rr5r6r7rr8r9r:shlex subprocessruntextwrapdedentstdoutstderr)r)r<r>rcmd_runrouteoutZeerrrrrtest_debugcapi_blds   "rc Cst|}d}|tdd|d|t|j tWdn1s)wY|\}}d|vs:J|tdd|d|dt|jt|\}}d|vs`JWddS1skwYdS)zoEnsures that fortran subroutine wrappers for F77 are included by default CLI :: --[no]-wrap-functions rbr3rmrAN Fortran 77 wrappers are saved toz --wrap-functionsrIryrrrtest_wrapfunc_defs    "rc Crx)zeEnsures that fortran subroutine wrappers for F77 can be disabled CLI :: --no-wrap-functions rbr3rmrAz --no-wrap-functionsrNrIryrrrtest_nowrapfuncr{rc Cst|}d}|tdd|d|dt|j>ttd|d}| }d|vs5Jd |vs;JWd n1sEwYWd d SWd d S1s]wYd S) zcAdd to the include directories CLI :: -include TODO: Document this in the help string rbr3rmrAz' -include -include r|r z#include z#include Nr)r;r)r<r>rrZocmrrrrtest_inclheaders$ "rcCdS)z?Add to the include directories CLI :: --include-paths Nrrrrr test_inclpath(rcCr)z;Add to the include directories CLI :: --help-link Nrrrrr test_hlink1rrcCst|}|tdd|dt|j0t|\}}d|vs'Jd|vs-Jd|vs3Jd|vs9Jd|vs?JWd d S1sJwYd S) zOCheck that Fortran-to-Python KIND specs can be passed CLI :: --f2cmap r3 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 changesNrI)r;r0r<r>rFrGrrr test_f2cmap:s     "rcCsvt|}|tdd|dt|jt|\}}t |dks)JWddS1s4wYdS)z)Reduce verbosity CLI :: --quiet r3rz --quietrN) rr5r6r7rr8r9r:rDlenrJrrr test_quietLs "rcCsrt|}|tdd|dt|jt|\}}d|vs'JWddS1s2wYdS)z-Increase verbosity CLI :: --verbose r3rz --verboseZ analyzelineNrIrJrrr test_verboseZs "rcCsn|tddttt|\}}ddl}|j | ks%JWddS1s0wYdS)z"Ensure version CLI :: -v r3zf2py -vrN) r5r6r7rOrPrQr:rDnumpy __version__strip)r;r<rFrGnprrr test_versionhs  "rcCst|}|tdd|dt|j"ttd}t j |ddd}d}|j |ks1JWd d S1srrrrrr test_npdistopvs "rcCr)z CLI :: -c --fcompiler Nrrrrrtest_npd_fcompilerrcCr)z CLI :: -c --compiler Nrrrrrtest_npd_compilerrrcCr)z$ CLI :: -c --help-fcompiler Nrrrrrtest_npd_help_fcompilerrrcCr)z CLI :: -c --f77exec Nrrrrrtest_npd_f77execrrcCr)z CLI :: -c --f90exec Nrrrrrtest_npd_f90execrrcCr)z CLI :: -c --f77flags Nrrrrrtest_npd_f77flagsrrcCr)z CLI :: -c --f90flags Nrrrrrtest_npd_f90flagsrrcCr)z CLI :: -c --opt Nrrrrr test_npd_optrrcCr)z CLI :: -c --arch Nrrrrr test_npd_archrrcCr)z CLI :: -c --noopt Nrrrrrtest_npd_nooptrrcCr)z CLI :: -c --noarch Nrrrrrtest_npd_noarchrrcCr)z CLI :: -c --debug Nrrrrrtest_npd_debugrrcCr)z% CLI :: -c --link- Nrrrrrtest_npd_link_autorrcCr)z/ CLI :: -c -L/path/to/lib/ -l Nrrrrr test_npd_librrcCr)z CLI :: -D Nrrrrrtest_npd_definerrcCr)z CLI :: -U Nrrrrrtest_npd_undefinerrcCr)z$ CLI :: -I/path/to/include/ Nrrrrr test_npd_incl rrcCr)z8 CLI :: .o .so .a Nrrrrrtest_npd_linkerrr)r)Hrrgr6rrpathlibr collectionsrrOrZnumpy.f2py.f2py2err:rrZfixturer)r+r-r.r0r@rHrKrSmarkZxfailr^rardrfrkrlrnrorwrzr}rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrs(       '             '