Your IP : 216.73.216.213
���W � @ s| d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m
Z
d d l m Z y d d l
Z d d � Z Wn$ e k
r� d d � Z d Z Yn Xy, d d l m Z d
d � Z d d
� Z Wn( e k
r#d d � Z d Z Z Yn Xe � Z d d d � Z d d � Z d d � Z e e � Z d d � Z e e � Z d S)z
Small utilities for testing.
� N)�mp)�SkipTest)�
with_setupc C s | S)z*A decorator to skip tests requiring numpy.� )�funcr r �C/opt/alt/python35/lib/python3.5/site-packages/joblib/test/common.py�
with_numpy s r c C s d d � } | S)z*A decorator to skip tests requiring numpy.c S s t j d � � d S)NzTest requires numpy)�noser r r r r �my_func s zwith_numpy.<locals>.my_funcr )r r
r r r r s )� memory_usagec C s | S)z4A decorator to skip tests requiring memory_profiler.r )r r r r �with_memory_profiler% s r c O s9 t j � t | | | f d d �} t | � t | � S)z)Compute memory usage when executing func.Zintervalg����MbP?)�gcZcollectr �max�min)r �args�kwargsZmem_user r r � memory_used) s
r c C s d d � } | S)z4A decorator to skip tests requiring memory_profiler.c S s t j d � � d S)Nz Test requires memory_profiler.)r r r r r r �
dummy_func2 s z(with_memory_profiler.<locals>.dummy_funcr )r r r r r r 0 s � c C s` d t j k s d t j k r" d St | � d d � } t j | | � t | <} | j � d S)z�Timeout 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--pdbNc S s_ t j � } t d | � t j | t j � t j d � t d | � t j | t j � d S)Nz1Timeout exceeded: terminating stalled process: %dg �?z-Timeout exceeded: killing stalled process: %d) �os�getpid�print�kill�signal�SIGTERM�timeZsleep�SIGKILL)�pidr r r �autokillM s
z setup_autokill.<locals>.autokill) r �environ�sys�argv�teardown_autokill� threadingZTimer�_KILLER_THREADS�start)� module_nameZtimeoutr �tr r r �setup_autokill? s
r( c C s) t j | � } | d k r% | j � d S)z)Cancel a previously started killer threadN)r$ �getZcancel)r& Zkillerr r r r" [ s r" c C s t d k r t d � � d S)NzNeed multiprocessing to run)r r r r r r �check_multiprocessingb s r* c C s" t j j d � s t d � � d S)Nz/dev/shmz1This test requires the /dev/shm shared memory fs.)r �path�existsr r r r r �setup_if_has_dev_shmj s r- ) �__doc__r# r r r r r r
Z joblib._multiprocessing_helpersr r Z
nose.toolsr ZnumpyZnpr � ImportErrorZmemory_profilerr r r �dictr$ r( r" r* Zwith_multiprocessingr- Z with_dev_shmr r r r �<module> s<