^Mh#/dZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z ddl Z ddlZ ddlZddlmZddlmZdZejeejekrdZn #e$rdZYnwxYwgdZdZejd pd Zd evrd Zd ejdvZGddeZ GddZ!GddZ"dZ#dZ$dZ%dZ&dZ'dZ(dS)z Generic test utilities. N)module_from_specspec_from_file_location)version)_pep440z3.0.8) PytestTestercheck_free_memory_TestPythranFuncIS_MUSLF HOST_GNU_TYPEmuslTeditableceZdZdZdS)FPUModeChangeWarningzWarning about FPU mode changeN)__name__ __module__ __qualname____doc__U/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/scipy/_lib/_testutils.pyrr3s''Drrc$eZdZdZdZ ddZdS) ra" Run tests for this namespace ``scipy.test()`` runs tests for all of SciPy, with the default settings. When used from a submodule (e.g., ``scipy.cluster.test()``, only the tests for that namespace are run. Parameters ---------- label : {'fast', 'full'}, optional Whether to run only the fast tests, or also those marked as slow. Default is 'fast'. verbose : int, optional Test output verbosity. Default is 1. extra_argv : list, optional Arguments to pass through to Pytest. doctests : bool, optional Whether to run doctests or not. Default is False. coverage : bool, optional Whether to run tests with code coverage measurements enabled. Default is False. tests : list of str, optional List of module names to run tests for. By default, uses the module from which the ``test`` function is called. parallel : int, optional Run tests in parallel with pytest-xdist, if number given is larger than 1. Default is 1. c||_dSN) module_name)selfrs r__init__zPytestTester.__init__Vs&rfastNFcddl}tj|j} tj| jd} ddg} |r| t|z } |r/t|dkr| ddt|dz zzgz } |r | d| zgz } |dkr| d d gz } n |d kr| d |gz } ||jg}|D|dkr>tr| d t|gz } nddl } | d d| dgt|zz } || } n#t$r}|j} Yd}~nd}~wwxYw| dkS)Nrz --showlocalsz --tb=shortr-vz--cov=rz-mznot slowfullz-nzMCould not run tests in parallel because pytest-xdist plugin is not available.) stacklevelz--pyargs)pytestsysmodulesrospathabspath__path__listint_pytest_has_xdiststrwarningswarnmain SystemExitcode)rlabelverbose extra_argvdoctestscoveragetestsparallelr&module module_path pytest_argsr1r5excs r__call__zPytestTester.__call__Ys T-.goofoa&899 %|4  , 4 ++ +K  8s7||a'' C#s7||A~"667 7K  4 H{23 3K F?? D*- -KK f__ D%= (K =%&E  HqLL "" ,c(mm44  F)*,,,  |d5kk11  ;;{++DD   8DDDDDD  sD22 E <EE )rrNFFNN)rrrrrrArrrrr8sK:'''KP6:******rrceZdZdZejejejejej ej gZ ej ej gZejejgZdZdZdZdZdZdZdZd S) r aL These are situations that can be tested in our pythran tests: - A function with multiple array arguments and then other positional and keyword arguments. - A function with array-like keywords (e.g. `def somefunc(x0, x1=None)`. Note: list/tuple input is not yet tested! `self.arguments`: A dictionary which key is the index of the argument, value is tuple(array value, all supported dtypes) `self.partialfunc`: A function used to freeze some non-array argument that of no interests in the original function c0i|_d|_d|_dSr) arguments partialfuncexpected)rs r setup_methodz_TestPythranFunc.setup_methods rctj|}i}|jD]'\}}|jtjjur |j||<(|Sr)inspect signature parametersitemsdefault Parameterempty)rfuncrJ optional_argskr"s rget_optional_argsz"_TestPythranFunc.get_optional_argssa%d++  (..00 - -DAqy 1 777#$9 a rcnd}|jD]*}t|j|d}||kr|}+|S)Nrr)rDlen)rmax_lenarg_idxcur_lens rget_max_dtype_list_lengthz*_TestPythranFunc.get_max_dtype_list_lengthsG~ " "G$.1!455G  !rcN|t|dz kr|dS||S)Nr)rU)r dtype_list dtype_idxs r get_dtypez_TestPythranFunc.get_dtypes/ s:q( ( (b> !i( (rc:t|D]x}g}|jD]b}||j|d|}||j|d|c|j|ydS)Nrr)rangerYrDr^appendastype pythranfunc)rtype_idx args_arrayrW new_dtypes rtest_all_dtypesz _TestPythranFunc.test_all_dtypessd<<>>?? * *HJ> P P NN4>'+B1+E+355 !!$."9!"<"C"CI"N"NOOOO D j ) ) )  * *rcg}|jD]:}||j|ddddddd;|j|dS)Nrr[)rDrarcrrerWs r test_viewsz_TestPythranFunc.test_viewssi ~ F FG   dnW5a82>tttD E E E E*%%%%rcg}|jD]F}|tj|j|ddddddG|j|dS)Nrr$)axis)rDranprepeatrcris r test_stridedz_TestPythranFunc.test_strideds ~ 9 9G   biw(?(B()333363Q38 9 9 9 9*%%%%rN)rrrrrmint8int16int32int64intcintp ALL_INTEGERfloat32float64 ALL_FLOAT complex64 complex128 ALL_COMPLEXrGrSrYr^rgrjrorrrr r s  7BHbh"'27KKRZ(I</K )))***&&& &&&&&rr c(ddlm}|dduS)zQ Check if the pytest-xdist plugin is installed, providing parallel tests r find_specxdistN)importlib.utilrr~s rr/r/s+ )((((( 9W  T ))rc\ddl} ttjd}d|tjd}n@#t $r3t }||d|d|dz d}YnwxYw||dzkr||dSdS) zJ Check *free_mb* of memory is available, otherwise do pytest.skip rNSCIPY_AVAILABLE_MEMz={} MB memory required, but environment SCIPY_AVAILABLE_MEM={}zlCould not determine available memory; set SCIPY_AVAILABLE_MEM variable to free memory in MB to run the test.z MB memory required, but .Az MB available)r& _parse_sizer)environformatKeyError_get_mem_availableskip)free_mbr&mem_freemsgs rrrsMMM Orz*?@AAMTT RZ 5688 OOO%''   KKI J J JNN8C<NNN O'C- C sAA :B B cdddddddddddddd d }tjd d ||tj}|r|d |vrtdt|d||d zS)Nrg?@@geAgmBg@g0AgAgpB)r brRMGTkbMbGbTbkibMibGibTibz^\s*(\d+)\s*({})\s*$|r$zInvalid size stringr) rematchrjoinkeysIgroup ValueErrorfloat)size_strsuffixesms rrrss4Cti ) UUH (//0I0IJJ  A 0 (**.///   x 3 33rc ddl}|jS#ttf$rYnwxYwt jdri}td5}|D]\}| }t|ddz||d d <] dddn #1swxYwYd|vr|dS|d |d zSdS) zD Get information about memory available, not counting swap. rNlinuxz /proc/meminforr: memavailablememfreecached) psutilvirtual_memory available ImportErrorAttributeErrorr'platform startswithopensplitrstriplower)rinfoflineps rrrsc  $$&&00  (      |w'' 4 / " " Ba B BJJLL05ad c0AQqTZZ__**,,-- B B B B B B B B B B B B B B B B T ! !' ' ?T(^3 3 4s33&A CCCc  ddl} tjddgn%#t$r|dYnwxYw|t t jz }tj |d}tj |||z ||z dz dz }|d z tj d t |d z }t|d 5}|d|dt jddddn #1swxYwYt jdkr+tjdddd|dt |g n(tjddd|t |g tjgd t'jd fd}|d|dfS)z Helper function to test building and importing Cython modules that make use of the Cython APIs for BLAS, LAPACK, optimize, and special. rNmesonz --versionzNo usable 'meson' foundrr;_cython_examplesbuildT)exist_okzinterpreter-native-file.iniwz [binaries] z python = ''win32setupz--buildtype=releasez --native-filez--vsenv)cwd)rcompilez-vv EXT_SUFFIXc|z }t||}t|}|j||Sr) with_suffixrrloader exec_module)modnamesospecmodsuffix target_dirs rloadz$_test_cython_extension..loadAsR7" / / 7 7&w33t$$ $$$ r extending extending_cpp)r& subprocess check_callFileNotFoundErrorrr0 threading get_identr)r*rshutilcopytreemakedirsrwriter' executabler sysconfigget_config_var) tmp_pathsrcdirr&mod_name build_dir native_filerrrrs @@r_test_cython_extensionrs MMM/w 45555 /// -...../#i133444Hw}}V$$Q'H OFHx/0008#g-0BBIW$JK T****i"??@@K k3  01  .S^...///000000000000000 |ww4. (#i..:#-       w. S^^M",    555:FFFF %l 3 3F 4  dd?33 33s!AA,9D11D58D5c(tj|fd}g}td|D]0}|tj||f|z|1|D]}||D]}|dS)a Run a given function concurrently across a given number of threads. This is equivalent to using a ThreadPoolExecutor, but using the threading primitives instead. This function ensures that the closure passed by parameter gets called concurrently by setting up a barrier before it gets called before any of the threads. Arguments --------- n_workers: int Number of concurrent threads to spawn. fn: callable Function closure to execute concurrently. Its first argument will be the thread id. *args: tuple Variable number of positional arguments to pass to the function. **kwargs: dict Keyword arguments to pass to the function. cH|g|Ri|dSr)wait)iargskwargsbarrierfns rclosurez(_run_concurrent_barrier..closurecs6  1tvrr)targetrrN)rBarrierr`raThreadstartr) n_workersrrrrworkersrworkerrs ` @r_run_concurrent_barrierrLs* **GG 1i ..y'V--- . . . .  r))rrIr)rrrr'rrrrrnumpyrmscipycythonCython.Compiler.Versionrcython_version scipy._librrequired_versionparseVersionr__all__r r_vr, IS_EDITABLERuntimeWarningrrr r/rrrrrrrrrse   DDDDDDDD  MMM #"""""w}^$$w7G'H'HHH FFF O N N  Yo..4" R<<GEN1--      >   KKKKKKKK\E&E&E&E&E&E&E&E&P**** 4 4 42545454p%%%%%s A++A54A5