a d6M@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_inpmnamebpathre/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/numpy/f2py/tests/test_f2py2e.py get_io_pathssrsession)ZscopecCs2tdddd}|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  r8cCsnt|}|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.r/r0r1r3r#r4r5r6r7rrrtest_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.r/r pytestraises SystemExitr0r1)r3r#r4r5r7errrrrtest_gen_pyf_no_overwritens  rCc 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.r/r0r1) r3r&r4foutlr5toskip remainingr6rBskeyrkeyrrrtest_f2py_skip}s$     rRc 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: rDrErFrGr'rHz -m test only: rrr)N) rr+r,r-rr.r/r r0r1r:rrrtest_overwrite-s rcc 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 rVr'rar(z --latex-docDocumentation is saved to file module.tex\documentclassN rr+r,r-rr.r/r0r1openreadr3r#r4r5rr6r7Zotexrrr test_latexdoc>s  rkc Cszt|}d}|tdd|d|dt|j.t|\}}d|vsXJWdn1sl0YdS)zMEnsures that TeX documentation is written out CLI :: --no-latex-doc rVr'rar(z --no-latex-docrdNr9r3r#r4r5rr6r7rrrtest_nolatexdocPs rmc 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 rVr'rar(z --latex-doc --short-latexrd./rerfNrgrjrrrtest_shortlatex`s  roc 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 rVr'rar(z --rest-doc#ReST Documentation is saved to filernz module.restz.. -*- rest -*-Nrg)r3r#r4r5rr6r7Zorstrrr test_restdocvs  rqc Cszt|}d}|tdd|d|dt|j.t|\}}d|vsXJWdn1sl0YdS)zLEnsures that TeX documentation is written out CLI :: --no-rest-doc rVr'rar(z --no-rest-docrpNr9rlrrrtest_norestexdocs rrc Cst|}d}|tdd|d|dt|jZttd|d }d| vsfJWd n1sz0YWd n1s0Yd S) zIEnsures that debugging wrappers are written CLI :: --debug-capi rVr'rar(z --debug-capirnr z#define DEBUGCFUNCSN rr+r,r-rr.r/r0rhri)r3r#r4r5rocmodrrrtest_debugcapisruzConsistently 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 rVr'rar(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.r/r0shlex subprocessruntextwrapdedentstdoutstderr)r#r4r5rcmd_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 rVr'rar(N Fortran 77 wrappers are saved toz --wrap-functionsr9rlrrrtest_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 rVr'rar(z --no-wrap-functionsrNr9rlrrrtest_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 rVr'rar(z' -include -include rnr z#include z#include Nrs)r3r#r4r5rrtZocmrrrrtest_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.r/r0r1lenr:rrr test_quiet!s  rcCspt|}|tdd|dt|j.t|\}}d|vsNJWdn1sb0YdS)z-Increase verbosity CLI :: --verbose r'rz --verboseZ analyzelineNr9r:rrr test_verbose/s  rcCsl|tddtt<t|\}}ddl}|j | ksJJWdn1s^0YdS)z"Ensure version CLI :: -v r'zf2py -vrN) r+r,r-r?r@rAr0r1numpy __version__strip)r3r4r6r7nprrr 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()"TrwrxrzN) rr+r,r-rr.r/r0r{r|r}r)r#r4r5rrrrrr 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)Er~r[r,r|r{pathlibr collectionsrr?rZnumpy.f2py.f2py2err0rrZfixturer#r%r&r8r;rCmarkZxfailrRrUrXrZr_r`rbrcrkrmrorqrrrurrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrs|(      '        '