o Dfc@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 jj5dLd)dMdNZ6dOdPZ7dQdRZ8dSdTZ9dUdVZ:dWdXZ;dYdZZd_d`Z?e jj5dLd)dadbZ@dcddZAdedfZBdgdhZCdidjZDdkdlZEdmdnZFdodpZGdqdrZHdsdtZIdudvZJdwdxZKdydzZLd{d|ZMd}d~ZNddZOddZPddZQddZRdS)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) fname_inpmnamebpathrU/home/ubuntu/webapp/venv/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.f90asciiencodingrgetpath read_text getbasetemp write_texttmpdir_factoryfdatfnrrrhello_world_f909 r,cC2tdddd}|d}|j|dd|S)z(F90 file for testing warnings in gh23598rr crackfortranzgh23598Warn.f90r r!r#r(rrr gh23598_warnBr-r0cCr.)z8F90 file for testing disallowed CLI arguments in ghff819rrrz gh_22819.pyfr r!r#r(rrr gh22819_cliKr-r1cCr)'Generates a single f77 file for testingrrrzhi77.fzhello.fr r!r#r(rrrhello_world_f77Tr-r3cCr)r2rr return_realzfoo77.fzfoo.fr r!r#r(rrr retreal_f77]r-r5cCs`tdddd}tdddd}|d}|d}|j|dd |j|dd |S) rrrf2cmapzisoFortranEnvMap.f90z .f2py_f2cmapz f2cmap.f90mapfiler r!r#)r)r*r6r+fmaprrr f2cmap_f90es  r9cCst|}|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)capfdr1 monkeypatchipath gen_pathsrrrtest_gh22819_cliqs    "rMc Cst|}|tdd|dt|j+tjt dd t Wdn1s-wYWddSWddS1sEwYdS)zBOnly one .pyf file allowed gh-22819 CLI :: .pyf files r:r;z hello.pyfzOnly one .pyf file per call)matchN) rrBrCrDrrErFpytestraises ValueErrorrG)rIr1rJrKrrrtest_gh22819_many_pyfs"rRcCszt|dd}|j}|tdd|dt|jt|j }d|vs+JWddS1s6wYdS)Ntestrr:f2py z -m testzintproductf2pywrap, intpr) rr rBrCrDrrErFrGrr%)rIr0rJfoutlrKwrapperrrrtest_gh23598_warns  "rXcCst|}t|jd}|tdd|d|t|j!t| \}}d|vs0Jt| s9JWddS1sDwYdS)zIEnsures that a signature file is generated via the CLI CLI :: -h r r:zf2py -h  Saving signatures to fileN) rstemrBrCrDrrErFrG readouterrexists)rIr,rJrKopathout_rrr test_gen_pyfs  "racCs|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 rZzfunction hi() ! in N rrBrCrDrrErFrGr\rIr,rJrKr_r`rrrtest_gen_pyf_stdouts  "rdc 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) rrBrCrDrrErFr'rOrP SystemExitrGr\)rIr,rJrKr`errrrrtest_gen_pyf_no_overwrites  "rjLinux) zCompiler and 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.cNrbrcrrrtest_untitled_clis "roc 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 hirTr:rUz -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) rr rBrCrDrrErFrGr\)rIr,rJMNAMErVrKr_r`rrr!test_no_py312_distutils_fcompilers8     "rsc Cst|dd}|j}d}d}|tdd|d|t|j4t| \}}|D] } d| d |vsTest that functions can be kept by only: CLI :: only: rSrTrtrur:rUz -m test only: -h pyf_file BUG: numpy-gh #20520 rprTr:rUrz -h hi.pyfzhi.pyfzpython module hiN) rr rBrCrDrrErFrGropenread)rIr3rJrrrVrKhipyfpyfdatrrrtest_mod_gen_gh25263Vs  "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 rSrTzHI\(\)zhi\(\)r:rUz -m test --lowerNz -m test --no-lower) rr recompilerBrCrDrrErFrGr\search rIr3rJrVrKcapshicapslor_r`rrrtest_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 rSrTz Block: HIz Block: hir:rUz -h z -m test --overwrite-signatureNz) -m test --overwrite-signature --no-lower)rr rrrBrCr rDrrErFrGr\rrrrrtest_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 rYz --build-dir zWrote C/API module "rxNrb)rIr,rJrKrodirr_r`rrrtest_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:rez --overwrite-signaturerfrgr r!rZN) rrBrCrDrrErFr'rGr\rcrrrtest_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:rrYz --latex-docDocumentation is saved to file module.tex\documentclassN rrBrCrDrrErFrGr\rrrIr,rJrKrr_r`otexrrr 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:rrYz --no-latex-docrNrbrIr,rJrKrr_r`rrrtest_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:rrYz --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 rr:rrYz --rest-doc#ReST Documentation is saved to filerz module.restz.. -*- rest -*-Nr)rIr,rJrKrr_r`orstrrr test_restdocs  "rc Cr)zLEnsures that TeX documentation is written out CLI :: --no-rest-doc rr:rrYz --no-rest-docrNrbrrrrtest_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 rr:rrYz --debug-capirr z#define DEBUGCFUNCSN rrBrCrDrrErFrGrr)rIr,rJrKrocmodrrrtest_debugcapi#s"rz2Consistently fails on CI; noisy so skip not xfail.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 rr:rrYz -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)rrBrCrDrrErFrGshlex subprocessruntextwrapdedentstdoutstderr)r,rJrKrcmd_runrouteouteerrrrrtest_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 rr:rrYN Fortran 77 wrappers are saved toz --wrap-functionsrbrrrrtest_wrapfunc_defQs    "rc Cr)zeEnsures that fortran subroutine wrappers for F77 can be disabled CLI :: --no-wrap-functions rr:rrYz --no-wrap-functionsrNrbrrrrtest_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 rr:rrYz' -include -include rr z#include z#include Nr)rIr,rJrKrrocmrrrrtest_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 r:r;z --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 changesNrb)rIr9rJrKr_r`rrr test_f2cmaps     "rcCsvt|}|tdd|dt|jt|\}}t |dks)JWddS1s4wYdS)z)Reduce verbosity CLI :: --quiet r:r;z --quietrN) rrBrCrDrrErFrGr\lenrcrrr test_quiets "rcCsrt|}|tdd|dt|jt|\}}d|vs'JWddS1s2wYdS)z-Increase verbosity CLI :: --verbose r:r;z --verbose analyzelineNrbrcrrr test_verboses "rcCsn|tddttt|\}}ddl}|j | ks%JWddS1s0wYdS)z"Ensure version CLI :: -v r:zf2py -vrN) rBrCrDrOrPrhrGr\numpy __version__strip)rIrJr_r`nprrr test_versions  "rcCst|}|tdd|dt|j"ttd}t j |ddd}d}|j |ks1JWd d S1s 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)SrrrCrrpathlibr collectionsrplatformrOrnumpy.f2py.f2py2errGrrfixturer,r0r1r3r5r9rMrRrXrardrjmarkskipifsystem version_infororsxfailr~rrrrrrrrrrrrrrskiprrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrs(       '                  '