o dBT@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) fname_inpmnamebpathra/var/www/html/visualizacion-main/env/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.f90asciiencodingrgetpath read_text getbasetemp write_texttmpdir_factoryfdatfnrrrhello_world_f908 r,cCs2tdddd}|d}|j|dd|S)z(F90 file for testing warnings in gh23598rr crackfortranzgh23598Warn.f90r r!r#r(rrr gh23598_warnAr-r/cCr)'Generates a single f77 file for testingrrrzhi77.fzhello.fr r!r#r(rrrhello_world_f77Jr-r1cCr)r0rr return_realzfoo77.fzfoo.fr r!r#r(rrr retreal_f77Sr-r3cCs`tdddd}tdddd}|d}|d}|j|dd |j|dd |S) rrrf2cmapzisoFortranEnvMap.f90z .f2py_f2cmapz f2cmap.f90mapfiler r!r#)r)r*r4r+fmaprrr f2cmap_f90[s  r7cCszt|dd}|j}|tdd|dt|jt|j }d|vs+JWddS1s6wYdS)Ntestrargvf2py z -m testzintproductf2pywrap, intpr) rr setattrsyssplitr switchdirparentf2pyclirr%)capfdr/ monkeypatchfoutlipathwrapperrrrtest_gh23598_warngs  "rGcCst|}t|jd}|tdd|d|t|j!t| \}}d|vs0Jt| s9JWddS1sDwYdS)zIEnsures that a signature file is generated via the CLI CLI :: -h r r:zf2py -h  Saving signatures to fileN) rstemr<r=r>rr?r@rA readouterrexists)rBr,rCrEopathout_rrr test_gen_pyfts  "rPcCs|t|}|tdd|t|jt|\}}d|vs&Jd|vs,JWddS1s7wYdS)zHEnsures that a signature file can be dumped to stdout CLI :: -h r:zf2py -h stdout rIzfunction hi() ! in N rr<r=r>rr?r@rArKrBr,rCrErNrOrrrtest_gen_pyf_stdouts  "rSc Cst|}|tdd|t|j>tdjdddt t t | \}}d|vs5JWdn1s?wYWddSWddS1sWwYdS) 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) rr<r=r>rr?r@r'pytestraises SystemExitrArK)rBr,rCrErOerrrrrtest_gen_pyf_no_overwrites  "r[c Cst|dd}|j}d}d}|tdd|d|t|j4t| \}}|D] } d| d |vsrr?r@rArK) rBr3rCrDrEtoskip remainingrNrZskeyrkeyrrrtest_f2py_skips(     "rfc Cst|dd}|j}d}d}|tdd|d|t|j4t| \}}|D] } d| d |vsTest that functions can be kept by only: CLI :: only: r8r9r\r]r:r;z -m test only: rr?r@rArK)rBr,r3rCrDrErbipath2rhrrNrZrdrerrrtest_file_processing_switchs.     "rmcCsd}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 hir9r:r;rkN) rr r<r=r>rr?r@rArrLrr)rBr,rCMNAMErDrErrrtest_mod_gen_f77s rpc 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 r8r9zHI\(\)zhi\(\)r:r;z -m test --lowerNz -m test --no-lower) rr recompiler<r=r>rr?r@rArKsearch rBr1rCrDrEcapshicapslorNrOrrrtest_lower_cmods(     "rwc 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 r8r9z Block: HIz Block: hir:r;z -h z -m test --overwrite-signatureNz) -m test --overwrite-signature --no-lower)rr rqrrr<r=r r>rr?r@rArKrsrtrrrtest_lower_sigs6     "rxc 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 rjtttmpr:f2py -m rHz --build-dir zWrote C/API module "r`NrQ)rBr,rCrErodirrNrOrrrtest_build_dir>s "r|cCst|}|tdd|dt|j!tdjdddt| \}}d|vs0JWd d S1s;wYd S) zXEnsures that the build directory can be specified CLI :: --overwrite-signature r:rTz --overwrite-signaturerUrVr r!rIN) rr<r=r>rr?r@r'rArKrRrrrtest_overwriteOs "r}c 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 rjr:rzrHz --latex-docDocumentation is saved to file module.tex\documentclassN rr<r=r>rr?r@rArKopenreadrBr,rCrErrNrOotexrrr test_latexdoc`s  "rc C|t|}d}|tdd|d|dt|jt|\}}d|vs,JWddS1s7wYdS)zMEnsures that TeX documentation is written out CLI :: --no-latex-doc rjr:rzrHz --no-latex-docr~NrQrBr,rCrErrNrOrrrtest_nolatexdocr "rc 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 rjr:rzrHz --latex-doc --short-latexr~./rrNrrrrrtest_shortlatexs"  "rc 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 rjr:rzrHz --rest-doc#ReST Documentation is saved to filerz module.restz.. -*- rest -*-Nr)rBr,rCrErrNrOorstrrr test_restdocs  "rc Cr)zLEnsures that TeX documentation is written out CLI :: --no-rest-doc rjr:rzrHz --no-rest-docrNrQrrrrtest_norestexdocrrc 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 rjr:rzrHz --debug-capirr z#define DEBUGCFUNCSN rr<r=r>rr?r@rArr)rBr,rCrErocmodrrrtest_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 rjr:rzrHz -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)rr<r=r>rr?r@rAshlex subprocessruntextwrapdedentstdoutstderr)r,rCrErcmd_runrouteouteerrrrrtest_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 rjr:rzrHN Fortran 77 wrappers are saved toz --wrap-functionsrQrrrrtest_wrapfunc_defs    "rc Cr)zeEnsures that fortran subroutine wrappers for F77 can be disabled CLI :: --no-wrap-functions rjr:rzrHz --no-wrap-functionsrNrQrrrrtest_nowrapfuncrrc 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 rjr:rzrHz' -include -include rr z#include z#include Nr)rBr,rCrErrocmrrrrtest_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 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 changesNrQ)rBr7rCrErNrOrrr test_f2cmap:s     "rcCsvt|}|tdd|dt|jt|\}}t |dks)JWddS1s4wYdS)z)Reduce verbosity CLI :: --quiet r:rz --quietrN) rr<r=r>rr?r@rArKlenrRrrr test_quietLs "rcCsrt|}|tdd|dt|jt|\}}d|vs'JWddS1s2wYdS)z-Increase verbosity CLI :: --verbose r:rz --verbose analyzelineNrQrRrrr test_verboseZs "rcCsn|tddttt|\}}ddl}|j | ks%JWddS1s0wYdS)z"Ensure version CLI :: -v r:zf2py -vrN) r<r=r>rWrXrYrArKnumpy __version__strip)rBrCrNrOnprrr test_versionhs  "rcCst|}|tdd|dt|j"ttd}t j |ddd}d}|j |ks1JWd d S1srr?r@rArrrr)r,rCrErrrrrr 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)Hrrqr=rrpathlibr collectionsrrWrnumpy.f2py.f2py2errArrfixturer,r/r1r3r7rGrPrSr[markxfailrfrirmrprwrxr|r}rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrs(       '             '