U mdpQ@s\ddlZddlZddlZddlZddlZddlmZddlmZddl Z ddl m Z ddl m ZeddZdtd d Ze jd d ddZe jd d ddZe jd d ddZe jd d ddZddZddZddZe jjddZddZd d!Zd"d#Zd$d%Zd&d'Z d(d)Z!d*d+Z"d,d-Z#d.d/Z$d0d1Z%d2d3Z&d4d5Z'd6d7Z(e jjd8d9d:d;Z)dd?Z+d@dAZ,dBdCZ-dDdEZ.dFdGZ/dHdIZ0dJdKZ1dLdMZ2e jjd8d9dNdOZ3dPdQZ4dRdSZ5dTdUZ6dVdWZ7dXdYZ8dZd[Z9d\d]Z:d^d_Z;d`daZdfdgZ?dhdiZ@djdkZAdldmZBdndoZCdpdqZDdrdsZEdS)uN)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_inpmnamebpathrU/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/numpy/f2py/tests/test_f2py2e.py get_io_pathssrsession)scopecCs2tdddd}|d}|j|dd|S)'Generates a single f90 file for testingtestssrccliz hiworld.f90z hello.f90asciiencodingrZgetpath read_textZ getbasetemp write_texttmpdir_factoryfdatfnrrrhello_world_f908s r'cCs2tdddd}|d}|j|dd|S)'Generates a single f77 file for testingrrrzhi77.fzhello.frrr r#rrrhello_world_f77As r)cCs2tdddd}|d}|j|dd|S)r(rrZ return_realzfoo77.fzfoo.frrr r#rrr retreal_f77Js r*cCs`tdddd}tdddd}|d}|d}|j|dd |j|dd |S) rrrf2cmapzisoFortranEnvMap.f90z .f2py_f2cmapz f2cmap.f90Zmapfilerrr )r$r%r+r&Zfmaprrr f2cmap_f90Rs  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_pyf^s  r?c Csft|}|tdd|t|j0t|\}}d|ksLt d|ksXt W5QRXdS)zHEnsures that a signature file can be dumped to stdout CLI :: -h r-zf2py -h stdout r/zfunction hi() ! in N rr1r2r3rr4r5r6r7r8r:r'r;r<r=r>rrrtest_gen_pyf_stdoutms  rBc 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)rr1r2r3rr4r5r"pytestraises SystemExitr6r7r8)r:r'r;r<r>errrrrtest_gen_pyf_no_overwritezs  rJc 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 r1r2r3rr4r5r6r7r8) r:r*r;foutlr<toskip remainingr=rIskeyrkeyrrrtest_f2py_skips&     rYc 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: rKrLrMrNr-rOz -m test only: rrrtest_lower_cmods$     rfc Cs t|dd}|j}td}td}|tdd|d|jdt |j <t | \}}| |d k szt| |d kstW5QRX|tdd|d|jd t |j <t | \}}| |d kst| |d k stW5QRXd S) z^Lowers cases in signature files by flag or when -h is present CLI :: --[no-]lower -h rKrLz Block: HIz Block: hir-rOz -h z -m test --overwrite-signatureNz) -m test --overwrite-signature --no-lower)rr rbrcr1r2r r3rr4r5r6r7rdr8rerrrtest_lower_sigs0     rgc Csvt|}d}d}|tdd|d|d|t|j,t|\}}d|d|ksht W5QRXd S) zNEnsures that the build directory can be specified CLI :: --build-dir r]Ztttmpr-f2py -m r.z --build-dir zWrote C/API module "rRNr@)r:r'r;r<rZodirr=r>rrrtest_build_dir(s ric Csnt|}|tdd|dt|j6tdjdddt| \}}d|ks`t W5QRXd S) zXEnsures that the build directory can be specified CLI :: --overwrite-signature r-rCz --overwrite-signaturerDrErrr/N) rr1r2r3rr4r5r"r6r7r8rArrrtest_overwrite9s rjc 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 r]r-rhr.z --latex-docDocumentation is saved to file module.tex\documentclassN rr1r2r3rr4r5r6r7r8openreadr:r'r;r<rr=r>Zotexrrr test_latexdocJs  rrc Csft|}d}|tdd|d|dt|j$t|\}}d|ksXt W5QRXdS)zMEnsures that TeX documentation is written out CLI :: --no-latex-doc r]r-rhr.z --no-latex-docrkNr@r:r'r;r<rr=r>rrrtest_nolatexdoc\s rtc 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 r]r-rhr.z --latex-doc --short-latexrk./rlrmNrnrqrrrtest_shortlatexls  rvc 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 r]r-rhr.z --rest-doc#ReST Documentation is saved to fileruz module.restz.. -*- rest -*-Nrn)r:r'r;r<rr=r>Zorstrrr test_restdocs  rxc Csft|}d}|tdd|d|dt|j$t|\}}d|ksXt W5QRXdS)zLEnsures that TeX documentation is written out CLI :: --no-rest-doc r]r-rhr.z --no-rest-docrwNr@rsrrrtest_norestexdocs ryc Cs~t|}d}|tdd|d|dt|j<ttd|d}d| ksft W5QRXW5QRXd S) zIEnsures that debugging wrappers are written CLI :: --debug-capi r]r-rhr.z --debug-capirur z#define DEBUGCFUNCSN rr1r2r3rr4r5r6rorpr8)r:r'r;r<rocmodrrrtest_debugcapisr|zConsistently 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 r]r-rhr.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)rr1r2r3rr4r5r6shlex subprocessruntextwrapdedentstdoutr8stderr)r'r;r<rcmd_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 r]r-rhr. Fortran 77 wrappers are saved toz --wrap-functionsNr@rsrrrtest_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 r]r-rhr.z --no-wrap-functionsrNr@rsrrrtest_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 r]r-rhr.z' -include -include rur z#include z#include Nrz)r:r'r;r<rr{Zocmrrrrtest_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_hlinksrc Cst|}|tdd|dt|jTt|\}}d|ksNt d|ksZt d|ksft d|ksrt d|ks~t W5QRXd S) zOCheck that Fortran-to-Python KIND specs can be passed CLI :: --f2cmap r- f2py -m blah z --f2cmap mapfilez!Reading f2cmap from 'mapfile' ...z&Mapping "real(kind=real32)" to "float"z'Mapping "real(kind=real64)" to "double"z,Mapping "integer(kind=int64)" to "long_long"z0Successfully applied user defined f2cmap changesNr@)r:r,r;r<r=r>rrr test_f2cmap$s     rc Cs`t|}|tdd|dt|j(t|\}}t |dksRt W5QRXdS)z)Reduce verbosity CLI :: --quiet r-rz --quietrN) rr1r2r3rr4r5r6r7lenr8rArrr test_quiet6s  rc Cs\t|}|tdd|dt|j$t|\}}d|ksNt W5QRXdS)z-Increase verbosity CLI :: --verbose r-rz --verboseZ analyzelineNr@rArrr test_verboseDs  rc CsX|tddtt2t|\}}ddl}|j | ksJt W5QRXdS)z"Ensure version CLI :: -v r-zf2py -vrN) r1r2r3rFrGrHr6r7numpy __version__stripr8)r:r;r=r>nprrr test_versionRs   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()"Tr~rrN) rr1r2r3rr4r5r6rrrrr8)r'r;r<rrrrrr test_npdistop`s rcCsdS)z CLI :: -c --fcompiler Nrrrrrtest_npd_fcompilertsrcCsdS)z CLI :: -c --compiler Nrrrrrtest_npd_compiler|srcCsdS)z$ CLI :: -c --help-fcompiler Nrrrrrtest_npd_help_fcompilersrcCsdS)z CLI :: -c --f77exec Nrrrrrtest_npd_f77execsrcCsdS)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)Frrbr2rrpathlibr collectionsrrFrZnumpy.f2py.f2py2err6rrZfixturer'r)r*r,r?rBrJmarkZxfailrYr\r_rarfrgrirjrrrtrvrxryr|rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrs(      '          '