U
|
Z ±dø ã @ sd d dl Z d dlZddlmZ ddlmZmZ ej d¡Z dd Z
|
dd Z d
|
d Z d d Z dd ZdS )é Né )Ú_core)Úassert_checkpointsÚwait_all_tasks_blockedz3ignore:.*UnboundedQueue:trio.TrioDeprecationWarningc à sð t ¡ } | d¡ | ¡ I d H dgks*tt t j¡ | ¡ W 5 Q R X | d¡ | d¡ | d¡ | ¡ dddgks~t| ¡ st|
|
¡ dkst| d ¡ | ¡ r°t|
|
¡ dksÀt| ¡ }|j
|
dksÖt|j dksätt | d S )NÚhié é r r )r ÚUnboundedQueueÚ
|
put_nowaitÚ get_batchÚAssertionErrorÚpytestÚraisesÚ
|
WouldBlockÚget_batch_nowaitÚemptyÚqsizeÚ
|
statisticsÚ tasks_waitingÚrepr)ÚqÚstats© r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/_core/tests/test_unbounded_queue.pyÚtest_UnboundedQueue_basic s$
|
|
|
|
|
r c sú g t ¡ fdd} fdd}| |fD ]Ä} ¡ t ¡ 4 I d H }| |¡ t ¡ I d H ¡ }|jdks|t|j dkst
|
d¡
|
d¡ t ¡ I d H
|
d ¡ t ¡ I d H ddgd ggksÚt|j ¡ W 5 Q I d H R X q0d S )
|
Nc s& ¡ I d H } | st | ¡ q d S ©N)r r Úappend©Úbatch©r Úrecordr r Úget_batch_consumer* s z8test_UnboundedQueue_blocking.<locals>.get_batch_consumerc s* 2 z 3 d H W } | st | ¡ q6 d S r )r r r r r r Úaiter_consumer0 s z4test_UnboundedQueue_blocking.<locals>.aiter_consumerr r é
|
é é ) r r ÚclearÚ open_nurseryÚ
|
start_soonr r r r r r
|
Ú cancel_scopeÚcancel)r! r" ZconsumerÚnurseryr r r r Ú test_UnboundedQueue_blocking& s&
|
|
|
|
r, c
|
s t ¡ d¡ d¡ ¡ ddgks0td fdd} t ¡ 4 I d H T}| | ¡ t ¡ I d H d¡ d¡ t t j
|
¡ ¡ W 5 Q R X W 5 Q I d H R X ddgksÀtg fdd}t ¡ 4 I d H l}| |d ¡ t ¡ I d H | |d
|
¡ t ¡ I d H t d D ] } |¡ t ¡ I d H q |j ¡ W 5 Q I d H R X ttt d ¡d d t d D kstd S )Nr r c s | ¡ I d H d S r )r )r )Úresultr r r Q s z/test_UnboundedQueue_fairness.<locals>.get_batchr é c s | ¡ I d H f¡ q d S r )r r )Únamer r r Úreaderb s z,test_UnboundedQueue_fairness.<locals>.readerÚaÚbé Úabc S s g | ]
|
}|gqS r r )Ú.0Úir r r Ú
|
<listcomp>r s z0test_UnboundedQueue_fairness.<locals>.<listcomp>)r r r
|
r r r' r( r r r r Úranger) r* ÚlistÚzipÚ itertoolsÚcycle)r r+ r0 r6 r )r r r- r Ú test_UnboundedQueue_fairnessF s4
|
|
|
|
"
|
r= c à sj t ¡ } | d ¡ t | ¡ I d H W 5 Q R X | d ¡ t | 2 z3 d H W } q\qH6 W 5 Q R X d S r )r r r
|
r r )r Ú_r r r Ú"test_UnboundedQueue_trivial_yieldsu s
|
|
r? c
|
sº g fdd} t ¡ 4 I d H }t ¡ }| | |d¡ t I d H | | |d¡ t I d H tdD ]}| |¡ qft I d H dttdfgkst|j
|
¡ W 5 Q I d H R X d S )Nc s | ¡ I d H } ||f¡ d S r )r r )r r6 Úgot©r r r Úgetter s z7test_UnboundedQueue_no_spurious_wakeups.<locals>.getterr r r# ) r r' r r( r r8 r
|
r9 r r) r* )rB r+ r r6 r rA r Ú'test_UnboundedQueue_no_spurious_wakeups s rC )r; r Ú r Ztestingr r ÚmarkÚfilterwarningsZ
|
pytestmarkr r, r= r? rC r r r r Ú<module> s ÿ /
|
