qKgcddlZddlZddlZddlZddlZddlmZddlmZddl Z ddl Z ddl m Z ddl mZeddZdKd Ze j$d d Ze j$d d Ze j$d dZe j$d dZe j$d dZe j$d dZdZdZdZdZdZdZe j>jAe jBdk7xsejDdkddZ#e j>jAe jBdk7xsejDdkddZ$e j>jJdZ&dZ'd Z(d!Z)d"Z*d#Z+d$Z,d%Z-d&Z.d'Z/d(Z0d)Z1d*Z2d+Z3d,Z4e j>jkd-d.Z6d/Z7d0Z8d1Z9d2Z:d3Z;d4Zd7Z?e j>jkd-d8Z@d9ZAd:ZBd;ZCd<ZDd=ZEd>ZFd?ZGd@ZHdAZIdBZJdCZKdDZLdEZMdFZNdGZOdHZPdIZQdJZRy)LN)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 `/home/alanp/www/video.onchill/myenv/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-9sI <<} = 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_warnr0BsK <<8I J T T VD  # # %(9 9BMM$M) Irctjddddj}|jdz }|j |d|S)z8F90 file for testing disallowed CLI arguments in ghff819rrr z gh_22819.pyfr!r"r$r)s r gh22819_clir2KsI <<~ > H H JD  # # % 6BMM$M) Irctjddddj}|jdz }|j |d|S)'Generates a single f77 file for testingrrr zhi77.fzhello.fr!r"r$r)s rhello_world_f77r5TsI <<x 8 B B DD  # # % 1BMM$M) Irctjddddj}|jdz }|j |d|S)r4rr return_realzfoo77.fzfoo.fr!r"r$r)s r retreal_f77r8]sI << 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+r:r,fmaps r f2cmap_f90r=es <<2H I S S UD \\'5(N C M M OF  # # % 4B  % % ') 3DMM$M)OOFWO- Irct|}|jtdd|jt j |j 5t|j jdDcgc]}|js|j!}}d|vsJd|vsJd|vsJd|vsJ dddycc}w#1swYyxYw) zCheck that module names are handled correctly gh-22819 Essentially, the -m name cannot be used to import the module, so the module named in the .pyf needs to be used instead CLI :: -m and a .pyf file argv f2py -m blah *z blahmodule.czblah-f2pywrappers.fztest_22819-f2pywrappers.fztest_22819module.cN) rsetattrsyssplitr switchdirparentf2pyclirglobis_filename)capfdr2 monkeypatchipathitem gen_pathss rtest_gh22819_clirPqs  EV}UG%<%B%B%DE  % +0<<+=+=c+BU+B4dllnTYY+B UY...$I555*i777#y000 & %U & %s$(C;CCCCCcHt|}|jtdd|djt j |j 5tjtd5tddddddy#1swYxYw#1swYyxYw)zBOnly one .pyf file allowed gh-22819 CLI :: .pyf files r?r@z hello.pyfzOnly one .pyf file per call)matchN) rrBrCrDrrErFpytestraises ValueErrorrG)rKr2rLrMs rtest_gh22819_many_pyfrVst  EV}UG:%F%L%L%NO  % ]]:-J K IL & % K K & %s$B0 B ;B B BB!cFt|d}|j}|jtdd|dj t j |j5t|jj}d|vsJ dddy#1swYyxYw)Ntestrr?f2py z -m testzintproductf2pywrap, intpr) rr rBrCrDrrErFrGrr&)rKr0rLfoutlrMwrappers rtest_gh23598_warnr]s 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) rstemrBrCrDrrErFrG readouterrexists)rKr-rLrMopathout_s r test_gen_pyfrgs  !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 r`zfunction hi() ! in N rrBrCrDrrErFrGrbrKr-rLrMrerfs rtest_gen_pyf_stdoutrks  !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) rrBrCrDrrErFr(rSrT SystemExitrGrb)rKr-rLrMrferrs rtest_gen_pyf_no_overwriterrs  !EV'9%%A%G%G%IJ  % [$$[7$C ]]: & I%%'FAs;sB BB' & % & & & %s$6C  $B?.C ?C C  CLinux) zCompiler and 3.12 required)reasonct|}|jtdd|jt j |j 5t|j\}}d|vsJ dddy#1swYyxYw)z@Check that modules are named correctly CLI :: defaults r?zf2py --backend meson -c zuntitledmodule.cNrirjs rtest_untitled_clirxsr  !EV'?w%G%M%M%OP  % !!#Q!S((( & % %s $BB c Fd}t||}|j}|jtdd|d|j t j |j5t|j\}}d|vsJ ddd|jtddj t j |j5t|j\}}d |vsJ dddd }|jtdd|d |d j t j |j5t|j\}}d |vsJ dddy#1swYxYw#1swYxYw#1swYyxYw)zqCheck that no distutils imports are performed on 3.12 CLI :: --fcompiler --help-link --backend distutils hirYr?rZz -c --fcompiler=gfortran -m z%--fcompiler cannot be used with mesonNzf2py --help-linkzUse --dep for meson buildshi2z -c -m z --backend distutilsz.Cannot use distutils backend with Python>=3.12) rr rBrCrDrrErFrGrb)rKr-rLMNAMEr[rMrerfs r!test_no_py312_distutils_fcompilerr}sr E  6E LLE VuUG#?wGMMO  % !!#Q6#=== & V'..0  % !!#Q+s222 & E VuUG75'1EFLLN  % !!#Q?3FFF & % & % & % & %s$&$E?$F $F?F FF ct|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: rXrYt0 t4 t8 sd s8 s4td s0r?rZz -m test skip: z=buildmodule: Could not found the body of interfaced routine " ". Skipping.Constructing wrapper function ""N rr rBrCrDrrErFrGrb) rKr8rLr[rMtoskip remainingrerqskeyrkeys rtest_f2py_skiprs F 3E JJE FI V wofX.4468  % ##%SLLNDOPTvUab #OO%D4TF!<C CC& & % %(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: rXrYrrr?rZz -m test only: -h pyf_file BUG: numpy-gh #20520 rzrYr?rZrz -h hi.pyfzhi.pyfzpython module hiN) rr rBrCrDrrErFrGropenread)rKr5rLr|r[rMhipyfpyfdats rtest_mod_gen_gh25263rVs E  6E JJEVuUG4wj%I%O%O%QR  % (^ "eZZ\F%/ //# & % " " & %s$'$C B4#C4B= 9CC ct|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 rXrYzHI\(\)zhi\(\)r?rZz -m test --lowerNz -m test --no-lower) rr recompilerBrCrDrrErFrGrbsearch rKr5rLr[rMcapshicapslorerfs rtest_lower_cmodrfsF  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 rXrYz Block: HIz Block: hir?rZz -h z -m test --overwrite-signatureNz) -m test --overwrite-signature --no-lower)rr rrrBrCrrDrrErFrGrbrrs rtest_lower_sigrsd  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 rtttmpr?f2py -m r_z --build-dir zWrote C/API module "rNri)rKr-rLrMrodirrerfs rtest_build_dirrs  !E E DV"5'5'tfEKKMO  % !!#Q&ugR0C777 & % %s (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?rmz --overwrite-signaturernror!r"r`N) rrBrCrDrrErFr(rGrbrjs rtest_overwriters  !E V UG#9:@@BD  % [$$[7$C !!#Q*c111 & % %s 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 rr?rr_z --latex-docDocumentation is saved to file module.tex\documentclassN rrBrCrDrrErFrGrbrrrKr-rLrMrrerfotexs r test_latexdocrs  !E EV"5'5'>DDFH  % !!#Q/3666 UG:& ' , , .$$ 3 33/ & %/ . & %s%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 rr?rr_z --no-latex-docrNrirKr-rLrMrrerfs rtest_nolatexdocrs  !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 rr?rr_z --latex-doc --short-latexr./rrNrrs rtest_shortlatexrs  !E E  5'5'!;<BBD  % !!#Q/3666 BugZ( ) . . 0D$DIIK7 771 & %1 0 & %%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 rr?rr_z --rest-doc#ReST Documentation is saved to filerz module.restz.. -*- rest -*-Nr)rKr-rLrMrrerforsts r test_restdocrs  !E EV"5'5'=CCEG  % !!#Q4;;; Bug[) * / / 1T%4 442 & %2 1 & %rc &t|}d}|jtdd|d|djt j |j 5t|j\}}d|vsJ dddy#1swYyxYw)zLEnsures that TeX documentation is written out CLI :: --no-rest-doc rr?rr_z --no-rest-docrNrirs 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 rr?rr_z --debug-capirr z#define DEBUGCFUNCSN rrBrCrDrrErFrGrr)rKr-rLrMrocmods rtest_debugcapir#s  !E EV"5'5'?EEGI  % BugX& ' , , .%)UZZ\9 99/ & % . . & %s$(B4B(B4(B1 -B44B=z2Consistently fails on CI; noisy so skip not xfail.c 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 rr?rr_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)rrBrCrDrrErFrGshlex subprocessruntextwrapdedentstdoutstderr)r-rLrMrcmd_runrouteouteerrs rtest_debugcapi_bldr3s  !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 rr?rr_N Fortran 77 wrappers are saved toz --wrap-functionsrirs rtest_wrapfunc_defrQs  !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 rr?rr_z --no-wrap-functionsrNrirs rtest_nowrapfuncrjs  !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 rr?rr_z' -include -include rr z#include z#include Nr)rKr-rLrMrrocmrs rtest_inclheaderrzs  !E E  5'5'!HI    % BugX& ' , , .%:: Nrrrrtest_npd_link_autorSrrcy)z/ CLI :: -c -L/path/to/lib/ -l Nrrrr test_npd_libr[rrcy)z CLI :: -D Nrrrrtest_npd_definercrrcy)z CLI :: -U Nrrrrtest_npd_undefinerkrrcy)z$ CLI :: -I/path/to/include/ Nrrrr test_npd_inclrsrrcy)z8 CLI :: .o .so .a Nrrrrtest_npd_linkerr{rr)untitled)SrrrCrrpathlibr collectionsrplatformrSrnumpy.f2py.f2py2errGrrfixturer-r0r2r5r8r=rPrVr]rgrkrrmarkskipifsystem version_inforxr}xfailrrrrrrrrrrrrrrrskiprrrrrrrrrrrrrrrrrrrrr r rrrrrrrrrrr(s+++" - HH INi !i !i !i !i !i !(  : ) , C_X__&'1Ss7G7G77R79 )9 )_X__&'1Ss7G7G77R\xyGzG>DD.D.D<%" 0 .4$.N8"2"4$ ; 8,5$ @ : MN#O#::2 > 0.  I$  $ -MN #O #&                  r