U f6M@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_inpmnamebpathrC/tmp/pip-target-lpfmz8o1/lib/python/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'c Cst|}t|jd}|tdd|d|t|j6t| \}}d|ks`t t| srt W5QRXdS)zIEnsures that a signature file is generated via the CLI CLI :: -h r argvzf2py -h  Saving signatures to fileN) rstemsetattrsyssplitr switchdirparentf2pycli readouterrAssertionErrorexists)capfdr$ monkeypatchipathZopathout_rrr test_gen_pyfSs  r:c CsZt|}|tdd|t|j$t|\}}d|ksLt W5QRXdS)zHEnsures that a signature file can be dumped to stdout CLI :: -h r(zf2py -h stdout r*N rr,r-r.rr/r0r1r2r3r5r$r6r7r8r9rrrtest_gen_pyf_stdoutbs  r=c Cst|}|tdd|t|jLtdjdddt t $t | \}}d|ksjt W5QRXW5QRXdS) 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 SystemExitr1r2r3)r5r$r6r7r9errrrrtest_gen_pyf_no_overwritens  rEc Cst|dd}|j}d}d}|tdd|d|t|j\t| \}}|D]} d| d |ks`t q`|D]} d | d |kst qW5QRXd 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/r0r1r2r3) r5r'r6foutlr7toskip remainingr8rDskeyrkeyrrrtest_f2py_skip}s&     rTc Cst|dd}|j}d}d}|tdd|d|t|j\t| \}}|D]} d| d |ks`t q`|D]} d | d |kst qW5QRXd S) z>Test that functions can be kept by only: CLI :: only: rFrGrHrIr(rJz -m test only: z --overwrite-signaturer?r@rrr*N) rr,r-r.rr/r0r!r1r2r3r<rrrtest_overwrite-s rec Cst|}d}|tdd|d|dt|jRt|\}}d|ksXt t|d }d| ks|t W5QRXW5QRXd S) zJEnsures that TeX documentation is written out CLI :: --latex-doc rXr(rcr)z --latex-docDocumentation is saved to file module.tex\documentclassN rr,r-r.rr/r0r1r2r3openreadr5r$r6r7rr8r9Zotexrrr test_latexdoc>s  rmc Csft|}d}|tdd|d|dt|j$t|\}}d|ksXt W5QRXdS)zMEnsures that TeX documentation is written out CLI :: --no-latex-doc rXr(rcr)z --no-latex-docrfNr;r5r$r6r7rr8r9rrrtest_nolatexdocPs roc Cst|}d}|tdd|d|dt|jTt|\}}d|ksXt td|d }d | ks~t W5QRXW5QRXd S) zEnsures that truncated documentation is written out TODO: Test to ensure this has no effect without --latex-doc CLI :: --latex-doc --short-latex rXr(rcr)z --latex-doc --short-latexrf./rgrhNrirlrrrtest_shortlatex`s  rqc Cst|}d}|tdd|d|dt|jTt|\}}d|ksXt td|d }d | ks~t W5QRXW5QRXd S) zIEnsures that RsT documentation is written out CLI :: --rest-doc rXr(rcr)z --rest-doc#ReST Documentation is saved to filerpz module.restz.. -*- rest -*-Nri)r5r$r6r7rr8r9Zorstrrr test_restdocvs  rsc Csft|}d}|tdd|d|dt|j$t|\}}d|ksXt W5QRXdS)zLEnsures that TeX documentation is written out CLI :: --no-rest-doc rXr(rcr)z --no-rest-docrrNr;rnrrrtest_norestexdocs rtc Cs~t|}d}|tdd|d|dt|j<ttd|d}d| ksft W5QRXW5QRXd S) zIEnsures that debugging wrappers are written CLI :: --debug-capi rXr(rcr)z --debug-capirpr z#define DEBUGCFUNCSN rr,r-r.rr/r0r1rjrkr3)r5r$r6r7rocmodrrrtest_debugcapisrwzConsistently fails on CI.)reasonc Cst|}d}|tdd|d|dt|jPttd}t j |ddd }d }t d }|j |ksvt|j|kstW5QRXd S) zEEnsures that debugging wrappers work CLI :: --debug-capi -c rXr(rcr)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 subprocessruntextwrapdedentstdoutr3stderr)r$r6r7rcmd_runrouteoutZeerrrrrtest_debugcapi_blds  rc Cst|}d}|tdd|d|t|j tW5QRX|\}}d|ks`t |tdd|d|dt|j$t|\}}d|kst W5QRXdS)zoEnsures that fortran subroutine wrappers for F77 are included by default CLI :: --[no]-wrap-functions rXr(rcr) Fortran 77 wrappers are saved toz --wrap-functionsNr;rnrrrtest_wrapfunc_defs   rc Csft|}d}|tdd|d|dt|j$t|\}}d|ksXt W5QRXdS)zeEnsures that fortran subroutine wrappers for F77 can be disabled CLI :: --no-wrap-functions rXr(rcr)z --no-wrap-functionsrNr;rnrrrtest_nowrapfuncs rc Cst|}d}|tdd|d|dt|jLttd|d&}| }d|ksjt d |ksvt W5QRXW5QRXd S) zcAdd to the include directories CLI :: -include TODO: Document this in the help string rXr(rcr)z' -include -include rpr z#include z#include Nru)r5r$r6r7rrvZocmrrrrtest_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_f2cmapsrc Cs`t|}|tdd|dt|j(t|\}}t |dksRt W5QRXdS)z)Reduce verbosity CLI :: --quiet r( f2py -m blah z --quietrN) rr,r-r.rr/r0r1r2lenr3r<rrr test_quiet!s  rc Cs\t|}|tdd|dt|j$t|\}}d|ksNt W5QRXdS)z-Increase verbosity CLI :: --verbose r(rz --verboseZ analyzelineNr;r<rrr test_verbose/s  rc CsX|tddtt2t|\}}ddl}|j | ksJt W5QRXdS)z"Ensure version CLI :: -v r(zf2py -vrN) r,r-r.rArBrCr1r2numpy __version__stripr3)r5r6r8r9nprrr test_version=s   rc Cspt|}|tdd|dt|j8ttd}t j |ddd}d}|j |ksbt W5QRXd S) z CLI :: -c r(rz -cz"python -c "import blah; blah.hi()"Tryrzr|N) rr,r-r.rr/r0r1r}r~rrr3)r$r6r7rrrrrr 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 collectionsrrArZnumpy.f2py.f2py2err1rrZfixturer$r&r'r:r=rEmarkZxfailrTrWrZr\rarbrdrermrorqrsrtrwrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrs|(      '        '