U
|
K ±då ã @ s¼ d dl Z ddlmZmZmZmZ edd Zdd Zdd Zd
|
Z z e
|
d W n e k
|
rd d Z Y nX e j j e d ddd Zdd Zdd Zdd Zdd Zdd Zdd Zd d ZdS ) é Né )ÚaclosingÚasync_generatorÚyield_Úasynccontextmanagerc à sB z t | D ]}t|I d H q
|
W n tk
|
r< d|d< Y nX d S )NTr )Úranger Ú GeneratorExit)ÚcountÚ closed_slotÚi© r úWd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\async_generator/_tests/test_util.pyÚ async_range s
|
r c
|
à s dg} t td| 4 I d H B}ttd}|2 z(3 d H W }|t|ksJt|dkr. qZq.6 W 5 Q I d H R X | d svtdg} zjt td| 4 I d H D}ttd}|2 z*3 d H W }|t|ksÂt|dkr¦t q¦6 W 5 Q I d H R X W n tk
|
rú Y nX | d s
|
td S )NFé
|
é r )r r Úiterr ÚnextÚAssertionErrorÚ
|
ValueError)r
|
ÚgenÚitÚitemr r r Ú test_aclosing s& r c à s¢ t tdd } t t¡*}| 4 I d H
|
tW 5 Q I d H R X W 5 Q R X |jjd dks^tt |jj
|
tsptt t¡2}| 4 I d H t d}|W 5 Q I d H R X W 5 Q R X |jjd dksÆt|jj
|
}|jd dksàt|j
|
|ksîtt t ¡.}| 4 I d H t dW 5 Q I d H R X W 5 Q R X |jjd dksBt|jj
|
d ksTtt tdd }t t ¡* | 4 I d H
|
t W 5 Q I d H R X W 5 Q R X d S ) Nc
|
Ó sB zt I d H W n, tk
|
r< } ztd| W 5 d } ~ X Y nX d S )Núissue29692:Chained)r Ú ExceptionÚ RuntimeError)Úexcr r r Úmanager_issue29692' s z[test_contextmanager_do_not_unchain_non_stopiteration_exceptions.<locals>.manager_issue29692r r úissue29692:Unchainedz generator raised StopIterationc Ó s t I d H d S ©N©r r r r r Únoop_async_context_managerF s zctest_contextmanager_do_not_unchain_non_stopiteration_exceptions.<locals>.noop_async_context_manager) r r ÚpytestÚraisesr ÚZeroDivisionErrorÚvalueÚargsr Ú
|
isinstanceÚ __cause__Ú StopIterationÚStopAsyncIteration)r Úexcinfor Úcauser! r r r Ú?test_contextmanager_do_not_unchain_non_stopiteration_exceptions& s6 " r- Tz¦
|
@asynccontextmanager
|
async def manager_issue29692_2():
|
try:
|
yield
|
except Exception as exc:
|
raise RuntimeError('issue29692:Chained') from exc
|
Fz1Python < 3.6 doesn't have native async generators)Úreasonc à sÔ t t¡*} t 4 I d H
|
tW 5 Q I d H R X W 5 Q R X | jjd dksNtt| jj ts`tt
|
t fD ]f}t |¡.} t 4 I d H |dW 5 Q I d H R X W 5 Q R X | jjd dks¾t| jj d kshtqhd S )Nr r r ) r" r# r Zmanager_issue29692_2r$ r% r& r r' r( r) r* )r+ Úclsr r r ÚFtest_native_contextmanager_do_not_unchain_non_stopiteration_exceptionsa s "r0 c à s~ t tdd } tttfD ]>}t |¡* | 4 I d H
|
|W 5 Q I d H R X W 5 Q R X q| 4 I d H W 5 Q I d H R X d S )Nc Ó s t I d H d S r r r r r r r! x s zRtest_asynccontextmanager_exception_passthrough.<locals>.noop_async_context_manager)r r r* r r r" r# )r! Úexc_typer r r Ú.test_asynccontextmanager_exception_passthroughv s r2 c
|
à s8 t tdd } | 4 I d H
|
tW 5 Q I d H R X d S )Nc Ó s& t t¡ t I d H W 5 Q R X d S r )r" r# r r r r r r Úcatch_it s z<test_asynccontextmanager_catches_exception.<locals>.catch_it)r r r )r3 r r r Ú*test_asynccontextmanager_catches_exception s
|
r4 c à sN t tdd } t t¡* | 4 I d H
|
tW 5 Q I d H R X W 5 Q R X d S )Nc Ó s. zt I d H W n tk
|
r( tY nX d S r )r ÚKeyErrorr r r r r Ú switch_it s z?test_asynccontextmanager_different_exception.<locals>.switch_it)r r r" r# r r5 )r6 r r r Ú,test_asynccontextmanager_different_exception s r7 c
|
à sp t tdd } | }t t¡}| W 5 Q R X W 5 Q R X dt|jksNt|4 I d H W 5 Q I d H R X d S )Nc Ó s t I d H d S r r r r r r Úxxx¡ s z@test_asynccontextmanager_nice_message_on_sync_enter.<locals>.xxxz
|
async with)r r r" r# r Ústrr% r )r8 Úcmr+ r r r Ú3test_asynccontextmanager_nice_message_on_sync_enter s r; c à sd t tdd } t t¡.}| 4 I d H ds4tW 5 Q I d H R X W 5 Q R X dt|jks`td S )Nc Ó s d S r r r r r r Úyeehaw³ s z1test_asynccontextmanager_no_yield.<locals>.yeehawFz didn't yield)r r r" r# r r r9 r% )r< r+ r r r Ú!test_asynccontextmanager_no_yield² s "r= c sÈ d t t fdd} t t¡&}| 4 I d H W 5 Q I d H R X W 5 Q R X dt|jks`t dksltt t¡*}| 4 I d H
|
tW 5 Q I d H R X W 5 Q R X dt|jks¸t dksÄtd S )Nr c sF zt I d H W n tk
|
r$ Y nX zt I d H W 5 d7 X d S )Né )r r r ©Z closed_countr r Ú doubleyield s z=test_asynccontextmanager_too_many_yields.<locals>.doubleyieldz didn't stopr> zdidn't stop after athrowr ) r r r" r# r r9 r% r r )r@ r+ r r? r Ú(test_asynccontextmanager_too_many_yields¿ s rA c à s& t t¡ tdd } W 5 Q R X d S )Nc s s
|
d V d S r r r r r r Úsyncgená s zCtest_asynccontextmanager_requires_asyncgenfunction.<locals>.syncgen)r" r# Ú TypeErrorr )rB r r r Ú2test_asynccontextmanager_requires_asyncgenfunctionÞ s rD )r" Ú r r r r r r r- ZnativeasyncgeneratorsÚexecÚ SyntaxErrorÚmarkZskipifr0 r2 r4 r7 r; r= rA rD r r r r Ú<module> s0
|
+ÿ
|
|
þ
|
|
