U .d @sTdZddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z z ddl Z Wne k rtdZ YnXzddlZddZWn e k rddZdZYnXz ddlmZd d Zd d ZWn$e k rd d ZdZZYnXeZdddZddZe edkddZe ejd ddZe e dkddZe e dk ddZdS)z Small utilities for testing. N)mp)SkipTestskipifcCs|S)*A decorator to skip tests requiring numpy.funcrr6/tmp/pip-unpacked-wheel-ho8k9mhl/joblib/test/common.py with_numpysr cCs dd}|S)rcSs tddS)NzTest requires numpyrrrrr my_funcszwith_numpy..my_funcr)rr rrr r s) memory_usagecCs|S)4A decorator to skip tests requiring memory_profiler.rrrrr with_memory_profiler*srcOs*tt|||fdd}t|t|S)z)Compute memory usage when executing func.gMbP?)interval)gcZcollectr maxmin)rargskwargsZmem_userrr memory_used.srcCs dd}|S)rcSs tddS)NzTest requires memory_profiler.r rrrr dummy_func7sz(with_memory_profiler..dummy_funcr)rrrrr r5scCsHdtjksdtjkrdSt|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. Z 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) renvironsysargvteardown_autokill threadingTimer_KILLER_THREADSstart) module_nametimeoutr#trrr setup_autokillDs  r/cCst|}|dk r|dS)z)Cancel a previously started killer threadN)r*getcancel)r,Zkillerrrr r'`s r'zNeeds 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__r(rrrr%rZjoblib._multiprocessing_helpersrZjoblib.testingrrZlz4 ImportErrorZnumpynpr Zmemory_profilerr rrdictr*r/r'Zwith_multiprocessingpathexistsZ with_dev_shmZwith_lz4Z without_lz4rrrr sR