o ʩZhOc@sddlZddlZddlZddlZddlZddlmZddlmZddl Z ddl Z ddl m Z ddl mZeddZdd d Ze jd d ddZe jd d ddZe jd d ddZe jd d ddZe jd d ddZe jd d ddZddZddZddZd d!Zd"d#Zd$d%Ze jj e !d&kpej"d'kd(d)d*d+Z#e jj e !d&kpej"d'kd(d)d,d-Z$e jj%d.d/Z&d0d1Z'd2d3Z(d4d5Z)d6d7Z*d8d9Z+d:d;Z,dd?Z.d@dAZ/dBdCZ0dDdEZ1dFdGZ2dHdIZ3dJdKZ4e jj%dLd)dMdNZ5dOdPZ6dQdRZ7dSdTZ8dUdVZ9dWdXZ:dYdZZ;d[d\Ze jj%dLd)dadbZ?dcddZ@dedfZAdgdhZBdidjZCdkdlZDdmdnZEdodpZFdqdrZGdsdtZHdudvZIdwdxZJdydzZKd{d|ZLd}d~ZMddZNddZOddZPddZQdS)N)Path) namedtuple)util)mainPPathsz(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_inpmnamebpathrS/var/www/html/lang_env/lib/python3.10/site-packages/numpy/f2py/tests/test_f2py2e.py get_io_pathssrsession)scopecC2tdddd}|d}|j|dd|S)'Generates a single f90 file for testingtestssrccliz hiworld.f90z hello.f90asciiencodingrZgetpath read_textZ getbasetemp write_texttmpdir_factoryfdatfnrrrhello_world_f909 r)cC2tdddd}|d}|j|dd|S)z(F90 file for testing warnings in gh23598rrZ crackfortranzgh23598Warn.f90rr r"r%rrr gh23598_warnBr*r,cCr+)z8F90 file for testing disallowed CLI arguments in ghff819rrrz gh_22819.pyfrr r"r%rrr gh22819_cliKr*r-cCr)'Generates a single f77 file for testingrrrzhi77.fzhello.frr r"r%rrrhello_world_f77Tr*r/cCr)r.rrZ return_realzfoo77.fzfoo.frr r"r%rrr retreal_f77]r*r0cCs`tdddd}tdddd}|d}|d}|j|dd |j|dd |S) rrrf2cmapzisoFortranEnvMap.f90z .f2py_f2cmapz f2cmap.f90Zmapfilerr r")r&r'r1r(Zfmaprrr f2cmap_f90es  r2cCst|}|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).0itemrrr }sz$test_gh22819_cli..*z blahmodule.czblah-f2pywrappers.fztest_22819-f2pywrappers.fztest_22819module.cz Ignoring blahN) rsetattrsyssplitr switchdirparentf2pyclirglob)capfdr- monkeypatchipathZ gen_pathsrrrtest_gh22819_cliqs    "rEc 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?pytestraises ValueErrorr@)rBr-rCrDrrrtest_gh22819_many_pyfs"rJcCszt|dd}|j}|tdd|dt|jt|j }d|vs+JWddS1s6wYdS)Ntestrr3f2py z -m testzintproductf2pywrap, intpr) rr r;r<r=rr>r?r@rr#)rBr,rCfoutlrDwrapperrrrtest_gh23598_warns  "rPcCst|}t|jd}|tdd|d|t|j!t| \}}d|vs0Jt| s9JWddS1sDwYdS)zIEnsures that a signature file is generated via the CLI CLI :: -h r r3zf2py -h  Saving signatures to fileN) rstemr;r<r=rr>r?r@ readouterrexists)rBr)rCrDZopathout_rrr test_gen_pyfs  "rXcCs|t|}|tdd|t|jt|\}}d|vs&Jd|vs,JWddS1s7wYdS)zHEnsures that a signature file can be dumped to stdout CLI :: -h r3zf2py -h stdout rRzfunction hi() ! in N rr;r<r=rr>r?r@rTrBr)rCrDrVrWrrrtest_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 r3f2py -h faker.pyf faker.pyf Fake newsrr z&Use --overwrite-signature to overwriteN) rr;r<r=rr>r?r$rGrH SystemExitr@rT)rBr)rCrDrWerrrrrtest_gen_pyf_no_overwrites  "raLinux) zCompiler and 3.12 required)reasoncCspt|}|tdd|t|jt|\}}d|vs&JWddS1s1wYdS)z@Check that modules are named correctly CLI :: defaults r3zf2py --backend meson -c zuntitledmodule.cNrYrZrrrtest_untitled_clis "rfc 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 hirLr3rMz -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) rr r;r<r=rr>r?r@rT)rBr)rCMNAMErNrDrVrWrrr!test_no_py312_distutils_fcompilers8     "ric Cst|dd}|j}d}d}|tdd|d|t|j4t| \}}|D] } d| d |vsr?r@rT) rBr0rCrNrDtoskip remainingrVr`skeyrkeyrrrtest_f2py_skips&     "rtc Cst|dd}|j}d}d}|tdd|d|t|j4t| \}}|D] } d| d |vsTest that functions can be kept by only: CLI :: only: rKrLrjrkr3rMz -m test only: r?r@rT)rBr)r0rCrNrDrpZipath2rvrrVr`rrrsrrrtest_file_processing_switch's,     "rzcCsd}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 rgrLr3rMryN) rr r;r<r=rr>r?r@rrUrr)rBr)rCrhrNrDrrrtest_mod_gen_f77Es r{c 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 rgrLr3rMryz -h hi.pyfzhi.pyfzpython module hiN) rr r;r<r=rr>r?r@ropenread)rBr/rCrhrNrDZhipyfZpyfdatrrrtest_mod_gen_gh25263Vs  "r~c 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 rKrLzHI\(\)zhi\(\)r3rMz -m test --lowerNz -m test --no-lower) rr recompiler;r<r=rr>r?r@rTsearch rBr/rCrNrDZcapshiZcapslorVrWrrrtest_lower_cmodfs(     "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 rKrLz Block: HIz Block: hir3rMz -h z -m test --overwrite-signatureNz) -m test --overwrite-signature --no-lower)rr rrr;r<r r=rr>r?r@rTrrrrrtest_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 rxZtttmpr3f2py -m rQz --build-dir zWrote C/API module "rnNrY)rBr)rCrDrZodirrVrWrrrtest_build_dirs "rcCst|}|tdd|dt|j!tdjdddt| \}}d|vs0JWd d S1s;wYd S) zXEnsures that the build directory can be specified CLI :: --overwrite-signature r3r\z --overwrite-signaturer]r^rr rRN) rr;r<r=rr>r?r$r@rTrZrrrtest_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 rxr3rrQz --latex-docDocumentation is saved to file module.tex\documentclassN rr;r<r=rr>r?r@rTr|r}rBr)rCrDrrVrWZotexrrr test_latexdocs  "rc C|t|}d}|tdd|d|dt|jt|\}}d|vs,JWddS1s7wYdS)zMEnsures that TeX documentation is written out CLI :: --no-latex-doc rxr3rrQz --no-latex-docrNrYrBr)rCrDrrVrWrrrtest_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 rxr3rrQz --latex-doc --short-latexr./rrNrrrrrtest_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 rxr3rrQz --rest-doc#ReST Documentation is saved to filerz module.restz.. -*- rest -*-Nr)rBr)rCrDrrVrWZorstrrr test_restdocs  "rc Cr)zLEnsures that TeX documentation is written out CLI :: --no-rest-doc rxr3rrQz --no-rest-docrNrYrrrrtest_norestexdocrrc 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 rxr3rrQz --debug-capirr z#define DEBUGCFUNCSN rr;r<r=rr>r?r@r|r})rBr)rCrDrocmodrrrtest_debugcapi#s"rzConsistently fails on CI.c Cst|}d}|tdd|d|dt|j.ttd}t j |ddd }d }t d }|j |ks;J|j|ksBJWd d S1sMwYd S) zEEnsures that debugging wrappers work CLI :: --debug-capi -c rxr3rrQz -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)rr;r<r=rr>r?r@shlex subprocessruntextwrapdedentstdoutstderr)r)rCrDrcmd_runrouteoutZeerrrrrtest_debugcapi_bld3s   "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 rxr3rrQN Fortran 77 wrappers are saved toz --wrap-functionsrYrrrrtest_wrapfunc_defQs    "rc Cr)zeEnsures that fortran subroutine wrappers for F77 can be disabled CLI :: --no-wrap-functions rxr3rrQz --no-wrap-functionsrNrYrrrrtest_nowrapfuncjrrc 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 rxr3rrQz' -include -include rr z#include z#include Nr)rBr)rCrDrrZocmrrrrtest_inclheaderzs$ "rcCdS)z?Add to the include directories CLI :: --include-paths Nrrrrr test_inclpathrcCr)z;Add to the include directories CLI :: --help-link Nrrrrr 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 changesNrY)rBr2rCrDrVrWrrr test_f2cmaps     "rcCsvt|}|tdd|dt|jt|\}}t |dks)JWddS1s4wYdS)z)Reduce verbosity CLI :: --quiet r3r4z --quietrN) rr;r<r=rr>r?r@rTlenrZrrr test_quiets "rcCsrt|}|tdd|dt|jt|\}}d|vs'JWddS1s2wYdS)z-Increase verbosity CLI :: --verbose r3r4z --verboseZ analyzelineNrYrZrrr test_verboses "rcCsn|tddttt|\}}ddl}|j | ks%JWddS1s0wYdS)z"Ensure version CLI :: -v r3zf2py -vrN) r;r<r=rGrHr_r@rTnumpy __version__strip)rBrCrVrWnprrr test_versions  "rcCst|}|tdd|dt|j"ttd}t j |ddd}d}|j |ks1JWd d S1sr?r@rrrr)r)rCrDrrrrrr test_npdistops "rcCr)z CLI :: -c --fcompiler Nrrrrrtest_npd_fcompilerrcCr)z CLI :: -c --compiler Nrrrrrtest_npd_compilerrrcCr)z$ CLI :: -c --help-fcompiler Nrrrrrtest_npd_help_fcompilerrrcCr)z CLI :: -c --f77exec Nrrrrrtest_npd_f77exec rrcCr)z CLI :: -c --f90exec Nrrrrrtest_npd_f90execrrcCr)z CLI :: -c --f77flags Nrrrrrtest_npd_f77flagsrrcCr)z CLI :: -c --f90flags Nrrrrrtest_npd_f90flags#rrcCr)z CLI :: -c --opt Nrrrrr test_npd_opt+rrcCr)z CLI :: -c --arch Nrrrrr test_npd_arch3rrcCr)z CLI :: -c --noopt Nrrrrrtest_npd_noopt;rrcCr)z CLI :: -c --noarch Nrrrrrtest_npd_noarchCrrcCr)z CLI :: -c --debug Nrrrrrtest_npd_debugKrrcCr)z% CLI :: -c --link- Nrrrrrtest_npd_link_autoSrrcCr)z/ CLI :: -c -L/path/to/lib/ -l Nrrrrr test_npd_lib[rrcCr)z CLI :: -D Nrrrrrtest_npd_definecrrcCr)z CLI :: -U Nrrrrrtest_npd_undefinekrrcCr)z$ CLI :: -I/path/to/include/ Nrrrrr test_npd_inclsrrcCr)z8 CLI :: .o .so .a Nrrrrrtest_npd_linker{rr)r)Rrrr<rrpathlibr collectionsrplatformrGrZnumpy.f2py.f2py2err@rrZfixturer)r,r-r/r0r2rErJrPrXr[ramarkZskipifsystem version_inforfriZxfailrtrwrzr{r~rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrs(       '                  '