gedZddlZddlZddlZddlZddlZgdZGddeZ GddZ Gdd Z d Z d Z d Zd ZdS)z Generic test utilities. N) PytestTestercheck_free_memory_TestPythranFuncceZdZdZdS)FPUModeChangeWarningzWarning about FPU mode changeN)__name__ __module__ __qualname____doc__5lib/python3.11/site-packages/scipy/_lib/_testutils.pyrrs''Dr rc$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__3s&r fastNFcddl}tj|j} tj| jd} ddg} |rtd|r| t|z } |r/t|dkr| ddt|dz zzgz } |r | d| zgz } |d kr| d d gz } n |d kr| d |gz } ||jg}|B|dkr !i( (r c:t|D]x}g}|jD]b}||j|d|}||j|d|c|j|ydS)Nrr)rangerOr:rTappendastype pythranfunc)rtype_idx args_arrayrM new_dtypes rtest_all_dtypesz _TestPythranFunc.test_all_dtypessd<<>>?? * *HJ> P P NN4>'+B1+E+355 !!$."9!"<"C"CI"N"NOOOO D j ) ) )  * *r cg}|jD]:}||j|ddddddd;|j|dS)NrrQ)r:rWrYrr[rMs r test_viewsz_TestPythranFunc.test_viewssi ~ F FG   dnW5a82>tttD E E E E*%%%%r cg}|jD]F}|tj|j|ddddddG|j|dS)Nr)axis)r:rWnprepeatrYr_s r test_stridedz_TestPythranFunc.test_strideds ~ 9 9G   biw(?(B()333363Q38 9 9 9 9*%%%%r N)rr r r rdint8int16int32int64intcintp ALL_INTEGERfloat32float64 ALL_FLOAT complex64 complex128 ALL_COMPLEXr=rIrOrTr]r`rfr r rrres  7BHbh"'27KKRZ(I</K )))***&&& &&&&&r rc(ddlm}|dduS)zQ Check if the pytest-xdist plugin is installed, providing parallel tests r find_specxdistN)importlib.utilrvrus rr%r%s+ )((((( 9W  T ))r cxddl} ttjd}d|tjd}nN#t $rAt }||dd||dz }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 {} MB available.A)r _parse_sizerenvironformatKeyError_get_mem_availableskip)free_mbrmem_freemsgs rrrsMMM #rz*?@AAMTT RZ 5688 ###%''   KKI J J J:AA Xc\## #'C- C sAA ABBcdddddddddddddd d }tjd d ||tj}|r|d |vrtdt|d||d zS)Nr{g?@@geAgmBg@g0AgAgpB)brHMGTkbMbGbTbkibMibGibTibz^\s*(\d+)\s*({})\s*$|rbzInvalid size stringr) rematchr~joinkeysIgroupr"float)size_strsuffixesms rr|r|ss4Cti ) UUH (//0I0IJJ  A 0 (**.///   x 3 33r c 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 ImportErrorAttributeErrorrplatform 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 CCC)r rrrnumpyrdr?__all__RuntimeWarningrrrr%rr|rr r rrs!   D C C     >   MMMMMMMM`E&E&E&E&E&E&E&E&P***, 4 4 4r