B td @sTdZddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z y ddl Z Wne k rtdZ YnXyddlZddZWn e k rddZdZYnXy 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)z*A decorator to skip tests requiring numpy.)funcrr_/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/joblib/test/common.py with_numpysrcCs dd}|S)z*A decorator to skip tests requiring numpy.cSs tddS)NzTest requires numpy)rrrrrmy_funcszwith_numpy..my_funcr)rr rrrrs) memory_usagecCs|S)z4A decorator to skip tests requiring memory_profiler.r)rrrrwith_memory_profiler*sr cOs*tt|||fdd}t|t|S)z)Compute memory usage when executing func.gMbP?)interval)gcZcollectr maxmin)rargskwargsZmem_userrr memory_used.srcCs dd}|S)z4A decorator to skip tests requiring memory_profiler.cSs tddS)NzTest requires memory_profiler.)rrrrr dummy_func7sz(with_memory_profiler..dummy_funcr)rrrrrr 5scCsHdtjksdtjkrdSt|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)pidrrrautokillRs    z setup_autokill..autokill) renvironsysargvteardown_autokill threadingTimer_KILLER_THREADSstart) module_nametimeoutrtrrrsetup_autokillDs  r+cCst|}|dk r|dS)z)Cancel a previously started killer threadN)r&getcancel)r(Zkillerrrrr#`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!r Zjoblib._multiprocessing_helpersrZjoblib.testingrrZlz4 ImportErrornumpynprZmemory_profilerr r rdictr&r+r#Zwith_multiprocessingpathexistsZ with_dev_shmZwith_lz4Z without_lz4rrrrsH