a p=ic6M@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)finpf90inppyfwrap77Zwrap90cmodf)rr with_suffix with_name)Z fname_inpmnamebpathrm/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/numpy/f2py/tests/test_f2py2e.py get_io_pathssrsession)scopecCs2tdddd}|d}|j|dd|S)z'Generates a single f90 file for testingtestssrccliz hiworld.f90z hello.f90asciiencodingrZgetpath read_textZ getbasetemp write_textZtmpdir_factoryZfdatfnrrrhello_world_f908s r$cCs2tdddd}|d}|j|dd|S)'Generates a single f77 file for testingrrrzhi77.fzhello.frrrr"rrrhello_world_f77As r&cCs2tdddd}|d}|j|dd|S)r%rrZ return_realzfoo77.fzfoo.frrrr"rrr retreal_f77Js r'cCst|}t|jd}|tdd|d|t|j@t| \}}d|vs`Jt| srJWdn1s0YdS)zIEnsures that a signature file is generated via the CLI CLI :: -h r argvzf2py -h  Saving signatures to fileN) rstemsetattrsyssplitr switchdirparentf2pycli readouterrexists)capfdr$ monkeypatchipathZopathout_rrr test_gen_pyfSs  r9cCsnt|}|tdd|t|j.t|\}}d|vsLJWdn1s`0YdS)zHEnsures that a signature file can be dumped to stdout CLI :: -h r(zf2py -h stdout r*N rr,r-r.rr/r0r1r2r4r$r5r6r7r8rrrtest_gen_pyf_stdoutbs  r<c Cst|}|tdd|t|jjtdjdddt t .t | \}}d|vsjJWdn1s~0YWdn1s0YdS) zjEnsures that the CLI refuses to overwrite signature files CLI :: -h without --overwrite-signature r(f2py -h faker.pyf faker.pyf Fake newsrrz&Use --overwrite-signature to overwriteN) rr,r-r.rr/r0r!pytestraises SystemExitr1r2)r4r$r5r6r8errrrrtest_gen_pyf_no_overwritens  rDc Cst|dd}|j}d}d}|tdd|d|t|jft| \}}|D]} d| d |vs`Jq`|D]} d | d |vsJqWd n1s0Yd S) z9Tests that functions can be skipped CLI :: skip: testrt0 t4 t8 sd s8 s4td s0r(f2py z -m test skip: z=buildmodule: Could not found the body of interfaced routine " ". Skipping.Constructing wrapper function ""N rr r,r-r.rr/r0r1r2) r4r'r5foutlr6toskip remainingr7rCskeyrkeyrrrtest_f2py_skip}s$     rSc Cst|dd}|j}d}d}|tdd|d|t|jft| \}}|D]} d| d |vs`Jq`|D]} d | d |vsJqWd n1s0Yd S) z>Test that functions can be kept by only: CLI :: only: rErFrGrHr(rIz -m test only: r?rrr*N) rr,r-r.rr/r0r!r1r2r;rrrtest_overwrite-s rdc Cst|}d}|tdd|d|dt|jpt|\}}d|vsXJt|d }d| vs|JWd n1s0YWd n1s0Yd S) zJEnsures that TeX documentation is written out CLI :: --latex-doc rWr(rbr)z --latex-docDocumentation is saved to file module.tex\documentclassN rr,r-r.rr/r0r1r2openreadr4r$r5r6rr7r8Zotexrrr test_latexdoc>s  rlc Cszt|}d}|tdd|d|dt|j.t|\}}d|vsXJWdn1sl0YdS)zMEnsures that TeX documentation is written out CLI :: --no-latex-doc rWr(rbr)z --no-latex-docreNr:r4r$r5r6rr7r8rrrtest_nolatexdocPs rnc Cst|}d}|tdd|d|dt|jrt|\}}d|vsXJtd|d }d | vs~JWd n1s0YWd n1s0Yd S) zEnsures that truncated documentation is written out TODO: Test to ensure this has no effect without --latex-doc CLI :: --latex-doc --short-latex rWr(rbr)z --latex-doc --short-latexre./rfrgNrhrkrrrtest_shortlatex`s  rpc Cst|}d}|tdd|d|dt|jrt|\}}d|vsXJtd|d }d | vs~JWd n1s0YWd n1s0Yd S) zIEnsures that RsT documentation is written out CLI :: --rest-doc rWr(rbr)z --rest-doc#ReST Documentation is saved to fileroz module.restz.. -*- rest -*-Nrh)r4r$r5r6rr7r8Zorstrrr test_restdocvs  rrc Cszt|}d}|tdd|d|dt|j.t|\}}d|vsXJWdn1sl0YdS)zLEnsures that TeX documentation is written out CLI :: --no-rest-doc rWr(rbr)z --no-rest-docrqNr:rmrrrtest_norestexdocs rsc Cst|}d}|tdd|d|dt|jZttd|d }d| vsfJWd n1sz0YWd n1s0Yd S) zIEnsures that debugging wrappers are written CLI :: --debug-capi rWr(rbr)z --debug-capiror z#define DEBUGCFUNCSN rr,r-r.rr/r0r1rirj)r4r$r5r6rocmodrrrtest_debugcapisrvzConsistently fails on CI.)reasonc Cst|}d}|tdd|d|dt|jZttd}t j |ddd }d }t d }|j |ksvJ|j|ksJWd n1s0Yd S) zEEnsures that debugging wrappers work CLI :: --debug-capi -c rWr(rbr)z -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/r0r1shlex subprocessruntextwrapdedentstdoutstderr)r$r5r6rcmd_runrouteoutZeerrrrrtest_debugcapi_blds  rc Cst|}d}|tdd|d|t|jtWdn1sR0Y|\}}d|vstJ|tdd|d|dt|j.t|\}}d|vsJWdn1s0YdS)zoEnsures that fortran subroutine wrappers for F77 are included by default CLI :: --[no]-wrap-functions rWr(rbr)N Fortran 77 wrappers are saved toz --wrap-functionsr:rmrrrtest_wrapfunc_defs$   rc Cszt|}d}|tdd|d|dt|j.t|\}}d|vsXJWdn1sl0YdS)zeEnsures that fortran subroutine wrappers for F77 can be disabled CLI :: --no-wrap-functions rWr(rbr)z --no-wrap-functionsrNr:rmrrrtest_nowrapfuncs rc Cst|}d}|tdd|d|dt|jjttd|d0}| }d|vsjJd |vsvJWd n1s0YWd n1s0Yd S) zcAdd to the include directories CLI :: -include TODO: Document this in the help string rWr(rbr)z' -include -include ror z#include z#include Nrt)r4r$r5r6rruZocmrrrrtest_inclheaders rcCsdS)z?Add to the include directories CLI :: --include-paths Nrrrrr test_inclpathsrcCsdS)z;Add to the include directories CLI :: --help-link Nrrrrr test_hlinksrcCsdS)zOCheck that Fortran-to-Python KIND specs can be passed CLI :: --f2cmap Nrrrrr test_f2cmapsrcCstt|}|tdd|dt|j2t|\}}t |dksRJWdn1sf0YdS)z)Reduce verbosity CLI :: --quiet r( f2py -m blah z --quietrN) rr,r-r.rr/r0r1r2lenr;rrr test_quiet!s  rcCspt|}|tdd|dt|j.t|\}}d|vsNJWdn1sb0YdS)z-Increase verbosity CLI :: --verbose r(rz --verboseZ analyzelineNr:r;rrr test_verbose/s  rcCsl|tddtt<t|\}}ddl}|j | ksJJWdn1s^0YdS)z"Ensure version CLI :: -v r(zf2py -vrN) r,r-r.r@rArBr1r2numpy __version__strip)r4r5r7r8nprrr test_version=s   rcCst|}|tdd|dt|jBttd}t j |ddd}d}|j |ksbJWd n1sv0Yd S) z CLI :: -c r(rz -cz"python -c "import blah; blah.hi()"Trxryr{N) rr,r-r.rr/r0r1r|r}r~r)r$r5r6rrrrrr test_npdistopKs rcCsdS)z CLI :: -c --fcompiler Nrrrrrtest_npd_fcompiler_srcCsdS)z CLI :: -c --compiler Nrrrrrtest_npd_compilergsrcCsdS)z$ CLI :: -c --help-fcompiler Nrrrrrtest_npd_help_fcompilerosrcCsdS)z CLI :: -c --f77exec Nrrrrrtest_npd_f77execwsrcCsdS)z CLI :: -c --f90exec Nrrrrrtest_npd_f90execsrcCsdS)z CLI :: -c --f77flags Nrrrrrtest_npd_f77flagssrcCsdS)z CLI :: -c --f90flags Nrrrrrtest_npd_f90flagssrcCsdS)z CLI :: -c --opt Nrrrrr test_npd_optsrcCsdS)z CLI :: -c --arch Nrrrrr test_npd_archsrcCsdS)z CLI :: -c --noopt Nrrrrrtest_npd_nooptsrcCsdS)z CLI :: -c --noarch Nrrrrrtest_npd_noarchsrcCsdS)z CLI :: -c --debug Nrrrrrtest_npd_debugsrcCsdS)z% CLI :: -c --link- Nrrrrrtest_npd_link_autosrcCsdS)z/ CLI :: -c -L/path/to/lib/ -l Nrrrrr test_npd_libsrcCsdS)z CLI :: -D Nrrrrrtest_npd_definesrcCsdS)z CLI :: -U Nrrrrrtest_npd_undefinesrcCsdS)z$ CLI :: -I/path/to/include/ Nrrrrr test_npd_inclsrcCsdS)z8 CLI :: .o .so .a Nrrrrrtest_npd_linkersr)r)Err\r-r}r|pathlibr collectionsrr@rZnumpy.f2py.f2py2err1rrZfixturer$r&r'r9r<rDmarkZxfailrSrVrYr[r`rarcrdrlrnrprrrsrvrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrs|(      '        '