U QipQ@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_inpmnamebpathr@/tmp/pip-unpacked-wheel-vgsdxuwo/numpy/f2py/tests/test_f2py2e.py get_io_pathssrsession)ZscopecCs2tdddd}|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.frrrr"rrrhello_world_f77As r(cCs2tdddd}|d}|j|dd|S)r'rrZ return_realzfoo77.fzfoo.frrrr"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 rr0r1r2rr3r4r5r6r7r9r&r:r;r<r=rrrtest_gen_pyf_stdoutms  rAc 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)rr0r1r2rr3r4r!pytestraises SystemExitr5r6r7)r9r&r:r;r=errrrrtest_gen_pyf_no_overwritezs  rIc 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 r0r1r2rr3r4r5r6r7) r9r)r:foutlr;toskip remainingr<rHskeyrkeyrrrtest_f2py_skips&     rXc 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: rJrKrLrMr,rNz -m test only: -include rtr z#include z#include Nry)r9r&r:r;rrzZocmrrrrtest_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?)r9r+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) rr0r1r2rr3r4r5r6lenr7r@rrr test_quiet6s  rc Cs\t|}|tdd|dt|j$t|\}}d|ksNt W5QRXdS)z-Increase verbosity CLI :: --verbose r,rz --verboseZ analyzelineNr?r@rrr test_verboseDs  rc CsX|tddtt2t|\}}ddl}|j | ksJt W5QRXdS)z"Ensure version CLI :: -v r,zf2py -vrN) r0r1r2rErFrGr5r6Znumpy __version__stripr7)r9r: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}r~rN) rr0r1r2rr3r4r5rrrrr7)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)Frrar1rrpathlibr collectionsrrErZnumpy.f2py.f2py2err5rrZfixturer&r(r)r+r>rArImarkZxfailrXr[r^r`rerfrhrirqrsrurwrxr{rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrs(      '          '