o Tjl@s"ddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl Z zddlZWn eyGdZYnwd@ddZddZe jje dde jd d gd d Zd dZddZddZe jjeduddddZdddDZejdkredddD7ZgdZdddDZejdkreddd D7Zd!d"Zejdkrd#gZngZe j d$d%d&Z!gd'Z"ejdkre"gd(7Z"d)d*Z#d+d,Z$e jj%e&d-d.vd/dd0d1Z'd2d3Z(e j d4d5d6Z)d7d8Z*d9d:Z+d;e,fdd?Z.dS)AN)IS_WASMcCsj|dur|j}i}t|D]%}|dkrq t||}t|dr2t|dr2|j|kr2|jd|j||<q |S)zEReturns a mapping of all objects with the wrong __module__ attribute.Ncore __module____name__.)rdirgetattrhasattrr)module module_nameresultsnameitemr^/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/numpy/tests/test_public_api.py check_dirs   rcCs ddi}tt}||ksJdS)Nrecarrayznumpy.rec.recarray)rnp) allowlist bad_resultsrrrtest_numpy_namespace&srzcan't start subprocess)reasonr testingcCs6tjdd|f}t|}|rJ|ttvsJdS)aMake sure we can actually use the modules we lazy load. While not exported as part of the public API, it was accessible. With the use of __getattr__ and __dir__, this isn't always true It can happen that an infinite recursion may happen. This is the only way I found that would force the failure to appear on the badly implemented code. We also test for the presence of the lazily imported modules in dir z-czimport numpy; numpy.N)sys executable subprocess check_outputrr)r exeresultrrrtest_import_lazy_import1s rcCs$tttttttksJdS)zWAssert that output of dir has only one "testing/tester" attribute without duplicateN)lenrrsetrrrrtest_dir_testingHs$r"cCttj}|iks JdSN)rrlinalgrrrrtest_numpy_linalgN r'cCr#r$)rrfftr&rrrtest_numpy_fftSr(r*z#ctypes not available in this pythoncCs0ttjjj}t|dd}|dusJddS)Ntest_not_exportedzG'test_not_exported' is mistakenly exported, NPY_NO_EXPORT does not work)ctypesCDLLr_core_multiarray_tests__file__r)cdllfrrrtest_NPY_NO_EXPORTXs r3cCg|]}d|qSnumpy.r.0srrr nr:) ctypeslibdtypes exceptionsf2pyr)libzlib.array_utilsz lib.formatzlib.introspectz lib.mixinsz lib.npyiozlib.recfunctionsz lib.scimathzlib.stride_tricksr%maz ma.extrasz ma.mrecords polynomialzpolynomial.chebyshevzpolynomial.hermitezpolynomial.hermite_ezpolynomial.laguerrezpolynomial.legendrezpolynomial.polynomialrandomstringsrztesting.overridestypingztyping.mypy_pluginversion) cCr4r5rr7rrrr:) distutilszdistutils.cpuinfozdistutils.exec_commandzdistutils.misc_utilz distutils.logzdistutils.system_info) numpy.charz numpy.emath numpy.reccCr4r5rr7rrrr:r;),compatz compat.py3kconftestrzcore.multiarrayz core.numericz core.umathzcore.arrayprintzcore.defchararrayzcore.einsumfunczcore.fromnumericzcore.function_basezcore.getlimitszcore.numerictypeszcore.overridesz core.recordszcore.shape_basez f2py.auxfuncszf2py.capi_mapsz f2py.cb_rulesz f2py.cfuncszf2py.common_ruleszf2py.crackfortranz f2py.diagnosez f2py.f2py2ezf2py.f90mod_ruleszf2py.func2subrz f2py.rulesz f2py.symboliczf2py.use_rulesz fft.helperzlib.user_arrayzlinalg.lapack_litez linalg.linalgzma.corez ma.testutilszma.timer_comparisonmatlib matrixlibzmatrixlib.defmatrixzpolynomial.polyutilsz random.mtrandzrandom.bit_generatorztesting.print_coercion_tablescCr4r5rr7rrrr:rI)7zdistutils.armccompilerzdistutils.fujitsuccompilerzdistutils.ccompilerzdistutils.ccompiler_optzdistutils.commandzdistutils.command.autodistzdistutils.command.bdist_rpmzdistutils.command.buildzdistutils.command.build_clibzdistutils.command.build_extzdistutils.command.build_pyzdistutils.command.build_scriptszdistutils.command.build_srczdistutils.command.configz!distutils.command.config_compilerzdistutils.command.developzdistutils.command.egg_infozdistutils.command.installzdistutils.command.install_clibzdistutils.command.install_dataz!distutils.command.install_headerszdistutils.command.sdistzdistutils.conv_templatezdistutils.corezdistutils.extensionzdistutils.fcompilerzdistutils.fcompiler.absoftzdistutils.fcompiler.armzdistutils.fcompiler.compaqzdistutils.fcompiler.environmentzdistutils.fcompiler.g95zdistutils.fcompiler.gnuzdistutils.fcompiler.hpuxzdistutils.fcompiler.ibmzdistutils.fcompiler.intelzdistutils.fcompiler.laheyzdistutils.fcompiler.mipszdistutils.fcompiler.nagzdistutils.fcompiler.nonezdistutils.fcompiler.pathf95zdistutils.fcompiler.pgzdistutils.fcompiler.nvzdistutils.fcompiler.sunzdistutils.fcompiler.vastzdistutils.fcompiler.fujitsuzdistutils.from_templatezdistutils.intelccompilerzdistutils.lib2defzdistutils.line_endingszdistutils.mingw32ccompilerzdistutils.msvccompilerzdistutils.npy_pkg_configzdistutils.numpy_distributionzdistutils.pathccompilerzdistutils.unixccompilercCsDd|vs d|vs d|vrdS|tvrdS|tvrdS|tvr dSdS)z%Check if this needs to be considered.z._z.testsz.setupFT)PUBLIC_MODULESPUBLIC_ALIASED_MODULESPRIVATE_BUT_PRESENT_MODULESr rrr is_unexpectedsrUznumpy.distutils.msvc9compilerz'ignore:.*np.compat.*:DeprecationWarningcCsXg}tjtjtjdddD]\}}}t|r |tvr ||q|r*td|dS)z| Test that we don't add anything that looks like a new public module by accident. Check is based on filenames. rN)pathprefixonerrorzFound unexpected modules: ) pkgutil walk_packagesr__path__rrU SKIP_LISTappendAssertionError)modnames_modnameispkgrrrtest_all_modules_are_expected)s rc) znumpy.lib.mathznumpy.matlib.charznumpy.matlib.recznumpy.matlib.emathznumpy.matlib.exceptionsznumpy.matlib.mathznumpy.matlib.linalgznumpy.matlib.fftznumpy.matlib.randomznumpy.matlib.ctypeslibznumpy.matlib.ma)znumpy.distutils.log.sysznumpy.distutils.log.loggingznumpy.distutils.log.warningscCs>dd}|d}tD] }|||q |rtd|dS)a Method checking all objects. The pkgutil-based method in `test_all_modules_are_expected` does not catch imports into a namespace, only filenames. So this test is more thorough, and checks this like: import .lib.scimath as emath To check if something in a module is (effectively) public, one can check if there's anything in that namespace that's a public function/object but is not exposed in a higher-level namespace. For example for a `numpy.lib` submodule:: mod = np.lib.mixins for obj in mod.__all__: if obj in np.__all__: continue elif obj in np.lib.__all__: continue else: print(obj) cSsxg}t|}t|dr|j}nt|}|D]#}|ds9|d|}tt||tj r9t |r9|t vr9| |q|S)N__all__r`r) importlib import_moduler rdr startswith isinstancertypes ModuleTyperU SKIP_LIST_2r])mod_namemembersr objnamesobjname fullobjnamerrrfind_unexpected_membersns     z@test_all_modules_are_expected_2..find_unexpected_membersnumpyz5Found unexpected object(s) that look like modules: {}N)rQextendr^format)rqunexpected_membersrarrrtest_all_modules_are_expected_2Usrvc Csdd}g}tD] }||s||q|rtd|tD]}zt|Wqty5||Yqw|r?td|tjdd$}tj dt dtj dt dt D] }||sa||qVWd n1slwY|rztd |d S) a  Check that all submodules listed higher up in this file can be imported Note that if a PRIVATE_BUT_PRESENT_MODULES entry goes missing, it may simply need to be removed from the list (deprecation may or may not be needed - apply common sense). c Ss*zt|WdSttfyYdSw)NFT)rerf ImportErrorAttributeError)r rrrcheck_importables  z-test_api_importable..check_importablez5Modules in the public API that cannot be imported: {}z1Modules in the public API that were not found: {}T)recordalways)categoryNzPModules that are not really public but looked public and can not be imported: {}) rQr]r^rtrRevalrxwarningscatch_warningsfilterwarningsDeprecationWarning ImportWarningrS)ry module_namesr wrrrtest_api_importablesD    rPy_DEBUG)Nr0zNumPy possibly built with `USE_DEBUG=True ./tools/travis-test.sh`, which does not expose the `array_api` entry point. See https://github.com/numpy/numpy/pull/19800cCstdtjv}tj}z|jdd}Wnty$| dg}Ynwt |dkr6|r4d}t |ddSz t dd|D}Wnt yT|rQd }t |dYdSw|jd kr\dS|}d |jd }|tusoJ|dS) zx Entry point for Array API implementation can be found with importlib and returns the main numpy namespace. platlib array_api)grouprz%No entry points for 'array_api' foundNcss|] }|jdkr|VqdS)rrNrT)r8eprrr sz-test_array_api_entry_point..z%'numpy' not in array_api entry pointsznumpy.array_apiznumpy entry point value 'z0' does not point to our Array API implementation) sysconfigget_pathrr0remetadata entry_pointsselectrxgetr r^next StopIterationvalueloadrr)numpy_in_sitepackagesepsxp_epsmsgrxprrrtest_array_api_entry_points4        rcCsVdd}dd}|tj}||}|t}||}||ks)Jd||dS)z Checks if `dir(np)` and `np.__all__` are consistent and return the same content, excluding exceptions and private members. cSsdd|DS)NcSsh|] }|ds|qS)r`)rg)r8mrrr szYtest_main_namespace_all_dir_coherence.._remove_private_members..r member_setrrr_remove_private_membersszFtest_main_namespace_all_dir_coherence.._remove_private_memberscSs |dhS)Nbool) differencerrrr_remove_exceptionsszAtest_main_namespace_all_dir_coherence.._remove_exceptionszMembers that break symmetry: N)rrd__dir__symmetric_difference)rr all_members dir_membersrrr%test_main_namespace_all_dir_coherences   rz:ignore:numpy.core(\.\w+)? is deprecated:DeprecationWarningcCsddlm}ttjD]Q}|ds|dvsd|tvrq ttj|}t |rS|}|}t|D] }|dr9q1t||}t d||gd}|t||usQJq1q |t||us\Jq dS) zp Check that all "semi-public" members of `numpy._core` are also accessible from `numpy.core` shims. rNr`)testsrDr6__z numpy.core.)fromlist) numpy.corerrrr.rgrRrinspectismodule __import__)r member_namemember submodulesubmodule_namesubmodule_member_namesubmodule_membercore_submodulerrrtest_core_shims_coherence s2       rcCsddlm}m}m}m}m}m}ddlm}t h}t }t } g} t g} t | dkr| } t| D]~} t| | }t|r`d|jvr`| ds`d|jvr`| dvr`||vr`| |||q2t|smt||t jfr||vr|jdvr|| jdkr|q2|jd kr| jd rq2|jd vr| jd krq2| jd krq2| |j| j| |fq2||| j| |<q2t | dks*~~~ t | dksJ| dS)z Check that each public function is available from one location only. Test performs BFS search traversing NumPy's public API. It flags any function-like object that is accessible from more that one place. r)AnyCallableDictListSetTuple)_ArrayFunctionDispatcherrrr` numpy._core)r?rArr)absolutearccosarccosharcsinarcsinharctanarctan2arctanh left_shift right_shift conjugateinvert remainderdivide concatenatepower transposetrimcoefznumpy.polynomialaddequal not_equalgreater greater_equalless less_equal numpy.stringsrKN)rErrrrrrnumpy._core._multiarray_umathrrr!dictr poprrrrrrgr]r isfunctionrhufunc)rrrrrrdispatched_functionvisited_modulesvisited_functionsfunctions_original_pathsduplicated_functions modules_queuer rrrrrtest_functions_single_location:sd                   VrcCs`tg}th}t}g}t|dkr|}t|D]}t||}t|rGd|jvrG| dsGd|jvrG|dvrG||vrG| || |qt|st |dr|j ds|j |jkr||vr|jdvrl|jdkrlq|jd krv|jdks|jd kr|jd krq|jd vrq| t|j|j |jd | |qt|dks|rt|dksJ|dSdS)Nrrrr`r) charrr<r?rA lapack_litemrecordsrrrBrEmtrand bit_generatorrrrrrzrL) BitGenerator GeneratorMT19937PCG64 PCG64DXSMPhilox RandomStateSFC64 SeedSequence)Funcactualexpected)rr!r rrrrrrrgr]rr rrrrrincorrect_entriesr rrrrrtest___module___attributes`              9rreturncCsb|j}|j}|j}||ddksJtj|}tt|d|}||up0| do0|j|kS)Nrz numpy.random) __qualname__rrsplitrmodules functoolsreducerrg)objqualnamer r r actual_objrrr_check___qualname__s  rcCstg}th}t}g}t|dkr{|}t|D][}t||}t|rGd|jvrG| dsG|dvrGd|jvrG||vrG| || |qt|stt |drt|j dst| dstt |st||vrt| t|j|jd| |qt|dks|rt|dksJ|dSdS)Nrrrr`)r?rArrrErr<rrr)rr)rr!r rrrrrrrgr]rr rrrrrrrtest___qualname___attributesR          !rr$)/rrrrrYrirerr~rrr numpy.testingrpytestr,rwrrmarkskipif parametrizerr"r'r*r3rQ version_inforRrSrUr\rrcrkrvrxfailget_config_varrrrrrrrrrrrrs         "  .=    40  1 ,sC