2iBT@ddlZddlZddlZddlZddlZddlmZddlmZddl Z ddl m Z ddl m ZeddZdBd Ze j"d d Ze j"d d Ze j"d dZe j"d dZe j"d dZdZdZdZdZe j6j8dZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'd Z(d!Z)d"Z*e j6j9d#$d%Z+d&Z,d'Z-d(Z.d)Z/d*Z0d+Z1d,Z2d-Z3d.Z4e j6j9d#$d/Z5d0Z6d1Z7d2Z8d3Z9d4Z:d5Z;d6Zd9Z?d:Z@d;ZAd<ZBd=ZCd>ZDd?ZEd@ZFdAZGy)CN)Path) namedtuple)util)mainPPathsz(finp, f90inp, pyf, wrap77, wrap90, cmodfc t|}t|jd|jd|jd|j|d|j|d|j|dS)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)finpf90inppyfwrap77wrap90cmodf)rr with_suffix with_name) fname_inpmnamebpaths S/var/www/html/acx/venv/lib/python3.12/site-packages/numpy/f2py/tests/test_f2py2e.py get_io_pathsrs~. OE    t $  (   f %%89%(:;<oox01  session)scopectjddddj}|jdz }|j |d|S)'Generates a single f90 file for testingtestssrccliz hiworld.f90z hello.f90asciiencodingrgetpath read_text getbasetemp write_texttmpdir_factoryfdatfns rhello_world_f90r-8sI <<} = G G ID  # # % 3BMM$M) Irctjddddj}|jdz }|j |d|S)z(F90 file for testing warnings in gh23598rr crackfortranzgh23598Warn.f90r!r"r$r)s r gh23598_warnr0AsK <<8I J T T VD  # # %(9 9BMM$M) Irctjddddj}|jdz }|j |d|S)'Generates a single f77 file for testingrrr zhi77.fzhello.fr!r"r$r)s rhello_world_f77r3JsI <<x 8 B B DD  # # % 1BMM$M) Irctjddddj}|jdz }|j |d|S)r2rr return_realzfoo77.fzfoo.fr!r"r$r)s r retreal_f77r6SsI << y A K K MD  # # % /BMM$M) Irc6tjddddj}tjddddj}|jdz }|jdz }|j |d |j |d |S) rrrf2cmapzisoFortranEnvMap.f90z .f2py_f2cmapz f2cmap.f90mapfiler!r"r$)r*r+r8r,fmaps r f2cmap_f90r;[s <<2H I S S UD \\'5(N C M M OF  # # % 4B  % % ') 3DMM$M)OOFWO- IrcFt|d}|j}|jtdd|dj t j |j5t|jj}d|vsJ dddy#1swYyxYw)Ntestrargvf2py z -m testzintproductf2pywrap, intpr) rr setattrsyssplitr switchdirparentf2pyclirr&)capfdr0 monkeypatchfoutlipathwrappers rtest_gh23598_warnrLgs V 4E LLE V wh%%')  %: ,,((**'999:::s "+BB ct|}t|jdz}|jtdd|d|j t j |j5t|j\}}d|vsJt|jsJ dddy#1swYyxYw)zIEnsures that a signature file is generated via the CLI CLI :: -h r r?zf2py -h  Saving signatures to fileN) rstemrArBrCrrDrErF readouterrexists)rGr-rHrJopathout_s r test_gen_pyfrVts  !E  ! & & /EVxwaw%?%E%E%GH  %) !!#Q*c111ug&&((( )))s .AB88Cc&t|}|jtdd|jt j |j 5t|j\}}d|vsJd|vsJ dddy#1swYyxYw)zHEnsures that a signature file can be dumped to stdout CLI :: -h r?zf2py -h stdout rOzfunction hi() ! in N rrArBrCrrDrErFrQrGr-rHrJrTrUs rtest_gen_pyf_stdoutrZs  !EVug%>%D%D%FG  %, !!#Q*c111$+++ ,,,s *BBct|}|jtdd|jt j |j 5tdjddtjt5t|j\}}d|vsJ ddddddy#1swYxYw#1swYyxYw) zjEnsures that the CLI refuses to overwrite signature files CLI :: -h without --overwrite-signature r?f2py -h faker.pyf faker.pyf Fake newsr!r"z&Use --overwrite-signature to overwriteN) rrArBrCrrDrEr(pytestraises SystemExitrFrQ)rGr-rHrJrUerrs rtest_gen_pyf_no_overwritercs  !EV'9%%A%G%G%IJ  %C [$$[7$C ]]: & C I%%'FAs;sB BB CCC C CCCs$6C  $B?.C ?C C  Cct|d}|j}d}d}|jtdd|d|j t j |j5t|j\}}|j D] } d| d |vr J|j D] } d | d |vr J d d d y #1swYy xYw) z9Tests that functions can be skipped CLI :: skip: r=r>t0 t4 t8 sd s8 s4td s0r?r@z -m test skip: z=buildmodule: Could not found the body of interfaced routine " ". Skipping.Constructing wrapper function ""N rr rArBrCrrDrErFrQ) rGr6rHrIrJtoskip remainingrTrbskeyrkeys rtest_f2py_skipros F 3E JJE FI V wofX.4468  %D ##%SLLN DOPTvUab  OO% DD4TF!<C CC DDDD(9C"CCCct|d}|j}d}d}|jtdd|d|j t j |j5t|j\}}|j D] } d| d |vr J|j D] } d | d |vr J d d d y #1swYy xYw) z>Test that functions can be kept by only: CLI :: only: r=r>rerfr?r@z -m test only: reztd s0 hiblahr?r@ -m z only: z : rrrgrhriN) rr rrArBrCrrDrErFrQ)rGr-r6rHrIrJrkipath2rsrrTrbrmrns rtest_file_processing_switchrys F 3E JJE F / "F F E  wd5'F8<BB   %D ##%SLLN DNtfT`a  LLN DD4TF!<C CC DDDDs;9C!5C!C!!C*cd}t||}|j}|jtdd|d|j t j |j5tdddtj|jsJtj|jsJy#1swYLxYw)zHChecks the generation of files based on a module name CLI :: -m hir>r?r@rwN) rr rArBrCrrDrErFrrRrr)rGr-rHMNAMErIrJs rtest_mod_gen_f77r}s E  6E LLEVuUG4w%?%E%E%GH  %  ;;u{{ ## # ;;u|| $$ $ s & B<<Cct|d}|j}tjd}tjd}|j t dd|dj tj|j5t|j\}}|j|J|j|J ddd|j t dd|d j tj|j5t|j\}}|j|J|j|J dddy#1swYxYw#1swYyxYw) zHLowers cases by flag or when -h is present CLI :: --[no-]lower r=r>zHI\(\)zhi\(\)r?r@z -m test --lowerNz -m test --no-lower) rr recompilerArBrCrrDrErFrQsearch rGr3rHrIrJcapshicapslorTrUs rtest_lower_cmodrsV  7E JJE ZZ "F ZZ "FVuUG3C%D%J%J%LM  %* !!#Q}}S!---}}S!))) * Vw&9:@@BD  %. !!#Q}}S!)))}}S!--- ..**..s AE/!AE;/E8;Fc Bt|d}|j}tjd}tjd}|j t dd|d|j djtj|j5t|j\}}|j|J|j|J d d d |j t dd|d|j d jtj|j5t|j\}}|j|J|j|J d d d y #1swYxYw#1swYy xYw) z^Lowers cases in signature files by flag or when -h is present CLI :: --[no-]lower -h r=r>z Block: HIz Block: hir?r@z -h z -m test --overwrite-signatureNz) -m test --overwrite-signature --no-lower)rr rrrArBrrCrrDrErFrQrrs rtest_lower_sigrst  7E JJE ZZ %F ZZ %F  wd599+%CDJJL  %* !!#Q}}S!---}}S!))) *  wd599+%NO    %. !!#Q}}S!)))}}S!--- ..**..sAF ;AF FFc 6t|}d}d}|jtdd|d|d|jt j |j 5t|j\}}d|d|vsJ d d d y #1swYy xYw) zNEnsures that the build directory can be specified CLI :: --build-dir rvtttmpr?f2py -m rNz --build-dir zWrote C/API module "riNrX)rGr-rHrJrodirrTrUs rtest_build_dirr>s  !E E DV"5'5'tfEKKMO  %8 !!#Q&ugR0C777888s (BBcTt|}|jtdd|djt j |j 5tdjddt|j\}}d|vsJ d d d y #1swYy xYw) zXEnsures that the build directory can be specified CLI :: --overwrite-signature r?r\z --overwrite-signaturer]r^r!r"rON) rrArBrCrrDrEr(rFrQrYs rtest_overwriterOs  !E V UG#9:@@BD  %2 [$$[7$C !!#Q*c111 222s ABB'c t|}d}|jtdd|d|djt j |j 5t|j\}}d|vsJt|dj5}d|jvsJ d d d d d d y #1swYxYw#1swYy xYw) zJEnsures that TeX documentation is written out CLI :: --latex-doc rvr?rrNz --latex-docDocumentation is saved to file module.tex\documentclassN rrArBrCrrDrErFrQopenreadrGr-rHrJrrTrUotexs r test_latexdocr`s  !E EV"5'5'>DDFH  %4 !!#Q/3666 UG:& ' , , . 4$$ 3 33 4 44 4 4 44s%AC C/C C C  Cc &t|}d}|jtdd|d|djt j |j 5t|j\}}d|vsJ dddy#1swYyxYw)zMEnsures that TeX documentation is written out CLI :: --no-latex-doc rvr?rrNz --no-latex-docrNrXrGr-rHrJrrTrUs rtest_nolatexdocrrs  !E EV"5'5'AGGIK  %; !!#Q/s:::;;; $BBc t|}d}|jtdd|d|djt j |j 5t|j\}}d|vsJtd|dj5}d |jvsJ d d d d d d y #1swYxYw#1swYy xYw) zEnsures that truncated documentation is written out TODO: Test to ensure this has no effect without --latex-doc CLI :: --latex-doc --short-latex rvr?rrNz --latex-doc --short-latexr./rrNrrs rtest_shortlatexrs  !E E  5'5'!;<BBD  %8 !!#Q/3666 BugZ( ) . . 0 8D$DIIK7 77 8 88 8 8 88%AC C0C C C  Cc t|}d}|jtdd|d|djt j |j 5t|j\}}d|vsJtd|dj5}d |jvsJ d d d d d d y #1swYxYw#1swYy xYw) zIEnsures that RsT documentation is written out CLI :: --rest-doc rvr?rrNz --rest-doc#ReST Documentation is saved to filerz module.restz.. -*- rest -*-Nr)rGr-rHrJrrTrUorsts r test_restdocrs  !E EV"5'5'=CCEG  %5 !!#Q4;;; Bug[) * / / 1 5T%4 44 5 55 5 5 55rc &t|}d}|jtdd|d|djt j |j 5t|j\}}d|vsJ dddy#1swYyxYw)zLEnsures that TeX documentation is written out CLI :: --no-rest-doc rvr?rrNz --no-rest-docrNrXrs rtest_norestexdocrs  !E EV"5'5'@FFHJ  %@ !!#Q4C???@@@rc t|}d}|jtdd|d|djt j |j 5ttd|dj5}d|jvsJ d d d d d d y #1swYxYw#1swYy xYw) zIEnsures that debugging wrappers are written CLI :: --debug-capi rvr?rrNz --debug-capirr z#define DEBUGCFUNCSN rrArBrCrrDrErFrr)rGr-rHrJrocmods rtest_debugcapirs  !E EV"5'5'?EEGI  %: BugX& ' , , . :%)UZZ\9 99 ::: : :::s$(B4B(B4(B1 -B44B=zConsistently fails on CI.)reasonc t|}d}|jtdd|d|djt j |j 5ttjd}tj|dd }d }tjd }|j|k(sJ|j|k(sJ d d d y #1swYy xYw) zEEnsures that debugging wrappers work CLI :: --debug-capi -c rvr?rrNz -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)rrArBrCrrDrErFshlex subprocessruntextwrapdedentstdoutstderr)r-rHrJrcmd_runrouteouteerrs rtest_debugcapi_bldrs  !E EV"5'5'1ABHHJL  %# ++EF~~gdWM  {{d"""{{d"""!###s A1CCc *t|}d}|jtdd|d|jt j |j 5tddd|j\}}d|vsJ|jtdd|d|djt j |j 5t|j\}}d|vsJ dddy#1swYxYw#1swYyxYw)zoEnsures that fortran subroutine wrappers for F77 are included by default CLI :: --[no]-wrap-functions rvr?rrNN Fortran 77 wrappers are saved toz --wrap-functionsrXrs rtest_wrapfunc_defrs  !E EVxwaw%?%E%E%GH  %     FC .# 55 5V"5'5'1BCIIKM  %: !!#Q2c999::::s C=$D =D Dc &t|}d}|jtdd|d|djt j |j 5t|j\}}d|vsJ dddy#1swYyxYw)zeEnsures that fortran subroutine wrappers for F77 can be disabled CLI :: --no-wrap-functions rvr?rrNz --no-wrap-functionsrNrXrs rtest_nowrapfuncrs  !E EV"5'5'1EFLLNP  %> !!#Q2#===>>>rc t|}d}|jtdd|d|djt j |j 5ttd|dj5}|j}d|vsJd |vsJ d d d d d d y #1swYxYw#1swYy xYw) zcAdd to the include directories CLI :: -include TODO: Document this in the help string rvr?rrNz' -include -include rr z#include z#include Nr)rGr-rHrJrrocmrs rtest_inclheaderrs  !E E  5'5'!HI    %0 BugX& ' , , . 0%:: Nrrrrtest_npd_link_autorrrcy)z/ CLI :: -c -L/path/to/lib/ -l Nrrrr test_npd_librrrcy)z CLI :: -D Nrrrrtest_npd_definerrrcy)z CLI :: -U Nrrrrtest_npd_undefinerrrcy)z$ CLI :: -I/path/to/include/ Nrrrr test_npd_inclr rrcy)z8 CLI :: .o .so .a Nrrrrtest_npd_linkerrrr)untitled)HrrrBrrpathlibr collectionsrr_rnumpy.f2py.f2py2errFrrfixturer-r0r3r6r;rLrVrZrcmarkxfailrortryr}rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr s+++" - HH INi !i !i !i !i ! : ) , CDD.D.D<%".4$.N8"2"4$ ; 8,5$ @ : 56#7#::2 > 0.  I$  $ -56 #7 #&                  r