hdZddlZddlZddlZddlZddlZgdZGddeZ 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__Q/var/www/html/Sam_Eipo/venv/lib/python3.11/site-packages/scipy/_lib/_testutils.pyrrs''Dr rc$eZdZdZdZ ddZdS) rz) Pytest test runner entry point. c||_dSN) module_name)selfrs r__init__zPytestTester.__init__s&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_dtypes}sd<<>>?? * *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 rrrLs  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?{0} MB memory required, but environment SCIPY_AVAILABLE_MEM={1}zlCould not determine available memory; set SCIPY_AVAILABLE_MEM variable to free memory in MB to run the test.z,{0} MB memory required, but {1} MB available.A)r _parse_sizerenvironformatKeyError_get_mem_availableskip)free_mbrmem_freemsgs rrrsMMM #rz*?@AAOVV RZ 5688 ###%''   KKI J J J<CC Xc\## #'C- C sAA ABBcdddddddddddddd d }tjd d ||tj}|r|d |vrtdt|d||d zS)Nr{g?@@geAgmBg@g0AgAgpB)brHMGTkbMbGbTbkibMibGibTibz^\s*(\d+)\s*({0})\s*$|rbzInvalid size stringr) rematchr~joinkeysIgroupr"float)size_strsuffixesms rr|r|ss4Cti ) UUH )00(--//1J1JKK  A 0 (**.///   x 3 33r c ddl}|jS#ttf$rYnwxYwt jdri}tdd5}|D]\}| }t|ddz||d d <] dddn #1swxYwYd |vr|d S|d |d zSdS) zD Get information about memory available, not counting swap. rNlinuxz /proc/meminforrr: memavailablememfreecached) psutilvirtual_memory available ImportErrorAttributeErrorrplatform startswithopensplitrstriplower)rinfoflineps rrrse  $$&&00  (      |w'' 4 /3 ' ' B1 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     >   44444444nE&E&E&E&E&E&E&E&P***, 4 4 4r