o `^h#/@sPdZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z ddl Z ddlZz ddlZddlmZWn eyIdZYnwddlmZdZeeeekr^dZgdZdZed pjd Zd evrqd Zd ejdvZGdddeZ GdddZ!GdddZ"ddZ#ddZ$ddZ%ddZ&ddZ'ddZ(dS) z Generic test utilities. N)module_from_specspec_from_file_location)version)_pep440z3.0.8) PytestTestercheck_free_memory_TestPythranFuncIS_MUSLF HOST_GNU_TYPEmuslTeditablec@seZdZdZdS)FPUModeChangeWarningzWarning about FPU mode changeN)__name__ __module__ __qualname____doc__rrS/home/air/shanriGPT/back/venv/lib/python3.10/site-packages/scipy/_lib/_testutils.pyr3src@s&eZdZdZddZ  d dd ZdS) 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. cCs ||_dSN) module_name)selfrrrr__init__Vs zPytestTester.__init__fastNFc CsJddl}tj|j} tj| jd} ddg} |r| t|7} |r4t |dkr4| ddt |dg7} |r=| d| g7} |dkrH| d d g7} n |d krR| d |g7} |durZ|jg}|dury|dkryt rn| d t |g7} n ddl } | j d dd| dgt|7} z || } W| dkSty}z |j} WYd}~| dkSd}~ww)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_argsr+r/excrrr__call__Ys@    zPytestTester.__call__)rrNFFNN)rrrrrr;rrrrr8s rc@s|eZdZdZejejejejej ej gZ ej ej gZejejgZddZddZddZdd Zd d Zd d ZddZdS)raL 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 cCsi|_d|_d|_dSr) arguments partialfuncexpected)rrrr setup_methods z_TestPythranFunc.setup_methodcCs>t|}i}|jD]\}}|jtjjur|j||<q |Sr)inspect signature parametersitemsdefault Parameterempty)rfuncrA optional_argskrrrrget_optional_argss  z"_TestPythranFunc.get_optional_argscCs2d}|jD]}t|j|d}||kr|}q|S)Nrr)r<len)rmax_lenarg_idxcur_lenrrrget_max_dtype_list_lengths z*_TestPythranFunc.get_max_dtype_list_lengthcCs |t|dkr |dS||S)Nr)rK)r dtype_list dtype_idxrrr get_dtypesz_TestPythranFunc.get_dtypecCs`t|D]'}g}|jD]}||j|d|}||j|d|q |j|qdS)Nrr)rangerOr<rSappendastype pythranfunc)rtype_idx args_arrayrM new_dtyperrrtest_all_dtypess  z _TestPythranFunc.test_all_dtypescCsFg}|jD]}||j|dddddddq|j|dS)NrrP)r<rUrWrrYrMrrr test_viewss *z_TestPythranFunc.test_viewscCsHg}|jD]}|tj|j|dddddddq|j|dS)Nrr)axis)r<rUnprepeatrWr\rrr test_strideds z_TestPythranFunc.test_stridedN)rrrrr_int8int16int32int64intcintp ALL_INTEGERfloat32float64 ALL_FLOAT complex64 complex128 ALL_COMPLEXr?rJrOrSr[r]rarrrrrs     rcCsddlm}|dduS)zQ Check if the pytest-xdist plugin is installed, providing parallel tests r find_specxdistN)importlib.utilrprorrrr)s  r)cCsddl}zttjd}d|tjd}Wnty5t}|dur)|d|d|dd}Ynw||dkrC||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_freemsgrrrrs    rcCsxdddddddddddddd d }td d ||tj}|r*|d |vr.tdt|d||d S)Nrtg?@@geAgmBg@g0AgAgpB)r brIMGTkbMbGbTbkibMibGibTibz^\s*(\d+)\s*({})\s*$|rzInvalid size stringr) rematchrwjoinkeysIgroup ValueErrorfloat)size_strsuffixesmrrrrusruc Csz ddl}|jWSttfyYnwtjdr\i}td"}|D]}| }t |dd||d d <q%Wdn1sGwYd|vrT|dS|d |d SdS) zD Get information about memory available, not counting swap. rNlinuxz /proc/meminforr~: memavailablememfreecached) psutilvirtual_memory available ImportErrorAttributeErrorr!platform startswithopensplitrstriplower)rinfoflineprrrrys$   $ryc sdddl}z tddgWnty|dYnw|tt}tj |d}t |||||dd}|d tj d d t|d }t|d }|d|dtjdWdn1slwYtjdkrtjdddd|dt|gdntjddd|t|gdtjgddtdfdd}|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' foundrr5_cython_examplesbuildT)exist_okzinterpreter-native-file.iniwz [binaries] z python = ''win32setupz--buildtype=releasez --native-filez--vsenv)cwd)rcompilez-vv EXT_SUFFIXcs0|}t||}t|}|j||Sr) with_suffixrrloader exec_module)modnamesospecmodsuffix target_dirrrloadAs   z$_test_cython_extension..load extending extending_cpp)r subprocess check_callFileNotFoundErrorrzr* threading get_identr#r$rshutilcopytreemakedirsrwriter! executabler sysconfigget_config_var)tmp_pathsrcdirr mod_name build_dir native_filerrrrr_test_cython_extensionsD       rcspt|fdd}g}td|D]}|tj||f||dq|D]}|q&|D]}|q/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. cs"|g|Ri|dSr)wait)iargskwargsbarrierfnrrclosurecsz(_run_concurrent_barrier..closurer)targetrrN)rBarrierrTrUThreadstartr) n_workersrrrrworkersrworkerrrr_run_concurrent_barrierLs     r))rr@r#rrrr!rrrrrrnumpyr_scipycythonCython.Compiler.Versionrcython_versionr scipy._librrequired_versionparseVersion__all__r r_vr& IS_EDITABLERuntimeWarningrrrr)rruryrrrrrrsJ  NH  8