o Tjl@s`ddlZddlZddlZddlZddlZddlmZddlmZddl Z ddl Z ddl m Z ddl mZddlmZdd Zed d Zdd dZe jddddZe jddddZe jddddZe jddddZe jddddZe jddddZddZdd Zd!d"Zd#d$Zd%d&Z d'd(Z!e j"j#ej$d)kd*d+d,d-Z%e j"j#e &d.kpej$d)kd/d+d0d1Z'e j"j(d2d3Z)d4d5Z*d6d7Z+d8d9Z,d:d;Z-dd?Z/d@dAZ0dBdCZ1dDdEZ2dFdGZ3dHdIZ4dJdKZ5dLdMZ6dNdOZ7e j"j8dPd+dQdRZ9dSdTZ:dUdVZ;dWdXZd]d^Z?d_d`Z@dadbZAdcddZBe j"j8dPd+dedfZCe j"j#e &d.kpHej$d)kdgd+dhdiZDe j"j#e &d.kp^ej$d)kdgd+djdkZEdldmZFdndoZGdpdqZHdrdsZIdtduZJdvdwZKdxdyZLdzd{ZMd|d}ZNd~dZOddZPddZQddZRddZSddZTddZUddZVddZWdS)N)Path) namedtuple)util)main) NOGIL_BUILDcCs ts tddStdS)Nz$CLI command needs a Fortran compiler)rhas_fortran_compilerpytestskipf2pyclir r _/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/numpy/f2py/tests/test_f2py2e.pycompiler_check_f2pyclis rPPathsz(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)finpf90inppyfwrap77wrap90cmodf)rr with_suffix with_name) fname_inpmnamebpathr r r get_io_paths$srsession)scopecC2tdddd}|d}|j|dd|S)'Generates a single f90 file for testingtestssrccliz hiworld.f90z hello.f90asciiencodingrgetpath read_text getbasetemp write_texttmpdir_factoryfdatfnr r r hello_world_f90K r2cC2tdddd}|d}|j|dd|S)z(F90 file for testing warnings in gh23598r#r$ crackfortranzgh23598Warn.f90r&r'r)r.r r r gh23598_warnTr3r6cCr4)z8F90 file for testing disallowed CLI arguments in ghff819r#r$r%z gh_22819.pyfr&r'r)r.r r r gh22819_cli]r3r7cCr!)'Generates a single f77 file for testingr#r$r%zhi77.fzhello.fr&r'r)r.r r r hello_world_f77fr3r9cCr!)r8r#r$ return_realzfoo77.fzfoo.fr&r'r)r.r r r retreal_f77or3r;cCs`tdddd}tdddd}|d}|d}|j|dd |j|dd |S) r"r#r$f2cmapzisoFortranEnvMap.f90z .f2py_f2cmapz f2cmap.f90mapfiler&r'r))r/r0r<r1fmapr r r f2cmap_f90ws  r?cCst|}|tdd|t|j0tdd|jdD}d|vs+Jd|vs1Jd|vs7Jd |vs=J Wd d S1sIwYd 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 cSsg|] }|r|jqSr )is_filename).0itemr r r sz$test_gh22819_cli..*z blahmodule.czblah-f2pywrappers.fztest_22819-f2pywrappers.fztest_22819module.cz Ignoring blahN) rsetattrsyssplitr switchdirparentr rglob)capfdr7 monkeypatchipath gen_pathsr r r test_gh22819_clis    "rRc Cst|}|tdd|dt|j+tjt dd t Wdn1s-wYWddSWddS1sEwYdS)zBOnly one .pyf file allowed gh-22819 CLI :: .pyf files r@rAz hello.pyfzOnly one .pyf file per call)matchN) rrHrIrJrrKrLr raises ValueErrorr )rNr7rOrPr r r test_gh22819_many_pyfs"rVcCszt|dd}|j}|tdd|dt|jt|j }d|vs+JWddS1s6wYdS)Ntestrr@f2py z -m testzintproductf2pywrap, intpr) rrrHrIrJrrKrLr rr+)rNr6rOfoutlrPwrapperr r r test_gh23598_warns  "r\cCst|}t|jd}|tdd|d|t|j!t| \}}d|vs0Jt| s9JWddS1sDwYdS)zIEnsures that a signature file is generated via the CLI CLI :: -h rr@zf2py -h  Saving signatures to fileN) rstemrHrIrJrrKrLr readouterrexists)rNr2rOrPopathout_r r r test_gen_pyfs  "recCs|t|}|tdd|t|jt|\}}d|vs&Jd|vs,JWddS1s7wYdS)zHEnsures that a signature file can be dumped to stdout CLI :: -h r@zf2py -h stdout r^zfunction hi() ! in N rrHrIrJrrKrLr r`rNr2rOrPrcrdr r r test_gen_pyf_stdouts  "rhc 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 newsr&r'z&Use --overwrite-signature to overwriteN) rrHrIrJrrKrLr-r rT SystemExitr r`)rNr2rOrPrderrr r r test_gen_pyf_no_overwrites  "rn) zPython 3.12 required)reasoncCspt|}|tdd|t|jt|\}}d|vs&JWddS1s1wYdS)z@Check that modules are named correctly CLI :: defaults r@zf2py --backend meson -c zuntitledmodule.cN) rrHrIrJrrKrLrr`rgr r r test_untitled_clis "rrLinuxzCompiler and 3.12 requiredc CsNd}t||d}|j}|tdd|d|t|jt| \}}d|vs0JWdn1s:wY|tddt|jt | \}}d |vs^JWdn1shwYd }|tdd|d |d t|jt | \}}d |vsJWddS1swYdS)zqCheck that no distutils imports are performed on 3.12 CLI :: --fcompiler --help-link --backend distutils hirXr@rYz -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) rrrHrIrJrrKrLrr`r )rNr2rOMNAMErZrPrcrdr r r !test_no_py312_distutils_fcompilers8     "rwc Cst|dd}|j}d}d}|tdd|d|t|j4t| \}}|D] } d| d |vsTest that functions can be kept by only: CLI :: only: rWrXrxryr@rYz -m test only: -h pyf_file BUG: numpy-gh #20520 rtrXr@rYrz -h hi.pyfzhi.pyfzpython module hiN) rrrHrIrJrrKrLr ropenread)rNr9rOrvrZrPhipyfpyfdatr r r test_mod_gen_gh25263js  "rc Cs$t|dd}|j}td}td}|tdd|dt|j #t | \}}| |dus9J| |dusBJWdn1sLwY|tdd|d t|j $t | \}}| |duswJ| |dusJWddS1swYdS) zHLowers cases by flag or when -h is present CLI :: --[no-]lower rWrXzHI\(\)zhi\(\)r@rYz -m test --lowerNz -m test --no-lower) rrrecompilerHrIrJrrKrLr r`search rNr9rOrZrPcapshicapslorcrdr r r test_lower_cmodzs(     "rc Cs4t|dd}|j}td}td}|tdd|d|jdt |j #t | \}}| |d us=J| |d usFJWd n1sPwY|tdd|d|jd t |j $t | \}}| |d usJ| |d usJWd d S1swYd S) z^Lowers cases in signature files by flag or when -h is present CLI :: --[no-]lower -h rWrXz Block: HIz Block: hir@rYz -h z -m test --overwrite-signatureNz) -m test --overwrite-signature --no-lower)rrrrrHrIrrJrrKrLr r`rrr r r test_lower_sigs6     "rc Cst|}d}d}|tdd|d|d|t|jt|\}}d|d|vs4JWd d S1s?wYd S) zNEnsures that the build directory can be specified CLI :: --build-dir rtttmpr@f2py -m r]z --build-dir zWrote C/API module "r|Nrf)rNr2rOrProdirrcrdr r r test_build_dirs "rcCst|}|tdd|dt|j!tdjdddt| \}}d|vs0JWd d S1s;wYd S) zXEnsures that the build directory can be specified CLI :: --overwrite-signature r@riz --overwrite-signaturerjrkr&r'r^N) rrHrIrJrrKrLr-r r`rgr r r test_overwrites "rc 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 rr@rr]z --latex-docDocumentation is saved to file module.tex\documentclassN rrHrIrJrrKrLr r`rrrNr2rOrPrrcrdotexr r r test_latexdocs  "rc C|t|}d}|tdd|d|dt|jt|\}}d|vs,JWddS1s7wYdS)zMEnsures that TeX documentation is written out CLI :: --no-latex-doc rr@rr]z --no-latex-docrNrfrNr2rOrPrrcrdr r r test_nolatexdoc "rc 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 rr@rr]z --latex-doc --short-latexr./rrNrrr r r test_shortlatexs"  "rc 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 rr@rr]z --rest-doc#ReST Documentation is saved to filerz module.restz.. -*- rest -*-Nr)rNr2rOrPrrcrdorstr r r test_restdocs  "rc Cr)zLEnsures that TeX documentation is written out CLI :: --no-rest-doc rr@rr]z --no-rest-docrNrfrr r r test_norestexdoc'rrc 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 rr@rr]z --debug-capirrz#define DEBUGCFUNCSN rrHrIrJrrKrLr rr)rNr2rOrProcmodr r r test_debugcapi7s"rz2Consistently fails on CI; noisy so skip not xfail.c Cst|}d}|tdd|d|dt|j2tttj d}t j |ddd }d }t d }|j|ks?J|j|ksFJWd d S1sQwYd S) zEEnsures that debugging wrappers work CLI :: --debug-capi -c rr@rr]z -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)rrHrIrJrrKrLr shlex executable subprocessruntextwrapdedentstdoutstderr)r2rOrPrcmd_runrouteouteerrr r r test_debugcapi_bldGs  "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 rr@rr]N Fortran 77 wrappers are saved toz --wrap-functionsrfrr r r test_wrapfunc_defes    "rc Cr)zeEnsures that fortran subroutine wrappers for F77 can be disabled CLI :: --no-wrap-functions rr@rr]z --no-wrap-functionsrNrfrr r r test_nowrapfunc~rrc 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 rr@rr]z' -include -include rrz#include z#include Nr)rNr2rOrPrrocmrr r r test_inclheaders$ "rcCdS)z?Add to the include directories CLI :: --include-paths Nr r r r r test_inclpathrcCr)z;Add to the include directories CLI :: --help-link Nr r r r r test_hlinkrrcCst|}|tdd|dt|j0t|\}}d|vs'Jd|vs-Jd|vs3Jd|vs9Jd|vs?JWd d S1sJwYd S) zOCheck that Fortran-to-Python KIND specs can be passed CLI :: --f2cmap r@rAz --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 changesNrf)rNr?rOrPrcrdr r r test_f2cmaps     "rcCsvt|}|tdd|dt|jt|\}}t |dks)JWddS1s4wYdS)z)Reduce verbosity CLI :: --quiet r@rAz --quietrN) rrHrIrJrrKrLr r`lenrgr r r test_quiets "rcCsrt|}|tdd|dt|jt|\}}d|vs'JWddS1s2wYdS)z-Increase verbosity CLI :: --verbose r@rAz --verbose analyzelineNrfrgr r r test_verboses "rcCsn|tddttt|\}}ddl}|j | ks%JWddS1s0wYdS)z"Ensure version CLI :: -v r@zf2py -vrN) rHrIrJr rTrlr r`numpy __version__strip)rNrOrcrdnpr r r test_versions  "rcCst|}|tdd|dt|j&tttj d}t j |ddd}d}|j |ks5JWd d S1s@wYd S) z CLI :: -c r@rAz -crTrrrN) rrHrIrJrrKrLr rrrrr)r2rOrPrrrr r r test_npdistops"rz*Compiler and Python 3.12 or newer requiredcCst|}|tdd|dt|jCttjd}t r(|d7}n|d7}t |}t j |ddd }d }|j |ksBJt rKd |jvsKJ|jd ksRJWd d S1s]wYd S)z. CLI :: --no-freethreading-compatible r@rAz! -c --no-freethreading-compatible -c "import blah; blah.hi();z1import sys; assert sys._is_gil_enabled() is True"r|TrrrzHThe global interpreter lock (GIL) has been enabled to load module 'blah'rNrrHrIrJrrKrLrrrrrrrr returncoder2rOrPcmdrrrr r r test_no_freethreading_compatibles    "rcCst|}|tdd|dt|jAttjd}t r(|d7}n|d7}t |}t j |ddd }d }|j |ksBJ|jd ksIJ|jd ksPJWd d S1s[wYd S)z+ CLI :: --freethreading_compatible r@rAz -c --freethreading-compatiblerz2import sys; assert sys._is_gil_enabled() is False"r|TrrrrNrrr r r test_freethreading_compatibles   "rcCr)z CLI :: -c --fcompiler Nr r r r r test_npd_fcompiler7rcCr)z CLI :: -c --compiler Nr r r r r test_npd_compiler?rrcCr)z$ CLI :: -c --help-fcompiler Nr r r r r test_npd_help_fcompilerGrrcCr)z CLI :: -c --f77exec Nr r r r r test_npd_f77execOrrcCr)z CLI :: -c --f90exec Nr r r r r test_npd_f90execWrrcCr)z CLI :: -c --f77flags Nr r r r r test_npd_f77flags_rrcCr)z CLI :: -c --f90flags Nr r r r r test_npd_f90flagsgrrcCr)z CLI :: -c --opt Nr r r r r test_npd_optorrcCr)z CLI :: -c --arch Nr r r r r test_npd_archwrrcCr)z CLI :: -c --noopt Nr r r r r test_npd_nooptrrcCr)z CLI :: -c --noarch Nr r r r r test_npd_noarchrrcCr)z CLI :: -c --debug Nr r r r r test_npd_debugrrcCr)z% CLI :: -c --link- Nr r r r r test_npd_link_autorrcCr)z/ CLI :: -c -L/path/to/lib/ -l Nr r r r r test_npd_librrcCr)z CLI :: -D Nr r r r r test_npd_definerrcCr)z CLI :: -U Nr r r r r test_npd_undefinerrcCr)z$ CLI :: -I/path/to/include/ Nr r r r r test_npd_inclrrcCr)z8 CLI :: .o .so .a Nr r r r r test_npd_linkerrr)r)XrrrrIrpathlibr collectionsrplatformr rrnumpy.f2py.f2py2err numpy.testing._private.utilsrrrrfixturer2r6r7r9r;r?rRrVr\rerhrnmarkskipif version_inforrsystemrwxfailrrrrrrrrrrrrrrrr rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr r r r s       '                 '