U j6M@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_inpmnamebpathr@/tmp/pip-unpacked-wheel-16f7874t/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&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  r9c 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.r/r0r1r2r4r#r5r6r7r8rrrtest_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.r/r pytestraises SystemExitr0r1r2)r4r#r5r6r8errrrrtest_gen_pyf_no_overwritens  rDc 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.r/r0r1r2) r4r&r5foutlr6toskip remainingr7rCskeyrkeyrrrtest_f2py_skip}s&     rSc 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: rErFrGrHr'rIz -m test only: r?rrr)N) rr+r,r-rr.r/r r0r1r2r;rrrtest_overwrite-s rdc 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 rWr'rbr(z --latex-docDocumentation is saved to file module.tex\documentclassN rr+r,r-rr.r/r0r1r2openreadr4r#r5r6rr7r8Zotexrrr test_latexdoc>s  rlc Csft|}d}|tdd|d|dt|j$t|\}}d|ksXt W5QRXdS)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|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 rWr'rbr(z --latex-doc --short-latexre./rfrgNrhrkrrrtest_shortlatex`s  rpc 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 rWr'rbr(z --rest-doc#ReST Documentation is saved to fileroz module.restz.. -*- rest -*-Nrh)r4r#r5r6rr7r8Zorstrrr test_restdocvs  rrc Csft|}d}|tdd|d|dt|j$t|\}}d|ksXt W5QRXdS)zLEnsures that TeX documentation is written out CLI :: --no-rest-doc rWr'rbr(z --no-rest-docrqNr:rmrrrtest_norestexdocs rsc Cs~t|}d}|tdd|d|dt|j<ttd|d}d| ksft W5QRXW5QRXd S) zIEnsures that debugging wrappers are written CLI :: --debug-capi rWr'rbr(z --debug-capiror z#define DEBUGCFUNCSN rr+r,r-rr.r/r0rirjr2)r4r#r5r6rocmodrrrtest_debugcapisrvzConsistently 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 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.r/r0shlex subprocessruntextwrapdedentstdoutr2stderr)r#r5r6rcmd_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 rWr'rbr( Fortran 77 wrappers are saved toz --wrap-functionsNr:rmrrrtest_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 rWr'rbr(z --no-wrap-functionsrNr:rmrrrtest_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 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_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.r/r0r1lenr2r;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-r@rArBr0r1Znumpy __version__stripr2)r4r5r7r8nprrr 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()"Trxryr{N) rr+r,r-rr.r/r0r|r}r~rr2)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.f2py2err0rrZfixturer#r%r&r9r<rDmarkZxfailrSrVrYr[r`rarcrdrlrnrprrrsrvrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrs|(      '        '