G/PhlddlZddlZddlZddlZddlZddlmZddlmZddl Z ddl Z ddl m Z ddl mZddlmZdZed d ZdRd Ze jd dZe jd dZe jd dZe jd dZe jd dZe jd dZdZdZdZdZdZ dZ!e j"#ej$dkddZ%e j"#e j&dkp ej$dkd d!Z'e j"j(d"Z)d#Z*d$Z+d%Z,d&Z-d'Z.d(Z/d)Z0d*Z1d+Z2d,Z3d-Z4d.Z5d/Z6d0Z7e j"8d1d2Z9d3Z:d4Z;d5Zd8Z?d9Z@d:ZAd;ZBe j"8d1d<ZCe j"#e j&dkp ej$dkd=d>ZDe j"#e j&dkp ej$dkd=d?ZEd@ZFdAZGdBZHdCZIdDZJdEZKdFZLdGZMdHZNdIZOdJZPdKZQdLZRdMZSdNZTdOZUdPZVdQZWdS)SN)Path) namedtuple)util)main) NOGIL_BUILDcttjstjddSt dS)Nz$CLI command needs a Fortran compiler)rhas_fortran_compilerpytestskipf2pycli\/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numpy/f2py/tests/test_f2py2e.pycompiler_check_f2pyclirs7  $ & & :;;;;; rPPathsz(finp, f90inp, pyf, wrap77, wrap90, cmodfuntitledc @t|}t|d|d|d||d||d||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 r get_io_pathsr"s. OOE    t $ $  ((   f % %%88899%;;;<<oo00011    rsession)scopectjdddd}|dz }||d|S)'Generates a single f90 file for testingtestssrccliz hiworld.f90z hello.f90asciiencodingrgetpath read_text getbasetemp write_texttmpdir_factoryfdatfns rhello_world_f90r6FsV <} = = G G I ID  # # % % 3BMM$M))) Irctjdddd}|dz }||d|S)z(F90 file for testing warnings in gh23598r'r( crackfortranzgh23598Warn.f90r*r+r-r2s r gh23598_warnr9OsX <8I J J T T V VD  # # % %(9 9BMM$M))) Irctjdddd}|dz }||d|S)z8F90 file for testing disallowed CLI arguments in ghff819r'r(r)z gh_22819.pyfr*r+r-r2s r gh22819_clir;XsV <~ > > H H J JD  # # % % 6BMM$M))) Irctjdddd}|dz }||d|S)'Generates a single f77 file for testingr'r(r)zhi77.fzhello.fr*r+r-r2s rhello_world_f77r>asV <x 8 8 B B D DD  # # % % 1BMM$M))) Irctjdddd}|dz }||d|S)r=r'r( return_realzfoo77.fzfoo.fr*r+r-r2s r retreal_f77rAjsV < y A A K K M MD  # # % % /BMM$M))) Ircbtjdddd}tjdddd}|dz }|dz }||d ||d |S) r&r'r(f2cmapzisoFortranEnvMap.f90z .f2py_f2cmapz f2cmap.f90mapfiler*r+r-)r3r4rCr5fmaps r f2cmap_f90rFrs <2H I I S S U UD \'5(N C C M M O OF  # # % % 4B  % % ' ') 3DMM$M)))OOFWO--- Irct|}|tdd|t j|j5td|jdD}d|vsJd|vsJd|vsJd|vsJ d d d d S#1swxYwYd S) 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 cDg|]}||jSr)is_filename).0items r z$test_gh22819_cli..s'UUU4dllnnUTYUUUr*z blahmodule.czblah-f2pywrappers.fztest_22819-f2pywrappers.fztest_22819module.cz Ignoring blahN) rsetattrsyssplitr switchdirparentr rglob)capfdr; monkeypatchipath gen_pathss rtest_gh22819_clir[s'   EV%u%>%>%D%D%F%FGGG  % %,, !!##Q*c1111$+++++ ,,,,,,,,,,,,,,,,,,s2BB!Bct|}|tdd|t j|j5tdddtj t5t| \}}d|vsJ dddn #1swxYwYddddS#1swxYwYdS) zjEnsures that the CLI refuses to overwrite signature files CLI :: -h without --overwrite-signature rHf2py -h faker.pyf faker.pyf Fake newsr*r+z&Use --overwrite-signature to overwriteN) rrQrRrSrrTrUr1r r^ SystemExitr rl)rWr6rXrYrperrs rtest_gen_pyf_no_overwriter|s  ! !EV%A%%A%A%G%G%I%IJJJ  % %CC [$$[7$CCC ]: & & C C III%%''FAs;sBBBBB C C C C C C C C C C C C C C CCCCCCCCCCCCCCCCCCCs6>C),C C)C C)C C))C-0C-) zPython 3.12 required)reasoncBt|}|tdd|t j|j5t|\}}d|vsJ ddddS#1swxYwYdS)z@Check that modules are named correctly CLI :: defaults rHzf2py --backend meson -c zuntitledmodule.cN) rrQrRrSrrTrUrrlrts rtest_untitled_clirs  ! !EV%G%G%G%M%M%O%OPPP  % %))   !!##Q!S((((())))))))))))))))))s,BBBLinuxzCompiler and 3.12 requiredc d}t||}|j}|tdd|d|t j|j5t| \}}d|vsJ dddn #1swxYwY|tddt j|j5t| \}}d |vsJ dddn #1swxYwYd }|tdd|d |d t j|j5t| \}}d |vsJ ddddS#1swxYwYdS)zqCheck that no distutils imports are performed on 3.12 CLI :: --fcompiler --help-link --backend distutils hircrHrdz -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) r"rrQrRrSrrTrUrrlr )rWr6rXMNAMErerYrorps r!test_no_py312_distutils_fcompilerrs E  6 6 6E LE VGUGGGGMMOO  % %>>   !!##Q6#=====>>>>>>>>>>>>>>> V(..00  % %33 !!##Q+s22222333333333333333 E VFUFF5FFFLLNN  % %GG !!##Q?3FFFFFGGGGGGGGGGGGGGGGGGs6(,B!!B%(B%3,D,,D03D0,GGGct|d}|j}d}d}|tdd|d|t j|j5t| \}}|D] } d| d |vsJ |D] } d | d |vsJ d d d d S#1swxYwYd S) z9Tests that functions can be skipped CLI :: skip: rbrct0 t4 t8 sd s8 s4td s0rHrdz -m test skip: z=buildmodule: Could not found the body of interfaced routine " ". Skipping.Constructing wrapper function ""N r"rrQrRrSrrTrUr rl) rWrArXrerYtoskip remainingror{skeyrkeys rtest_f2py_skiprs F 3 3 3E JE FI V...f..4466888  % %DD ##%%SLLNN  DbPTbbbOO%% D DDTest that functions can be kept by only: CLI :: only: rbrcrrrHrdz -m test only: C;cd}t||}|j}|tdd|d|t j|j5tdddn #1swxYwYtj |j sJtj |j sJdS)zHChecks the generation of files based on a module name CLI :: -m rrcrHrdrN) r"rrQrRrSrrTrUr rrmrr)rWr6rXrrerYs rtest_mod_gen_f77rTs E  6 6 6E LEV%?U%?%?%?%?%E%E%G%GHHH  % %  ;u{ # #### ;u| $ $$$$$$s(BB Bc d}t||}|j}|tdd|d|dt j|j5ttd 5}| }d|vsJ d d d n #1swxYwYd d d d S#1swxYwYd S) zCheck that pyf files are correctly generated with module structure CLI :: -m -h pyf_file BUG: numpy-gh #20520 rrcrHrdrz -h hi.pyfzhi.pyfzpython module hiN) r"rrQrRrSrrTrUr ropenread)rWr>rXrrerYhipyfpyfdats rtest_mod_gen_gh25263resd E  6 6 6E JEV%IU%I%I%I%I%I%O%O%Q%QRRR  % %00 (^^ " " 0eZZ\\F%///// 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0000000000000000000s6)0CC5 CC CC CC Ccdt|d}|j}tjd}tjd}|t dd|dtj|j 5t| \}}| |J| |J dddn #1swxYwY|t dd|d tj|j 5t| \}}| |J| |J ddddS#1swxYwYdS) zHLowers cases by flag or when -h is present CLI :: --[no-]lower rbrczHI\(\)zhi\(\)rHrdz -m test --lowerNz -m test --no-lower) r"rrecompilerQrRrSrrTrUr rlsearch rWr>rXrerYcapshicapslororps rtest_lower_cmodrus%  7 7 7E JE Z " "F Z " "FV%DU%D%D%D%J%J%L%LMMM  % %** !!##Q}}S!!---}}S!!)))) *************** V::::@@BBDDD  % %.. !!##Q}}S!!)))}}S!!---- ..................s& AC--C14C1AF%%F),F)c t|d}|j}tjd}tjd}|t dd|d|jdtj |j 5t| \}}| |J| |J d d d n #1swxYwY|t dd|d|jd tj |j 5t| \}}| |J| |J d d d d S#1swxYwYd S) z^Lowers cases in signature files by flag or when -h is present CLI :: --[no-]lower -h rbrcz Block: HIz Block: hirHrdz -h z -m test --overwrite-signatureNz) -m test --overwrite-signature --no-lower)r"rrrrQrRrrSrrTrUr rlrrs rtest_lower_sigrsC  7 7 7E JE Z % %F Z % %F DDD59DDDJJLL  % %** !!##Q}}S!!---}}S!!)))) *************** OOO59OOO    % %.. !!##Q}}S!!)))}}S!!---- ..................s&AC55C9<C9AF55F9<F9c ^t|}d}d}|tdd|d|d|t j|j5t|\}}d|d|vsJ d d d d S#1swxYwYd S) zNEnsures that the build directory can be specified CLI :: --build-dir rtttmprHf2py -m riz --build-dir zWrote C/API module "rNrs)rWr6rXrYr odirrorps rtest_build_dirrs  ! !E E DVE5EE5EEtEEKKMMOOO  % %88 !!##Q0u000C77777888888888888888888s$0B""B&)B&ct|}|tdd|dt j|j5tdddt| \}}d|vsJ d d d d S#1swxYwYd S) zXEnsures that the build directory can be specified CLI :: --overwrite-signature rHrwz --overwrite-signaturerxryr*r+rjN) rrQrRrSrrTrUr1r rlrts rtest_overwriters  ! !E V:U:::@@BBDDD  % %22 [$$[7$CCC !!##Q*c11111 222222222222222222sAB99B=B=c t|}d}|tdd|d|dt j|j5t|\}}d|vsJt|d 5}d| vsJ d d d n #1swxYwYd d d d S#1swxYwYd S) zJEnsures that TeX documentation is written out CLI :: --latex-doc rrHrriz --latex-docDocumentation is saved to file module.tex\documentclassN rrQrRrSrrTrUr rlrrrWr6rXrYr rorpotexs r test_latexdocrs  ! !E EV>5>>5>>>DDFFHHH  % %44 !!##Q/36666 U&&& ' ' , , . . 4$$ 33333 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 444444444444444444s7 AC.0C C.C C.C C..C25C2c Nt|}d}|tdd|d|dt j|j5t|\}}d|vsJ ddddS#1swxYwYdS)zMEnsures that TeX documentation is written out CLI :: --no-latex-doc rrHrriz --no-latex-docrNrsrWr6rXrYr rorps rtest_nolatexdocrs  ! !E EVA5AA5AAAGGIIKKK  % %;; !!##Q/s:::::;;;;;;;;;;;;;;;;;; ,BB!Bc t|}d}|tdd|d|dt j|j5t|\}}d|vsJtd|d 5}d | vsJ d d d n #1swxYwYd d d d S#1swxYwYd S) zEnsures that truncated documentation is written out TODO: Test to ensure this has no effect without --latex-doc CLI :: --latex-doc --short-latex rrHrriz --latex-doc --short-latexr./rrNrrs rtest_shortlatexrs  ! !E E <5<<5<<<BBDD  % %88 !!##Q/36666 (u((( ) ) . . 0 0 8D$DIIKK77777 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8888888888888888887 AC/1C C/C C/C C//C36C3c t|}d}|tdd|d|dt j|j5t|\}}d|vsJtd|d 5}d | vsJ d d d n #1swxYwYd d d d S#1swxYwYd S) zIEnsures that RsT documentation is written out CLI :: --rest-doc rrHrriz --rest-doc#ReST Documentation is saved to filerz module.restz.. -*- rest -*-Nr)rWr6rXrYr rorporsts r test_restdocrs  ! !E EV=5==5===CCEEGGG  % %55 !!##Q4;;;; )u))) * * / / 1 1 5T%44444 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 555555555555555555rc Nt|}d}|tdd|d|dt j|j5t|\}}d|vsJ ddddS#1swxYwYdS)zLEnsures that TeX documentation is written out CLI :: --no-rest-doc rrHrriz --no-rest-docrNrsrs rtest_norestexdocr"s  ! !E EV@5@@5@@@FFHHJJJ  % %@@ !!##Q4C?????@@@@@@@@@@@@@@@@@@rc t|}d}|tdd|d|dt j|j5ttd|d5}d| vsJ d d d n #1swxYwYd d d d S#1swxYwYd S) zIEnsures that debugging wrappers are written CLI :: --debug-capi rrHrriz --debug-capirrz#define DEBUGCFUNCSN rrQrRrSrrTrUr rr)rWr6rXrYr ocmods rtest_debugcapir2sl  ! !E EV?5??5???EEGGIII  % %:: &u&&& ' ' , , . . :%)UZZ\\99999 : : : : : : : : : : : : : : :::::::::::::::::::s6 4CB:. C:B> >CB> CCCz2Consistently fails on CI; noisy so skip not xfail.c t|}d}|tdd|d|dt j|j5ttjtj d}tj |dd }d }tj d }|j|ksJ|j|ksJ d d d d S#1swxYwYd S) zEEnsures that debugging wrappers work CLI :: --debug-capi -c rrHrriz -c --debug-capi -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)rrQrRrSrrTrUr shlex executable subprocessruntextwrapdedentstdoutstderr)r6rXrYr cmd_runrouteouteerrs rtest_debugcapi_bldrBsN  ! !E EVB5BB5BBBHHJJLLL  % %## +OOOPP~gdWMMM   {d""""{d"""""!##################s A7C%%C),C)c nt|}d}|tdd|d|t j|j5tdddn #1swxYwY|\}}d|vsJ|tdd|d|dt j|j5t|\}}d|vsJ ddddS#1swxYwYdS)zoEnsures that fortran subroutine wrappers for F77 are included by default CLI :: --[no]-wrap-functions rrHrriN Fortran 77 wrappers are saved toz --wrap-functionsrsrs rtest_wrapfunc_defr`s  ! !E EV%?%?%?%?%?%E%E%G%GHHH  % %      FC .# 5 5 5 5VC5CC5CCCIIKKMMM  % %:: !!##Q2c99999::::::::::::::::::s$A::A>A>0,D**D.1D.c Nt|}d}|tdd|d|dt j|j5t|\}}d|vsJ ddddS#1swxYwYdS)zeEnsures that fortran subroutine wrappers for F77 can be disabled CLI :: --no-wrap-functions rrHrriz --no-wrap-functionsrNrsrs rtest_nowrapfuncrys  ! !E EVF5FF5FFFLLNNPPP  % %>> !!##Q2#=====>>>>>>>>>>>>>>>>>>rc t|}d}|tdd|d|dt j|j5ttd|d5}| }d|vsJd |vsJ d d d n #1swxYwYd d d d S#1swxYwYd S) zcAdd to the include directories CLI :: -include TODO: Document this in the help string rrHrriz' -include -include rrz#include z#include Nr)rWr6rXrYr rocmrs rtest_inclheaderrs~  ! !E E I5II5III    % %00 &u&&& ' ' , , . . 0%::<>>   G GCC 4KC+c""~gdWMMM{d""""  m]aealllll!##### $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $sB C22C69C6ct|}|tdd|dt j|j5ttjd}tr|dz }n|dz }tj|}tj |dd }d }|j |ksJ|jd ksJ|jd ksJ d d d d S#1swxYwYd S)z+ CLI :: --freethreading_compatible rHrIz -c --freethreading-compatiblerz2import sys; assert sys._is_gil_enabled() is False"rTrrrrNrrs rtest_freethreading_compatibler sW  ! !EV%ZU%Z%Z%Z%`%`%b%bccc  % % $ $   >>>   H HCC 4KC+c""~gdWMMM{d""""{b    !##### $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $sBC--C14C1cdS)z CLI :: -c --fcompiler Nrrrrtest_npd_fcompilerr 2  DrcdS)z CLI :: -c --compiler Nrrrrtest_npd_compilerr:r rcdS)z$ CLI :: -c --help-fcompiler Nrrrrtest_npd_help_fcompilerrBr rcdS)z CLI :: -c --f77exec Nrrrrtest_npd_f77execrJr rcdS)z CLI :: -c --f90exec Nrrrrtest_npd_f90execrRr rcdS)z CLI :: -c --f77flags Nrrrrtest_npd_f77flagsrZr rcdS)z CLI :: -c --f90flags Nrrrrtest_npd_f90flagsrbr rcdS)z CLI :: -c --opt Nrrrr test_npd_optrjr rcdS)z CLI :: -c --arch Nrrrr test_npd_archrrr rcdS)z CLI :: -c --noopt Nrrrrtest_npd_nooptrzr rcdS)z CLI :: -c --noarch Nrrrrtest_npd_noarchr!r rcdS)z CLI :: -c --debug Nrrrrtest_npd_debugr#r rcdS)z% CLI :: -c --link- Nrrrrtest_npd_link_autor%r rcdS)z/ CLI :: -c -L/path/to/lib/ -l Nrrrr test_npd_libr'r rcdS)z CLI :: -D Nrrrrtest_npd_definer)r rcdS)z CLI :: -U Nrrrrtest_npd_undefiner+r rcdS)z$ CLI :: -I/path/to/include/ Nrrrr test_npd_inclr-r rcdS)z8 CLI :: .o .so .a Nrrrrtest_npd_linkerr/r r)r)XrrrRrrpathlibr collectionsrplatformr r rnumpy.f2py.f2py2err numpy.testing._private.utilsrrrr"fixturer6r9r;r>rArFr[r`rgrqrur|markskipif version_inforsystemrxfailrrrrrrrrrrrrrrrr rrrrrrrrrrrrr r rrrrrrrrrr!r#r%r'r)r+r-r/rrrr;s++++++++++++++++++++"""""" ------444444 HH I INi   ! i   ! i   ! i   ! i   ! i   ! (    : : : ) ) ) , , , C C CC$/8NOO ) )PO )_X_&&'1Ss7G77R\xyyGGzyG>DDD.DDD.DDD<%%%" 0 0 0 ...4$.$.$.N888"222"444$ ; ; ; 888,555$ @ @ @ : : : MNN##ON#::::2 > > > 000.      III$    $ $ $ - - -MNN # #ON #_X_&&'1Qc6F'6QGII$$II$._X_&&'1Qc6F'6QGII$$II$2                                                        r