o tBh @sNdZddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z zddl Z Wn e y9dZ Ynwz ddlZddZWne ySddZdZYnwzddlmZd d Zd d ZWne yud d ZdZZYnweZdddZddZe eduddZe ejd ddZe e duddZe e duddZdS)z Small utilities for testing. N)mp)SkipTestskipifcC|S)*A decorator to skip tests requiring numpy.funcrri/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/joblib/test/common.py with_numpyr cC dd}|S)rcStd)NzTest requires numpyrrrrr my_funczwith_numpy..my_funcr)r rrrr r ) memory_usagecCr)4A decorator to skip tests requiring memory_profiler.rrrrr with_memory_profiler*r rcOs*tt|||fdd}t|t|S)z)Compute memory usage when executing func.gMbP?)interval)gccollectrmaxmin)r argskwargsmem_userrr memory_used.srcCr )rcSr)NzTest requires memory_profiler.rrrrr dummy_func7rz(with_memory_profiler..dummy_funcr)r rrrr r5rcCsHdtjvs dtjvr dSt|dd}t||t|<}|dS)zTimeout based suiciding thread to kill the test runner process If some subprocess dies in an unexpected way we don't want the parent process to block indefinitely. NO_AUTOKILLz--pdbNcSsJt}td|t|tjtdtd|t|tjdS)Nz1Timeout exceeded: terminating stalled process: %dg?z-Timeout exceeded: killing stalled process: %d) osgetpidprintkillsignalSIGTERMtimesleepSIGKILL)pidrrr autokillRs    z setup_autokill..autokill) r"environsysargvteardown_autokill threadingTimer_KILLER_THREADSstart) module_nametimeoutr,trrr setup_autokillDs  r8cCs"t|}|dur|dSdS)z)Cancel a previously started killer threadN)r3getcancel)r5killerrrr r0`s  r0zNeeds multiprocessing to run.)reasonz/dev/shmz5This test requires a large /dev/shm shared memory fs.zNeeds lz4 compression to runz$Needs lz4 not being installed to run)r )__doc__r1r&r(r"r.rjoblib._multiprocessing_helpersrjoblib.testingrrlz4 ImportErrornumpynpr memory_profilerrrrdictr3r8r0with_multiprocessingpathexists with_dev_shmwith_lz4 without_lz4rrrr sV