File: //usr/local/lib/python3.7/test/__pycache__/test_asyncgen.cpython-37.opt-1.pyc
B
��gӄ � @ s� d dl Z d dlZd dlZd dlmZ ed�ZG dd� de�Zejdd�dd ��Z d
d � Z
d d
� Z G dd� dej �Z
G dd� dej �ZG dd� dej �Zedkr�e�� dS )� N)�
import_module�asyncioc @ s e Zd ZdS )�AwaitExceptionN)�__name__�
__module__� __qualname__� r r �./usr/local/lib/python3.7/test/test_asyncgen.pyr s r F)�throwc c s | r dV ndV d S )N)r
)�resultr )r
r r r � awaitable
s r c
C sf d}x\y"|r d}| � t�}n
| �d �}W n( tk
rP } z
|jd S d }~X Y nX |dkrd}qW d S )NFr )r
T)r
r �send�
StopIteration�args)�coro�exc�fut�exr r r �run_until_complete s r c s � fdd�}t |� �S )Nc � s> g } x4� 2 y 3 d H }W n
t k
s.X | �|� q
Y W | S )N)�StopAsyncIteration�append)�res�i)�genr r �iterate& s zto_list.<locals>.iterate)r )r r r )r r �to_list% s r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �AsyncGenSyntaxTestc C s, d}| � td�� t|i i � W d Q R X d S )NzJasync def foo():
await abc
yield from 123
zyield from.*inside async)�assertRaisesRegex� SyntaxError�exec)�self�coder r r �test_async_gen_syntax_011 s z+AsyncGenSyntaxTest.test_async_gen_syntax_01c C s, d}| � td�� t|i i � W d Q R X d S )Nz4async def foo():
yield from 123
zyield from.*inside async)r r r )r r! r r r �test_async_gen_syntax_02: s z+AsyncGenSyntaxTest.test_async_gen_syntax_02c C s, d}| � td�� t|i i � W d Q R X d S )NzXasync def foo():
await abc
yield
return 123
zreturn.*value.*async gen)r r r )r r! r r r �test_async_gen_syntax_03B s z+AsyncGenSyntaxTest.test_async_gen_syntax_03c C s, d}| � td�� t|i i � W d Q R X d S )NzBasync def foo():
yield
return 123
zreturn.*value.*async gen)r r r )r r! r r r �test_async_gen_syntax_04L s z+AsyncGenSyntaxTest.test_async_gen_syntax_04c C s, d}| � td�� t|i i � W d Q R X d S )NzWasync def foo():
if 0:
yield
return 12
zreturn.*value.*async gen)r r r )r r! r r r �test_async_gen_syntax_05U s z+AsyncGenSyntaxTest.test_async_gen_syntax_05N)r r r r" r# r$ r% r& r r r r r / s
r c @ st e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS ) � AsyncGenTestc C s8 dd� }dd� }dd� }||�}||�}| � ||� |S )Nc
S st g }xjy|� | �� � W q tk
r8 |� d� P Y q tk
rj } z|� tt|��� W d d }~X Y qX qW |S )N�STOP)r �__next__r � Exception�str�type)�gr r r r r � sync_iteratec s
(z5AsyncGenTest.compare_generators.<locals>.sync_iteratec
S s� g }x�| � � }y�x�y |�� W q tk
rf } z*|jrJ|�|jd � P n |�d� P W d d }~X Y q tk
rz � Y q tk
r� } z|�tt|��� P W d d }~X Y qX qW W q tk
r� |�d� P Y qX qW |S )Nr zEMPTY StopIterationr( ) � __anext__r) r r r r r* r+ r, )r- r Zanr r r r �
async_iterateo s,
z6AsyncGenTest.compare_generators.<locals>.async_iteratec
S s� g }x�y| � � �� W q tk
r6 |�d� P Y q tk
rz } z(|jr^|�|jd � n |�d� P W d d }~X Y q tk
r� } z|�tt|��� W d d }~X Y qX qW |S )Nr( r zEMPTY StopIteration) r/ r) r r r r r* r+ r, )r- r r r r r r0 � s
()� assertEqual)r �sync_gen� async_genr. r0 Zsync_gen_resultZasync_gen_resultr r r �compare_generatorsb s z AsyncGenTest.compare_generatorsc s( � fdd�}� � t|� �dddg� d S )Nc sF t � I d H dV } � �| d � t � I d H dV t � I d H dV d S )N�{ i� i )r ZassertIs)�a)r r r r � s z5AsyncGenTest.test_async_gen_iteration_01.<locals>.genr5 i� i )r1 r )r r r )r r �test_async_gen_iteration_01� s z(AsyncGenTest.test_async_gen_iteration_01c
C s� dd� }|� }|� � }| �|�� �� d� y|�� �� W n4 tk
rn } z| �|jd d� W d d }~X Y n X | �d� | �|�� �� d� y|�� �� W n. tk
r� } z| �|j� W d d }~X Y n X | �d� d S )Nc S s" t � I d H dV t � I d H d S )Nr5 )r r r r r r � s z5AsyncGenTest.test_async_gen_iteration_02.<locals>.gen)r r r5 z StopIteration was not raisedz!StopAsyncIteration was not raised) � __aiter__r1 r/ r) r r �failr � assertFalse)r r r- �air r r r �test_async_gen_iteration_02� s $
z(AsyncGenTest.test_async_gen_iteration_02c C s, dd� }| � t�� t|� � W d Q R X d S )Nc S s, t � I d H dV t dd�I d H dV d S )Nr5 T)r
i� )r r r r r r � s z5AsyncGenTest.test_async_gen_exception_03.<locals>.gen)� assertRaisesr r )r r r r r �test_async_gen_exception_03� s z(AsyncGenTest.test_async_gen_exception_03c
C s� dd� }|� }|� � }| �|�� �� d� y|�� �� W n4 tk
rn } z| �|jd d� W d d }~X Y n X | �d� | �t�� |�� �� W d Q R X d S )Nc S s t � I d H dV dd d S )Nr5 � r )r r r r r r � s z5AsyncGenTest.test_async_gen_exception_04.<locals>.gen)r r r5 z StopIteration was not raised) r8 r1 r/ r) r r r9 r= �ZeroDivisionError)r r r- r; r r r r �test_async_gen_exception_04� s $
z(AsyncGenTest.test_async_gen_exception_04c C s. dd� }| � td�� t|� � W d Q R X d S )Nc S s dV t �d S )Nr5 )r r r r r r � s z5AsyncGenTest.test_async_gen_exception_05.<locals>.genz#async generator.*StopAsyncIteration)r � RuntimeErrorr )r r r r r �test_async_gen_exception_05� s z(AsyncGenTest.test_async_gen_exception_05c C s. dd� }| � td�� t|� � W d Q R X d S )Nc S s dV t �d S )Nr5 )r r r r r r � s z5AsyncGenTest.test_async_gen_exception_06.<locals>.genz async generator.*StopIteration)r rB r )r r r r r �test_async_gen_exception_06� s z(AsyncGenTest.test_async_gen_exception_06c C s$ dd� }dd� }| � |� |� � d S )Nc s s, zdV dd W d dV dV X dV d S )Nr? r � � �d r r r r r r2 � s z:AsyncGenTest.test_async_gen_exception_07.<locals>.sync_genc S s, zdV dd W d dV dV X dV d S )Nr? r rE rF rG r r r r r r3 s z;AsyncGenTest.test_async_gen_exception_07.<locals>.async_gen)r4 )r r2 r3 r r r �test_async_gen_exception_07� s
z(AsyncGenTest.test_async_gen_exception_07c C s$ dd� }dd� }| � |� |� � d S )Nc s s, z
dV W d dV dd dV X dV d S )Nr? rE r rF rG r r r r r r2
s
z:AsyncGenTest.test_async_gen_exception_08.<locals>.sync_genc S sD zdV t � I d H W d t � I d H dV dd dV X dV d S )Nr? rE r rF rG )r r r r r r3 s z;AsyncGenTest.test_async_gen_exception_08.<locals>.async_gen)r4 )r r2 r3 r r r �test_async_gen_exception_08 s
z(AsyncGenTest.test_async_gen_exception_08c C s$ dd� }dd� }| � |� |� � d S )Nc s s, zdV dd W d dV dV X dV d S )Nr? r rE rF rG r r r r r r2 & s z:AsyncGenTest.test_async_gen_exception_09.<locals>.sync_genc S sD z t � I d H dV dd W d dV t � I d H dV X dV d S )Nr? r rE rF rG )r r r r r r3 0 s z;AsyncGenTest.test_async_gen_exception_09.<locals>.async_gen)r4 )r r2 r3 r r r �test_async_gen_exception_09% s
z(AsyncGenTest.test_async_gen_exception_09c C s4 dd� }| � td�� |� �� �d� W d Q R X d S )Nc S s
dV d S )Nr5 r r r r r r ? s z5AsyncGenTest.test_async_gen_exception_10.<locals>.genz!non-None value .* async generatorrG )r � TypeErrorr/ r
)r r r r r �test_async_gen_exception_10> s z(AsyncGenTest.test_async_gen_exception_10c s< dd� ��fdd�}dd� � � fdd�}| � |� |� � d S ) Nc s s dV dV d S )N�
� r r r r r r2 F s z:AsyncGenTest.test_async_gen_exception_11.<locals>.sync_genc 3 sL dV � � } | � d � y| �t� � W n tk
r@ dV Y nX dV d S )Nr? rE rF )r
r
�
GeneratorExit)�sg)r2 r r �sync_gen_wrapperJ s
zBAsyncGenTest.test_async_gen_exception_11.<locals>.sync_gen_wrapperc S s dV dV d S )NrM rN r r r r r r3 T s z;AsyncGenTest.test_async_gen_exception_11.<locals>.async_genc sX dV � � } | � d �I d H y| �t� �I d H W n tk
rL dV Y nX dV d S )Nr? rE rF )�asend�athrowrO )Zasg)r3 r r �async_gen_wrapperX s zCAsyncGenTest.test_async_gen_exception_11.<locals>.async_gen_wrapper)r4 )r rQ rT r )r3 r2 r �test_async_gen_exception_11E s
z(AsyncGenTest.test_async_gen_exception_11c C s� dd� }|� }| � |jd� d|_| � |jd� | �d|j� d|_| � |jd� | �|j� | �|jtj � | �
|j � | �|j tj
� | �t�|�� �� d S )Nc S s
dV d S )Nr5 r r r r r r e s z/AsyncGenTest.test_async_gen_api_01.<locals>.genr Z123z.gen)r1 r �assertInr Z assertIsNone�ag_await�assertIsInstance�ag_frame�types� FrameTyper: �
ag_running�ag_code�CodeTypeZ
assertTrue�inspectZ isawaitable�aclose)r r r- r r r �test_async_gen_api_01d s z"AsyncGenTest.test_async_gen_api_01N)r r r r4 r7 r< r> rA rC rD rH rI rJ rL rU ra r r r r r' ` s =
r' c @ s$ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAdB� Z#dCdD� Z$dEdF� Z%dGS )H�AsyncGenAsyncioTestc C s t �� | _t �d � d S )N)r Znew_event_loop�loopZset_event_loop)r r r r �setUp| s
zAsyncGenAsyncioTest.setUpc C s | j �� d | _ d S )N)rc �close)r r r r �tearDown� s
z AsyncGenAsyncioTest.tearDownc � s> g }x4|2 y 3 d H }W n
t k
s.X |�|� q
Y W |S )N)r r )r r r r r r r r � s zAsyncGenAsyncioTest.to_listc s4 � fdd�}� j �� �|� ��}� �|ddg� d S )Nc s< dV t jd� jd�I d H dV t jd� jd�I d H d S )Nr? g{�G�z�?)rc rE rF )r �sleeprc r )r r r r � s z:AsyncGenAsyncioTest.test_async_gen_asyncio_01.<locals>.genr? rE )rc r r r1 )r r r r )r r �test_async_gen_asyncio_01� s z-AsyncGenAsyncioTest.test_async_gen_asyncio_01c s: � fdd�}� � t�� � j�� �|� �� W d Q R X d S )Nc s4 dV t jd� jd�I d H dV dd dV d S )Nr? g{�G�z�?)rc rE r rF )r rg rc r )r r r r � s
z:AsyncGenAsyncioTest.test_async_gen_asyncio_02.<locals>.gen)r= r@ rc r r )r r r )r r �test_async_gen_asyncio_02� s z-AsyncGenAsyncioTest.test_async_gen_asyncio_02c s> | j � G � fdd�d�}� �| �|� ��}| �|ddg� d S )Nc s e Zd Z� fdd�ZdS )z:AsyncGenAsyncioTest.test_async_gen_asyncio_03.<locals>.Genc s$ dV t jd� d�I d H dV d S )Nr? g{�G�z�?)rc rE )r rg )r )rc r r r8 � s zDAsyncGenAsyncioTest.test_async_gen_asyncio_03.<locals>.Gen.__aiter__N)r r r r8 r )rc r r �Gen� s rj r? rE )rc r r r1 )r rj r r )rc r �test_async_gen_asyncio_03� s z-AsyncGenAsyncioTest.test_async_gen_asyncio_03c sH �fdd�� � �fdd�}� �fdd�}�j �|� � �j �|� � d S )Nc sh dV t jd� jd�I d H ydV dV W n tk
rF dV Y nX t jd� jd�I d H dV d S )Nr? g{�G�z�?)rc rE rF i� � )r rg rc r@ r )r r r �foo� s
z@AsyncGenAsyncioTest.test_async_gen_asyncio_anext_04.<locals>.fooc � s� � � � � } ��| �� I d H d� ��| �� I d H d� ��| �� I d H d� ��| �� I d H d� ��t�� | �� I d H W d Q R X ��t�� | �� I d H W d Q R X d S )Nr? rE rF rl )r8 r1 r/ r= r )�it)rm r r r �run1� s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_anext_04.<locals>.run1c
� s� � � � � } ��| �� I d H d� ��| �� I d H d� y| �� �t� W n4 tk
r| } z��|jd d� W d d }~X Y n X ��d� ��| �� I d H d� ��t �� | �� I d H W d Q R X d S )Nr? rE r i� z StopIteration was not raisedrl )
r8 r1 r/ r
r@ r r r9 r= r )rn r )rm r r r �run2� s
$
zAAsyncGenAsyncioTest.test_async_gen_asyncio_anext_04.<locals>.run2)rc r )r ro rp r )rm r r � test_async_gen_asyncio_anext_04� s
z3AsyncGenAsyncioTest.test_async_gen_asyncio_anext_04c s( dd� � � �fdd�}�j �|� � d S )Nc S s dV } | V } | d V d S )Nr? rG r )�vr r r rm � s z@AsyncGenAsyncioTest.test_async_gen_asyncio_anext_05.<locals>.fooc
� s( � � � � } y| �� �d � W n4 tk
rP } z��|jd d� W d d }~X Y n X ��d� y| �� �d� W n4 tk
r� } z��|jd d� W d d }~X Y n X ��d� y| �� �d� W n4 tk
r� } z��|jd d� W d d }~X Y n X ��d� ��t�� | �� I d H W d Q R X d S )Nr r? z StopIteration was not raisedrM � i� ) r8 r/ r
r r1 r r9 r= r )rn r )rm r r r �run� s$
$
$
$
z@AsyncGenAsyncioTest.test_async_gen_asyncio_anext_05.<locals>.run)rc r )r rt r )rm r r � test_async_gen_asyncio_anext_05� s z3AsyncGenAsyncioTest.test_async_gen_asyncio_anext_05c sv d� dd� }|� }|� d � ��t�� |� d � W d Q R X � fdd��� ��fdd�}�j�|� � ��� d� d S ) Nr c s s y
d V W n Y nX d S )Nr r r r r rm � s
z@AsyncGenAsyncioTest.test_async_gen_asyncio_anext_06.<locals>.fooc s y
d V W n Y nX d� d S )Nr? r r )�DONEr r r
s
z@AsyncGenAsyncioTest.test_async_gen_asyncio_anext_06.<locals>.genc � sH �� } | � d �I d H ��t�� | � d �I d H W d Q R X � d7 � d S )NrM )rR r= r )r- )rv r r r r rt s
z@AsyncGenAsyncioTest.test_async_gen_asyncio_anext_06.<locals>.run� )r
r= r rc r r1 )r rm r- rt r )rv r r r � test_async_gen_asyncio_anext_06� s
z3AsyncGenAsyncioTest.test_async_gen_asyncio_anext_06c s( dd� � � �fdd�}�j �|� � d S )Nc S s* y
dV W n t k
r$ dV Y nX d S )N)r? )rE )r@ r r r r rm s
zCAsyncGenAsyncioTest.test_async_gen_asyncio_anext_tuple.<locals>.fooc � s� � � � � } ��| �� I d H d� ��t��}| �� �t� W d Q R X ��|jjd d� ��t �� | �� I d H W d Q R X d S )N)r? r )rE )
r8 r1 r/ r= r r
r@ � exceptionr r )rn �cm)rm r r r rt $ s
zCAsyncGenAsyncioTest.test_async_gen_asyncio_anext_tuple.<locals>.run)rc r )r rt r )rm r r �"test_async_gen_asyncio_anext_tuple s
z6AsyncGenAsyncioTest.test_async_gen_asyncio_anext_tuplec s( dd� � � �fdd�}�j �|� � d S )Nc S s2 yt d�V W n tk
r, t d�V Y nX d S )Nr? rF )r r@ r r r r rm 1 s zKAsyncGenAsyncioTest.test_async_gen_asyncio_anext_stopiteration.<locals>.fooc � s� � � � � } | �� I d H }��|t� ��|jd� ��t��}| �� �t� W d Q R X |j j
d }��|t� ��|jd� ��t �� | �� I d H W d Q R X d S )Nr? r rF ) r8 r/ rX r r1 �valuer= r
r@ ry r r )rn rr rz )rm r r r rt 7 s
zKAsyncGenAsyncioTest.test_async_gen_asyncio_anext_stopiteration.<locals>.run)rc r )r rt r )rm r r �*test_async_gen_asyncio_anext_stopiteration0 s z>AsyncGenAsyncioTest.test_async_gen_asyncio_anext_stopiterationc sB �fdd�� � fdd�}�� td�� �j�|� � W d Q R X d S )Nc
s6 zdV dd W d t jd� jd�I d H dV X d S )Nr? r g{�G�z�?)rc rs )r rg rc r )r r r rm H s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_06.<locals>.fooc � s. � � } | � � }|�� I d H | �� I d H d S )N)r8 r/ r` )r rn )rm r r rt P s zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_06.<locals>.runz%async generator ignored GeneratorExit)r rB rc r )r rt r )rm r r � test_async_gen_asyncio_aclose_06G s z4AsyncGenAsyncioTest.test_async_gen_asyncio_aclose_06c s< d� � �fdd���fdd�}�j �|� � ��� d� d S )Nr c
sV zdV dd W d t jd�jd�I d H t jd�jd�I d H � d7 � X � d7 � d S )Nr? r g{�G�z�?)rc i� )r rg rc r )rv r r r rm ^ s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_07.<locals>.fooc � s. � � } | � � }|�� I d H | �� I d H d S )N)r8 r/ r` )r rn )rm r r rt i s zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_07.<locals>.runr? )rc r r1 )r rt r )rv rm r r � test_async_gen_asyncio_aclose_07[ s
z4AsyncGenAsyncioTest.test_async_gen_asyncio_aclose_07c sv d� t j�jd��� ��fdd����fdd�}�j�|� �}��� d� ��� |�� �j�t jd�jd�� d S ) Nr )rc c
sf z"dV �I d H � d7 � dV W d t jd�jd�I d H t jd�jd�I d H � d7 � X � d7 � d S )Nr? i� rE g{�G�z�?)rc )r rg rc r )rv r r r r rm w s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_08.<locals>.fooc � s\ � � } | � � }��|�� I d H d� �j�|�� �}tjd�jd�I d H | �� I d H |S )Nr? g{�G�z�?)rc )r8 r1 r/ rc � create_taskr rg r` )r rn �t)rm r r r rt � s zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_08.<locals>.runr? g{�G�z�?)r ZFuturerc r r1 �cancelrg )r rt r� r )rv rm r r r � test_async_gen_asyncio_aclose_08r s
z4AsyncGenAsyncioTest.test_async_gen_asyncio_aclose_08c s> d� � �fdd����fdd�}�j �|� � ��� d� d S )Nr c
sH zx
dV qW W d t jd�jd�I d H t jd�jd�I d H d� X d S )Nr? g{�G�z�?)rc )r rg rc r )rv r r r r � s zDAsyncGenAsyncioTest.test_async_gen_asyncio_gc_aclose_09.<locals>.genc � s> � � } | � � I d H | � � I d H ~ tjd�jd�I d H d S )Ng�������?)rc )r/ r rg rc )r- )r r r r rt � s
zDAsyncGenAsyncioTest.test_async_gen_asyncio_gc_aclose_09.<locals>.runr? )rc r r1 )r rt r )rv r r r �#test_async_gen_asyncio_gc_aclose_09� s
z7AsyncGenAsyncioTest.test_async_gen_asyncio_gc_aclose_09c s\ d� dd� }|� }|� d � |�� � fdd��� �fdd�}| j�|� � | �� d� d S ) Nr c s s y
d V W n Y nX d S )Nr r r r r rm � s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_10.<locals>.fooc s y
d V W n Y nX d� d S )Nr? r r )rv r r r � s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_10.<locals>.genc � s0 �� } | � d �I d H | �� I d H � d7 � d S )NrM )rR r` )r- )rv r r r rt � s zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_10.<locals>.runrw )r
re rc r r1 )r rm r- rt r )rv r r � test_async_gen_asyncio_aclose_10� s
z4AsyncGenAsyncioTest.test_async_gen_asyncio_aclose_10c sv d� dd� }|� }|� d � ��td�� |�� W d Q R X � fdd��� ��fdd�}�j�|� � ��� d � d S )
Nr c s s" y
d V W n Y nX d V d S )Nr r r r r rm � s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_11.<locals>.foozignored GeneratorExitc s* y
d V W n Y nX d V � d7 � d S )Nr? r r )rv r r r � s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_11.<locals>.genc � sH �� } | � d �I d H ��td�� | �� I d H W d Q R X � d7 � d S )Nzignored GeneratorExitrM )rR r rB r` )r- )rv r r r r rt � s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_11.<locals>.runrM )r
r rB re rc r r1 )r rm r- rt r )rv r r r � test_async_gen_asyncio_aclose_11� s
z4AsyncGenAsyncioTest.test_async_gen_asyncio_aclose_11c sD d� dd� �� �fdd���fdd�}| j �|� � | �� d� d S ) Nr c � s t �d�I d H dd d S )Ng{�G�z�?r? r )r rg r r r r �target� s zDAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_12.<locals>.targetc sF t ��� �} z
dV W d y| I d H W n tk
r> d� Y nX X d S )Nr? )r r� r@ )Ztask)rv r� r r rm � s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_12.<locals>.fooc � s. � � } | � � }|�� I d H | �� I d H d S )N)r8 r/ r` )r rn )rm r r rt s zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_12.<locals>.runr? )rc r r1 )r rt r )rv rm r� r � test_async_gen_asyncio_aclose_12� s z4AsyncGenAsyncioTest.test_async_gen_asyncio_aclose_12c sx d� dd� }|� }|� d �}��|d� |� d�}��|d� � �fdd����fd d
�}�j�|� � ��� d� d S ) Nr c s s dV } | d V d S )Nr? rE r )rr r r r �sgen s zAAsyncGenAsyncioTest.test_async_gen_asyncio_asend_01.<locals>.sgenr? rG �� c
s� zVt jd�jd�I d H dV } t jd�jd�I d H | d V t jd�jd�I d H d S t jd�jd�I d H t jd�jd�I d H d� X d S )Ng{�G�z�?)rc r? rE )r rg rc )rr )rv r r r r s
z@AsyncGenAsyncioTest.test_async_gen_asyncio_asend_01.<locals>.genc � sh � � } | � d �I d H }��|d� | � d�I d H }��|d� ��t�� | � d �I d H W d Q R X d S )Nr? rG r� )rR r1 r= r )r- rr )r r r r rt ) s z@AsyncGenAsyncioTest.test_async_gen_asyncio_asend_01.<locals>.run)r
r1 rc r )r r� rP rr rt r )rv r r r � test_async_gen_asyncio_asend_01 s
z3AsyncGenAsyncioTest.test_async_gen_asyncio_asend_01c sb d� �fdd��� ��fdd����fdd�}�� t�� �j�|� � W d Q R X ��� d� d S ) Nr c � s" t j| � jd�I d H dd d S )N)rc r? r )r rg rc )�delay)r r r �
sleep_n_crash; s zJAsyncGenAsyncioTest.test_async_gen_asyncio_asend_02.<locals>.sleep_n_crashc
sx z@t jd�jd�I d H dV } �d�I d H � d7 � | d V W d t jd�jd�I d H t jd�jd�I d H d� X d S )Ng{�G�z�?)rc r? i� rE )r rg rc )rr )rv r r� r r r ? s z@AsyncGenAsyncioTest.test_async_gen_asyncio_asend_02.<locals>.genc � s6 � � } | � d �I d H }��|d� | � d�I d H d S )Nr? rG )rR r1 )r- rr )r r r r rt L s z@AsyncGenAsyncioTest.test_async_gen_asyncio_asend_02.<locals>.runr? )r= r@ rc r r1 )r rt r )rv r r r� r � test_async_gen_asyncio_asend_028 s
z3AsyncGenAsyncioTest.test_async_gen_asyncio_asend_02c sd d� �fdd��� ��fdd����fdd�}�� tj�� �j�|� � W d Q R X ��� d� d S ) Nr c � s@ t jt j| �jd��jd�� �j�| d � fdd�� � I d H S )N)rc rE c s � � � S )N)r� r )r r r �<lambda>^ � z\AsyncGenAsyncioTest.test_async_gen_asyncio_asend_03.<locals>.sleep_n_crash.<locals>.<lambda>)r �
ensure_futurerg rc �
call_later)r� )r )r r r� [ s
zJAsyncGenAsyncioTest.test_async_gen_asyncio_asend_03.<locals>.sleep_n_crashc
sx z@t jd�jd�I d H dV } �d�I d H � d7 � | d V W d t jd�jd�I d H t jd�jd�I d H d� X d S )Ng{�G�z�?)rc r? i� rE )r rg rc )rr )rv r r� r r r a s z@AsyncGenAsyncioTest.test_async_gen_asyncio_asend_03.<locals>.genc � s6 � � } | � d �I d H }��|d� | � d�I d H d S )Nr? rG )rR r1 )r- rr )r r r r rt n s z@AsyncGenAsyncioTest.test_async_gen_asyncio_asend_03.<locals>.runr? )r= r �CancelledErrorrc r r1 )r rt r )rv r r r� r � test_async_gen_asyncio_asend_03X s
z3AsyncGenAsyncioTest.test_async_gen_asyncio_asend_03c s� d� G dd� dt ���fdd�}|� }|�d �}��|d� |���}��|d� ��t�� |�d � W d Q R X � ��fdd �����fd
d �}�j�|� � ��� d� d S ) Nr c @ s e Zd ZdS )zCAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_01.<locals>.FooErN)r r r r r r r �FooEr} s r� c 3 s2 y
dV } W n � k
r" d} Y nX | d V d S )Nr? i� rE r )rr )r� r r r� � s
zBAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_01.<locals>.sgenr? i� c
s� ztt jd�jd�I d H y
dV } W n. �k
rP d} t jd�jd�I d H Y nX | d V t jd�jd�I d H W d t jd�jd�I d H t jd�jd�I d H d� X d S )Ng{�G�z�?)rc r? i� rE )r rg rc )rr )rv r� r r r r � s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_01.<locals>.genc � sh �� } | � d �I d H }��|d� | �� �I d H }��|d� ��t�� | � d �I d H W d Q R X d S )Nr? i� )rR r1 rS r= r )r- rr )r� r r r r rt � s zAAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_01.<locals>.run)r* r
r1 r
r= r rc r )r r� rP rr rt r )rv r� r r r � test_async_gen_asyncio_athrow_01z s
z4AsyncGenAsyncioTest.test_async_gen_asyncio_athrow_01c sz d� G dd� dt ���fdd��� ���fdd��� ���fdd �}��tj�� �j�|� � W d Q R X ��� d
� d S ) Nr c @ s e Zd ZdS )zCAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_02.<locals>.FooErN)r r r r r r r r� � s r� c � s@ t jt j| �jd��jd�� �j�| d � fdd�� � I d H S )N)rc rE c s � � � S )N)r� r )r r r r� � r� z]AsyncGenAsyncioTest.test_async_gen_asyncio_athrow_02.<locals>.sleep_n_crash.<locals>.<lambda>)r r� rg rc r� )r� )r )r r r� � s
zKAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_02.<locals>.sleep_n_crashc
s� zht jd�jd�I d H y
dV } W n" �k
rD �d�I d H Y nX | d V t jd�jd�I d H W d t jd�jd�I d H t jd�jd�I d H d� X d S )Ng{�G�z�?)rc r? rE )r rg rc )rr )rv r� r r� r r r � s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_02.<locals>.genc � sj �� } | � d �I d H }��|d� y| ���I d H W n$ tjk
rZ ��� d� � Y n X ��d� d S )Nr? z CancelledError was not raised)rR r1 rS r r� r9 )r- rr )rv r� r r r r rt � s zAAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_02.<locals>.runr? )r* r= r r� rc r r1 )r rt r )rv r� r r r� r � test_async_gen_asyncio_athrow_02� s z4AsyncGenAsyncioTest.test_async_gen_asyncio_athrow_02c sv d� dd� }|� }|� d � ��t�� |�t� W d Q R X � fdd��� ��fdd�}�j�|� � ��� d� d S ) Nr c s s y
d V W n Y nX d S )Nr r r r r rm � s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_03.<locals>.fooc s y
d V W n Y nX d� d S )Nr? r r )rv r r r � s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_03.<locals>.genc � sH �� } | � d �I d H ��t�� | �t�I d H W d Q R X � d7 � d S )NrM )rR r= r rS �
ValueError)r- )rv r r r r rt � s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_03.<locals>.runrw )r
r= r r
r� rc r r1 )r rm r- rt r )rv r r r � test_async_gen_asyncio_athrow_03� s
z4AsyncGenAsyncioTest.test_async_gen_asyncio_athrow_03c s( dd� � � �fdd�}�j �|� � d S )Nc S s* y
dV W n t k
r$ dV Y nX d S )Nr? )rE )r@ r r r r r s
zDAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_tuple.<locals>.genc � sh � � } | � d �I d H }��|d� | �t�I d H }��|d� ��t�� | � d �I d H W d Q R X d S )Nr? )rE )rR r1 rS r@ r= r )r- rr )r r r r rt s zDAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_tuple.<locals>.run)rc r )r rt r )r r r �#test_async_gen_asyncio_athrow_tuple s z7AsyncGenAsyncioTest.test_async_gen_asyncio_athrow_tuplec s( dd� � � �fdd�}�j �|� � d S )Nc S s. y
dV W n t k
r( td�V Y nX d S )Nr? rE )r@ r r r r r r s
zLAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_stopiteration.<locals>.genc � sv � � } | � d �I d H }��|d� | �t�I d H }��|t� ��|jd� ��t�� | � d �I d H W d Q R X d S )Nr? rE ) rR r1 rS r@ rX r r| r= r )r- rr )r r r r rt s zLAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_stopiteration.<locals>.run)rc r )r rt r )r r r �+test_async_gen_asyncio_athrow_stopiteration s
z?AsyncGenAsyncioTest.test_async_gen_asyncio_athrow_stopiterationc s� d� � �fdd���fdd�}�j �|� �}�j �|� �}�j �tjd�j d�� �j ��j �� � ��� d� |�� |�� �j �tjd�j d�� d S ) Nr c
sF z t j| �jd�I d H dV W d t jd�jd�I d H � d7 � X d S )N)rc r? r )r rg rc )�timeout)� finalizedr r r �waiter( s
zFAsyncGenAsyncioTest.test_async_gen_asyncio_shutdown_01.<locals>.waiterc � s4 x.� d�2 y 3 d H } W q
t k
s$X q
Y W d S )Nr? )r )�_)r� r r �wait1 s "zDAsyncGenAsyncioTest.test_async_gen_asyncio_shutdown_01.<locals>.waitg�������?)rc rE )rc r� r r rg �shutdown_asyncgensr1 r� )r r� Zt1Zt2r )r� r r� r �"test_async_gen_asyncio_shutdown_01% s z6AsyncGenAsyncioTest.test_async_gen_asyncio_shutdown_01c s� d� � �fdd�}�fdd���fdd�}�j �|� �}�j �tjd�j d �� �j �|� �j ��j �� � ��� d
� |�� �j �tjd�j d �� d S ) Nr c s( �� d|� d}||d kr$� d7 � d S )NZasyncgenz0an error occurred during closing of asynchronous�messager? )rV )rc �contextZexpected)�loggedr r r �loggerE s zFAsyncGenAsyncioTest.test_async_gen_asyncio_shutdown_02.<locals>.loggerc s0 z t j| � jd�I d H dV W d dd X d S )N)rc r? r )r rg rc )r� )r r r r� L s
zFAsyncGenAsyncioTest.test_async_gen_asyncio_shutdown_02.<locals>.waiterc � s4 x.� d�2 y 3 d H } W q
t k
s$X q
Y W d S )Nr? )r )r� )r� r r r� S s "zDAsyncGenAsyncioTest.test_async_gen_asyncio_shutdown_02.<locals>.waitg�������?)rc r? ) rc r� r r rg �set_exception_handlerr� r1 r� )r r� r� r� r )r� r r� r �"test_async_gen_asyncio_shutdown_02B s z6AsyncGenAsyncioTest.test_async_gen_asyncio_shutdown_02c sP �fdd�� � fdd���fdd�}�j �|� �}��|dd� td �D �� d S )
Nc s2 x,t | �D ] }tjd� jd�I d H |V q
W d S )Ng{�G�z�?)rc )�ranger rg rc )�nr )r r r �aranged s z@AsyncGenAsyncioTest.test_async_gen_expression_01.<locals>.arangec s dd� � | �2 �S )Nc S s4 | y 3 d H }W n
t k
s&X |d V q Y d S )NrE )r )�.0r r r r � <genexpr>k s
zXAsyncGenAsyncioTest.test_async_gen_expression_01.<locals>.make_arange.<locals>.<genexpr>r )r� )r� r r � make_arangei s zEAsyncGenAsyncioTest.test_async_gen_expression_01.<locals>.make_arangec � s dd� � d�2 �I d H S )Nc
� s. g | y 3 d H }W n
t k
s"X |�q Y S )N)r )r� r r r r �
<listcomp>n s zQAsyncGenAsyncioTest.test_async_gen_expression_01.<locals>.run.<locals>.<listcomp>rM r r )r� r r rt m s z=AsyncGenAsyncioTest.test_async_gen_expression_01.<locals>.runc S s g | ] }|d �qS )rE r )r� r r r r r� q s zDAsyncGenAsyncioTest.test_async_gen_expression_01.<locals>.<listcomp>rM )rc r r1 r� )r rt r r )r� r� r r � test_async_gen_expression_01c s
z0AsyncGenAsyncioTest.test_async_gen_expression_01c sR �fdd���fdd�� � fdd�}�j �|� �}��|dd� td d
�D �� d S ) Nc � s t jd� jd�I d H | S )Ng{�G�z�?)rc )r rg rc )r� )r r r �wrapt s z>AsyncGenAsyncioTest.test_async_gen_expression_02.<locals>.wrapc s � fdd�t | �D �S )Nc s$ | ] }� |�I d H r|d V qd S )NrE r )r� r )r� r r r� z s zXAsyncGenAsyncioTest.test_async_gen_expression_02.<locals>.make_arange.<locals>.<genexpr>)r� )r� )r� r r r� x s zEAsyncGenAsyncioTest.test_async_gen_expression_02.<locals>.make_arangec � s dd� � d�2 �I d H S )Nc
� s. g | y 3 d H }W n
t k
s"X |�q Y S )N)r )r� r r r r r� } s zQAsyncGenAsyncioTest.test_async_gen_expression_02.<locals>.run.<locals>.<listcomp>rM r r )r� r r rt | s z=AsyncGenAsyncioTest.test_async_gen_expression_02.<locals>.runc S s g | ] }|d �qS )rE r )r� r r r r r� � s zDAsyncGenAsyncioTest.test_async_gen_expression_02.<locals>.<listcomp>r? rM )rc r r1 r� )r rt r r )r� r r� r � test_async_gen_expression_02s s
z0AsyncGenAsyncioTest.test_async_gen_expression_02c sB g ��fdd��dd� � � �fdd�}t �|� � | �g �� d S )Nc s � � |� d S )N)r )rc r� )�messagesr r �exception_handler� s z]AsyncGenAsyncioTest.test_asyncgen_nonstarted_hooks_are_cancellable.<locals>.exception_handlerc S s dV dV d S )Nr? rE r r r r r r0 � s zYAsyncGenAsyncioTest.test_asyncgen_nonstarted_hooks_are_cancellable.<locals>.async_iteratec � sF t �� } | ��� x.� � 2 y 3 d H }W n
tk
s6X P q Y W d S )N)r Zget_running_loopr� r )rc r )r0 r� r r �main� s
zPAsyncGenAsyncioTest.test_asyncgen_nonstarted_hooks_are_cancellable.<locals>.main)r rt r1 )r r� r )r0 r� r� r �.test_asyncgen_nonstarted_hooks_are_cancellable� s zBAsyncGenAsyncioTest.test_asyncgen_nonstarted_hooks_are_cancellablec s( dd� � � �fdd�}�j �|� � d S )Nc S s dV dV d S )Nr? rE r r r r r r0 � s zUAsyncGenAsyncioTest.test_async_gen_await_same_anext_coro_twice.<locals>.async_iteratec � sL � � } | � � }|I d H ��td�� |I d H W d Q R X | �� I d H d S )Nz4cannot reuse already awaited __anext__\(\)/asend\(\))r/ r rB r` )rn �nxt)r0 r r r rt � s
zKAsyncGenAsyncioTest.test_async_gen_await_same_anext_coro_twice.<locals>.run)rc r )r rt r )r0 r r �*test_async_gen_await_same_anext_coro_twice� s z>AsyncGenAsyncioTest.test_async_gen_await_same_anext_coro_twicec s( dd� � � �fdd�}�j �|� � d S )Nc S s dV dV d S )Nr? rE r r r r r r0 � s zVAsyncGenAsyncioTest.test_async_gen_await_same_aclose_coro_twice.<locals>.async_iteratec � s> � � } | � � }|I d H ��td�� |I d H W d Q R X d S )Nz2cannot reuse already awaited aclose\(\)/athrow\(\))r` r rB )rn r� )r0 r r r rt � s
zLAsyncGenAsyncioTest.test_async_gen_await_same_aclose_coro_twice.<locals>.run)rc r )r rt r )r0 r r �+test_async_gen_await_same_aclose_coro_twice� s
z?AsyncGenAsyncioTest.test_async_gen_await_same_aclose_coro_twicec s& dd� � � fdd�}| j �|� � d S )Nc S s dV dV d S )Nr? rE r r r r r r0 � s z[AsyncGenAsyncioTest.test_async_gen_aclose_twice_with_different_coros.<locals>.async_iteratec � s& � � } | � � I d H | � � I d H d S )N)r` )rn )r0 r r rt � s zQAsyncGenAsyncioTest.test_async_gen_aclose_twice_with_different_coros.<locals>.run)rc r )r rt r )r0 r �0test_async_gen_aclose_twice_with_different_coros� s zDAsyncGenAsyncioTest.test_async_gen_aclose_twice_with_different_corosc s& dd� � � fdd�}| j �|� � d S )Nc S s dV dV d S )Nr? rE r r r r r r0 � s zQAsyncGenAsyncioTest.test_async_gen_aclose_after_exhaustion.<locals>.async_iteratec � sD � � } x*| 2 y 3 d H }W q t k
s&X q Y W | �� I d H d S )N)r r` )rn r� )r0 r r rt � s zGAsyncGenAsyncioTest.test_async_gen_aclose_after_exhaustion.<locals>.run)rc r )r rt r )r0 r �&test_async_gen_aclose_after_exhaustion� s z:AsyncGenAsyncioTest.test_async_gen_aclose_after_exhaustionN)&r r r rd rf r rh ri rk rq ru rx 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 rb z sF *###!%* "5.# !
rb �__main__)r_ rZ ZunittestZ test.supportr r r* r � coroutiner r r ZTestCaser r' rb r r� r r r r �<module> s,
1 c