o k)hl@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)rZhas_fortran_compilerpytestskipf2pyclir r sz$test_gh22819_cli..*z blahmodule.czblah-f2pywrappers.fztest_22819-f2pywrappers.fztest_22819module.cz Ignoring blahN) rsetattrsyssplitr switchdirparentr Zrglob)capfdr- monkeypatchipathZ gen_pathsr r r test_gh22819_clis    "rCc Cst|}|tdd|dt|j+tjt dd t Wdn1s-wYWddSWddS1sEwYdS)zBOnly one .pyf file allowed gh-22819 CLI :: .pyf files r3r4z hello.pyfzOnly one .pyf file per call)matchN) rr;r<r=rr>r?rraises ValueErrorr )r@r-rArBr r r test_gh22819_many_pyfs"rGcCszt|dd}|j}|tdd|dt|jt|j }d|vs+JWddS1s6wYdS)Ntestrr3f2py z -m testzintproductf2pywrap, intpr) rrr;r<r=rr>r?r rr#)r@r,rAfoutlrBwrapperr r r test_gh23598_warns  "rMcCst|}t|jd}|tdd|d|t|j!t| \}}d|vs0Jt| s9JWddS1sDwYdS)zIEnsures that a signature file is generated via the CLI CLI :: -h rr3zf2py -h  Saving signatures to fileN) rZstemr;r<r=rr>r?r readouterrexists)r@r)rArBZopathout_r r r test_gen_pyfs  "rTcCs|t|}|tdd|t|jt|\}}d|vs&Jd|vs,JWddS1s7wYdS)zHEnsures that a signature file can be dumped to stdout CLI :: -h r3zf2py -h stdout rOzfunction hi() ! in N rr;r<r=rr>r?r rPr@r)rArBrRrSr r r test_gen_pyf_stdouts  "rWc Cst|}|tdd|t|j>tdjdddt t t | \}}d|vs5JWdn1s?wYWddSWddS1sWwYdS) zjEnsures that the CLI refuses to overwrite signature files CLI :: -h without --overwrite-signature r3f2py -h faker.pyf faker.pyf Fake newsrr z&Use --overwrite-signature to overwriteN) rr;r<r=rr>r?r$rrE SystemExitr rP)r@r)rArBrSerrr r r test_gen_pyf_no_overwrites  "r]) zPython 3.12 required)reasoncCspt|}|tdd|t|jt|\}}d|vs&JWddS1s1wYdS)z@Check that modules are named correctly CLI :: defaults r3zf2py --backend meson -c zuntitledmodule.cN) rr;r<r=rr>r?r rPrVr r r test_untitled_clis "raZLinuxzCompiler 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 hirIr3rJz -c --fcompiler=gfortran -m z%--fcompiler cannot be used with mesonNzf2py --help-linkzUse --dep for meson buildsZhi2z -c -m z --backend distutilsz.Cannot use distutils backend with Python>=3.12) rrr;r<r=rr>r?r rPr )r@r)rAMNAMErKrBrRrSr r r !test_no_py312_distutils_fcompilers8     "rdc Cst|dd}|j}d}d}|tdd|d|t|j4t| \}}|D] } d| d |vsr?r rP) r@r0rArKrBtoskipZ remainingrRr\skeyrkeyr r r test_f2py_skip s&     "rnc Cst|dd}|j}d}d}|tdd|d|t|j4t| \}}|D] } d| d |vsTest that functions can be kept by only: CLI :: only: rHrIrerfr3rJz -m test only: r?r rP)r@r)r0rArKrBrkZipath2rprrRr\rlrmr r r test_file_processing_switch;s,     "rtcCsd}t||d}|j}|tdd|d|t|j tWdn1s.wYt |j s;Jt |j sCJdS)zHChecks the generation of files based on a module name CLI :: -m rbrIr3rJrsN) rrr;r<r=rr>r?r rrQrr)r@r)rArcrKrBr r r test_mod_gen_f77Ys ruc Csd}t||d}|j}|tdd|d|dt|j4tt d }| }d|vs6JWd n1s@wYWd d SWd d S1sXwYd S) zCheck that pyf files are correctly generated with module structure CLI :: -m -h pyf_file BUG: numpy-gh #20520 rbrIr3rJrsz -h hi.pyfzhi.pyfzpython module hiN) rrr;r<r=rr>r?r ropenread)r@r/rArcrKrBZhipyfZpyfdatr r r test_mod_gen_gh25263js  "rxc 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 rHrIzHI\(\)zhi\(\)r3rJz -m test --lowerNz -m test --no-lower) rrrecompiler;r<r=rr>r?r rPsearch r@r/rArKrBZcapshiZcapslorRrSr r r test_lower_cmodzs(     "r}c 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 rHrIz Block: HIz Block: hir3rJz -h z -m test --overwrite-signatureNz) -m test --overwrite-signature --no-lower)rrryrzr;r<rr=rr>r?r rPr{r|r r r test_lower_sigs6     "r~c 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 rrZtttmpr3f2py -m rNz --build-dir zWrote C/API module "riNrU)r@r)rArBrZodirrRrSr 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 r3rXz --overwrite-signaturerYrZrr rON) rr;r<r=rr>r?r$r rPrVr 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 rrr3rrNz --latex-docDocumentation is saved to file module.tex\documentclassN rr;r<r=rr>r?r rPrvrwr@r)rArBrrRrSZotexr 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 rrr3rrNz --no-latex-docrNrUr@r)rArBrrRrSr 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 rrr3rrNz --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 rrr3rrNz --rest-doc#ReST Documentation is saved to filerz module.restz.. -*- rest -*-Nr)r@r)rArBrrRrSZorstr r r test_restdocs  "rc Cr)zLEnsures that TeX documentation is written out CLI :: --no-rest-doc rrr3rrNz --no-rest-docrNrUrr 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 rrr3rrNz --debug-capirrz#define DEBUGCFUNCSN rr;r<r=rr>r?r rvrw)r@r)rArBrocmodr 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 rrr3rrNz -c --debug-capi -c "import blah; blah.hi()"TUTF-8Zcapture_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=rr>r?r shlex executable subprocessruntextwrapdedentstdoutstderr)r)rArBrcmd_runrouteoutZeerrr 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 rrr3rrNN Fortran 77 wrappers are saved toz --wrap-functionsrUrr r r test_wrapfunc_defes    "rc Cr)zeEnsures that fortran subroutine wrappers for F77 can be disabled CLI :: --no-wrap-functions rrr3rrNz --no-wrap-functionsrNrUrr 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 rrr3rrNz' -include -include rrz#include z#include Nr)r@r)rArBrrZocmrr 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 r3r4z --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 changesNrU)r@r2rArBrRrSr r r test_f2cmaps     "rcCsvt|}|tdd|dt|jt|\}}t |dks)JWddS1s4wYdS)z)Reduce verbosity CLI :: --quiet r3r4z --quietrN) rr;r<r=rr>r?r rPlenrVr r r test_quiets "rcCsrt|}|tdd|dt|jt|\}}d|vs'JWddS1s2wYdS)z-Increase verbosity CLI :: --verbose r3r4z --verboseZ analyzelineNrUrVr r r test_verboses "rcCsn|tddttt|\}}ddl}|j | ks%JWddS1s0wYdS)z"Ensure version CLI :: -v r3zf2py -vrN) r;r<r=rrEr[r rPZnumpy __version__strip)r@rArRrSZnpr r r test_versions  "rcCst|}|tdd|dt|j&tttj d}t j |ddd}d}|j |ks5JWd d S1s@wYd S) z CLI :: -c r3r4z -crTrrrN) rr;r<r=rr>r?r rrrrr)r)rArBrrrr 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 r3r4z! -c --no-freethreading-compatible -c "import blah; blah.hi();z1import sys; assert sys._is_gil_enabled() is True"riTrrrzHThe global interpreter lock (GIL) has been enabled to load module 'blah'rNrr;r<r=rr>r?r rrrrrrr returncoder)rArBcmdrrrr 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 r3r4z -c --freethreading-compatiblerz2import sys; assert sys._is_gil_enabled() is False"riTrrrrNrrr 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)Xryrrr<rZpathlibr collectionsrplatformrrrZnumpy.f2py.f2py2err Znumpy.testing._private.utilsrr rrZfixturer)r,r-r/r0r2rCrGrMrTrWr]ZmarkZskipif version_inforasystemrdZxfailrnrqrtrurxr}r~rrrrrrrrr rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr r r r s       '                 '