o U=i"P@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)finpf90inppyfwrap77Zwrap90cmodfN)rr with_suffix with_name)Z fname_inpmnamebpathrbC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\numpy/f2py/tests/test_f2py2e.py get_io_pathssrsession)ZscopecC2tdddd}|d}|j|dd|S) z'Generates a single f90 file for testingtestssrccliz hiworld.f90z hello.f90asciiencodingNrZgetpath read_textZ getbasetemp write_textZtmpdir_factoryZfdatfnrrrhello_world_f908 r$cCr) 'Generates a single f77 file for testingrrrzhi77.fzhello.frrNrr"rrrhello_world_f77Ar%r'cCr) r&rrZ return_realzfoo77.fzfoo.frrNrr"rrr retreal_f77Jr%r(cCst|}t|jd}|tdd|d|t|j!t| \}}d|vs0Jt| s9JWddS1sDwYdS)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  "r:cCspt|}|tdd|t|jt|\}}d|vs&JWddS1s1wYdS)zHEnsures that a signature file can be dumped to stdout CLI :: -h r)zf2py -h stdout r+N rr-r.r/rr0r1r2r3r5r$r6r7r8r9rrrtest_gen_pyf_stdoutbs "r=c Cst|}|tdd|t|j>tdjdddt t t | \}}d|vs5JWdn1s?wYWddSWddS1sWwYdS) 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/rr0r1r!pytestraises SystemExitr2r3)r5r$r6r7r9errrrrtest_gen_pyf_no_overwritens  "rEc Cst|dd}|j}d}d}|tdd|d|t|j4t| \}}|D] } d| d |vsTest that functions can be kept by only: CLI :: only: rFrGrHrIr)rJz -m test only: z --overwrite-signaturer?r@rrr+N) rr-r.r/rr0r1r!r2r3r<rrrtest_overwrite-s "rec Cst|}d}|tdd|d|dt|jAt|\}}d|vs,Jt|d }d| vs>JWd n1sHwYWd d SWd d S1s`wYd 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/rr0r1r2r3openreadr5r$r6r7rr8r9Zotexrrr test_latexdoc>s  "rmc C|t|}d}|tdd|d|dt|jt|\}}d|vs,JWddS1s7wYdS)zMEnsures that TeX documentation is written out CLI :: --no-latex-doc rXr)rcr*z --no-latex-docrfNr;r5r$r6r7rr8r9rrrtest_nolatexdocP "rpc Cst|}d}|tdd|d|dt|jBt|\}}d|vs,Jtd|d }d | vs?JWd n1sIwYWd d SWd d S1sawYd 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"  "rsc Cst|}d}|tdd|d|dt|jBt|\}}d|vs,Jtd|d }d | vs?JWd n1sIwYWd d SWd d S1sawYd S) zIEnsures that RsT documentation is written out CLI :: --rest-doc rXr)rcr*z --rest-doc#ReST Documentation is saved to filerrz module.restz.. -*- rest -*-Nri)r5r$r6r7rr8r9Zorstrrr test_restdocvs  "ruc Crn)zLEnsures that TeX documentation is written out CLI :: --no-rest-doc rXr)rcr*z --no-rest-docrtNr;rorrrtest_norestexdocrqrvc Cst|}d}|tdd|d|dt|j6ttd|d}d| vs3JWd n1s=wYWd d SWd d S1sUwYd S) zIEnsures that debugging wrappers are written CLI :: --debug-capi rXr)rcr*z --debug-capirrr z#define DEBUGCFUNCSN rr-r.r/rr0r1r2rjrk)r5r$r6r7rocmodrrrtest_debugcapis"ryzConsistently fails on CI.)reasonc Cst|}d}|tdd|d|dt|j.ttd}t j |ddd }d }t d }|j |ks;J|j|ksBJWd d S1sMwYd 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/rr0r1r2shlex subprocessruntextwrapdedentstdoutstderr)r$r6r7rcmd_runrouteoutZeerrrrrtest_debugcapi_blds   "rc Cst|}d}|tdd|d|t|j tWdn1s)wY|\}}d|vs:J|tdd|d|dt|jt|\}}d|vs`JWddS1skwYdS)zoEnsures that fortran subroutine wrappers for F77 are included by default CLI :: --[no]-wrap-functions rXr)rcr*N Fortran 77 wrappers are saved toz --wrap-functionsr;rorrrtest_wrapfunc_defs    "rc Crn)zeEnsures that fortran subroutine wrappers for F77 can be disabled CLI :: --no-wrap-functions rXr)rcr*z --no-wrap-functionsrNr;rorrrtest_nowrapfuncrqrc Cst|}d}|tdd|d|dt|j>ttd|d}| }d|vs5Jd |vs;JWd n1sEwYWd d SWd d S1s]wYd S) zcAdd to the include directories CLI :: -include TODO: Document this in the help string rXr)rcr*z' -include -include rrr z#include z#include Nrw)r5r$r6r7rrxZocmrrrrtest_inclheaders$ "rcCdS)z?Add to the include directories CLI :: --include-paths Nrrrrr test_inclpathrcCr)z;Add to the include directories CLI :: --help-link Nrrrrr test_hlinkrrcCr)zOCheck that Fortran-to-Python KIND specs can be passed CLI :: --f2cmap Nrrrrr test_f2cmaprrcCsvt|}|tdd|dt|jt|\}}t |dks)JWddS1s4wYdS)z)Reduce verbosity CLI :: --quiet r) f2py -m blah z --quietrN) rr-r.r/rr0r1r2r3lenr<rrr test_quiet!s "rcCsrt|}|tdd|dt|jt|\}}d|vs'JWddS1s2wYdS)z-Increase verbosity CLI :: --verbose r)rz --verboseZ analyzelineNr;r<rrr test_verbose/s "rcCsn|tddttt|\}}ddl}|j | ks%JWddS1s0wYdS)z"Ensure version CLI :: -v r)zf2py -vrN) r-r.r/rArBrCr2r3Znumpy __version__strip)r5r6r8r9nprrr test_version=s  "rcCst|}|tdd|dt|j"ttd}t j |ddd}d}|j |ks1JWd d S1s Nrrrrrtest_npd_link_autorrcCr)z/ CLI :: -c -L/path/to/lib/ -l Nrrrrr test_npd_librrcCr)z CLI :: -D Nrrrrrtest_npd_definerrcCr)z CLI :: -U Nrrrrrtest_npd_undefinerrcCr)z$ CLI :: -I/path/to/include/ Nrrrrr test_npd_inclrrcCr)z8 CLI :: .o .so .a Nrrrrrtest_npd_linkerrr)r)Err]r.rrpathlibr collectionsrrArZnumpy.f2py.f2py2err2rrZfixturer$r'r(r:r=rEmarkZxfailrTrWrZr\rarbrdrermrprsrurvryrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrs~(       '       '