o lh @s`dZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z zddlZWn ey?dZYnwz ddlZddZWneyYddZdZYnwzdd lmZd d Zd d ZWney{dd ZdZZYnweZdddZddZe e duddZe ejd ddZe edupe ddZe eduddZ dS)z Small utilities for testing. N) PY3_OR_LATER)mp)SkipTestskipifcC|S)*A decorator to skip tests requiring numpy.funcrrD/var/www/html/env/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.rr rrr 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_func8rz(with_memory_profiler..dummy_funcr)r r rrr r6rcCsHdtjvs 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 autokillSs    z setup_autokill..autokill) r#environsysargvteardown_autokill threadingTimer_KILLER_THREADSstart) module_nametimeoutr-trrr setup_autokillEs  r9cCs"t|}|dur|dSdS)z)Cancel a previously started killer threadN)r4getcancel)r6killerrrr r1as  r1zNeeds 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__r2r'r)r#r/rjoblib._compatrjoblib._multiprocessing_helpersrjoblib.testingrrlz4 ImportErrornumpynpr memory_profilerrrrdictr4r9r1with_multiprocessingpathexists with_dev_shmwith_lz4 without_lz4rrrr s\