Your IP : 216.73.216.85
��Yf� � - @ sJ
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 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 Z d d l Z d d l Z d d l Z e j j d � Z e j j d � 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 y d d l m Z e j! Z" Wn e# k
r�d Z" Yn Xy d d l$ m% Z% m& Z& d Z' Wn e# k
r�d Z' Yn Xy d d l( Z( Wn e# k
r�d Z( Yn Xd d
� Z) e j* Z+ d Z, d Z- e- r>d d
d f \ Z. Z/ Z0 n d d d f \ Z. Z/ Z0 e1 e d d � Z2 e j3 d k Z4 d d l m5 Z5 d d � Z6 y e j7 d � Z8 Wn d Z8 Yn Xd d g Z9 y d d l: m; Z; m< Z< m= Z= Wn" e# k
re> Z; d Z< Z= Yn Xd d � Z? Gd d � d e> � Z@ Gd d � d e> � ZA d d � ZB Gd! d" � d" � ZC Gd# d$ � d$ eA � ZD Gd% d& � d& e jE � ZF Gd' d( � d( eA � ZG d) d* � ZH d+ d, � ZI Gd- d. � d. eA � ZJ Gd/ d0 � d0 eA � ZK Gd1 d2 � d2 eA � ZL Gd3 d4 � d4 eA � ZM Gd5 d6 � d6 eA � ZN Gd7 d8 � d8 e> � ZO d9 d: � ZP Gd; d< � d< e> � ZQ Gd= d> � d> e> � ZR Gd? d@ � d@ eA � ZS GdA dB � dB eA � ZT GdC dD � dD eA � ZU GdE dF � dF eA � ZV dG dH dI � ZW dJ dK � ZX dL dM � ZY GdN dO � dO e> � ZZ GdP dQ � dQ e[ � Z\ dR dS � Z] GdT dU � dU eA � Z^ dV dW � Z_ dX dY � Z` GdZ d[ � d[ eA � Za Gd\ d] � d] eA � Zb d d^ l mc Zc md Zd me Ze Gd_ d` � d` e> � Zf da db � Zg Gdc dd � dd ed � Zh Gde df � df ec � Zi ei jj dg dh ef �ei jj di dh ef dj dk dl f �ei jj db dh eg dm eh �Gdn do � do eA � Zk e jl � Zm dp dq � Zn Gdr ds � ds ec � Zo eo jj dq dh en �Gdt du � du ec � Zp ep jj dq � dv Zq Gdw dx � dx eA � Zr Gdy dz � dz eA � Zs e) d{ � Zt Gd| d} � d} eA � Zu Gd~ d � d eA � Zv Gd� d� � d� eA � Zw Gd� d� � d� eA � Zx e jy e" d� � Gd� d� � d� eA � � Zz Gd� d� � d� eA � Z{ Gd� d� � d� e; � Z| Gd� d� � d� eA � Z} Gd� d� � d� eA � Z~ Gd� d� � d� e j � Z� Gd� d� � d� eA � Z� Gd� d� � d� eA � Z� Gd� d� � d� e j � Z� Gd� d� � d� e j � Z� d� d� � Z� Gd� d� � d� e j � Z� d� d� � Z� d� d� � Z� d� d� � Z� d� d� � Z� Gd� d� � d� e> � Z� Gd� d� � d� e j � Z� Gd� d� � d� e j � Z� Gd� d� � d� e j � Z� Gd� d� � d� e j � Z� Gd� d� � d� e j � Z� Gd� d� � d� e j � Z� Gd� d� � d� e j � Z� Gd� d� � d� e j � Z� Gd� d� � d� e j � Z� Gd� d� � d� e j � Z� e j� e j3 d k d� � Gd� d� � d� e j � � Z� Gd� d� � d� e j � Z� Gd� d� � d� e> � Z� Gd� d� � d� e> � Z� Gd� d� � d� e> � Z� d� d� � Z� d S)�� N�_multiprocessingzmultiprocessing.synchronize)�util)� reductionF)�Value�copyTc C s
| j d � S)N�latin)�encode)�s� r
�?/opt/alt/python35/lib64/python3.5/test/_test_multiprocessing.pyr @ s r g�������?g=
ףp=�?gffffff�?gffffff�?�HAVE_BROKEN_SEM_GETVALUE�win32)�waitc C s. | d k r | d k r d } t | g | � S)Ng )r )Zhandle�timeoutr
r
r �wait_for_handle[ s r � SC_OPEN_MAX� �__main__z$test.test_multiprocessing_forkserver)� Structure�c_int�c_doublec C sk d } y t j d � } Wn t t f k
r7 d SYn X| d k sP | | k rT d St j d | � � d S)zACheck that the system supports enough semaphores to run the test.r �SC_SEM_NSEMS_MAXN� zHThe OS doesn't support enough semaphores to run the test (required: %d).���)�os�sysconf�AttributeError�
ValueError�unittest�SkipTest)Z nsems_minZnsemsr
r
r �check_enough_semaphoress s r c @ s( e Z d Z d d � Z d d � Z d S)�
TimingWrapperc C s | | _ d | _ d S)N)�func�elapsed)�selfr" r
r
r �__init__� s zTimingWrapper.__init__c O s; t j � } z | j | | � SWd t j � | | _ Xd S)N)�timer" r# )r$ �args�kwds�tr
r
r �__call__� s zTimingWrapper.__call__N)�__name__�
__module__� __qualname__r% r* r
r
r
r r! � s r! c @ s@ e Z d Z d Z d d � Z d d � Z d d � Z e Z d
S) � BaseTestCase� processes�manager�threadsc C s t r | j | | d � d S)Nr )�
CHECK_TIMINGS�assertAlmostEqual)r$ �a�br
r
r �assertTimingAlmostEqual� s z$BaseTestCase.assertTimingAlmostEqualc G s9 y | | � } Wn t k
r$ Yn X| j | | � Sd S)N)�NotImplementedError� assertEqual)r$ �valuer" r' �resr
r
r �assertReturnsIfImplemented� s
z'BaseTestCase.assertReturnsIfImplementedc G s t d � � d S)Nz#shouldn't try to pickle a test case)r7 )r$ r' r
r
r �
__reduce__� s zBaseTestCase.__reduce__N)r/ r0 r1 )r+ r, r- �
ALLOWED_TYPESr6 r; r< �
__reduce_ex__r
r
r
r r. � s
r. c C sm y | j � SWnX t k
rh y | j SWn8 t k
rc y | j SWn t k
r^ t � Yn XYn XYn Xd S)N)� get_valuer Z_Semaphore__valueZ_valuer7 )r$ r
r
r r? � s
r? c @ s e Z d Z d d � Z d S)�
DummyCallablec C s& t | t � s t � | j d � d S)N� )�
isinstancer@ �AssertionError�put)r$ �q�cr
r
r r* � s zDummyCallable.__call__N)r+ r, r- r* r
r
r
r r@ � s r@ c @ s� e Z d Z d Z d d � Z d d � Z e d d � � Z d d
� Z e d d � � Z d
d � Z
d d � Z d d � Z e d d � � Z
d d � Z e d d � � Z d d � Z d d � Z d S) � _TestProcessr/ r1 c C s� | j d k r( | j d j | j � � | j � } | j } | j | j � � | j | j � | j | t � | j t
| � d k � | j | j t
j � � | j | j d � d S)Nr1 ztest not appropriate for {}r )�TYPE�skipTest�format�current_process�authkey�
assertTrue�is_alive�daemon�assertIsInstance�bytes�lenr8 Zidentr �getpid�exitcode)r$ �currentrL r
r
r � test_current� s z_TestProcess.test_currentc C s� | j d k r( | j d j | j � � | j d | j � } | j | j | j � j � | j d | j d d � } | j | j � | j d | j d d � } | j | j � d S)Nr1 ztest not appropriate for {}�targetrO TF)
rH rI rJ �Process�_testr8 rO rK rM � assertFalse)r$ Zproc0Zproc1Zproc2r
r
r �test_daemon_argument� s z!_TestProcess.test_daemon_argumentc O so | j � } | j | � | j | � | j | j � | j d k rk | j t | j � � | j | j � d S)Nr1 )rK rD �namerH rQ rL �pid)�clsrE r' r( rU r
r
r rY � s
z_TestProcess._testc C s: | j d � } | j � } | d d f } d d d d i } d } | j d | j d | d
| d | � } d | _ | j � } | j d
k r� | j | j | j � | j | j � d � | j | j d � | j
| | j � � | j t
| j � � t k � | j | j d � | j � | j | j d � | j | j � d � | j | | j � � | j | j � | d d � � | j | j � | � | j | j � | j � | j d
k r�| j | j � | j � | j | j � | j � | j � | j | j d � | j | j � d � | j
| | j � � d S)Nr � �hello� ZbyegR� ��Q@Z SomeProcessrW r' �kwargsr\ Tr1 Fr )�Queue�EventrX rY rO rK rH r8 rL rN � assertNotIn�active_childrenrM �type�listrT �start�assertIn�getr\ r] �join)r$ rE �er' rb r\ �prU r
r
r � test_process� s<
z_TestProcess.test_processc C s t j d � d S)N�d )r& �sleep)r^ r
r
r �_test_terminate s z _TestProcess._test_terminatec s | j d k r( | j d j | j � � | j d | j � � d � _ � j � | j � j � d � | j � | j
� � | j � j d � t � j
� } | j | d � d � | j | j d � | j � j � d � | j | d
� d � | j | j d � | j � j � d � t j d � � j � t t d � r�� f d d
� } t j t j | � } z$ t j d � | j | � d � Wd t j d � t j t j | � Xn | j | � d � | j | j d � | j � j � d � | j � | j
� � � j
� d S)Nr1 ztest not appropriate for {}rW Tr g r �alarmc s t d � � � d S)Nzjoin took too long: %s)� RuntimeError)r' )rn r
r �handler9 s z,_TestProcess.test_terminate.<locals>.handler�
Fr )rH rI rJ rX rr rO ri r8 rN rj rf rT r! rl r6 r# r& rq � terminate�hasattr�signal�SIGALRMrs re )r$ rl ru Z old_handlerr
)rn r �test_terminate s<
z_TestProcess.test_terminatec C s[ y t j � } Wn t k
r* d } Yn X| j t | � t k � | j | d k � d S)Nr )�multiprocessing� cpu_countr7 rM rg �int)r$ Zcpusr
r
r �test_cpu_countO s
z_TestProcess.test_cpu_countc C s� | j t | j � � t � | j d t j d t f � } | j | | j � � d | _ | j
� | j | | j � � | j � | j | | j � � d S)NrW r' T)
r8 rg rf rh rX r& rq �DELTAre rO ri rj rl )r$ rn r
r
r �test_active_childrenW s
z!_TestProcess.test_active_childrenc C sv | j | � t | � d k rr xP t d � D]B } | j d | j d | | | g f � } | j � | j � q, Wd S)Nr_ rW r' )�sendrR �rangerX �_test_recursionri rl )r^ �wconn�id�irn r
r
r r� d s
z _TestProcess._test_recursionc C s� | j d d � \ } } | j | g � t j t � g } x# | j � r` | j | j � � q> Wg d g d d g d d g d g d d g d d g g } | j | | � d S)N�duplexFr r ) �Piper� r& rq r� �poll�append�recvr8 )r$ �rconnr� �result�expectedr
r
r �test_recursiono s
z_TestProcess.test_recursionc C s | j d � d S)Ng $@)r )r^ �eventr
r
r �_test_sentinel� s z_TestProcess._test_sentinelc C s� | j d k r( | j d j | j � � | j � } | j d | j d | f � } | j t � � | j Wd QRX| j � | j
| j � | j } | j | t
� | j t | d d �� | j � | j � | j t | d d �� d S)Nr1 ztest not appropriate for {}rW r' r g r )rH rI rJ rd rX r� � assertRaisesr �sentinelri �
addCleanuprl rP r~ rZ r �setrM )r$ r� rn r� r
r
r �
test_sentinel� s
z_TestProcess.test_sentinelc C s� t � } t j | � } | j � } | j d | d | | f � } ~ | j � | j � | j | � d � | j | j � d � d S)NrW r' rA )
r@ �weakref�refrc rX ri rl �assertIsr8 rk )r$ rF �wrrE rn r
r
r �test_lose_target_ref� s
z!_TestProcess.test_lose_target_refN)r/ r1 )r+ r, r- r= rV r[ � classmethodrY ro rr r{ r r� r� r� r� r� r� r
r
r
r rG � s
' 4
rG c @ s@ e Z d Z d d � Z d d � Z d d � Z d d � Z d S)
� _UpperCaserc C s, t j j | � t j � \ | _ | _ d S)N)r| rX r% r� �
child_conn� parent_conn)r$ r
r
r r% � s z_UpperCaser.__init__c C sT | j j � x3 t | j j d � D] } | j j | j � � q# W| j j � d S)N)r� �close�iterr� r� r� �upper)r$ r r
r
r �run� s
z_UpperCaser.runc C s5 t | � t k s t � | j j | � | j j � S)N)rg �strrC r� r� r� )r$ r r
r
r �submit� s z_UpperCaser.submitc C s. | j j d � | j j � | j j � d S)N)r� r� r� r� )r$ r
r
r �stop� s
z_UpperCaser.stopN)r+ r, r- r% r� r� r� r
r
r
r r� � s r� c @ s^ e Z d Z d
Z d d � Z d d � Z e d d � � Z e d d � � Z d
d � Z d S)�_TestSubclassingProcessr/ c C sf t � } d | _ | j � | j | j d � d � | j | j d � d � | j � | j � d S)NTr` ZHELLOZworldZWORLD)r� rO ri r8 r� r� rl )r$ Z
uppercaserr
r
r �test_subclassing� s
z(_TestSubclassingProcess.test_subclassingc
C s� | j d k r( | j d j | j � � t j j } | j t j j | � | j d | j d | f � } | j
� | j � t | d � �C } | j
� } | j d | � | j d | � | j d | � Wd QRXd S) Nr1 ztest not appropriate for {}rW r' �r�ZeroDivisionErrorztest_multiprocessing.pyz 1/0 # MARKER)rH rI rJ �test�support�TESTFNr� �unlinkrX �_test_stderr_flushri rl �open�readrj )r$ �testfn�proc�f�errr
r
r �test_stderr_flush� s
z)_TestSubclassingProcess.test_stderr_flushc C sG t j | t j t j Bt j B� } t | d d d �t _ d d d S)N�w�closefdFr r )r r� �O_WRONLY�O_CREAT�O_EXCL�sys�stderr)r^ r� �fdr
r
r r� � s #z*_TestSubclassingProcess._test_stderr_flushc C sL t j | t j t j Bt j B� } t | d d d �t _ t j | � d S)Nr� r� F)r r� r� r� r� r� r� �exit)r^ �reasonr� r� r
r
r �_test_sys_exit� s #z&_TestSubclassingProcess._test_sys_exitc C sr | j d k r( | j d j | j � � t j j } | j t j j | � x� d d d g d f D]� } | j d | j d | | f � } d | _
| j � | j d
� | j
| j d � t | d � � } | j � } Wd QRX| j
| j � t | � � t j | � q` Wx_ d D]W } | j d t j d | f � } d | _
| j � | j d
� | j
| j | � qWd S)Nr1 ztest not appropriate for {}r r_ � z ignore thisrW r' TrA r� F� )TFr� )rH rI rJ r� r� r� r� r� rX r� rO ri rl r8 rT r� r� �rstripr� r r� r� )r$ r� r� rn r� Zcontentr
r
r �
test_sys_exit� s,
!
z%_TestSubclassingProcess.test_sys_exitN)r/ )
r+ r, r- r= r� r� r� r� r� r� r
r
r
r r� � s r� c C s- t | d � r | j � S| j � d k Sd S)N�emptyr )rx r� �qsize)rE r
r
r � queue_empty s
r� c C s- t | d � r | j � S| j � | k Sd S)N�full)rx r� r� )rE �maxsizer
r
r �
queue_full s
r� c @ s� e Z d Z e d d � � Z d d � Z e d d � � Z d d � Z e d d
� � Z d d � Z d
d � Z
e d d � � Z d d � Z d d � Z
d d � Z d d � Z d S)�
_TestQueuec C s9 | j � x t d � D] } | j � q W| j � d S)N� )r r� rk r� )r^ �queue�child_can_start�parent_can_continuer� r
r
r � _test_put s
z_TestQueue._test_putc C s� d } | j d | � } | j � } | j � } | j d | j d | | | f � } d | _ | j � | j t | � d � | j t | | � d � | j d � | j d d � | j d d d � | j d
d � | j d d d � | j
d � t j t
� | j t | � d � | j t | | � d � t | j � } t | j
� } | j t j | d d � | j | j d
� | j t j | d d d � | j | j d
� | j t j | d � | j | j d
� | j t j | d d t � | j | j t � | j t j | d d t � | j | j d
� | j t j | d d d t �| j | j t � | j � | j � | j t | � d � | j t | | � d � | j � d S)Nr� r� rW r' TFr r_ r� � rA � r r )rc rd rX r� rO ri r8 r� r� rD �
put_nowaitr& rq r� r! r� �pyqueueZFullr6 r# �TIMEOUT1�TIMEOUT2�TIMEOUT3r� r rl )r$ ZMAXSIZEr� r� r� r� rD r� r
r
r �test_put" sN
z_TestQueue.test_putc C sL | j � | j d � | j d � | j d � | j d � | j � d S)Nr_ r� r� rA )r rD r� )r^ r� r� r� r
r
r � _test_get\ s
z_TestQueue._test_getc C sP | j � } | j � } | j � } | j d | j d | | | f � } d | _ | j � | j t | � d � | j � | j � t
j t � | j t | � d � | j | j
d d � d � | j | j
d � d � | j | j
d d � d � | j | j � d
� | j t | � d � t | j
� } t | j � } | j t j | d � | j | j d � | j t j | d d � | j | j d � | j t j | � | j | j d � | j t j | d t � | j | j t � | j t j | d t � | j | j d � | j t j | d t �| j | j t � | j � d S) NrW r' TFr_ r� r r r� rA r )rc rd rX r� rO ri r8 r� r� r r& rq r� rk �
get_nowaitr! r� r� �Emptyr6 r# r� r� r� rl )r$ r� r� r� r� rk r� r
r
r �test_getf sB
z_TestQueue.test_getc C s+ x$ t d d � D] } | j | � q Wd S)Nrv � )r� rD )r^ r� r� r
r
r �
_test_fork� s z_TestQueue._test_forkc C s� | j � } x! t d � D] } | j | � q Wt j t � | j d | j d | f � } d | _ | j � x* t d � D] } | j
| j � | � q{ W| j t
j | j d � | j � d S)Nrv rW r' Tr� F)rc r� rD r& rq r� rX r� rO ri r8 rk r� r� r� rl )r$ r� r� rn r
r
r � test_fork� s
z_TestQueue.test_forkc C s� | j � } y | j | j � d � Wn t k
rG | j d � Yn X| j d � | j | j � d � | j d � | j | j � d � | j � | j | j � d � | j � | j | j � d � d S)Nr z qsize method not implementedr rA r_ )rc r8 r� r7 rI rD rk )r$ rE r
r
r �
test_qsize� s
z_TestQueue.test_qsizec C s8 x1 t | j d � D] } t j t � | j � q Wd S)N)r� rk r& rq r� Z task_done)r^ rE �objr
r
r �_test_task_done� s
z_TestQueue._test_task_donec s� � j � � � � f d d � t d � D� } x! | D] } d | _ | j � q5 Wx! t d � D] } � j | � q_ W� j � x | D] } � j d � q� Wx | D] } | j � q� Wd S)Nc s. g | ]$ } � j d � j d � f � � q S)rW r' )rX r� )�.0r� )r� r$ r
r �
<listcomp>� s z-_TestQueue.test_task_done.<locals>.<listcomp>r� Trv )�
JoinableQueuer� rO ri rD rl )r$ Zworkersrn r� r
)r� r$ r �test_task_done� s
z_TestQueue.test_task_donec C s� t j j � �� d } t | d d � � } | j d � Wd QRXt j j t j � � �8 y t | � Wn" t j
k
r� | j d � Yn XWd QRXWd QRXd S)NZ imported_by_an_imported_modulez.pyr� z�if 1:
import multiprocessing
q = multiprocessing.Queue()
q.put('knock knock')
q.get(timeout=3)
q.close()
del q
z?Probable regression on import lock contention; see Issue #22853) r� r� Ztemp_cwdr� �writeZ
DirsOnSysPathr �getcwd�
__import__r� r� �fail)r$ Z module_namer� r
r
r � test_no_import_lock_contention� s z)_TestQueue.test_no_import_lock_contentionc C sX t j � } t j � } | j t j | j d d � t j � | } | j | d � d S)NTg�������?g�(\���?)r| rc r& r� r� r� rk �assertGreaterEqual)r$ rE ri �deltar
r
r � test_timeout� s
z_TestQueue.test_timeoutc C s� | j d k r( | j d j | j � � Gd d � d t � } t j j � �G | j � } | j | � � | j d � | j | j
d d � � Wd QRXd S)Nr/ ztest not appropriate for {}c @ s e Z d Z d d � Z d S)zF_TestQueue.test_queue_feeder_donot_stop_onexc.<locals>.NotSerializablec S s
t � d S)N)r )r$ r
r
r r< s zQ_TestQueue.test_queue_feeder_donot_stop_onexc.<locals>.NotSerializable.__reduce__N)r+ r, r- r< r
r
r
r �NotSerializable
s r� Tr g�������?) rH rI rJ �objectr� r� �captured_stderrrc rD rM rk )r$ r� rE r
r
r �"test_queue_feeder_donot_stop_onexc s
z-_TestQueue.test_queue_feeder_donot_stop_onexcN)r+ r, r- r� r� r� r� r� r� r� r� r� r� r� r� r� r
r
r
r r� s :
4 r� c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)� _TestLockc C sq | j � } | j | j � d � | j | j d � d � | j | j � d � | j t t j f | j � d S)NTF)�Lockr8 �acquire�releaser� r � threadingZ ThreadError)r$ �lockr
r
r � test_lock s
z_TestLock.test_lockc C s� | j � } | j | j � d � | j | j � d � | j | j � d � | j | j � d � | j | j � d � | j | j � d � | j t t f | j � d S)NT)�RLockr8 r� r� r� rC rt )r$ r� r
r
r �
test_rlock s z_TestLock.test_rlockc C s | j � � Wd QRXd S)N)r� )r$ r
r
r �test_lock_context* s
z_TestLock.test_lock_contextN)r+ r, r- r� r� r� r
r
r
r r� s
r� c @ s@ e Z d Z d d � Z d d � Z d d � Z d d � Z d S)
�_TestSemaphorec C s� | j d t | � | j | j � d � | j d t | � | j | j � d � | j d t | � | j | j d � d � | j d t | � | j | j � d � | j d t | � | j | j � d � | j d t | � d S)Nr_ Tr r F)r; r? r8 r� r� )r$ �semr
r
r �_test_semaphore1 s z _TestSemaphore._test_semaphorec C sr | j d � } | j | � | j | j � d � | j d t | � | j | j � d � | j d t | � d S)Nr_ r� r� )� Semaphorer r8 r� r; r? )r$ r r
r
r �test_semaphore> s
z _TestSemaphore.test_semaphorec C s | j d � } | j | � d S)Nr_ )�BoundedSemaphorer )r$ r r
r
r �test_bounded_semaphoreF s z%_TestSemaphore.test_bounded_semaphorec C s# | j d k r( | j d j | j � � | j d � } t | j � } | j | d � d � | j | j d � | j | d d � d � | j | j d � | j | d t � d � | j | j d � | j | d t
� d � | j | j t
� | j | d t � d � | j | j t � d S)Nr/ ztest not appropriate for {}r Fg Tr ) rH rI rJ r r! r� r8 r6 r# r� r� r� )r$ r r� r
r
r r� N s z_TestSemaphore.test_timeoutN)r+ r, r- r r r r� r
r
r
r r� / s
r� c @ s� e Z d Z e d d d � � Z d d � Z d d � Z d d � Z d
d � Z e d d
� � Z e
j e d � d d � � Z
e d d � � Z e
j e d � d d � � Z e d d � � Z d d � Z d S)�_TestConditionNc C s9 | j � | j � | j | � | j � | j � d S)N)r� r� r )r^ �cond�sleeping�wokenr r
r
r r� g s
z_TestCondition.fc C sq | j d k rm yI | j j � | j j � } | j | d � | j | j j � d � Wn t k
rl Yn Xd S)Nr/ r )rH Z_sleeping_countr? Z _woken_countr8 Z_wait_semaphorer7 )r$ r Zsleepersr
r
r �check_invarianto s
z _TestCondition.check_invariantc C sc | j � } | j d � } | j d � } | j d | j d | | | f � } d | _ | j � t j d | j d | | | f � } d | _ | j � | j � | j � t j
t � | j d t
| � | j � | j � | j � t j
t � | j d t
| � | j � | j � | j � t j
t � | j d t
| � | j | � | j � d S)Nr rW r' Tr r_ )� Conditionr rX r� rO ri r� �Threadr� r& rq r� r; r? �notifyr� r
rl )r$ r r r rn r
r
r � test_notifyz s2 $
$
z_TestCondition.test_notifyc C sp | j � } | j d � } | j d � } x� t d � D]z } | j d | j d | | | t f � } d | _ | j � t j d | j d | | | t f � } d | _ | j � q7 Wx t d � D] } | j
� q� Wx t d � D] } | j
� q� W| j d t | � | j
| � x� t d � D]t } | j d | j d | | | f � } d | _ | j � t j d | j d | | | f � } d | _ | j � q$Wx t d � D] } | j
� q�Wt j t � | j d t | � | j
� | j � | j � xN t d � D]@ } y t | � d k r$PWn t k
r:PYn Xt j t � qW| j d t | � | j
| � d S)Nr r� rW r' Tr� rv )r r r� rX r� r� rO ri r� r r� r; r? r
r& rq r� Z
notify_allr� r7 )r$ r r r r� rn r) r
r
r �test_notify_all� sR
$
$
z _TestCondition.test_notify_allc C sb | j � } t | j � } | j � | t � } | j � | j | d � | j | j t � d S)NF) r r! r r� r� r� r8 r6 r# )r$ r r r: r
r
r r� � s
z_TestCondition.test_timeoutc sc | �V d � _ | j � | j � f d d � � } | sK � j d k rX t j d � Wd QRXd S)Nr c s
� j d k S)Nr� )r9 r
)�stater
r �<lambda>� s z0_TestCondition._test_waitfor_f.<locals>.<lambda>r� r )r9 r
�wait_forr� r� )r^ r r r� r
)r r �_test_waitfor_f� s
z _TestCondition._test_waitfor_fzneeds sharedctypesc s | j � } | j d d � � | j d | j d | � f � } d | _ | j � | �@ | j � f d d � � } | j | � | j � j d � Wd QRXxH t
d � D]: } t j d
� | � � j d 7_ | j
� Wd QRXq� W| j d � | j | j � � | j | j d � d S)
Nr� r rW r' Tc s
� j d k S)Nr )r9 r
)r r
r r s z-_TestCondition.test_waitfor.<locals>.<lambda>r r� g{�G�z�?rA r )r r rX r rO ri r rM r8 r9 r� r& rq r
rl rZ rN rT )r$ r rn r� r� r
)r r � test_waitfor� s" !
z_TestCondition.test_waitforc s� | j � | �| d } t j � } | j � f d d � d | �} t j � | } | r� | d | k oz | d k n r� d | _ Wd QRXd S)Ng�������?c s
� j d k S)Nr� )r9 r
)r r
r r s z8_TestCondition._test_waitfor_timeout_f.<locals>.<lambda>r g333333�?g $@T)r� r& r r9 )r^ r r �successr r� Zdtr� r
)r r �_test_waitfor_timeout_f s
!+z&_TestCondition._test_waitfor_timeout_fc C s� | j � } | j d d � } | j d d � } | j d � } | j d | j d | | | | f � } d | _ | j � | j | j d d � � xH t d � D]: } t
j d
� | � | j d 7_ | j
� Wd QRXq� W| j d � | j | j � d S)
Nr� r FrW r' Tr rv r� g{�G�z�?r rA )r r r rX r rO ri rM r� r� r& rq r9 r
rl )r$ r r r r rn r� r
r
r �test_waitfor_timeout s
z#_TestCondition.test_waitfor_timeoutc
C sH | � | j � Wd QRXt j d � | d k rD t j | t j � d S)Nr )r
r& rq r �killry �SIGINT)r^ rF r] r
r
r �_test_wait_result5 s
z _TestCondition._test_wait_resultc
C s� t | t � r- t j d k r- t j � } n d } | j � } | �� | j | j d � � | j | j d � � | j d | j
d | | f � } | j � | j | j d � � | d k r� | j
t | j d � | j � Wd QRXd S)Nr
r g�������?rW r' rv )rB �ProcessesMixinr� �platformr rS r rZ r rX r ri rM r� �KeyboardInterruptrl )r$ r] rF rn r
r
r �test_wait_result= s !
z _TestCondition.test_wait_result)r+ r, r- r� r� r
r r r� r r �
skipUnless�HAS_SHAREDCTYPESr r r r r r
r
r
r r e s + A r c @ s. e Z d Z e d d � � Z d d � Z d S)�
_TestEventc C s t j t � | j � d S)N)r& rq r� r� )r^ r� r
r
r � _test_eventT s
z_TestEvent._test_eventc C sD | j � } t | j � } | j | j � d � | j | d � d � | j | j d � | j | t � d � | j | j t � | j � | j | j � d � | j | � d � | j | j d � | j | t � d � | j | j d � | j � | j
d | j d | f � } d | _ | j
� | j | � d � d S)NFg TrW r' )rd r! r r8 �is_setr6 r# r� r� �clearrX r" rO ri )r$ r� r rn r
r
r �
test_eventY s$
z_TestEvent.test_eventN)r+ r, r- r� r" r% r
r
r
r r! R s r! c @ sL e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d S) �
_DummyListc C sK t j j t j d � � } t j � } | j | | f � d | j d <d S)Nr� r )r| �heap�
BufferWrapper�structZcalcsizer� � __setstate__�
_lengthbuf)r$ �wrapperr� r
r
r r% � s z_DummyList.__init__c C s1 | \ | _ | _ | j j � j d � | _ d S)Nr� )�_wrapper�_lockZcreate_memoryview�castr+ )r$ r r
r
r r* � s z_DummyList.__setstate__c C s | j | j f S)N)r- r. )r$ r
r
r � __getstate__� s z_DummyList.__getstate__c C s& | j � | j d d 7<Wd QRXd S)Nr r )r. r+ )r$ �_r
r
r r� � s
z_DummyList.appendc C s | j � | j d SWd QRXd S)Nr )r. r+ )r$ r
r
r �__len__� s
z_DummyList.__len__N)r+ r, r- r% r* r0 r� r2 r
r
r
r r& � s
r& c C s t j d � d S)Ng{�G�z�?)r& rq r
r
r
r �_wait� s r3 c @ sa e Z d Z d Z d d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d S)�Bunchz
A bunch of threads.
Fc
C s� | | _ | | _ | | _ | j � | _ | j � | _ | j � | _ | s[ | j j � g } xI t | � D]; } | j
d | j � } d | _ | j
� | j | � qn Wd d � } t j | | | � | _ d S)z�
Construct a bunch of `n` threads running the same function `f`.
If `wait_before_exit` is True, the threads won't terminate until
do_finish() is called.
rW Tc S s x | D] } | j � q Wd S)N)rl )r1 rn r
r
r �finalize� s
z Bunch.__init__.<locals>.finalizeN)r� r' �n� DummyList�started�finishedrd � _can_exitr� r� rX �taskrO ri r� r� r5 �
_finalizer)
r$ � namespacer� r' r6 Zwait_before_exitr1 r� rn r5 r
r
r r% � s
zBunch.__init__c C sm t j � } | j j | � z | j | j � Wd | j j | � | j j d � | j j � sh t
� Xd S)N� ) r rS r8 r� r� r' r9 r: r r# rC )r$ r] r
r
r r; � s z
Bunch.taskc C s* x# t | j � | j k r% t � q Wd S)N)rR r8 r6 r3 )r$ r
r
r �wait_for_started� s zBunch.wait_for_startedc C s* x# t | j � | j k r% t � q Wd S)N)rR r9 r6 r3 )r$ r
r
r �wait_for_finished� s zBunch.wait_for_finishedc C s | j j � d S)N)r: r� )r$ r
r
r � do_finish� s zBunch.do_finishc C s | j � d S)N)r< )r$ r
r
r r� � s z Bunch.closeN)
r+ r, r- �__doc__r% r; r? r@ rA r� r
r
r
r r4 � s
r4 c @ s( e Z d Z d d � Z d d � Z d S)�
AppendTruec C s
| | _ d S)N)r� )r$ r� r
r
r r% � s zAppendTrue.__init__c C s | j j d � d S)NT)r� r� )r$ r
r
r r* � s zAppendTrue.__call__N)r+ r, r- r% r* r
r
r
r rC � s rC c @ s{ e Z d Z d Z d Z d Z d d � Z d d � Z d d � Z d
d � Z e
d d
� � Z d d d � Z d d � Z
e
d d � � Z d d � Z e
d d � � Z d d � Z e
d d � � Z d d � Z e
d d � � Z d! d" � Z e
d# d$ � � Z d% d& � Z e
d' d( � � Z d) d* � Z e
d+ d, � � Z d- d. � Z d/ d0 � Z e
d1 d2 � � Z d3 d4 � Z d5 S)6� _TestBarrierz$
Tests for Barrier objects.
rA g >@c C s" | j | j d | j �| _ d S)Nr )�Barrier�N�defaultTimeout�barrier)r$ r
r
r �setUp� s z_TestBarrier.setUpc C s | j j � d | _ d S)N)rH �abort)r$ r
r
r �tearDown� s
z_TestBarrier.tearDownc C s: | j d k r g S| j d k r/ | j j � St � Sd S)Nr1 r0 )rH r0 rh r& )r$ r
r
r r7 � s
z_TestBarrier.DummyListc
C sF t | | | | j d � } z | | � | j � Wd | j � Xd S)Nr )r4 rF r@ r� )r$ r� r' r5 r
r
r � run_threads� s
z_TestBarrier.run_threadsc C s� | j } | | j k s t � x� t | � D]� } | d j d � t | d � | | k sb t � | j � | d j d � t | d � | d | k s� t � | j � q+ Wy | j d k s� t � Wn t k
r� Yn X| j s� t � d S)Nr Tr )
�partiesrF rC r� r� rR r � n_waitingr7 �broken)r^ rH �resultsr6 �mr� r
r
r � multipass� s
$
z_TestBarrier.multipassr c C s; | j � | j � g } | j | j | j | | f � d S)z;
Test that a barrier is passed in lockstep
N)r7 rL rR rH )r$ �passesrP r
r
r � test_barrier s z_TestBarrier.test_barrierc C s
| j d � S)zC
Test that a barrier works for 10 consecutive runs
rv )rT )r$ r
r
r �test_barrier_10 s z _TestBarrier.test_barrier_10c C s | j � } | j | � d S)N)r rD )r^ rH r� r: r
r
r �_test_wait_return_f s z _TestBarrier._test_wait_return_fc sg | j � � | j | j | j � f � � f d d � t | j � D� } | j | j d � d � d S)z9
test the return value from barrier.wait
c s g | ] } � j � � q Sr
)rk )r� r� )r� r
r r� ' s z1_TestBarrier.test_wait_return.<locals>.<listcomp>r r N)rc rL rV rH r� rF r8 �count)r$ rP r
)r� r �test_wait_return! s "z _TestBarrier.test_wait_returnc C s& | j � t | � d k r" t � d S)Nr )r rR rt )r^ rH rP r
r
r �_test_action_f* s
z_TestBarrier._test_action_fc C s] | j � } | j | j d t | � �} | j | j | | f � | j t | � d � d S)z,
Test the 'action' callback
�actionr N)r7 rE rF rC rL rY r8 rR )r$ rP rH r
r
r � test_action0 s z_TestBarrier.test_actionc C s� y@ | j � } | | j d k r( t � | j � | j d � Wn= t j k
rd | j d � Yn t k
r | j � Yn Xd S)Nr_ T)r rF rt r� r� �BrokenBarrierErrorrJ )r^ rH �results1�results2r� r
r
r �
_test_abort_f9 s
z_TestBarrier._test_abort_fc C s� | j � } | j � } | j | j | j | | f � | j t | � d � | j t | � | j d � | j | j j � d S)zK
Test that an abort will put the barrier in a broken state
r r N) r7 rL r_ rH r8 rR rF rM rO )r$ r] r^ r
r
r �
test_abortF s z_TestBarrier.test_abortc C s� | j � } | | j d k rV x' | j | j d k rH t j d � q" W| j � n@ y | j � | j d � Wn" t j k
r� | j d � Yn X| j � | j d � d S)Nr_ r g����MbP?T) r rF rN r& rq �resetr� r� r\ )r^ rH r] r^ �results3r� r
r
r �
_test_reset_fR s
z_TestBarrier._test_reset_fc C s� | j � } | j � } | j � } | j | j | j | | | f � | j t | � d � | j t | � | j d � | j t | � | j � d S)zL
Test that a 'reset' on a barrier frees the waiting threads
r r N)r7 rL rc rH r8 rR rF )r$ r] r^ rb r
r
r �
test_resetd s z_TestBarrier.test_resetc C s� y@ | j � } | | j d k r( t � | j � | j d � Wn= t j k
rd | j d � Yn t k
r | j � Yn X| j � | j d k r� | j � | j � | j � | j d � d S)Nr_ T)r rF rt r� r� r\ rJ ra )r^ rH �barrier2r] r^ rb r� r
r
r �_test_abort_and_reset_fq s
z$_TestBarrier._test_abort_and_reset_fc C s� | j � } | j � } | j � } | j | j � } | j | j | j | | | | f � | j t | � d � | j t | � | j d � | j t | � | j � d S)zF
Test that a barrier can be reset after being broken.
r r N)r7 rE rF rL rf rH r8 rR )r$ r] r^ rb re r
r
r �test_abort_and_reset� s z!_TestBarrier.test_abort_and_resetc C sf | j � } | | j d k r, t j d � y | j d � Wn" t j k
ra | j d � Yn Xd S)Nr_ g �?g �?T)r rF r& rq r� r\ r� )r^ rH rP r� r
r
r �_test_timeout_f� s
z _TestBarrier._test_timeout_fc C sH | j � } | j | j | j | f � | j t | � | j j � d S)z$
Test wait(timeout)
N)r7 rL rh rH r8 rR rM )r$ rP r
r
r r� � s z_TestBarrier.test_timeoutc C si | j | j � } | | j d k r2 t j d � y | j � Wn" t j k
rd | j d � Yn Xd S)Nr_ g �?T)r rG rF r& rq r� r\ r� )r^ rH rP r� r
r
r �_test_default_timeout_f� s
z$_TestBarrier._test_default_timeout_fc C sZ | j | j d d �} | j � } | j | j | | f � | j t | � | j � d S)z4
Test the barrier's default timeout
r g �?N)rE rF r7 rL ri r8 rR rM )r$ rH rP r
r
r �test_default_timeout� s z!_TestBarrier.test_default_timeoutc C s' | j d � } | j � | j � d S)Nr )rE r )r$ r5 r
r
r �test_single_thread� s
z _TestBarrier.test_single_threadc
C s@ x9 t | � D]+ } | j � | � | j | � Wd QRXq
Wd S)N)r� r r� )r^ rH rS �connr� r� r
r
r �_test_thousand_f� s
z _TestBarrier._test_thousand_fc C s� | j d k r( | j d j | j � � d } | j � } | j d � \ } } xK t | j � D]: } | j d | j d | j | | | f � } | j
� q_ WxD t | � D]6 } x- t | j � D] } | j | j � | � q� Wq� Wd S)Nr0 ztest not appropriate for {}i� FrW r' )
rH rI rJ r� r� r� rF rX rm rH ri r8 r� )r$ rS r� rl r� �jrn r� r
r
r �
test_thousand� s z_TestBarrier.test_thousandN) r+ r, r- rB rF rG rI rK r7 rL r� rR rT rU rV rX rY r[ r_ r` rc rd rf rg rh r� ri rj rk rm ro r
r
r
r rD � s6
rD c @ s� e Z d Z d Z d d d d e d � e d
� f g Z d d � Z e d d � � Z d d d � Z d d � Z
d d � Z d S) �
_TestValuer/ r� � �^ �d�
@g @�h�� �� rF �x�yc C s t s | j d � d S)Nz%requires multiprocessing.sharedctypes)r rI )r$ r
r
r rI � s z_TestValue.setUpc C s4 x- t | | j � D] \ } } | d | _ q Wd S)Nr_ )�zip� codes_valuesr9 )r^ �values�sv�cvr
r
r rY � s z_TestValue._testFc s� | r% � f d d � � j D� } n � f d d � � j D� } x7 t | � j � D]# \ } } � j | j | d � qT W� j d � j d | f � } d | _ | j � | j � x7 t | � j � D]# \ } } � j | j | d � q� Wd S) Nc s+ g | ]! \ } } } � j | | � � q Sr
)�RawValue)r� �coder9 r1 )r$ r
r r� � s z)_TestValue.test_value.<locals>.<listcomp>c s+ g | ]! \ } } } � j | | � � q Sr
)r )r� r� r9 r1 )r$ r
r r� � s r rW r' Tr_ ) r{ rz r8 r9 rX rY rO ri rl )r$ �rawr| r} r~ r� r
)r$ r �
test_value� s
z_TestValue.test_valuec C s | j d d � d S)Nr� T)r� )r$ r
r
r �
test_rawvalue s z_TestValue.test_rawvaluec
C sK | j d d � } | j � } | j � } | j d d d d �} | j � } | j � } | j � } | j d d d | �} | j � } | j � }
| j | | � | j d d d d �} | j t | d � � | j t | d � � | j t | j d d d d �| j d d � } | j t | d � � | j t | d � � d S)Nr� rA r� F�get_lock�get_objZnavalue)
r r� r� r� r8 rZ rx r� r r )
r$ Zval1�lock1�obj1Zval2�lock2�obj2r� Zval3�lock3�obj3�arr4�arr5r
r
r �test_getobj_getlock s$ z _TestValue.test_getobj_getlockN)r/ )r� rq rr � �)rs rt r� ����)ru r� rw ) r+ r, r- r= r r{ rI r� rY r� r� r� r
r
r
r rp � s rp c @ s� e Z d Z d Z e d d � � Z e j e d k d � d d d � � Z e j e d k d � d d
� � Z
e j e d k d � d d � � Z e j e d k d � d
d � � Z d S)�
_TestArrayr/ c C s: x3 t d t | � � D] } | | | | d 7<q Wd S)Nr )r� rR )r^ �seqr� r
r
r r� % s z _TestArray.fNzrequires _ctypesFc
C sw d d d d d d d d d d
g
} | r? | j d | � } n | j d | � } | j t | � t | � � | j | d | d � | j t | d
d � � t | d
d � � � t j d d d
d d g � | d d � <| d d � <| j t | d d � � | � | j | � | j d | j d | f � } d | _ | j � | j
� | j t | d d � � | � d S)Ni� ir i� i5 � �� i$ i� i� i? r� r� r_ r� r r� r� rW r' T) �RawArray�Arrayr8 rR rh �arrayr� rX rO ri rl )r$ r� r� �arrrn r
r
r �
test_array* s $ 06
z_TestArray.test_arrayc C s� d } x� t d � D]� } | j d | � } | j t | � | � | j t | � d g | � t d � | d d � <| j t | � t t d � � � ~ q Wd S)Nrv r� r� r )r� r� r8 rR rh )r$ �sizer1 r� r
r
r �test_array_from_sizeC s "z _TestArray.test_array_from_sizec C s | j d d � d S)Nr� T)r� )r$ r
r
r �
test_rawarrayR s z_TestArray.test_rawarrayc
C s� | j d t t d � � � } | j � } | j � } | j d t t d � � d d �} | j � } | j � } | j � } | j d t t d � � d | �} | j � } | j � }
| j | | � | j d t d � d d �} | j t | d � � | j t | d � � | j t
| j d t d � d d �| j d t d � � } | j t | d � � | j t | d � � d S)Nr� rv r� Fr� r� Znotalock) r� rh r� r� r� r� r8 rZ rx r� r r� )
r$ Zarr1r� r� Zarr2r� r� r� Zarr3r� r� r� r� r
r
r �test_getobj_getlock_objV s& $ $ z"_TestArray.test_getobj_getlock_obj)r/ )
r+ r, r- r= r� r� r �skipIfr r� r� r� r� r
r
r
r r� ! s $$r� c @ s: e Z d Z d Z d d � Z d d � Z d d � Z d S)
�_TestContainersr0 c C s[ | j t t d � � � } | j | d d � t t d � � � | j � } | j | d d � g � | j t t d � � � | j | d d � t t d � � � | j | d d � | j | d d � d d d g � | d 9} | j | d d � d d d d d d d d d d g
� | j | d d g d d d d d d d d d d d d g � | j | d d � t t d � � � | | g } | j | � } | j | d d � d d d d d d d d d
d g
d d d d d d d d d d g
g � | j | g � } | j d � | j | d d � d d d d d d d d d
d d g g � d S)
Nrv rA r_ r� r� r r r� r� r� � r` )rh r� r8 �extendr� )r$ r4 r5 rs rm r� r
r
r � test_listx s( & &#
8>&
I
z_TestContainers.test_listc C s� | j � } t t d d � � } x | D] } t | � | | <q( W| j | j � t d d � | D� � � | j t | j � � | � | j t | j � � d d � | D� � | j t | j � � d d � | D� � d S)N�A �F c s s! | ] } | t | � f Vq d S)N)�chr)r� r� r
r
r � <genexpr>� s z,_TestContainers.test_dict.<locals>.<genexpr>c S s g | ] } t | � � q Sr
)r� )r� r� r
r
r r� � s z-_TestContainers.test_dict.<locals>.<listcomp>c S s" g | ] } | t | � f � q Sr
)r� )r� r� r
r
r r� � s )
�dictrh r� r� r8 r �sorted�keysr| �items)r$ rs �indicesr� r
r
r � test_dict� s
) )z_TestContainers.test_dictc C s� | j � } d | _ d | _ d | _ | j | j | j f d � | ` | j t | � d � | j t | d � � | j t | d � � d S)N�Bob�BuilderZhiddenzNamespace(name='Bob')r\ �job)r� r� )� Namespacer\ r� Z_hiddenr8 r� rM rx )r$ r6 r
r
r �test_namespace� s z _TestContainers.test_namespaceN)r0 )r+ r, r- r= r� r� r� r
r
r
r r� t s
r� g c C s t j | � | | S)N)r& rq )rx r r
r
r �sqr� s
r� c C s | | S)Nr
)rx ry r
r
r �mul� s r� c C s | S)Nr
)rx r
r
r �identity� s r� c @ s. e Z d Z d Z d d � Z d d � Z d S)�
CountedObjectr c C s | j d 7_ t j | � S)Nr )� n_instancesr� �__new__)r^ r
r
r r� � s zCountedObject.__new__c C s t | � j d 8_ d S)Nr )rg r� )r$ r
r
r �__del__� s zCountedObject.__del__N)r+ r, r- r� r� r� r
r
r
r r� � s r� c @ s e Z d Z d S)� SayWhenErrorN)r+ r, r- r
r
r
r r� � s r� c c sP | d k r t d � � x1 t | � D]# } | | k rC t d � � | Vq% Wd S)Nr zSomebody said whenr )r� r� )ZtotalZwhenr� r
r
r � exception_throwing_generator� s r� c sl e Z d Z e � f d d � � Z e � f d d � � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) d* � Z e d+ d, � � Z d- d. � Z e d/ d0 � � Z d1 d2 � Z d3 d4 � Z � S)5� _TestPoolc s# t � j � | j d � | _ d S)Nr� )�super�
setUpClass�Pool�pool)r^ )� __class__r
r r� � s
z_TestPool.setUpClassc s4 | j j � | j j � d | _ t � j � d S)N)r� rw rl r� �
tearDownClass)r^ )r� r
r r� � s
z_TestPool.tearDownClassc C sZ | j j } | j | t d � t d � � | j | t f d d i � t d d � � d S)NrA rx r� )rA )r� �applyr8 r� )r$ Zpapplyr
r
r �
test_apply� s z_TestPool.test_applyc C s� | j j } | j | t t t d � � � t t t t t d � � � � � | j | t t t d � � d d �t t t t t d � � � � � d S)Nrv rp � chunksizer� )r� �mapr8 r� rh r� )r$ Zpmapr
r
r �test_map� s @$z_TestPool.test_mapc C s� | j j } t t t d � t d d d � � � } | j | t | � t t j t | � � � t t t d � t d d
d � � � } | j | t | d d �t t j t | � � � d S) Nrv r� r rp �c r� r� r r r r )r� �starmaprh rz r� r8 r� � itertools)r$ Zpsmap�tuplesr
r
r � test_starmap� s ''z_TestPool.test_starmapc C sb t t t d � t d d d � � � } | j | j j t | � j � t t j t | � � � d S)Nrp r� r r r )
rh rz r� r8 r� Z
starmap_asyncr� rk r� r� )r$ r� r
r
r �test_starmap_async� s ' z _TestPool.test_starmap_asyncc C sP | j | j j t t t d � � � j � t t t t t d � � � � � d S)Nrv )r8 r� � map_asyncr� rh r� rk r� )r$ r
r
r �test_map_async� s *z_TestPool.test_map_asyncc C s� | j d k r | j j � n g } | j j t d g d | j d | j �j � | j d t | � � | j d g | d � | j j t d g d | j d | j �j � | j d t | � � | j
| d t � d S) Nr0 �1�callback�error_callbackr r r4 r_ ) rH r0 rh r� r� r~ r� r r8 rR rP r )r$ � call_argsr
r
r �test_map_async_callbacks� s $ z"_TestPool.test_map_async_callbacksc C sv | j d k r( | j d j | j � � Gd d � d t � } | j t � �" | j j t | � g d � Wd QRXd S)Nr1 ztest not appropriate for {}c @ s e Z d Z d d � Z d S)z*_TestPool.test_map_unplicklable.<locals>.Ac S s t d � � d S)Nz
cannot pickle)rt )r$ r
r
r r< s z5_TestPool.test_map_unplicklable.<locals>.A.__reduce__N)r+ r, r- r< r
r
r
r �A s r� rv ) rH rI rJ r� r� rt r� r� r� )r$ r� r
r
r �test_map_unplicklable s
z _TestPool.test_map_unplicklablec C sR y) | j j t g d d �j d t � Wn" t j k
rM | j d � Yn Xd S)Nr� r r z2pool.map_async with chunksize stalled on null list)r� r� r� rk r� r| � TimeoutErrorr� )r$ r
r
r �test_map_chunksize s )z _TestPool.test_map_chunksizec ' C sA | j d k r( | j d j | j � � | j t � �$ | j j t t d d � d � Wd QRX| j t � �$ | j j t t d d � d � Wd QRX| j t � �$ | j j t t d d � d � Wd QRXGd d � d � } | j t � � | j j t | � d � Wd QRX| j t � � | j j t | � d � Wd QRXd S)
Nr0 ztest not appropriate for {}r rv r� c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)zE_TestPool.test_map_handle_iterable_exception.<locals>.SpecialIterablec S s | S)Nr
)r$ r
r
r �__iter__& s zN_TestPool.test_map_handle_iterable_exception.<locals>.SpecialIterable.__iter__c S s
t � d S)N)r� )r$ r
r
r �__next__( s zN_TestPool.test_map_handle_iterable_exception.<locals>.SpecialIterable.__next__c S s d S)Nr r
)r$ r
r
r r2 * s zM_TestPool.test_map_handle_iterable_exception.<locals>.SpecialIterable.__len__N)r+ r, r- r� r� r2 r
r
r
r �SpecialIterable% s r� r r ) rH rI rJ r� r� r� r� r� r� )r$ r� r
r
r �"test_map_handle_iterable_exception s &&& z,_TestPool.test_map_handle_iterable_exceptionc C sT | j j t d t f � } t | j � } | j | � d � | j | j t � d S)Nr� �1 ) r� � apply_asyncr� r� r! rk r8 r6 r# )r$ r: rk r
r
r �
test_async1 s z_TestPool.test_asyncc C s^ | j j t d t d f � } t | j � } | j t j | d t �| j | j
t � d S)Nr� g �?r ) r� r� r� r� r! rk r� r| r� r6 r# )r$ r: rk r
r
r �test_async_timeout7 s z _TestPool.test_async_timeoutc C s& | j j t t t d � � � } | j t | � t t t t t d � � � � � | j j t t t d � � � } x. t d � D] } | j t | � | | � q� W| j t | j
� | j j t t t d � � d d �} x. t d � D] } | j t | � | | � q� W| j t | j
� d S)Nrv i� r� rp ) r� �imapr� rh r� r8 r� �nextr� �
StopIterationr� )r$ �itr� r
r
r � test_imap= s !1! ' z_TestPool.test_imapc C s� | j d k r( | j d j | j � � | j j t t d d � d � } | j t | j � | j j t t d d � d � } | j t | j � | j j t t d d � d � } x. t
d � D] } | j t | � | | � q� W| j t | j � | j j t t d d � d � } x. t
d � D] } | j t | � | | � q#W| j t | j � | j j t t d d � d
� } x. t
d
� D] } | j t | � | | � q�W| j t | j � d S)
Nr0 ztest not appropriate for {}r rv r� r� r� r_ r� r� r r )
rH rI rJ r� r� r� r� r� r� r� r� r8 r� )r$ r� r� r
r
r �#test_imap_handle_iterable_exceptionK s$ !!! ! ! z-_TestPool.test_imap_handle_iterable_exceptionc C s� | j j t t t d � � � } | j t | � t t t t t d � � � � � | j j t t t d � � d d �} | j t | � t t t t t d � � � � � d S)Ni� r� �5 )r� �imap_unorderedr� rh r� r8 r� r� )r$ r� r
r
r �test_imap_unorderede s !1'z _TestPool.test_imap_unorderedc C s� | j d k r( | j d j | j � � | j j t t d d � d � } | j t | j � | j j t t d d
� d � } | j t | j � | j j t t d d � d � } t
t t t
t d � � � � } | j t � �E x= t d � D]/ } t
| � } | j | | � | j | � q� WWd QRX| j j t t d d � d � } t
t t t
t d � � � � } | j t � �E x= t d � D]/ } t
| � } | j | | � | j | � q�WWd QRXd S) Nr0 ztest not appropriate for {}r rv r� r� r� r_ r r )rH rI rJ r� r� r� r� r� r� r� rh r� r� r� rj �remove)r$ r� Zexpected_valuesr� r9 r
r
r �-test_imap_unordered_handle_iterable_exceptionl s8 ! ! z7_TestPool.test_imap_unordered_handle_iterable_exceptionc
C s� | j d k r t n t } | j | | j d � | j | | j d � | j d k r� | j d � } z | j d t | j � � Wd | j � | j � Xd S)Nr0 r r r� r )
rH � RemoteErrorr r� r� r8 rR �_poolr� rl )r$ Zexpected_errorrn r
r
r �test_make_pool� s
z_TestPool.test_make_poolc C sn | j j t j d d � t d � D� d d �} | j j � t | j j � } | � | j | j d � d S)Nc S s g | ] } d � q S)g�������?r
)r� r� r
r
r r� � s z,_TestPool.test_terminate.<locals>.<listcomp>i' r� r g @)
r� r� r& rq r� rw r! rl �
assertLessr# )r$ r� rl r
r
r r{ � s (
z_TestPool.test_terminatec C s� | j d � } | j | j t g � g � | j t | j t g � � g � | j t | j t g � � g � | j | j t g � j � g � | j � | j
� d S)Nr ) r� r8 r� r� rh r� r� r� rk r� rl )r$ rn r
r
r �test_empty_iterable� s """
z _TestPool.test_empty_iterablec C s� | j d k r� t t d � � } d d � | D� } | j d � �/ } | j t | � } | j | j � | � Wd QRX| j t | j t | � d S)Nr/ rv c S s g | ] } t | � � q Sr
)r� )r� r� r
r
r r� � s z*_TestPool.test_context.<locals>.<listcomp>r_ )
rH rh r� r� r� r� r8 rk r� r )r$ �Lr� rn r� r
r
r � test_context� s z_TestPool.test_contextc C s t d � � d S)N�{ )rt )r^ r
r
r �_test_traceback� s z_TestPool._test_tracebackc < C s� | j d k r�| j d � �V } y | j | j � Wn+ t k
rb } z | } WYd d } ~ Xn X| j d � Wd QRX| j t | � t � | j | j
d � | j } | j t | � t j
j � | j d | j � t j j � �9 } y
| � Wn% t k
rt j t j � � Yn XWd QRX| j d | j � � | j d � �� } y | j t t d d � d � Wn+ t k
r�} z | } WYd d } ~ Xn X| j d � | j t | � t � | j | j d � Wd QRXd S) Nr/ r zexpected RuntimeErrorr� z&raise RuntimeError(123) # some commentzexpected SayWhenError)r� r ) rH r� r� r� � Exceptionr� r� rg rt r8 r' � __cause__r| r� ZRemoteTracebackrj �tbr� r� r� r� �
excepthook�exc_info�getvaluer� r� r� r� )r$ rn rm �exc�cause�f1r
r
r �test_traceback� s6
z_TestPool.test_tracebackc C s t d � � d S)N�foo)rt )r^ r
r
r �_test_wrapped_exception� s z!_TestPool._test_wrapped_exceptionc C sD | j d � �. } | j t � � | j | j � Wd QRXWd QRXd S)Nr )r� r� rt r� r� )r$ rn r
r
r �test_wrapped_exception� s z _TestPool.test_wrapped_exceptionc C s� d d � t d � D� } d d � | D� } | j j t | � ~ t j t � | j t d d � | D� � d h � | j t j
d � d S)Nc S s g | ] } t � � q Sr
)r� )r� r� r
r
r r� � s z4_TestPool.test_release_task_refs.<locals>.<listcomp>rv c S s g | ] } t j | � � q Sr
)r� r� )r� �or
r
r r� � s c s s | ] } | � Vq d S)Nr
)r� r� r
r
r r� � s z3_TestPool.test_release_task_refs.<locals>.<genexpr>r ) r� r� r� r� r& rq r� r8 r� r� r� )r$ ZobjsZrefsr
r
r �test_release_task_refs� s
&z _TestPool.test_release_task_refs) r+ r, r- r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r{ r� r� r� r� r� r r r
r
)r� r r� � s4
$
$ r� c C s t d � � d S)N�key)�KeyErrorr
r
r
r �raising� s r c C s
d d � S)Nc S s d S)N�* r
r
r
r
r r s z%unpickleable_result.<locals>.<lambda>r
r
r
r
r �unpickleable_result s r c @ s. e Z d Z d Z d d � Z d d � Z d S)�_TestPoolWorkerErrorsr/ c s� t j d � } d g � � f d d � } | j t d | �} | j t | j � | j � d � | j � d t � | j � | j
� d S)Nr_ c s | � d <d S)Nr r
)r� )�
scratchpadr
r �errback s z@_TestPoolWorkerErrors.test_async_error_callback.<locals>.errbackr� r ) r| r� r� r r� r rk rM rP r� rl )r$ rn r
r: r
)r r �test_async_error_callback s
z/_TestPoolWorkerErrors.test_async_error_callbackc s� d d l m } t j d � } x� t d � D]� } d g � � f d d � } | j t d | �} | j | | j � � d } | j | � | j
� d | � | j | j � | j | j
� q, W| j � | j � d S)Nr )�MaybeEncodingErrorr_ r� c s | � d <d S)Nr r
)r� )r r
r r
s z?_TestPoolWorkerErrors.test_unpickleable_result.<locals>.errbackr� )�multiprocessing.poolr r| r� r� r� r r� rk rM rP ZassertIsNotNoner� r9 r� rl )r$ r rn Z iterationr
r: �wrappedr
)r r �test_unpickleable_result s
z._TestPoolWorkerErrors.test_unpickleable_resultN)r/ )r+ r, r- r= r r r
r
r
r r s r c @ s. e Z d Z d Z d d � Z d d � Z d S)�_TestPoolWorkerLifetimer/ c C sq t j d d d �} | j d t | j � � d d � | j D� } g } x0 t d � D]" } | j | j t | f � � qW Wx6 t | � D]( \ } } | j | j
� t | � � q� W| j � d } x> | rt d d � | j D� � r| d
8} t
j t � q� Wd d � | j D� } | j d | � | j d | � | j t | � t | � � | j � | j � d S) Nr� �maxtasksperchildrv c S s g | ] } | j � q Sr
)r] )r� r� r
r
r r� 3 s zE_TestPoolWorkerLifetime.test_pool_worker_lifetime.<locals>.<listcomp>rp �2 c s s | ] } | j � Vq d S)N)rN )r� r� r
r
r r� A s zD_TestPoolWorkerLifetime.test_pool_worker_lifetime.<locals>.<genexpr>r c S s g | ] } | j � q Sr
)r] )r� r� r
r
r r� D s )r| r� r8 rR r� r� r� r� r� � enumeraterk Z_repopulate_pool�allr& rq r� re ZassertNotEqualr� r� rl ) r$ rn ZorigworkerpidsrP r� rn r: Z countdownZfinalworkerpidsr
r
r �test_pool_worker_lifetime0 s&
&
z1_TestPoolWorkerLifetime.test_pool_worker_lifetimec C s� t j d d d �} g } x3 t d � D]% } | j | j t | d f � � q( W| j � | j � x6 t | � D]( \ } } | j | j
� t | � � qr Wd S)Nr� r r r� g333333�?) r| r� r� r� r� r� r� rl r r8 rk )r$ rn rP r� rn r: r
r
r �%test_pool_worker_lifetime_early_closeM s #
z=_TestPoolWorkerLifetime.test_pool_worker_lifetime_early_closeN)r/ )r+ r, r- r= r r r
r
r
r r - s r )� BaseManager� BaseProxyr� c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�FooBarc C s d S)Nzf()r
)r$ r
r
r r� a s zFooBar.fc C s
t � d S)N)r )r$ r
r
r �gc s zFooBar.gc C s d S)Nz_h()r
)r$ r
r
r �_he s z FooBar._hN)r+ r, r- r� r r r
r
r
r r ` s r c c s$ x t d � D] } | | Vq
Wd S)Nrv )r� )r� r
r
r �bazh s r c @ s. e Z d Z d Z d d � Z d d � Z d S)�
IteratorProxyr� c C s | S)Nr
)r$ r
r
r r� n s zIteratorProxy.__iter__c C s
| j d � S)Nr� )� _callmethod)r$ r
r
r r� p s zIteratorProxy.__next__N)r� )r+ r, r- Z _exposed_r� r� r
r
r
r r l s r c @ s e Z d Z d S)� MyManagerN)r+ r, r- r
r
r
r r s s r �Foo�callable�BarZexposedr� r Z proxytypec @ sF e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
S) �_TestMyManagerr0 c C sD t � } | j � | j | � | j � | j | j j d � d S)Nr )r ri �common�shutdownr8 �_processrT )r$ r0 r
r
r �test_mymanager s
z _TestMyManager.test_mymanagerc
C s: t � � } | j | � Wd QRX| j | j j d � d S)Nr )r r$ r8 r& rT )r$ r0 r
r
r �test_mymanager_context� s z%_TestMyManager.test_mymanager_contextc
C sH t � } | j � | � | j | � Wd QRX| j | j j d � d S)Nr )r ri r$ r8 r& rT )r$ r0 r
r
r �!test_mymanager_context_prestarted� s
z0_TestMyManager.test_mymanager_context_prestartedc se | j � � | j � � | j � } � f d d � d D� } � f d d � d D� } | j | d d g � | j | d d g � | j � j � d � | j t � j � | j � j d � d � | j t � j d � | j � j � d � | j � j
� d � | j � j d � d � | j � j d � d � | j t | � d d � t d
� D� � d S)
Nc s% g | ] } t � | � r | � q Sr
)rx )r� r\ )r� r
r r� � s z)_TestMyManager.common.<locals>.<listcomp>r� r r c s% g | ] } t � | � r | � q Sr
)rx )r� r\ )�barr
r r� � s zf()z_h()c S s g | ] } | | � q Sr
r
)r� r� r
r
r r� � s rv )r� r r )r� r r )
r r" r r8 r� r� r r r r� r rh r� )r$ r0 r Z foo_methodsZ bar_methodsr
)r* r� r r$ � s z_TestMyManager.commonN)r0 )r+ r, r- r= r' r( r) r$ r
r
r
r r# { s
r# c C s t S)N)�_queuer
r
r
r � get_queue� s r, c @ s e Z d Z d Z d S)� QueueManagerz$manager class used by server processN)r+ r, r- rB r
r
r
r r- � s r- c @ s e Z d Z d Z d S)�
QueueManager2z@manager class which specifies the same interface as QueueManagerN)r+ r, r- rB r
r
r
r r. � s r. Z xmlrpclibc @ s_ e Z d Z d
Z d d d d d d d g Z e d d � Z e d d
� � Z d d � Z d S)�_TestRemoteManagerr0 z hello worldNTg @u hallå världenu привіт світs
hall� v�rldenc C sK t d | d | d t � } | j � | j � } | j t | j � � d S)N�addressrL �
serializer)r. �
SERIALIZER�connectr, rD �tupler| )r^ r0 rL r0 r� r
r
r �_putter� s
z_TestRemoteManager._putterc C s� t j d � } t d t j j d f d | d t � } | j � | j d | j d | j
| f � } d | _ | j � t d | j
d | d t � } | j
� | j � } | j | j � | j � | j t | j t j � ~ | j � d S) N� r0 r rL r1 rW r' T)r �urandomr- r� r� �HOSTr2 ri rX r5 r0 rO r. r3 r, r8 rk r� r� r� rD r& rq r% )r$ rL r0 rn Zmanager2r� r
r
r � test_remote� s !
$
z _TestRemoteManager.test_remote)r0 ) r+ r, r- r= r| r� r� r5 r9 r
r
r
r r/ � s
r/ c @ s. e Z d Z e d d � � Z d d � Z d S)�_TestManagerRestartc C sB t d | d | d t � } | j � | j � } | j d � d S)Nr0 rL r1 z hello world)r- r2 r3 r, rD )r^ r0 rL r0 r� r
r
r r5 � s
z_TestManagerRestart._putterc C sd t j d � } t d t j j d f d | d t � } | j � } | j } | j j
� | j � | j d | j
d | j | f � } d | _ | j � | j � } | j | j � d � ~ | j � t d | d | d t � } y | j � Wnb t k
rU} zB | j t j k r� t j d
� t d | d | d t � } WYd d } ~ Xn X| j � d S) Nr6 r0 r rL r1 rW r' Tz hello worldg �?)r r7 r- r� r� r8 r2 Z
get_serverr0 Zlistenerr� ri rX r5 rO r, r8 rk r% �OSError�errnoZ
EADDRINUSEr& rq )r$ rL r0 Zsrvr�addrrn r� rm r
r
r �test_rapid_restart� s2 !
$
(z&_TestManagerRestart.test_rapid_restartN)r+ r, r- r� r5 r> r
r
r
r r: � s r: � c @ s` e Z d Z d% Z e d d � � Z d d � Z d d � Z d d
� Z d d � Z e d
d � � Z
e d d d � � Z e j
e d � d d � � Z e j
e d � e j e j d k d � e j e d k d � e j
e e d � d � d d � � � � � Z e d d � � Z e j
e d � e j e j d k d � d d! � � � Z d" d# � Z d$ S)&�_TestConnectionr/ r1 c C s8 x' t | j t � D] } | j | � q W| j � d S)N)r� �
recv_bytes�SENTINEL�
send_bytesr� )r^ rl �msgr
r
r �_echo% s z_TestConnection._echoc C sx | j � \ } } | j d | j d | f � } d | _ | j � d d d g } t d � } | d } t j d t t d � � � } | j d
k r� | j
t | j � � t
� | j
| j | � d � | j
| j � | � | j
| j | � d � | j
| j � | � | j d
k r�t j d d g d � } t | � d g d t | � } | j
| j | � d � | j
| j | � t | � | j � | j
t | � | � t j d d g d � } d g d t | � d g d t | � } | j
| j | � d � | j
| j | d | j � t | � | j � | j
t | � | � t t d
d � � } | j
| j | � d � y | j | � }
Wn> t j k
r�} z | j
| j | f � WYd d } ~ Xn X| j d |
� t | j � } | j
| � d � | j | j d � | j
| d � d � | j | j d � | j
| t � d � | j | j t � | j d � t j d � | j
| t � d � | j | j d � | j
| j � d � t d � d }
| j |
� | j
| j � |
� | j t! � | j" � | j d
k rj| j
| j# d � | j
| j$ d � | j% t& | j � | j% t& | j � | j' � d S)NrW r' Tr g @z hello worldrv r� r� r/ r r� � �( z expected BufferTooShort, got %sFg�������?�Xi � r� r i i )(r� rX rE rO ri r r� rh r� rH r8 rg �filenor~ r� r� rC rA rR Zrecv_bytes_into�itemsize� bytearrayr| ZBufferTooShortr' r� r! r� r6 r# r� r& rq rB r� �readable�writabler� �EOFErrorrl )r$ rl r� rn r� rD Zlongmsgr� �bufferr� r: rm r� Zreally_big_msgr
r
r �test_connection+ sr
!, )
z _TestConnection.test_connectionc C s� | j d d � \ } } | j | j d � d � | j | j � d � | j d k r� | j | j d � | j | j d � | j | j d � | j | j d � | j t | j d � | j t | j � | j t | j � d S)Nr� Fr r/ Tr_ )
r� r8 r� r� rH rM rN r� r; r� )r$ �reader�writerr
r
r �test_duplex_falsez s z!_TestConnection.test_duplex_falsec C s� | j � \ } } | j d | j d | f � } d | _ | j � | j � t d � } | j | � | j | j � | � | j t
� | j � | j � d S)NrW r' Tr` ) r� rX rE rO ri r� r rC r8 rA rB rl )r$ rl r� rn rD r
r
r �test_spawn_close� s
z _TestConnection.test_spawn_closec C s� | j d k r( | j d j | j � � t d � } | j � \ } } | j | � | j | j � | � | j | d � | j | j � | d d � � | j | d d � | j | j � | d d � � | j | d � | j | j � t d � � | j | d d � | j | j � t d � � | j t | j | d
� | j t | j | d d � | j t | j | d d � | j t | j | d � | j t | j | d
d � d S)Nr/ ztest not appropriate for {}ZabcdefghijklmnopqrstuvwxyzrA r� r� � r? r � � r r� � r r )
rH rI rJ r r� rC r8 rA r� r )r$ rD r4 r5 r
r
r �test_sendbytes� s&
z _TestConnection.test_sendbytesc C sZ y t j | � Wn> t k
rQ } z | j t j k r< d S� WYd d } ~ Xn Xd Sd S)NFT)r �fstatr; r< �EBADF)r^ r� rm r
r
r �_is_fd_assigned� s z _TestConnection._is_fd_assignedFc C s� | rE x< t d d � D]+ } | j | � s t j | j � | � q Wt j | � } t ro t j | t j � } t j
| | � t j | � d S)Nr r ) r� r] r �dup2rJ r � recv_handle�msvcrtZopen_osfhandler� r� r� )r^ rl �dataZcreate_dummy_fdsr� r� r
r
r �_writefd� s z_TestConnection._writefdz$test needs multiprocessing.reductionc C s | j d k r | j d � | j d d � \ } } | j d | j d | d f � } d | _ | j � | j t j j
t j j � t t j j d � �> } | j
� } t r� t j | � } t j | | | j � Wd QRX| j � t t j j d � � } | j | j � d � Wd QRXd S)
Nr/ z only makes sense with processesr� TrW r' s foo�wb�rb)rH rI r� rX rb rO ri r� r� r� r� r� r� rJ r` Z
get_osfhandler � send_handler] rl r8 r� )r$ rl r� rn r� r� r
r
r �test_fd_transfer� s
!
z _TestConnection.test_fd_transferr
z*test semantics don't make sense on Windowsr z)largest assignable fd number is too smallr^ ztest needs os.dup2()c C sg | j d k r | j d � | j d d � \ } } | j d | j d | d d f � } d | _ | j � | j t j j
t j j � t t j j d � �� } | j
� } x4 t d t � D] } | j | � s� Pq� W| j d
� t j | | � z t j | | | j � Wd t j | � XWd QRX| j � t t j j d � � } | j | j � d � Wd QRXd S) Nr/ z only makes sense with processesr� TrW r' s barrc r z2could not find an unassigned large file descriptorrd )rH rI r� rX rb rO ri r� r� r� r� r� r� rJ r� �MAXFDr] r� r r^ r re r] r� rl r8 r� )r$ rl r� rn r� r� Znewfdr
r
r �test_large_fd_transfer� s(
$
z&_TestConnection.test_large_fd_transferc C s t j | j � d � d S)Ns )r r� rJ )r$ rl r
r
r �_send_data_without_fd
s z%_TestConnection._send_data_without_fdz doesn't make sense on Windowsc C s� | j d k r | j d � | j d d � \ } } | j d | j d | f � } d | _ | j � | j t t j
| � | j � d S)Nr/ z only makes sense with processesr� TrW r' ) rH rI r� rX ri rO ri r� rt r r_ rl )r$ rl r� rn r
r
r �test_missing_fd_transfer
s
z(_TestConnection.test_missing_fd_transferc C s� | j � \ } } | �e | �W | j d � | j | j � d � | j d k rr | j | j � | j | j � Wd QRXWd QRX| j d k r� | j | j � | j | j � | j t | j � | j t | j � d S)Ni� r/ )
r� r� r8 r� rH rZ �closedrM r� r; )r$ r4 r5 r
r
r r�
s
z _TestConnection.test_contextN)r/ r1 )r+ r, r- r= r� rE rQ rT rU rZ r] rb r r �
HAS_REDUCTIONrf r� r� r rg rx r rh ri rj r� r
r
r
r r@ ! s* O
*r@ c @ s. e Z d Z d Z d d � Z d d � Z d S)�
_TestListenerr/ c C s_ xX | j j D]J } | j j d | � } | j | j � | j t | j j | j | � q
Wd S)N�family)�
connection�families�Listenerr� r� r� r; r0 )r$ rn �lr
r
r �test_multiple_bind.
s
z _TestListener.test_multiple_bindc C s� | j j � �_ } | j j | j � �@ } | j � �* } | j d � | j | j � d � Wd QRXWd QRXWd QRX| j d k r� | j t
| j � d S)Ni� r/ ) ro rq �Clientr0 �acceptr� r8 r� rH r� r; )r$ rr rF rs r
r
r r� 5
s
+z_TestListener.test_contextN)r/ )r+ r, r- r= rs r� r
r
r
r rm *
s rm c @ sL e Z d Z d Z e d d � � Z d d � Z d d � Z d d
� Z d S)
�_TestListenerClientr/ r1 c C s- | j j | � } | j d � | j � d S)Nr` )ro rt r� r� )r^ r0 rl r
r
r rY C
s
z_TestListenerClient._testc C s� x� | j j D]� } | j j d | � } | j d | j d | j f � } d | _ | j � | j � } | j | j
� d � | j � | j � q
Wd S)Nrn rW r' Tr` )
ro rp rq rX rY r0 rO ri ru r8 r� rl r� )r$ rn rr rn rl r
r
r �test_listener_clientI
s !
z(_TestListenerClient.test_listener_clientc C s� | j j � } | j d | j d | j f � } d | _ | j � t j d � | j � } | j
| j � d � | j � | j
� | j � d S)NrW r' Tr r` )ro rq rX rY r0 rO ri r& rq ru r8 r� r� rl )r$ rr rn rl r
r
r �test_issue14725T
s !
z#_TestListenerClient.test_issue14725c C s� x� | j j D]} } | j j d | � } | j j | j � } | j � } | j d � | j | j d � � | j � | j � | j � q
Wd S)Nrn s hellor )
ro rp rq rt r0 ru rC rM r� r� )r$ �famrr rF r4 r
r
r �test_issue16955d
s
z#_TestListenerClient.test_issue16955N)r/ r1 ) r+ r, r- r= r� rY rw rx rz r
r
r
r rv ?
s
rv c @ s| e Z d Z d Z d d � Z e d d � � Z d d � Z e d d
� � Z d d � Z e d
d � � Z
d d � Z d S)� _TestPollr/ r1 c C se | j � \ } } | j | j � d � | j d � | j | j � d � | j | j � d � d S)NF� T)r� r8 r� rC )r$ r4 r5 r
r
r �test_empty_strings
s
z_TestPoll.test_empty_stringc C s9 x( | D] } t j d � | j | � q W| j � d S)Ng�������?)r& rq rC r� )r^ rl �stringsr r
r
r �_child_stringsz
s
z_TestPoll._child_stringsc C s� d } | j � \ } } | j d | j d | | f � } | j � xQ | D]I } x$ t d � D] } | j d
� r] Pq] W| j � } | j | | � qJ W| j � d S) N� hellor| � a� b� bye� loprW r' �� g{�G�z�?)r� r| r� r� r| r� r| r� ) r� rX r ri r� r� rA r8 rl )r$ r~ r4 r5 rn r r� rx r
r
r � test_strings�
s !
z_TestPoll.test_stringsc C s | j d � d S)NrA )r� )r^ r� r
r
r �_child_boundaries�
s z_TestPoll._child_boundariesc C s� | j d � \ } } | j d | j d | f � } | j � t j d � d d g } x | D] } | j | � q] W| j � | j � | j | j
� | � d S)NFrW r' r_ s firsts second) r� rX r� ri r& rq rC r� rl rj rA )r$ r� r� rn r� r� r
r
r �test_boundaries�
s
z_TestPoll.test_boundariesc C s+ | j d � | j d � | j d � d S)Nr� r� s cd)rC )r^ r5 r
r
r �_child_dont_merge�
s
z_TestPoll._child_dont_mergec C s9 | j � \ } } | j | j d � d � | j | j d � d � | j d | j d | f � } | j � | j | j � d � | j | j d � d � | j | j d � d � | j | j � d � | j | j d � d � | j | j d � d � | j | j d � d � | j | j � d
� | j � d S) Ng Fg�������?rW r' r� g �?Tr� s cd)r� r8 r� rX r� ri rA rl )r$ r4 r5 rn r
r
r �test_dont_merge�
s
z_TestPoll.test_dont_mergeN)r/ r1 ) r+ r, r- r= r} r� r r� r� r� r� r� r
r
r
r r{ o
s r{ z$test needs multiprocessing.reductionc @ sv e Z d Z d Z e d d � � Z e d d � � Z e d d � � Z d d � Z e d
d � � Z d d
� Z
d S)�_TestPicklingConnectionsr/ c C s$ d d l m } | j d d � d S)Nr )�resource_sharerr rA )r| r� r� )r^ r� r
r
r r� �
s z&_TestPicklingConnections.tearDownClassc C s� x` | D]X } | j j d | � } | j | j � | j � } | j | � | j � | j � q Wt j � } | j t j j
d f � | j � | j | j � � | j � \ } } | j | � | j � | j � | j
� d S)Nrn r )ro rq r� r0 ru r� �socket�bindr� r� r8 �listen� getsocknamer� )r^ rl rp ry rr �new_connr= r
r
r � _listener�
s
z"_TestPicklingConnections._listenerc C s� xO t | j d � D]; \ } } | j j | � } | j | j � � | j � q W| j � \ } } t j � } | j | � | j | j � � | j � | j � d S)N)
r� r� ro rt r� r� r� r� r3 �sendall)r^ rl r0 rD �clientr
r
r �_remote�
s
z _TestPicklingConnections._remotec C s� | j j } | j � \ } } | j d | j d | | f � } d | _ | j � | j � | j � \ } } | j d | j d | f � } d | _ | j � | j � xh | D]` } d | j d � } | j
� }
| j |
| f � | j
� } | j | j
� | j
� � q� W| j d � t d � } | j
� }
| j |
| f � | j
� } g } x' | j
d � }
|
swP| j |
� qaWd j | � } | j | | j
� � | j � | j d � | j � | j � | j � | j � d S) NrW r' Tz This connection uses family %s�asciiz$This connection uses a normal socketrp r| )ro rp r� rX r� rO ri r� r� r r� r� r8 r� r r� rl )r$ rp ZlconnZlconn0Zlpr� Zrconn0Zrpry rD r0 r� Zbufr r
r
r �
test_pickling�
sH !
z&_TestPicklingConnections.test_picklingc C sZ | j � } | j d � | j � | j � } | j � } | j | d � | j � d S)Nz all is wellr_ )r� r� r� )r^ rl r� r� rD r
r
r � child_access s
z%_TestPicklingConnections.child_accessc C s� | j � \ } } | j d | j d | f � } d | _ | j � | j � | j d d � \ } } | j | � | j � | j | j � d � | j � | j d d � \ } } | j | � | j � | j d � | j � | j | j � d � d S)
NrW r' Tr� Fz all is wellZfoobarr_ Z foobarfoobar) r� rX r� rO ri r� r� r8 r� )r$ rl r� rn r� r� r
r
r � test_access) s
z$_TestPicklingConnections.test_accessN)r/ ) r+ r, r- r= r� r� r� r� r� r� r� r
r
r
r r� �
s . r� c @ s. e Z d Z d Z d d � Z d d � Z d S)� _TestHeapr/ c C s d } d } g } xw t | � D]i } t t j d d � d � } t j j | � } | j | � t | � | k r t j | � } | | =q Wt j j j
} g } d } | j j � | j
| j j � xc t | j j � � D]L }
xC |
D]; \ } } }
| j | j j | � | |
|
| d f � q� Wq� WxT | j D]I \ } } }
| j | j j | � | |
|
| d f � | |
| 7} q7W| j � x� t t | � d � D]o } | | d d � \ } } }
| | d d d � \ } } } | j | | k r| d k p|
| k � q�Wd S) Ni� r r r i� Zfree�occupiedr� )r� r~ �randomZlognormvariater| r' r( r� rR Z randrangeZ_heapr. r� r� r� rh Z _len_to_seqr| Z_arenas�indexZ_allocated_blocks�sortrM )r$ Z
iterationsZ maxblocksZblocksr� r� r5 r' r r� r� Zarenari r� ZnarenaZnstartZnstopr
r
r � test_heapL s:
! z_TestHeap.test_heapc C s� t j � s& t j � | j t j � t j � } | j t j | � t j d � xJ t d � D]< } t j j
d � } t j j
d � } | | _ | | _ q_ Wd S)Nrv i� r ) �gc� isenabled�enabler� �disable�
get_threshold�
set_thresholdr� r| r' r( Zbuddy)r$ Z
thresholdsr� r4 r5 r
r
r �test_free_from_gcs s
z_TestHeap.test_free_from_gcN)r/ )r+ r, r- r= r� r� r
r
r
r r� H s 'r� c @ s( e Z d Z d e f d e f g Z d S)�_Foorx ry N)r+ r, r- r r Z_fields_r
r
r
r r� � s r� c @ s[ e Z d Z d Z d d � Z e d d � � Z d d d � Z d d
� Z d d � Z d
S)�_TestSharedCTypesr/ c C s t s | j d � d S)Nz%requires multiprocessing.sharedctypes)r rI )r$ r
r
r rI � s z_TestSharedCTypes.setUpc C sz | j d 9_ | j d 9_ | j d 9_ | j d 9_ | j d 9_ x( t t | � � D] } | | d 9<q^ Wd S)Nr_ )r9 rx ry r� rR )r^ rx ry r� r� �stringr� r
r
r �_double� s z_TestSharedCTypes._doubleFc C sl t d d d | �} t t d d | �} t t d d d | �} | j d t t d � � d | �} | j d
d d | �} t d � | _ | j d
| j d | | | | | f � } d | _
| j � | j � | j
| j d � | j | j d � | j
| j d � | j | j d � x, t d � D] } | j | | | d � q-W| j
| j t d � � d S)Nr� r� r� g �?g @r� r_ rs rv rF r� r` rW r' T� g @r� g @Z
hellohellogUUUUUU�?gUUUUUU�?)r r r� r� rh r� r r9 rX r� rO ri rl r8 r3 rx ry ) r$ r� rx ry r� r� r� rn r� r
r
r �test_sharedctypes� s" $*
z#_TestSharedCTypes.test_sharedctypesc C s | j d d � d S)Nr� T)r� )r$ r
r
r �test_synchronize� s z"_TestSharedCTypes.test_synchronizec C sW t d d � } t | � } d | _ d | _ | j | j d � | j | j d � d S)Nr_ g @r )r� r rx ry r8 r3 )r$ r� r* r
r
r � test_copy� s z_TestSharedCTypes.test_copyN)r/ )
r+ r, r- r= rI r� r� r� r� r� r
r
r
r r� � s
r� c @ sX e Z d Z d
Z d d � Z d d � Z e d d � � Z d d � Z d
d � Z d S)�
_TestFinalizer/ c C s# t j j � | _ t j j � d S)N)r �_finalizer_registryr �registry_backupr$ )r$ r
r
r rI � s z_TestFinalize.setUpc C s' | j t j � t j j | j � d S)N)rZ r r� �updater� )r$ r
r
r rK � s z_TestFinalize.tearDownc C s� Gd d � d t � } | � } t j | | j d d �~ | � } t j | | j d d �} | � | � ~ | � } t j | | j d d �| � } t j | | j d d d d �| � } t j | | j d d d d �| � } t j | | j d d d d �| � }
t j |
| j d d d d �t j d | j d d d d �t j d | j d d d d �t j � | j � t j d � d S) Nc @ s e Z d Z d S)z)_TestFinalize._test_finalize.<locals>.FooN)r+ r, r- r
r
r
r r � s r r' r4 r5 rF �d10� exitpriorityr �d01r �d02�d03rm rv �STOPrp )r4 )r5 )rF )r� )r� )r� )r� )rm i����)r� i����)r� r �Finalizer� Z_exit_functionr� r �_exit) r^ rl r r4 r5 Zclose_brF r� r� r� r� r
r
r �_test_finalize� s0
z _TestFinalize._test_finalizec C s� | j � \ } } | j d | j d | f � } d | _ | j � | j � d d � t | j d � D� } | j | d d d d
d d d
g � d S)NrW r' Tc S s g | ] } | � q Sr
r
)r� r� r
r
r r� s z/_TestFinalize.test_finalize.<locals>.<listcomp>r� r4 r5 r� r� r� r� rm ) r� rX r� rO ri rl r� r� r8 )r$ rl r� rn r� r
r
r �
test_finalize� s
z_TestFinalize.test_finalizec s, d d � � G� f d d � d t � � d � d � � � f d d � } � � � f d d � } t j � } t j � } z� t j d
� t j d d d � t j d | � t j d | � g } t j
j | � � t j
d
� d � Wd QRX� d k r� � � Wd t j | � t j | � t j � Xd S)Nc S s d S)Nr
r
r
r
r �cb s z,_TestFinalize.test_thread_safety.<locals>.cbc s" e Z d Z � f d d � Z d S)z-_TestFinalize.test_thread_safety.<locals>.Fooc s/ | | _ t j | � d t j d d � �d S)Nr� r rp )r� r r� r� Zrandint)r$ )r� r
r r% s z6_TestFinalize.test_thread_safety.<locals>.Foo.__init__N)r+ r, r- r% r
)r� r
r r s r Fc sd x] � s_ t j t j � d � y t j � Wq t k
r[ } z | � WYd d } ~ Xq Xq Wd S)Ng�������?)r& rq r� r Z_run_finalizersr� )rm )r� �finishr
r �run_finalizers s z8_TestFinalize.test_thread_safety.<locals>.run_finalizersc s i } xr � sz y0 � f d d � t d � D� | t j d � <Wq t k
rv } z | � | j � WYd d } ~ Xq Xq Wd S)Nc s h | ] } � � � q Sr
r
)r� r� )r r
r � <setcomp>& s zL_TestFinalize.test_thread_safety.<locals>.make_finalizers.<locals>.<setcomp>rv rA )r� r� Z getrandbitsr� r$ )rs rm )r r� r� r
r �make_finalizers s 0z9_TestFinalize.test_thread_safety.<locals>.make_finalizersg���ư>rA rW g @T)r� r� �getswitchintervalr� r� �setswitchintervalr� r� r r� r� Z
start_threadsr& rq �collect)r$ r� r� Z old_intervalZ
old_thresholdr1 r
)r r� r� r� r �test_thread_safety s*
z _TestFinalize.test_thread_safetyN)r/ )
r+ r, r- r= rI rK r� r� r� r� r
r
r
r r� � s & r� c @ s( e Z d Z d d � Z d d � Z d S)�_TestImportStarc C s� d d l } t j j t j � } t j j | d � } | j | � } d d � | D� } d d � | D� } | j d � | j d � | S)Nr z*.pyc S s6 g | ], } t j j t j j | � d � d � q S)r r )r �path�splitext�split)r� r� r
r
r r� H s z4_TestImportStar.get_module_names.<locals>.<listcomp>c S s g | ] } d | � q S)zmultiprocessing.r
)r� rQ r
r
r r� I s zmultiprocessing.__init__r| ) �globr r� �dirnamer| �__file__rl r� r� )r$ r� Zfolder�pattern�files�modulesr
r
r �get_module_namesC s
z _TestImportStar.get_module_namesc C s� | j � } t j d k rE | j d � | j d � | j d � n | j d � t se | j d � t d k r~ | j d � xu | D]m } t | � t j | } | j t | d � | � x4 | j
D]) } | j t | | � d | | f � q� Wq� Wd S) Nr
zmultiprocessing.popen_forkz multiprocessing.popen_forkserverz!multiprocessing.popen_spawn_posixz!multiprocessing.popen_spawn_win32z multiprocessing.sharedctypes�__all__z %r does not have attribute %r) r� r� r r� rl r r� r� rM rx r� )r$ r� r\ �mod�attrr
r
r � test_importN s$
z_TestImportStar.test_importN)r+ r, r- r� r� r
r
r
r r� A s r� c @ s@ e Z d Z d Z d d � Z e d d � � Z d d � Z d S)
� _TestLoggingr/ c C sZ t j � } | j t j � | j | d k � | j d � | j d � | j t � d S)Nzthis will not be printedz
nor will this) r| �
get_logger�setLevelr �
SUBWARNINGrM �debug�info� LOG_LEVEL)r$ �loggerr
r
r �test_enable_loggingp s
z _TestLogging.test_enable_loggingc C s# t j � } | j | j � � d S)N)r| r� r� ZgetEffectiveLevel)r^ rl r� r
r
r � _test_levelx s z_TestLogging._test_levelc C s d } d } t j � } t j � } | j } t j d d � \ } } | j | � | j d | j d | f � } d | _ | j
� | j | | j � � | j t j
� | j | � | j d | j d | f � } d | _ | j
� | j | | j � � | j | � | j d t � d S) Nr6 �% r� FrW r' T�level)r| r� �loggingZ getLoggerr� r� r� rX r� rO ri r8 r� ZNOTSETr� ) r$ ZLEVEL1ZLEVEL2r� Z root_loggerZ
root_levelrR rS rn r
r
r �
test_level} s&
z_TestLogging.test_levelN)r/ )r+ r, r- r= r� r� r� r� r
r
r
r r� l s r� c @ sO e Z d Z d Z e d d � � Z e j e e d � d � d d � � Z
d S)
�_TestPollEintrr/ c C s$ t j d � t j | t j � d S)Ng�������?)r& rq r r ry �SIGUSR1)r^ r] r
r
r �_killer� s
z_TestPollEintr._killerr� zrequires SIGUSR1c s� d g � � f d d � } t j � } t j t j | � } z� | j d | j d | f � } | j � z3 | j d t j d d � } | j � | j � Wd | j � X| j
� d � | j | j d � Wd t j t j | � Xd S) NFc s d � d <d S)NTr r
)r' )�
got_signalr
r �record� s z._TestPollEintr.test_poll_eintr.<locals>.recordrW r' r_ r )r_ )
r rS ry r� rX r� ri r& rq rl rM r8 rT )r$ r� r] Z
oldhandlerZkillerrn r
)r� r �test_poll_eintr� s
z _TestPollEintr.test_poll_eintrN)r/ ) r+ r, r- r= r� r� r r rx ry r� r
r
r
r r� � s r� c @ s. e Z d Z e j e d � d d � � Z d S)�TestInvalidHandlezskipped on Windowsc C so t j j d � } z- y | j � Wn t t f k
r= Yn XWd d | _ X| j t t f t j j d � d S)Ni�M�r r )r| ro Z
Connectionr� r r; Z_handler� )r$ rl r
r
r �test_invalid_handles� s
z&TestInvalidHandle.test_invalid_handlesN)r+ r, r- r r� �WIN32r� r
r
r
r r� � s r� c @ s( e Z d Z d d � Z d d � Z d S)� OtherTestc C s< Gd d � d t � } | j t j t j j | � d � d S)Nc @ s( e Z d Z d d � Z d d � Z d S)zFOtherTest.test_deliver_challenge_auth_failure.<locals>._FakeConnectionc S s d S)Ns something bogusr
)r$ r� r
r
r rA � s zQOtherTest.test_deliver_challenge_auth_failure.<locals>._FakeConnection.recv_bytesc S s d S)Nr
)r$ ra r
r
r rC � s zQOtherTest.test_deliver_challenge_auth_failure.<locals>._FakeConnection.send_bytesN)r+ r, r- rA rC r
r
r
r �_FakeConnection� s r� s abc)r� r� r| �AuthenticationErrorro Zdeliver_challenge)r$ r� r
r
r �#test_deliver_challenge_auth_failure� s z-OtherTest.test_deliver_challenge_auth_failurec C s< Gd d � d t � } | j t j t j j | � d � d S)Nc @ s4 e Z d Z d d � Z d d � Z d d � Z d S)zEOtherTest.test_answer_challenge_auth_failure.<locals>._FakeConnectionc S s
d | _ d S)Nr )rW )r$ r
r
r r% � s zNOtherTest.test_answer_challenge_auth_failure.<locals>._FakeConnection.__init__c S s? | j d 7_ | j d k r( t j j S| j d k r; d Sd S)Nr r_ s something bogusr| )rW r| ro Z CHALLENGE)r$ r� r
r
r rA � s
zPOtherTest.test_answer_challenge_auth_failure.<locals>._FakeConnection.recv_bytesc S s d S)Nr
)r$ ra r
r
r rC � s zPOtherTest.test_answer_challenge_auth_failure.<locals>._FakeConnection.send_bytesN)r+ r, r- r% rA rC r
r
r
r r� � s r� s abc)r� r� r| r� ro Zanswer_challenge)r$ r� r
r
r �"test_answer_challenge_auth_failure� s z,OtherTest.test_answer_challenge_auth_failureN)r+ r, r- r� r� r
r
r
r r� � s
r� c C s | j d 7_ d S)Nr )r� )�nsr
r
r � initializer
s r� c @ s@ e Z d Z d d � Z d d � Z d d � Z d d � Z d S)
�TestInitializersc C s1 t j � | _ | j j � | _ d | j _ d S)Nr )r| �Manager�mgrr� r� r� )r$ r
r
r rI
s zTestInitializers.setUpc C s | j j � | j j � d S)N)r� r% rl )r$ r
r
r rK
s
zTestInitializers.tearDownc C si t j j � } | j t | j d � | j t | j f � | j | j j d � | j
� | j � d S)Nr ) r| ZmanagersZ SyncManagerr� � TypeErrorri r� r� r8 r� r% rl )r$ rQ r
r
r �test_manager_initializer
s
z)TestInitializers.test_manager_initializerc C sb | j t t j d d �t j d t | j f � } | j � | j � | j | j j d � d S)Nr� r )
r� r� r| r� r� r� r� rl r8 r� )r$ rn r
r
r �test_pool_initializer
s
z&TestInitializers.test_pool_initializerN)r+ r, r- rI rK r� r� r
r
r
r r�
s r� c C s2 y | j d d � } Wn t j k
r- Yn Xd S)N�blockF)rk r� r� )rE �itemr
r
r �_this_sub_process&
s r� c C sH t j � } t j d t d | f � } d | _ | j � | j � d S)NrW r' T)r| rc rX r� rO ri rl )rE r� ZsubProcr
r
r �
_test_process,
s
r� c C s | | S)Nr
)rx r
r
r �_afunc3
s r� c C sQ t j d d � } | j t d d d d d d d g � } | j � | j � d S) Nr/ r� r r_ r� rA r� r� )r| r� r� r� r� rl )r� rx r
r
r �pool_in_process6
s '
r� c @ sF e Z d Z d d � Z e d d � � Z d d � Z d d � Z d S)
�
_file_likec C s | | _ d | _ d S)N)� _delegate�_pid)r$ Zdelegater
r
r r% =
s z_file_like.__init__c C s4 t j � } | | j k r- | | _ g | _ | j S)N)r rS r �_cache)r$ r] r
r
r �cacheA
s
z_file_like.cachec C s | j j | � d S)N)r r� )r$ ra r
r
r r� J
s z_file_like.writec C s) | j j d j | j � � g | _ d S)Nr? )r r� rl r r )r$ r
r
r �flushM
s z_file_like.flushN)r+ r, r- r% �propertyr r� r r
r
r
r r� <
s r� c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�TestStdinBadfiledescriptorc C s? t j � } t j d t d | f � } | j � | j � d S)NrW r' )r| rc rX r� ri rl )r$ r� r� r
r
r �test_queue_in_processS
s
z0TestStdinBadfiledescriptor.test_queue_in_processc C s* t j d t � } | j � | j � d S)NrW )r| rX r� ri rl )r$ rn r
r
r �test_pool_in_processY
s
z/TestStdinBadfiledescriptor.test_pool_in_processc si t j � } t | � � � j d � t j d � f d d � � } � j � | j � d k se t � d S)Nr� rW c s
� j � S)N)r r
)�fliker
r r b
s z:TestStdinBadfiledescriptor.test_flushing.<locals>.<lambda>) �io�StringIOr� r� r| rX r r� rC )r$ Zsior� r
)r r �
test_flushing^
s
z(TestStdinBadfiledescriptor.test_flushingN)r+ r, r- r r r r
r
r
r r Q
s r c @ s� e Z d Z e d d � � Z d d d � Z e d d � � Z d d d � Z d
d � Z d d
� Z d d � Z
e d d � � Z d d � Z d d � Z
d S)�TestWaitc C s[ xJ t d � D]< } | r0 t j t j � d � | j | t j � f � q
W| j � d S)Nrv g�������?)r� r& rq r� r� r rS r� )r^ r� �slowr� r
r
r �_child_test_waiti
s
zTestWait._child_test_waitFc si d d l m } g } g � g } x� t d � D]� } t j d d � \ } } t j d | j d | | f � } d | _ | j � | j � | j
| � � j
| � | j | j � q/ Wxj | r%x] | | � D]O } y | j
� } Wn) t k
r| j | � | j � Yq� X| j
| � q� Wq� W| j � t � f d d
� t d � D� � }
| j | |
� d S) Nr )r r� r� FrW r' Tc 3 s+ | ]! } � D] } | | j f Vq
q d S)N)r] )r� r� rn )�procsr
r r� �
s z%TestWait.test_wait.<locals>.<genexpr>rv )�multiprocessing.connectionr r� r| r� rX r rO ri r� r� r� rl r� rO r� r� r� r8 ) r$ r r �readersZmessagesr� r� r� rn rD r� r
)r r � test_waitq
s0 !
%zTestWait.test_waitc C su t j � } | j | � xK t d � D]= } | rI t j t j � d � | j d | j d � � q& W| j � d S)Nrv g�������?z%s
r� ) r� r3 r� r& rq r� r� r r� )r^ r0 r r r� r
r
r �_child_test_wait_socket�
s
z TestWait._child_test_wait_socketc C s� d d l m } t j � } | j t j j d f � | j � | j � } g } g } i } xe t d � D]W } t
j d | j d | | f � } d | _
| j � | j | � | j | j � qj Wx= t d � D]/ } | j � \ }
} | j |
� g | |
<q� W| j � xa | rrxT | | � D]F }
|
j d � } | sZ| j |
� |
j � q%| |
j | � q%WqWd j d d
� t d � D� � j d � }
x- | j � D] } | j d
j | � |
� q�Wd S)Nr )r r� rW r' Tr6 r? c s s | ] } d | Vq d S)z%s
Nr
)r� r� r
r
r r� �
s z,TestWait.test_wait_socket.<locals>.<genexpr>rv r� r| )r r r� r� r� r� r8 r� r� r� r| rX r rO ri r� r� rl ru r� r� r� r r| r8 )r$ r r rr r= r r Zdicr� rn r� r1 rD r� �vr
r
r �test_wait_socket�
s<
+zTestWait.test_wait_socketc C s | j d � d S)NT)r )r$ r
r
r �test_wait_slow�
s zTestWait.test_wait_slowc C s | j d � d S)NT)r )r$ r
r
r �test_wait_socket_slow�
s z TestWait.test_wait_socket_slowc C s� d d l m } d } t j � \ } } t j � } | | | g | � } t j � | } | j | g � | j | | d � | j | | d � | j d � t j � } | | | g d � } t j � | } | j | | g � | j | d � d S)Nr )r rA r_ g �?r� g�������?) r r r| r� r& r8 r� �
assertGreaterr� )r$ r r� r4 r5 ri r: r� r
r
r �test_wait_timeout�
s
zTestWait.test_wait_timeoutc C s | j � t j | � d S)N)r� r& rq )r^ r Zperiodr
r
r �signal_and_sleep�
s
zTestWait.signal_and_sleepc C s d d l m } d } d d � } t j d � } t j � \ } } t j d | j d | | f � } | j � | j | j t
� | j | j d d � � t
j
� } | | | j | g | d � } t
j
� | }
| j | | j g � | j |
| d
� | j |
| d
� | j d � t
j
� } | | | j | g d � } t
j
� | }
| j | | � | | j | g � � | j |
d � | j d � t
j
� } | | | j | g d � } t
j
� | }
| j | | � | | | j | g � � | j |
d � | j � | j � d S) Nr )r r� c S s t | d d d � �S)Nr c S s
t | � S)N)r� )rx r
r
r r �
s z>TestWait.test_wait_integer.<locals>.<lambda>.<locals>.<lambda>)r� )rr r
r
r r �
s z,TestWait.test_wait_integer.<locals>.<lambda>rW r' r r� r_ g�������?)r r r| r r� rX r ri rP r� r~ rM r� r& r8 r� r r� rw rl ) r$ r r� Zsorted_r r4 r5 rn ri r: r� r
r
r �test_wait_integer�
s<
%
(
zTestWait.test_wait_integerc C s� d d l m } t j � \ } } t j � } | | g d d �} t j � | } | j | g � | j | d � | j � | j � d S)Nr )r r r r )r r r| r� r& r8 r� r� )r$ r r4 r5 r) r: r
r
r �test_neg_timeout s
zTestWait.test_neg_timeoutN)r+ r, r- r� r r r r r r r r r r r
r
r
r r
g
s
% +r
c @ sL e Z d Z e j e d � d d � � Z e j e d � d d � � Z d S)�TestInvalidFamilyzskipped on Windowsc C s+ | j t � � t j j d � Wd QRXd S)Nz\\.\test)r� r r| ro rq )r$ r
r
r �test_invalid_family s z%TestInvalidFamily.test_invalid_familyz skipped on non-Windows platformsc C s+ | j t � � t j j d � Wd QRXd S)Nz/var/test.pipe)r� r r| ro rq )r$ r
r
r �test_invalid_family_win32! s z+TestInvalidFamily.test_invalid_family_win32N) r+ r, r- r r� r� r r r r
r
r
r r s r c @ s@ e Z d Z e d d � � Z e d d � � Z d d � Z d S)� TestFlagsc C s | j t t j � � d S)N)r� r4 r� �flags)r^ rl r
r
r �run_in_grandchild+ s zTestFlags.run_in_grandchildc C s� d d l } t j d d � \ } } t j d | j d | f � } | j � | j � } | j � | j � | j � t t
j � | f } t | j
| � � d S)Nr r� FrW r' )�jsonr| r� rX r# ri r� rl r� r4 r� r" �print�dumps)r^ r$ r� r� rn �grandchild_flagsr" r
r
r � run_in_child/ s
zTestFlags.run_in_childc C sx d d l } d d l } d d } | j t j d d d d | g � } | j | j d � � \ } } | j | | � d S) Nr z2from test._test_multiprocessing import TestFlags; zTestFlags.run_in_child()z-Ez-Sz-Oz-cr� )r$ �
subprocessZ check_outputr� �
executable�loads�decoder8 )r$ r$ r) �progra Z child_flagsr' r
r
r �
test_flags< s zTestFlags.test_flagsN)r+ r, r- r� r# r( r. r
r
r
r r! * s
r! c @ s. e Z d Z e d d � � Z d d � Z d S)� TestTimeoutsc C sQ t j d � | j d � | j � t j j | � } | j d � | j � d S)Nr r� i� )r& rq r� r� r| ro rt )r^ �childr0 rl r
r
r �
_test_timeoutK s
zTestTimeouts._test_timeoutc C s� t j � } z� t j d � t j d d � \ } } t j j d d � } t j d | j d | | j f � } | j
� | j � | j | j
� d � | j � | j � } | j | j
� d � | j � | j � | j d
� Wd t j | � Xd S) Ng�������?r� Trn �AF_INETrW r' r� i� rv )r� ZgetdefaulttimeoutZsetdefaulttimeoutr| r� ro rq rX r1 r0 ri r� r8 r� ru rl )r$ Z old_timeout�parentr0 rr rn rl r
r
r r� T s"
zTestTimeouts.test_timeoutN)r+ r, r- r� r1 r� r
r
r
r r/ J s r/ c @ s e Z d Z d d � Z d S)�TestNoForkBombc C s� t j � } t j j t j j t � d � } | d k r} t j j j
| | � \ } } } | j | d � | j d | � nG t j j j
| | � \ } } } | j | j � d � | j | d � d S)Nzmp_fork_bomb.py�forkr| s RuntimeErrors 123)r| �get_start_methodr r� rl r� r� r� r� �
script_helperZassert_python_failurer8 rj �assert_python_okr� )r$ Zsmr\ �rc�outr� r
r
r �test_noforkbombm s ! !!z TestNoForkBomb.test_noforkbombN)r+ r, r- r; r
r
r
r r4 l s r4 c @ s. e Z d Z e d d � � Z d d � Z d S)�TestForkAwareThreadLockc C s| | d k rX t j d | j d | d | f � } | j � | j � | j d d � n | j t t j � � | j � d S)Nr rW r' r rA )
r| rX r0 ri r� rl r� rR r �_afterfork_registry)r^ r6 rl rn r
r
r r0 � s %
z TestForkAwareThreadLock.childc C s� t j d � \ } } t j � } t t j � } t j d | j d d | f � } | j � | j � | j
� } | j d d � | j | | � d S)NFrW r' rA r )
r| r� r ZForkAwareThreadLockrR r= rX r0 ri r� r� rl ZassertLessEqual)r$ r� r� rr Zold_sizern Znew_sizer
r
r r� � s !
z!TestForkAwareThreadLock.test_lockN)r+ r, r- r� r0 r� r
r
r
r r< } s r< c @ sF e Z d Z d d � Z d d � Z e d d � � Z d d � Z d S)
� TestCloseFdsc C s� t r t j � j � St j � j � } g } x, | d k r\ | j | � t j | � } q1 Wx | D] } t j | � qd W| Sd S)Nr )r� r� �detachr� r �dupr� )r$ r� Zto_closerx r
r
r �get_high_socket_fd� s
z TestCloseFds.get_high_socket_fdc C s0 t r t j d | � j � n
t j | � d S)NrJ )r� r� r� r )r$ r� r
r
r r� � s zTestCloseFds.closec C so y t j | t j t j � } Wn2 t k
rS } z | j | � WYd d } ~ Xn X| j � | j d � d S)N)r� Zfromfdr2 Z SOCK_STREAMr� r� r� )r^ rl r� r rm r
r
r �_test_closefds� s
zTestCloseFds._test_closefdsc C s t s t j d � � t j � \ } } | j � } zU t j d | j d | | f � } | j � | j � | j
� } | j d d � Wd | j | � | j � | j � Xt j � d k r� | j
| d � nA d } | j | t � | j | j t j k p | j | k | � d S)Nzrequires fd picklingrW r' r rA r5 i6' )rl r r r| r� rA rX rB ri r� r� rl r6 r� rP r; rM r< r\ Zwinerror)r$ rR rS r� rn rm Z WSAENOTSOCKr
r
r � test_closefd� s(
zTestCloseFds.test_closefdN)r+ r, r- rA r� r� rB rC r
r
r
r r> � s
r> c @ s� e Z d Z e d d � � Z e j e e d � d � d d � � Z e d d � � Z
e j e e d � d � d d
� � Z d S) �TestIgnoreEINTRc C sZ d d � } t j t j | � | j d � | j � } | j | � | j d d � d S)Nc S s d S)Nr
)�signum�framer
r
r ru � s z-TestIgnoreEINTR._test_ignore.<locals>.handler�ready� xi i )ry r� r� r� rC )r^ rl ru rx r
r
r � _test_ignore� s
z TestIgnoreEINTR._test_ignorer� zrequires SIGUSR1c C s t j � \ } } z� t j d | j d | f � } d | _ | j � | j � | j | j � d � t j
d � t j | j
t j � t j
d � | j d � | j | j � d � t j
d � t j | j
t j � | j | j � d d � t j
d � | j � Wd | j � Xd S)
NrW r' TrG g�������?i� rH i i )r| r� rX rI rO ri r� r8 r� r& rq r r r] ry r� r� rA rl )r$ rl r� rn r
r
r � test_ignore� s&
zTestIgnoreEINTR.test_ignorec C se d d � } t j t j | � t j j � �0 } | j | j � | j � } | j d � Wd QRXd S)Nc S s d S)Nr
)rE rF r
r
r ru s z6TestIgnoreEINTR._test_ignore_listener.<locals>.handler�welcome)ry r� r| ro rq r� r0 ru )r^ rl ru rr r4 r
r
r �_test_ignore_listener s z%TestIgnoreEINTR._test_ignore_listenerc C s� t j � \ } } z� t j d | j d | f � } d | _ | j � | j � | j � } t j d � t
j | j t
j � t j d � t j j | � } | j | j � d � | j � Wd | j � Xd S)NrW r' Tg�������?rK )r| r� rX rL rO ri r� r� r& rq r r r] ry r� ro rt r8 rl )r$ rl r� rn r0 r� r
r
r �test_ignore_listener s
z$TestIgnoreEINTR.test_ignore_listenerN) r+ r, r- r� rI r r rx ry rJ rL rM r
r
r
r rD � s
'
rD c @ s^ e Z d Z e d d � � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
S)�TestStartMethodc C s | j t j � � d S)N)r� r| r6 )r^ rl r
r
r �_check_context s z TestStartMethod._check_contextc C s� | j d d � \ } } | j d | j d | f � } | j � | j � | j � } | j � | j � | j | | j � � d S)Nr� FrW r' ) r� rX rO ri r� r� rl r8 r6 )r$ �ctxr� r� rn Z child_methodr
r
r �
check_context# s
z TestStartMethod.check_contextc
C s� x� d D]� } y t j | � } Wn t k
r7 w Yn X| j | j � | � | j | j � | � | j t | j d � | j t | j d � | j | � q Wd S)Nr5 �spawn�
forkserver)r5 rR rS ) r| � get_contextr r8 r6 r� r� �set_start_methodrQ )r$ �methodrP r
r
r r� - s
z TestStartMethod.test_contextc C s$ t j t � d } t j � } z� x� d D]� } y t j | d d �Wn t k
r] w) Yn X| j t j � | � t j � } | j | j � | � | j t | � j
j � j | � � | j | j
j
j � j | � � | j t � | d 7} q) WWd t j | d d �X| j | d � d S) Nr r5 rR rS �forceTr )r5 rR rS )r| �set_forkserver_preload�PRELOADr6 rU r r8 rT rM rg r+ �lower�
startswithrX rQ r� )r$ rW Z
old_methodrV rP r
r
r � test_set_get9 s&
%
z TestStartMethod.test_set_getc C sc t j � } t j d k r1 | j | d g � n. | j | d d g k p[ | d d d g k � d S)Nr
rR r5 rS )r| Zget_all_start_methodsr� r r8 rM )r$ Zmethodsr
r
r � test_get_allO s
z TestStartMethod.test_get_allc C s� t j � d k r | j d � t j j t j j t � d � } t j j
j | � \ } } } | j � } | j � } | j
� d k s� | d k r� t | � t | � | j d � d S)NrS z*test only relevant for 'forkserver' methodz
mp_preload.py�okr? z(failed spawning forkserver or grandchild)r| r6 rI r r� rl r� r� r� r� r7 r8 r, r� r% r� )r$ r\ r9 r: r� r
r
r �test_preload_resourcesW s
!
z&TestStartMethod.test_preload_resourcesN)
r+ r, r- r� rO rQ r� r\ r] r_ r
r
r
r rN s
rN z*test semantics don't make sense on Windowsc @ s e Z d Z d d � Z d S)�TestSemaphoreTrackerc C s� d d l } d } t j � \ } } | j t j d | | | f g d | g d | j �} t j | � t | d d d ��= } | j � j
� j d � } | j � j
� j d � } Wd QRXt j
| � | j � | j � t j d
� | j t � � } t j
| � Wd QRX| j | j j t j t j f � | j j � j d � }
| j j � d } | j |
| � | j |
d
| � d S)Nr aM if 1:
import multiprocessing as mp, time, os
mp.set_start_method("spawn")
lock1 = mp.Lock()
lock2 = mp.Lock()
os.write(%d, lock1._semlock.name.encode("ascii") + b"\n")
os.write(%d, lock2._semlock.name.encode("ascii") + b"\n")
time.sleep(10)
z-cZpass_fdsr� rd r� Tr� g @zutf-8z9semaphore_tracker: There appear to be 2 leaked semaphoresz semaphore_tracker: %r: \[Errno) r) r �pipe�Popenr� r* �PIPEr� r� �readliner� r, r Z
sem_unlinkrw r r& rq r� r; rj Z exceptionr< �ENOENTZEINVALr� r� Z assertRegex) r$ r) �cmdr� r� rn r� Zname1Zname2rP r� r� r
r
r �test_semaphore_trackerk s.
"
"
z+TestSemaphoreTracker.test_semaphore_trackerN)r+ r, r- rg r
r
r
r r` h s r` c @ s. e Z d Z e d d � � Z d d � Z d S)�TestSimpleQueuec
C sF | j � z* | j | j � � | j | j � � Wd | j � Xd S)N)r rD r� r� )r^ r� r� r� r
r
r � _test_empty� s
zTestSimpleQueue._test_emptyc C s� t j � } t j � } t j � } t j d | j d | | | f � } d | _ | j � | j | j � � | j � | j
� | j | j � � | j | j
� d � | j | j
� d � | j | j � � | j � d S)NrW r' TF)r| Z SimpleQueuerd rX ri rO ri rM r� r� r rZ r8 rk rl )r$ r� r� r� r� r
r
r �
test_empty� s
zTestSimpleQueue.test_emptyN)r+ r, r- r� ri rj r
r
r
r rh � s
rh c @ s' e Z d Z d Z e j Z e j Z e e j � Z e e j � Z e e j
� Z
e e j � Z e e j � Z e e j
� Z
e e j � Z e e j � Z e e j � Z e e j � Z e e j � Z e e j � Z e e j � Z e e j � Z e e j � Z e e j � Z e e j � Z d S)r r/ N)r+ r, r- rH r| rX ro � staticmethodrK rf r� r� rc r� r� r� r r r rd rE r r� r r� r
r
r
r r � s( r c @ s{ e Z d Z d Z e j Z e e j d � � Z e e j d � � Z
e e j d � � Z e e j d � � Z e e j d � � Z
e e j d � � Z e e j d � � Z e e j d � � Z e e j d
� � Z e e j d � � Z e e j d � � Z e e j d
� � Z e e j d � � Z e e j d � � Z e d d � � Z e d d � � Z e d d � � Z d S)� ManagerMixinr0 z
manager.Queuezmanager.JoinableQueuez manager.Lockz
manager.RLockzmanager.Semaphorezmanager.BoundedSemaphorezmanager.Conditionz
manager.Eventzmanager.Barrierz
manager.Valuez
manager.Arrayz manager.listz manager.dictzmanager.Namespacec O s | j j | | � S)N)r0 r� )r^ r' r( r
r
r r� � s zManagerMixin.Poolc C s t j � | _ d S)N)r| r� r0 )r^ r
r
r r� � s zManagerMixin.setUpClassc C s� d } x? t t j � � d k rG | d k rG t j | � | d 9} q Wt j � | j j � d k r� t d � t | j j
� � | j j � | j j � d | _ d S)Ng{�G�z�?r rA r_ r z5Shared objects which still exist at manager shutdown:)
rR r| rf r& rq r� r� r0 Z_number_of_objectsr% Z _debug_infor% rl )r^ r) r
r
r r� � s '
zManagerMixin.tearDownClassN)r+ r, r- rH r| rX r �operator�
attrgetterrc r� r� r� r r r rd rE r r� rh r� r� r� r� r� r� r
r
r
r rl � s& rl c @ s< e Z d Z d Z e j j Z e j j Z e e j j � Z e e j j
� Z
e e j j � Z e e j j � Z e e j j
� Z
e e j j � Z e e j j � Z e e j j � Z e e j j � Z e e j j � Z e e j j � Z e e j j � Z e e j j � Z e e j j � Z e e j j � Z d S)� ThreadsMixinr1 N)r+ r, r- rH r| ZdummyrX ro rk rK rf r� r� rc r� r� r� r r r rd rE r r� r
r
r
r ro � s$ ro c
s� | d } t � } d d d h } xG| j � D]9\ } } t | t � sM q/ t | t � r| t k rk q/ t | j � | k s� t | j � � x� | j D]z } d | j � | d d � } | | j � d } Gd d � d | | t
j � }
| |
_ |
_
| |
_ |
| | <q� Wq/ t | t
j � r/ Gd
d � d | t � }
| |
_ |
_
| |
_ |
| | <q/ Wd d g � d g � � � � f d d � } � � f d
d � } | | d <| | d <d S)Nr+ r/ r1 r0 ZWithr �Mixinc @ s e Z d Z d S)z*install_tests_in_module_dict.<locals>.TempN)r+ r, r- r
r
r
r �Temp" s rq c @ s e Z d Z d S)z*install_tests_in_module_dict.<locals>.TempN)r+ r, r- r
r
r
r rq ( s c s t j t � t j j � t j j j � � d <t j j � � d <t j d d � � d <y t j � d d �Wn% t
k
r� t j � d � � Yn Xt
j j d � r� y t j � } Wn! t k
r� t j d � � Yn Xt � t j � t j � j t � d S) Nr r Z
allow_noneTrW z start method not supported�linuxz1OSError raises on RLock creation, see issue 3111!)r| rX rY �process�_cleanup� _danglingr r� r6 rU r r r r� r r[ r� r; r r Z get_temp_dirr� r� r� )r� )�dangling�old_start_method� start_methodr
r � setUpModule1 s$
z1install_tests_in_module_dict.<locals>.setUpModulec s� t j � d d d �t j d � t j j � t j � t t j j � t � d � } | rw t
d | d t j �~ t t
j � t � d � } | r� t
d | d t j �d S) Nr rW Tg �?zDangling processes:�filer zDangling threads:)r| rU r& rq rs rt r� r� r� ru r% r� r� r� )Ztmp)rv rw r
r �tearDownModuleG s
z4install_tests_in_module_dict.<locals>.tearDownModulery r{ )�globalsr� rB rg �
issubclassr. r� r= rC �
capitalizer �TestCaser+ r- r, r� )
Z remote_globsrx r, Z local_globsZ ALL_TYPESr\ �baseZtype_Znewnamerp rq ry r{ r
)rv rw rx r � install_tests_in_module_dict s8
$
r� )�r r� r� r& r
r� r� r r� r< ry r� r� r� r� r) rm r� Z test.supportr� Ztest.support.script_helperr� �
import_moduler r� Zmultiprocessing.dummyr| r Zmultiprocessing.managersZmultiprocessing.heapr
r r ZHAVE_SEND_HANDLErl � ImportErrorZ multiprocessing.sharedctypesr r r r` r r� r� r� r2 r� r� r� �getattrZ
HAVE_GETVALUEr r� r r r rg rY Zctypesr r r r� r r! r. r? r@ rG rX r� r� r� r� r� r� r� r r! r& r3 r4 rC rD rp r� r� r� r� r� r� r r� r� r� r r r r r r r� r r r r �registerr# rc r+ r, r- r. r2 r/ r: rB r@ rm rv r{ r r� r� r� r� r� r r� r� r� r� r� r� r� r� r� r� r� r� r r
r r! r/ r4 r< r> rD rN r� r` rh r rl ro r� r
r
r
r �<module> s<
�N �6�2 9�ES<
� 1 (1 7
0. �
0R �E6x+C"! � " B@J#(/