o ZZhl@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 K/var/www/auris/lib/python3.10/site-packages/numpy/f2py/tests/test_f2py2e.pycompiler_check_f2pyclis r PPathsz(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)Z fname_inpmnamebpathr r r get_io_paths$srsession)scopecC2tdddd}|d}|j|dd|S)'Generates a single f90 file for testingtestssrccliz hiworld.f90z hello.f90asciiencodingrZgetpath read_textZ getbasetemp write_texttmpdir_factoryfdatfnr r r hello_world_f90K r.cC2tdddd}|d}|j|dd|S)z(F90 file for testing warnings in gh23598r!r"Z crackfortranzgh23598Warn.f90r$r%r'r*r r r gh23598_warnTr/r1cCr0)z8F90 file for testing disallowed CLI arguments in ghff819r!r"r#z gh_22819.pyfr$r%r'r*r r r gh22819_cli]r/r2cCr)'Generates a single f77 file for testingr!r"r#zhi77.fzhello.fr$r%r'r*r r r hello_world_f77fr/r4cCr)r3r!r"Z return_realzfoo77.fzfoo.fr$r%r'r*r r r retreal_f77or/r5cCs`tdddd}tdddd}|d}|d}|j|dd |j|dd |S) r r!r"f2cmapzisoFortranEnvMap.f90z .f2py_f2cmapz f2cmap.f90Zmapfiler$r%r')r+r,r6r-Zfmapr r r f2cmap_f90ws  r7cCst|}|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)capfdr2 monkeypatchipathZ gen_pathsr r r test_gh22819_clis    "rIc Cst|}|tdd|dt|j+tjt dd t Wdn1s-wYWddSWddS1sEwYdS)zBOnly one .pyf file allowed gh-22819 CLI :: .pyf files r8r9z hello.pyfzOnly one .pyf file per call)matchN) rr@rArBrrCrDrraises ValueErrorr )rFr2rGrHr r r test_gh22819_many_pyfs"rMcCszt|dd}|j}|tdd|dt|jt|j }d|vs+JWddS1s6wYdS)Ntestrr8f2py z -m testzintproductf2pywrap, intpr) rrr@rArBrrCrDr rr()rFr1rGfoutlrHwrapperr r r test_gh23598_warns  "rScCst|}t|jd}|tdd|d|t|j!t| \}}d|vs0Jt| s9JWddS1sDwYdS)zIEnsures that a signature file is generated via the CLI CLI :: -h rr8zf2py -h  Saving signatures to fileN) rstemr@rArBrrCrDr readouterrexists)rFr.rGrHZopathout_r r r test_gen_pyfs  "r[cCs|t|}|tdd|t|jt|\}}d|vs&Jd|vs,JWddS1s7wYdS)zHEnsures that a signature file can be dumped to stdout CLI :: -h r8zf2py -h stdout rUzfunction hi() ! in N rr@rArBrrCrDr rWrFr.rGrHrYrZr r r test_gen_pyf_stdouts  "r^c Cst|}|tdd|t|j>tdjdddt t t | \}}d|vs5JWdn1s?wYWddSWddS1sWwYdS) zjEnsures that the CLI refuses to overwrite signature files CLI :: -h without --overwrite-signature r8f2py -h faker.pyf faker.pyf Fake newsr$r%z&Use --overwrite-signature to overwriteN) rr@rArBrrCrDr)rrK SystemExitr rW)rFr.rGrHrZerrr r r test_gen_pyf_no_overwrites  "rd) zPython 3.12 required)reasoncCspt|}|tdd|t|jt|\}}d|vs&JWddS1s1wYdS)z@Check that modules are named correctly CLI :: defaults r8zf2py --backend meson -c zuntitledmodule.cN) rr@rArBrrCrDr rWr]r r r test_untitled_clis "rhLinuxzCompiler 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 hirOr8rPz -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@rArBrrCrDr rWr )rFr.rGMNAMErQrHrYrZr r r !test_no_py312_distutils_fcompilers8     "rlc Cst|dd}|j}d}d}|tdd|d|t|j4t| \}}|D] } d| d |vsTest that functions can be kept by only: CLI :: only: rNrOrmrnr8rPz -m test only: -h pyf_file BUG: numpy-gh #20520 rjrOr8rPr|z -h hi.pyfzhi.pyfzpython module hiN) rrr@rArBrrCrDr ropenread)rFr4rGrkrQrHZhipyfZpyfdatr 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 rNrOzHI\(\)zhi\(\)r8rPz -m test --lowerNz -m test --no-lower) rrrecompiler@rArBrrCrDr rWsearch rFr4rGrQrHZcapshiZcapslorYrZr 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 rNrOz Block: HIz Block: hir8rPz -h z -m test --overwrite-signatureNz) -m test --overwrite-signature --no-lower)rrrrr@rArrBrrCrDr rWrrr 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 r{Ztttmpr8f2py -m rTz --build-dir zWrote C/API module "rqNr\)rFr.rGrHrZodirrYrZr 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 r8r_z --overwrite-signaturer`rar$r%rUN) rr@rArBrrCrDr)r rWr]r 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 r{r8rrTz --latex-docDocumentation is saved to file module.tex\documentclassN rr@rArBrrCrDr rWrrrFr.rGrHrrYrZZotexr 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 r{r8rrTz --no-latex-docrNr\rFr.rGrHrrYrZr 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 r{r8rrTz --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 r{r8rrTz --rest-doc#ReST Documentation is saved to filerz module.restz.. -*- rest -*-Nr)rFr.rGrHrrYrZZorstr r r test_restdocs  "rc Cr)zLEnsures that TeX documentation is written out CLI :: --no-rest-doc r{r8rrTz --no-rest-docrNr\rr 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 r{r8rrTz --debug-capirrz#define DEBUGCFUNCSN rr@rArBrrCrDr rr)rFr.rGrHrocmodr 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 r{r8rrTz -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)rr@rArBrrCrDr shlex executable subprocessruntextwrapdedentstdoutstderr)r.rGrHrcmd_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 r{r8rrTN Fortran 77 wrappers are saved toz --wrap-functionsr\rr r r test_wrapfunc_defes    "rc Cr)zeEnsures that fortran subroutine wrappers for F77 can be disabled CLI :: --no-wrap-functions r{r8rrTz --no-wrap-functionsrNr\rr 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 r{r8rrTz' -include -include rrz#include z#include Nr)rFr.rGrHrrZocmrr 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 r8r9z --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 changesNr\)rFr7rGrHrYrZr r r test_f2cmaps     "rcCsvt|}|tdd|dt|jt|\}}t |dks)JWddS1s4wYdS)z)Reduce verbosity CLI :: --quiet r8r9z --quietrN) rr@rArBrrCrDr rWlenr]r r r test_quiets "rcCsrt|}|tdd|dt|jt|\}}d|vs'JWddS1s2wYdS)z-Increase verbosity CLI :: --verbose r8r9z --verboseZ analyzelineNr\r]r r r test_verboses "rcCsn|tddttt|\}}ddl}|j | ks%JWddS1s0wYdS)z"Ensure version CLI :: -v r8zf2py -vrN) r@rArBrrKrbr rWnumpy __version__strip)rFrGrYrZnpr r r test_versions  "rcCst|}|tdd|dt|j&tttj d}t j |ddd}d}|j |ks5JWd d S1s@wYd S) z CLI :: -c r8r9z -crTrrrN) rr@rArBrrCrDr rrrrr)r.rGrHrrrr 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 r8r9z! -c --no-freethreading-compatible -c "import blah; blah.hi();z1import sys; assert sys._is_gil_enabled() is True"rqTrrrzHThe global interpreter lock (GIL) has been enabled to load module 'blah'rNrr@rArBrrCrDr rrrrrrr returncoder.rGrHcmdrrrr 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 r8r9z -c --freethreading-compatiblerz2import sys; assert sys._is_gil_enabled() is False"rqTrrrrNrrr 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)XrrrrArpathlibr collectionsrplatformrrrZnumpy.f2py.f2py2err Znumpy.testing._private.utilsrr rrZfixturer.r1r2r4r5r7rIrMrSr[r^rdmarkZskipif version_inforhsystemrlZxfailrwrzr}r~rrrrrrrrrrrr rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr r r r s       '                 '