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.�intervalg����MbP?)�gc�collectr �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�time�sleep�SIGKILL)�pidr r r �autokillM s
z setup_autokill.<locals>.autokill) r �environ�sys�argv�teardown_autokill� threading�Timer�_KILLER_THREADS�start)� module_name�timeoutr! �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( �get�cancel)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 r0 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 r3 ) �__doc__r& r r r r r# r Z joblib._multiprocessing_helpersr r �
nose.toolsr �numpy�npr � ImportErrorZmemory_profilerr r r �dictr( r- r% r0 Zwith_multiprocessingr3 Z with_dev_shmr r r r �<module> s<