B 0d@sfdZddlZddlZddlZddgZGdddeZGdddZdd Zd dZ d d Z d dZ dS)z Generic test utilities. N PytestTestercheck_free_memoryc@seZdZdZdS)FPUModeChangeWarningzWarning about FPU mode changeN)__name__ __module__ __qualname____doc__r r G/var/www/html/venv/lib/python3.7/site-packages/scipy/_lib/_testutils.pyrsrc@s"eZdZdZddZd dd ZdS) rz) Pytest test runner entry point. cCs ||_dS)N) module_name)selfr r r r __init__szPytestTester.__init__fastNFc CsNddl}tj|j} tj| jd} ddg} |r:td|rJ| t |7} |rtt |dkrt| ddt |dg7} |r| d| g7} |d kr| d d g7} n|d kr| d |g7} |dkr|jg}|dk r|dkrt r| d t |g7} nddl } | d| dgt |7} y|| } Wn*tk rD}z |j} Wdd}~XYnX| dkS)Nrz --showlocalsz --tb=shortzDoctests not supportedr-vz--cov=rz-mznot slowfullz-nzMCould not run tests in parallel because pytest-xdist plugin is not available.z--pyargs)pytestsysmodulesr ospathabspath__path__ ValueErrorlistint_pytest_has_xdiststrwarningswarnmain SystemExitcode)r labelverboseZ extra_argvZdoctestsZcoveragetestsparallelrmodule module_pathZ pytest_argsrr#excr r r __call__s:    zPytestTester.__call__)rrNFFNN)rrrrr r+r r r r rscCsddlm}|ddk S)zQ Check if the pytest-xdist plugin is installed, providing parallel tests r) find_specZxdistN)importlib.utilr,)r,r r r rJs rcCsddl}y$ttjd}d|tjd}Wn<tk rht}|dkrT|dd||d}YnX||dkr||dS)zJ Check *free_mb* of memory is available, otherwise do pytest.skip rNZSCIPY_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 availableg.A)r _parse_sizerenvironformatKeyError_get_mem_availableskip)Zfree_mbrZmem_freemsgr r r rSs  cCsxdddddddddddddd d }td d ||tj}|rT|d |kr\tdt|d||d S)Ng.Ag?g@@geAgmBg@g0AgAgpB)bkMGTkbZMbZGbZTbZkibZMibZGibZTibz^\s*(\d+)\s*({0})\s*$|zInvalid size stringr) rematchr0joinkeysIgrouprfloat)Zsize_strsuffixesmr r r r.isr.c Csyddl}|jSttfk r*YnXtjdri}tdd>}x6|D].}| }t |dd||d d <qNWWdQRXd |kr|d S|d |d SdS) zD Get information about memory available, not counting swap. rNlinuxz /proc/meminforrg@@:Z memavailableZmemfreecached) psutilZvirtual_memory available ImportErrorAttributeErrorrplatform startswithopensplitrDstriplower)rKinfoflinepr r r r2xs    0r2) rrr>r__all__RuntimeWarningrrrrr.r2r r r r s7