o tBh6M@sHddlZddlZddlZddlZddlZddlmZddlmZddl Z ddl m Z ddl m ZeddZdrd d Ze jd d ddZe jd d ddZe jd d ddZddZddZddZe jjddZddZddZd d!Zd"d#Zd$d%Zd&d'Z d(d)Z!d*d+Z"d,d-Z#d.d/Z$d0d1Z%d2d3Z&d4d5Z'e jjd6d7d8d9Z(d:d;Z)dd?Z+d@dAZ,dBdCZ-dDdEZ.dFdGZ/dHdIZ0dJdKZ1e jjd6d7dLdMZ2dNdOZ3dPdQZ4dRdSZ5dTdUZ6dVdWZ7dXdYZ8dZd[Z9d\d]Z:d^d_Z;d`daZdfdgZ?dhdiZ@djdkZAdldmZBdndoZCdpdqZDdS)sN)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_inpmnamebpathrs/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/numpy/f2py/tests/test_f2py2e.py get_io_pathssrsession)scopecC2tdddd}|d}|j|dd|S)z'Generates a single f90 file for testingtestssrccliz hiworld.f90z hello.f90asciiencodingrgetpath read_text getbasetemp write_texttmpdir_factoryfdatfnrrrhello_world_f908 r+cCr)'Generates a single f77 file for testingrrrzhi77.fzhello.frr r"r'rrrhello_world_f77Ar,r.cCr)r-rr return_realzfoo77.fzfoo.frr r"r'rrr retreal_f77Jr,r0cCst|}t|jd}|tdd|d|t|j!t| \}}d|vs0Jt| s9JWddS1sDwYdS)zIEnsures that a signature file is generated via the CLI CLI :: -h r argvzf2py -h  Saving signatures to fileN) rstemsetattrsyssplitr switchdirparentf2pycli readouterrexists)capfdr+ monkeypatchipathopathout_rrr test_gen_pyfSs  "rCcCspt|}|tdd|t|jt|\}}d|vs&JWddS1s1wYdS)zHEnsures that a signature file can be dumped to stdout CLI :: -h r1zf2py -h stdout r3N rr5r6r7rr8r9r:r;r=r+r>r?rArBrrrtest_gen_pyf_stdoutbs "rFc Cst|}|tdd|t|j>tdjdddt t t | \}}d|vs5JWdn1s?wYWddSWddS1sWwYdS) zjEnsures that the CLI refuses to overwrite signature files CLI :: -h without --overwrite-signature r1f2py -h faker.pyf faker.pyf Fake newsrr z&Use --overwrite-signature to overwriteN) rr5r6r7rr8r9r&pytestraises SystemExitr:r;)r=r+r>r?rBerrrrrtest_gen_pyf_no_overwritens  "rNc Cst|dd}|j}d}d}|tdd|d|t|j4t| \}}|D] } d| d |vsfoutlr?toskip remainingrArMskeyrkeyrrrtest_f2py_skip}s&     "r]c Cst|dd}|j}d}d}|tdd|d|t|j4t| \}}|D] } d| d |vsTest that functions can be kept by only: CLI :: only: rOrPrQrRr1rSz -m test only: rXr?rYtokeeprArMr[r\rrrtest_f2py_onlys&     "r`c 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 rOrPrQztd s0 hiblahr1rS -m z only: z : r^rTrUrVN) rr rr5r6r7rr8r9r:r;)r=r+r0r>rXr?rYipath2r_rrArMr[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 hirPr1rSrbN) rr r5r6r7rr8r9r:rr<rr)r=r+r>MNAMErXr?rrrtest_mod_gen_f77s rgc 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 rOrPzHI\(\)zhi\(\)r1rSz -m test --lowerNz -m test --no-lower) rr recompiler5r6r7rr8r9r:r;search r=r.r>rXr?capshicapslorArBrrrtest_lower_cmods(     "rnc 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 rOrPz Block: HIz Block: hir1rSz -h z -m test --overwrite-signatureNz) -m test --overwrite-signature --no-lower)rr rhrir5r6r r7rr8r9r:r;rjrkrrrtest_lower_sigs6     "roc 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 ratttmpr1f2py -m r2z --build-dir zWrote C/API module "rVNrD)r=r+r>r?rodirrArBrrrtest_build_dirs "rscCst|}|tdd|dt|j!tdjdddt| \}}d|vs0JWd d S1s;wYd S) zXEnsures that the build directory can be specified CLI :: --overwrite-signature r1rGz --overwrite-signaturerHrIrr r3N) rr5r6r7rr8r9r&r:r;rErrrtest_overwrite-s "rtc 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 rar1rqr2z --latex-docDocumentation is saved to file module.tex\documentclassN rr5r6r7rr8r9r:r;openreadr=r+r>r?rrArBotexrrr test_latexdoc>s  "r}c C|t|}d}|tdd|d|dt|jt|\}}d|vs,JWddS1s7wYdS)zMEnsures that TeX documentation is written out CLI :: --no-latex-doc rar1rqr2z --no-latex-docruNrDr=r+r>r?rrArBrrrtest_nolatexdocP "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 rar1rqr2z --latex-doc --short-latexru./rvrwNrxr{rrrtest_shortlatex`s"  "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 rar1rqr2z --rest-doc#ReST Documentation is saved to filerz module.restz.. -*- rest -*-Nrx)r=r+r>r?rrArBorstrrr test_restdocvs  "rc Cr~)zLEnsures that TeX documentation is written out CLI :: --no-rest-doc rar1rqr2z --no-rest-docrNrDrrrrtest_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 rar1rqr2z --debug-capirr z#define DEBUGCFUNCSN rr5r6r7rr8r9r:ryrz)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 rar1rqr2z -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_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 rar1rqr2N Fortran 77 wrappers are saved toz --wrap-functionsrDrrrrtest_wrapfunc_defs    "rc Cr~)zeEnsures that fortran subroutine wrappers for F77 can be disabled CLI :: --no-wrap-functions rar1rqr2z --no-wrap-functionsrNrDrrrrtest_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 rar1rqr2z' -include -include rr z#include z#include Nr)r=r+r>r?rrocmrrrrtest_inclheaders$ "rcCdS)z?Add to the include directories CLI :: --include-paths Nrrrrr test_inclpathrcCr)z;Add to the include directories CLI :: --help-link Nrrrrr test_hlinkrrcCr)zOCheck that Fortran-to-Python KIND specs can be passed CLI :: --f2cmap Nrrrrr test_f2cmaprrcCsvt|}|tdd|dt|jt|\}}t |dks)JWddS1s4wYdS)z)Reduce verbosity CLI :: --quiet r1 f2py -m blah z --quietrN) rr5r6r7rr8r9r:r;lenrErrr test_quiet!s "rcCsrt|}|tdd|dt|jt|\}}d|vs'JWddS1s2wYdS)z-Increase verbosity CLI :: --verbose r1rz --verbose analyzelineNrDrErrr test_verbose/s "rcCsn|tddttt|\}}ddl}|j | ks%JWddS1s0wYdS)z"Ensure version CLI :: -v r1zf2py -vrN) r5r6r7rJrKrLr:r;numpy __version__strip)r=r>rArBnprrr test_version=s  "rcCst|}|tdd|dt|j"ttd}t j |ddd}d}|j |ks1JWd d S1sr?rrrrrr test_npdistopKs "rcCr)z CLI :: -c --fcompiler Nrrrrrtest_npd_fcompiler_rcCr)z CLI :: -c --compiler Nrrrrrtest_npd_compilergrrcCr)z$ CLI :: -c --help-fcompiler Nrrrrrtest_npd_help_fcompilerorrcCr)z CLI :: -c --f77exec Nrrrrrtest_npd_f77execwrrcCr)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_inclrrcCr)z8 CLI :: .o .so .a Nrrrrrtest_npd_linkerrr)r)Errhr6rrpathlibr collectionsrrJrnumpy.f2py.f2py2err:rrfixturer+r.r0rCrFrNmarkxfailr]r`rdrgrnrorsrtr}rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrs~(       '       '