Your IP : 216.73.216.213
U
i�f�} � @ 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 �7/opt/alt/python38/lib64/python3.8/test/test_asyncgen.pyr s r F��throwc c s | r dV ndV d S )Nr ��resultr r r r r
� awaitable
s r c
C sj d}z"|rd}| � t�}n
| �d �}W n0 tk
rV } z|jd W Y �S d }~X Y nX |dkrd}qd 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 } � 2 z3 d H W }| � |� q6 | S �N��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 5 Q R X d S )NzJasync def foo():
await abc
yield from 123
�yield 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 5 Q R X d S )Nz4async def foo():
yield from 123
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 5 Q R X d S )NzXasync def foo():
await abc
yield
return 123
�return.*value.*async genr# 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 5 Q R X d S )NzBasync def foo():
yield
return 123
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 5 Q R X d S )NzWasync def foo():
if 0:
yield
return 12
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 s0 dd� }dd� }||�}||�}| � ||� |S )Nc
S sr g }z|� | �� � W q tk
r8 |� d� Y qnY q tk
rj } z|� tt|��� W 5 d }~X Y qX q|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 }| � � }z�z |�� W q tk
rn } z6|jrL|�|jd � W Y �q�n|�d� W Y �q�W 5 d }~X Y q tk
r� � Y q tk
r� } z |�tt|��� W Y �q�W 5 d }~X Y qX qW q tk
r� |�d� Y q�Y qX q|S )Nr zEMPTY StopIterationr1 ) � __anext__r2 r r r �StopAsyncIterationr3 r4 r5 )r6 r �anr r r r
�
async_iterateo s(
z6AsyncGenTest.compare_generators.<locals>.async_iterate)� assertEqual)r( �sync_gen� async_genr7 r; 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�{ �� � )r ZassertIs)�a�r( r r
r � s z5AsyncGenTest.test_async_gen_iteration_01.<locals>.genr@ rA rB )r< r �r( r r rD r
�test_async_gen_iteration_01� s z(AsyncGenTest.test_async_gen_iteration_01c
C s� dd� }|� }|� � }|�� }| �|�� d� z |�� W n4 tk
rn } z| �|jd d� W 5 d }~X Y n X | �d� |�� }| �|�� d� z |�� W n. tk
r� } z| �|j� W 5 d }~X Y n X | �d� d S )Nc S s" t � I d H dV t � I d H d S �Nr@ �r r r r r
r � s z5AsyncGenTest.test_async_gen_iteration_02.<locals>.genr
r r@ � StopIteration was not raisedz!StopAsyncIteration was not raised) � __aiter__r8 r< r2 r r �failr9 � assertFalse�r( r r6 Zair: r r r r
�test_async_gen_iteration_02� s" $
z(AsyncGenTest.test_async_gen_iteration_02c C s, dd� }| � t�� t|� � W 5 Q R X d S )Nc S s, t � I d H dV t dd�I d H dV d S )Nr@ Tr rA rH r r r r
r � s z5AsyncGenTest.test_async_gen_exception_03.<locals>.gen)� assertRaisesr r rE r r r
�test_async_gen_exception_03� s z(AsyncGenTest.test_async_gen_exception_03c
C s� dd� }|� }|� � }|�� }| �|�� d� z |�� W n4 tk
rn } z| �|jd d� W 5 d }~X Y n X | �d� | �t�� |�� �� W 5 Q R X d S )Nc S s t � I d H dV dd d S )Nr@ � r rH r r r r
r � s z5AsyncGenTest.test_async_gen_exception_04.<locals>.genr
r r@ rI ) rJ r8 r< r2 r r rK rO �ZeroDivisionErrorrM r r r
�test_async_gen_exception_04� s $
z(AsyncGenTest.test_async_gen_exception_04c C s. dd� }| � td�� t|� � W 5 Q R X d S )Nc S s dV t �d S rG )r9 r r r r
r � s z5AsyncGenTest.test_async_gen_exception_05.<locals>.genz#async generator.*StopAsyncIteration�r$ � RuntimeErrorr rE r r r
�test_async_gen_exception_05� s
�z(AsyncGenTest.test_async_gen_exception_05c C s. dd� }| � td�� t|� � W 5 Q R X d S )Nc S s dV t �d S rG )r r r r r
r � s z5AsyncGenTest.test_async_gen_exception_06.<locals>.genz async generator.*StopIterationrT rE 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 5 dV dV X dV d S �N� � rQ r �d r r r r r
r= � s z:AsyncGenTest.test_async_gen_exception_07.<locals>.sync_genc S s, zdV dd W 5 dV dV X dV d S rX r r r r r
r> � s z;AsyncGenTest.test_async_gen_exception_07.<locals>.async_gen�r? �r( r= r> 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 5 dV dd dV X dV d S �NrY rQ r rZ r[ r r r r r
r= � s
z:AsyncGenTest.test_async_gen_exception_08.<locals>.sync_genc S sD zdV t � I d H W 5 t � I d H dV dd dV X dV d S r_ rH r r r r
r> s z;AsyncGenTest.test_async_gen_exception_08.<locals>.async_genr\ r] 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 5 dV dV X dV d S rX r r r r r
r= s z:AsyncGenTest.test_async_gen_exception_09.<locals>.sync_genc S sD z t � I d H dV dd W 5 dV t � I d H dV X dV d S rX rH r r r r
r> " s z;AsyncGenTest.test_async_gen_exception_09.<locals>.async_genr\ r] r r r
�test_async_gen_exception_09 s
z(AsyncGenTest.test_async_gen_exception_09c C s4 dd� }| � td�� |� �� �d� W 5 Q R X d S )Nc S s
dV d S rG r r r r r
r 1 s z5AsyncGenTest.test_async_gen_exception_10.<locals>.genz!non-None value .* async generatorr[ )r$ � TypeErrorr8 r rE r r r
�test_async_gen_exception_100 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
r= 8 s z:AsyncGenTest.test_async_gen_exception_11.<locals>.sync_genc 3 sL dV � � } | � d � z| �t� � W n tk
r@ dV Y nX dV d S �NrQ rY rZ )r r �
GeneratorExit)�sg)r= r r
�sync_gen_wrapper< s
zBAsyncGenTest.test_async_gen_exception_11.<locals>.sync_gen_wrapperc S s dV dV d S rd r r r r r
r> F s z;AsyncGenTest.test_async_gen_exception_11.<locals>.async_genc sX dV � � } | � d �I d H z| �t� �I d H W n tk
rL dV Y nX dV d S rg )�asend�athrowrh )Zasg)r> r r
�async_gen_wrapperJ s zCAsyncGenTest.test_async_gen_exception_11.<locals>.async_gen_wrapperr\ )r( rj rm r )r> r= r
�test_async_gen_exception_117 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 rG r r r r r
r W s z/AsyncGenTest.test_async_gen_api_01.<locals>.genr Z123z.gen)r< r ZassertInr Z assertIsNone�ag_await�assertIsInstance�ag_frame�types� FrameTyperL �
ag_running�ag_code�CodeTypeZ
assertTrue�inspectZ isawaitable�aclose)r( r r6 r r r
�test_async_gen_api_01V s z"AsyncGenTest.test_async_gen_api_01N)r r r r? rF rN rP rS rV rW r^ r` ra rc rn ry r r r r
r0 ` s +
r0 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 r )r Znew_event_loop�loopZset_event_looprD r r r
�setUpn s
zAsyncGenAsyncioTest.setUpc C s | j �� d | _ t�d � d S r )r{ �closer Zset_event_loop_policyrD r r r
�tearDownr s
z AsyncGenAsyncioTest.tearDownc � s& g }|2 z3 d H W }|� |� q6 |S r r )r( r r r r r r
r w s zAsyncGenAsyncioTest.to_listc C s0 dd� }| j �| �|� ��}| �|ddg� d S )Nc S s0 dV t �d�I d H dV t �d�I d H d S )NrQ �{�G�z�?rY rZ �r �sleepr r r r
r ~ s z:AsyncGenAsyncioTest.test_async_gen_asyncio_01.<locals>.genrQ rY �r{ r r r< )r( r r r r r
�test_async_gen_asyncio_01} s z-AsyncGenAsyncioTest.test_async_gen_asyncio_01c C s6 dd� }| � t�� | j�| �|� �� W 5 Q R X d S )Nc S s. dV t �d�I d H dV dd dV d S )NrQ r rY r rZ r� r r r r
r � s
z:AsyncGenAsyncioTest.test_async_gen_asyncio_02.<locals>.gen)rO rR r{ r r rE r r r
�test_async_gen_asyncio_02� s z-AsyncGenAsyncioTest.test_async_gen_asyncio_02c C s: | j }G dd� d�}|�| �|� ��}| �|ddg� d S )Nc @ s e Zd Zdd� ZdS )z:AsyncGenAsyncioTest.test_async_gen_asyncio_03.<locals>.Genc S s dV t �d�I d H dV d S )NrQ r rY r� rD r r r
rJ � s zDAsyncGenAsyncioTest.test_async_gen_asyncio_03.<locals>.Gen.__aiter__N)r r r rJ r r r r
�Gen� s r� rQ rY r� )r( r{ r� r r r r
�test_async_gen_asyncio_03� s z-AsyncGenAsyncioTest.test_async_gen_asyncio_03c sD dd� � � �fdd�}� �fdd�}�j �|� � �j �|� � d S )Nc S s\ dV t �d�I d H zdV dV W n tk
r@ dV Y nX t �d�I d H dV d S )NrQ r rY rZ �� � )r r� rR 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 5 Q R X ��t�� | �� I d H W 5 Q R X d S )NrQ rY rZ r� )rJ r< r8 rO r9 ��it�r� r( r r
�run1� s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_anext_04.<locals>.run1c
� s� � � � � } ��| �� I d H d� ��| �� I d H d� z| �� �t� W n4 tk
r| } z��|jd d� W 5 d }~X Y n X ��d� ��| �� I d H d� ��t �� | �� I d H W 5 Q R X d S )NrQ rY r r� rI r� )
rJ r< r8 r rR r r rK rO r9 �r� r r� r r
�run2� s
$
zAAsyncGenAsyncioTest.test_async_gen_asyncio_anext_04.<locals>.run2�r{ r )r( r� r� r 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 �NrQ r[ r ��vr r r
r� � s z@AsyncGenAsyncioTest.test_async_gen_asyncio_anext_05.<locals>.fooc
� s( � � � � } z| �� �d � W n4 tk
rP } z��|jd d� W 5 d }~X Y n X ��d� z| �� �d� W n4 tk
r� } z��|jd d� W 5 d }~X Y n X ��d� z| �� �d� W n4 tk
r� } z��|jd d� W 5 d }~X Y n X ��d� ��t�� | �� I d H W 5 Q R X d S )Nr rQ rI re � i� ) rJ r8 r r r< r rK rO r9 r� r� r r
�run� s$
$
$
$
z@AsyncGenAsyncioTest.test_async_gen_asyncio_anext_05.<locals>.runr� �r( r� r r� r
� test_async_gen_asyncio_anext_05� s z3AsyncGenAsyncioTest.test_async_gen_asyncio_anext_05c sv d� dd� }|� }|� d � ��t�� |� d � W 5 Q R X � fdd��� ��fdd�}�j�|� � ��� d� d S ) Nr c s s z
d V W n Y nX d S r r r r r r
r� � s
z@AsyncGenAsyncioTest.test_async_gen_asyncio_anext_06.<locals>.fooc s z
d V W n Y nX d� d S �NrQ 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 5 Q R X � d7 � d S �Nre )rk rO r9 �r6 �r� r r( r r
r� s
z@AsyncGenAsyncioTest.test_async_gen_asyncio_anext_06.<locals>.run� )r rO r r{ r r< �r( r� r6 r� 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* z
dV W n t k
r$ dV Y nX d S )N�rQ �rY �rR r r r r
r� s
zCAsyncGenAsyncioTest.test_async_gen_asyncio_anext_tuple.<locals>.fooc � s� � � � � } ��| �� I d H d� ��t��}| �� �t� W 5 Q R X ��|jjd d� ��t �� | �� I d H W 5 Q R X d S )Nr� r r� )
rJ r< r8 rO r r rR � exceptionr r9 )r� �cmr� r r
r� s
zCAsyncGenAsyncioTest.test_async_gen_asyncio_anext_tuple.<locals>.runr� r� r 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 zt d�V W n tk
r, t d�V Y nX d S )NrQ rZ )r rR r r r r
r� $ s zKAsyncGenAsyncioTest.test_async_gen_asyncio_anext_stopiteration.<locals>.fooc � s� � � � � } | �� I d H }��|t� ��|jd� ��t��}| �� �t� W 5 Q R X |j j
d }��|t� ��|jd� ��t �� | �� I d H W 5 Q R X d S )NrQ r rZ ) rJ r8 rp r r< �valuerO r rR r� r r9 )r� r� r� r� r r
r� * s
zKAsyncGenAsyncioTest.test_async_gen_asyncio_anext_stopiteration.<locals>.runr� r� r r� r
�*test_async_gen_asyncio_anext_stopiteration# s z>AsyncGenAsyncioTest.test_async_gen_asyncio_anext_stopiterationc s> dd� � � fdd�}| � td�� | j�|� � W 5 Q R X d S )Nc S s0 zdV dd W 5 t �d�I d H dV X d S )Nr r� rQ r r� r r r r
r� ; s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_06.<locals>.fooc � s. � � } | � � }|�� I d H | �� I d H d S r �rJ r8 rx �r r� �r� r r
r� C s zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_06.<locals>.runz%async generator ignored GeneratorExit)r$ rU r{ r r� r r� r
� test_async_gen_asyncio_aclose_06: s �z4AsyncGenAsyncioTest.test_async_gen_asyncio_aclose_06c s: d� � fdd���fdd�}| j �|� � | �� d� d S )Nr c sJ zdV dd W 5 t �d�I d H t �d�I d H � d7 � X � d7 � d S )Nr rQ r r� r� r r� r r
r� Q s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_07.<locals>.fooc � s. � � } | � � }|�� I d H | �� I d H d S r r� r� r� r r
r� \ s zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_07.<locals>.runrQ �r{ r r< r� r )r� r� r
� test_async_gen_asyncio_aclose_07N s
z4AsyncGenAsyncioTest.test_async_gen_asyncio_aclose_07c sf d� t j�jd��� �fdd����fdd�}�j�|� � ��� d� ��� �j�t �d�� d S ) Nr �r{ c sZ z"dV �I d H � d7 � dV W 5 t �d�I d H t �d�I d H � d7 � X � d7 � d S �Nr rQ r� rY r� r )r� r r r
r� j s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_08.<locals>.fooc � s6 � � } | � � }��|�� I d H d� | �� I d H d S r� )rJ r< r8 rx r� r� r r
r� w s zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_08.<locals>.runrQ r )r ZFuturer{ r r< �cancelr� r� r )r� r� r r( r
� test_async_gen_asyncio_aclose_08e s
z4AsyncGenAsyncioTest.test_async_gen_asyncio_aclose_08c s: d� � fdd���fdd�}| j �|� � | �� d� d S )Nr c s8 z dV qW 5 t �d�I d H t �d�I d H d� X d S )Nr rQ r� r r� r r
r � s
zDAsyncGenAsyncioTest.test_async_gen_asyncio_gc_aclose_09.<locals>.genc � s8 � � } | � � I d H | � � I d H ~ t�d�I d H d S )N皙�����?)r8 r r� r� r r r
r� � s
zDAsyncGenAsyncioTest.test_async_gen_asyncio_gc_aclose_09.<locals>.runrQ r� r� r �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 z
d V W n Y nX d S r r r r r r
r� � s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_10.<locals>.fooc s z
d V W n Y nX d� d S r� r r r� r r
r � s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_10.<locals>.genc � s0 �� } | � d �I d H | �� I d H � d7 � d S r� )rk rx r� r� r r
r� � s zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_10.<locals>.runr� )r r} r{ r r< r� r r� r
� test_async_gen_asyncio_aclose_10� s
z4AsyncGenAsyncioTest.test_async_gen_asyncio_aclose_10c sv d� dd� }|� }|� d � ��td�� |�� W 5 Q R X � fdd��� ��fdd�}�j�|� � ��� d � d S )
Nr c s s" z
d V W n Y nX d V d S r r r r r r
r� � s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_11.<locals>.foo�ignored GeneratorExitc s* z
d V W n Y nX d V � d7 � d S r� r r r� r r
r � s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_11.<locals>.genc � sH �� } | � d �I d H ��td�� | �� I d H W 5 Q R X � d7 � d S )Nr� re )rk r$ rU rx r� r� r r
r� � s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_11.<locals>.runre )r r$ rU r} r{ r r< r� 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 )Nr rQ r r� r r r r
�target� s zDAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_12.<locals>.targetc sF t ��� �} z
dV W 5 z| I d H W n tk
r> d� Y nX X d S r� )r � create_taskrR )�task)r� r� r r
r� � s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_12.<locals>.fooc � s. � � } | � � }|�� I d H | �� I d H d S r r� r� r� r r
r� � s zAAsyncGenAsyncioTest.test_async_gen_asyncio_aclose_12.<locals>.runrQ r� r� r )r� r� r� r
� test_async_gen_asyncio_aclose_12� s z4AsyncGenAsyncioTest.test_async_gen_asyncio_aclose_12c sv d� dd� }|� }|� d �}��|d� |� d�}��|d� � fdd����fd d
�}�j�|� � ��� d� d S ) Nr c s s dV } | d V d S �NrQ rY r r� r r r
�sgen s zAAsyncGenAsyncioTest.test_async_gen_asyncio_asend_01.<locals>.sgenrQ r[ �� c st zHt �d�I d H dV } t �d�I d H | d V t �d�I d H W �d S t �d�I d H t �d�I d H d� X d S �Nr rQ rY r� r� 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 5 Q R X d S )NrQ r[ r� )rk r< rO r9 �r6 r� �r r( r r
r� s z@AsyncGenAsyncioTest.test_async_gen_asyncio_asend_01.<locals>.run)r r< r{ r �r( r� ri r� r� r r� r
� test_async_gen_asyncio_asend_01� s
z3AsyncGenAsyncioTest.test_async_gen_asyncio_asend_01c s\ d� dd� �� �fdd����fdd�}�� t�� �j�|� � W 5 Q R X ��� d� d S ) Nr c � s t �| �I d H dd d S )NrQ r r� �Zdelayr r r
�
sleep_n_crash* s zJAsyncGenAsyncioTest.test_async_gen_asyncio_asend_02.<locals>.sleep_n_crashc sf z:t �d�I d H dV } �d�I d H � d7 � | d V W 5 t �d�I d H t �d�I d H d� X d S r� r� r� �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 r� �rk r< r� r� r r
r� ; s z@AsyncGenAsyncioTest.test_async_gen_asyncio_asend_02.<locals>.runrQ )rO rR r{ r r< r� r �r� r r( r� r
� test_async_gen_asyncio_asend_02' s
z3AsyncGenAsyncioTest.test_async_gen_asyncio_asend_02c sb d� �fdd��� �fdd����fdd�}�� tj�� �j�|� � W 5 Q R X ��� d� d S ) Nr c � s: t jt �| ��jd�� �j�| d � fdd�� � I d H S )Nr� rY c s � � � S r �r� r �r r r
�<lambda>M � z\AsyncGenAsyncioTest.test_async_gen_asyncio_asend_03.<locals>.sleep_n_crash.<locals>.<lambda>�r Z
ensure_futurer� r{ Z
call_laterr� rD r� r
r� J s
�zJAsyncGenAsyncioTest.test_async_gen_asyncio_asend_03.<locals>.sleep_n_crashc sf z:t �d�I d H dV } �d�I d H � d7 � | d V W 5 t �d�I d H t �d�I d H d� X d S r� r� r� r� r r
r P s z@AsyncGenAsyncioTest.test_async_gen_asyncio_asend_03.<locals>.genc � s6 � � } | � d �I d H }��|d� | � d�I d H d S r� r� r� r� r r
r� ] s z@AsyncGenAsyncioTest.test_async_gen_asyncio_asend_03.<locals>.runrQ )rO r �CancelledErrorr{ r r< r� r r� r
� test_async_gen_asyncio_asend_03G s
z3AsyncGenAsyncioTest.test_async_gen_asyncio_asend_03c s� d� G dd� dt ���fdd�}|� }|�d �}��|d� |���}��|d� ��t�� |�d � W 5 Q R X � �fdd �����fd
d �}�j�|� � ��� d� d S ) Nr c @ s e Zd ZdS )zCAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_01.<locals>.FooErNr r r r r
�FooErl s r� c 3 s2 z
dV } W n � k
r" d} Y nX | d V d S )NrQ r� rY r r� )r� r r
r� p s
zBAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_01.<locals>.sgenrQ �� c s� zbt �d�I d H z
dV } W n( �k
rD d} t �d�I d H Y nX | d V t �d�I d H W 5 t �d�I d H t �d�I d H d� X d S r� r� r� )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 5 Q R X d S )NrQ r� )rk r< rl rO r9 r� )r� r r( r r
r� � s zAAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_01.<locals>.run)r3 r r< r rO r r{ r r� r �r� r� r r( r
� test_async_gen_asyncio_athrow_01i s
z4AsyncGenAsyncioTest.test_async_gen_asyncio_athrow_01c sx d� G dd� dt ���fdd��� ��fdd��� ���fdd �}��tj�� �j�|� � W 5 Q R X ��� d
� d S ) Nr c @ s e Zd ZdS )zCAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_02.<locals>.FooErNr r r r r
r� � s r� c � s: t jt �| ��jd�� �j�| d � fdd�� � I d H S )Nr� rY c s � � � S r r� r r� r r
r� � r� z]AsyncGenAsyncioTest.test_async_gen_asyncio_athrow_02.<locals>.sleep_n_crash.<locals>.<lambda>r� r� rD r� r
r� � s
�zKAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_02.<locals>.sleep_n_crashc s� z\t �d�I d H z
dV } W n" �k
r> �d�I d H Y nX | d V t �d�I d H W 5 t �d�I d H t �d�I d H d� X d S r� r� r� )r� r� r� r r
r � s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_02.<locals>.genc � sj �� } | � d �I d H }��|d� z| ���I d H W n$ tjk
rZ ��� d� � Y n X ��d� d S )NrQ z CancelledError was not raised)rk r< rl r r� rK r� r� r r
r� � s zAAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_02.<locals>.runrQ )r3 rO r r� r{ r r< r� r )r� 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 5 Q R X � fdd��� ��fdd�}�j�|� � ��� d� d S ) Nr c s s z
d V W n Y nX d S r r r r r r
r� � s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_03.<locals>.fooc s z
d V W n Y nX d� d S r� r r r� r r
r � s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_03.<locals>.genc � sH �� } | � d �I d H ��t�� | �t�I d H W 5 Q R X � d7 � d S r� )rk rO r9 rl �
ValueErrorr� r� r r
r� � s
zAAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_03.<locals>.runr� )r rO r r r� r{ r r< r� 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* z
dV W n t k
r$ dV Y nX d S �NrQ r� 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 5 Q R X d S r� )rk r< rl rR rO r9 r� r� r r
r� � s zDAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_tuple.<locals>.runr� 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. z
dV W n t k
r( td�V Y nX d S r� )rR 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 5 Q R X d S r� ) rk r< rl rR rp r r� rO r9 r� r� r r
r� s zLAsyncGenAsyncioTest.test_async_gen_asyncio_athrow_stopiteration.<locals>.runr� 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 �t�d�� |�� |�� | �tj�� | j �|� W 5 Q R X | �tj�� | j �|� W 5 Q R X | j �| j �� � | � � d� d S )Nr c s: zt �| �I d H dV W 5 t �d�I d H � d7 � X d S )Nr rQ r� )Ztimeout)� finalizedr r
�waiter s
zFAsyncGenAsyncioTest.test_async_gen_asyncio_shutdown_01.<locals>.waiterc � s � d�2 z 3 d H W } q6 d S r� r )�_)r� r r
�wait s zDAsyncGenAsyncioTest.test_async_gen_asyncio_shutdown_01.<locals>.waitr� rY )
r{ r� r r r� r� rO r� Zshutdown_asyncgensr< )r( r� �t1�t2r )r� r� r
�"test_async_gen_asyncio_shutdown_01 s z6AsyncGenAsyncioTest.test_async_gen_asyncio_shutdown_01c sL dd� � � fdd���fdd�}| j �|� �}| �|dd� td �D �� d S )
Nc S s( t | �D ]}t�d�I d H |V qd S �Nr )�ranger r� )�nr r r r
�arange7 s z@AsyncGenAsyncioTest.test_async_gen_expression_01.<locals>.arangec s dd� � | �2 �S )Nc S s | z3 d H W }|d V q6 d S �NrY r ��.0r r r r
� <genexpr>> s zXAsyncGenAsyncioTest.test_async_gen_expression_01.<locals>.make_arange.<locals>.<genexpr>r �r� )r� r r
� make_arange< s zEAsyncGenAsyncioTest.test_async_gen_expression_01.<locals>.make_arangec � s dd� � d�2 �I d H S )Nc � s g | z3 d H W }|�q6 S r r r� r r r
�
<listcomp>A s zQAsyncGenAsyncioTest.test_async_gen_expression_01.<locals>.run.<locals>.<listcomp>re r r �r� r r
r� @ s z=AsyncGenAsyncioTest.test_async_gen_expression_01.<locals>.runc S s g | ] }|d �qS r� r r� r r r
r� D s zDAsyncGenAsyncioTest.test_async_gen_expression_01.<locals>.<listcomp>re �r{ r r< r� �r( r� r r )r� r� r
� test_async_gen_expression_016 s
z0AsyncGenAsyncioTest.test_async_gen_expression_01c sN dd� ��fdd�� � fdd�}| j �|� �}| �|dd� td d
�D �� d S ) Nc � s t �d�I d H | S r� r� r� r r r
�wrapG 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 r� r r� �r� r r
r� M s zXAsyncGenAsyncioTest.test_async_gen_expression_02.<locals>.make_arange.<locals>.<genexpr>)r� r� r� r r
r� K s zEAsyncGenAsyncioTest.test_async_gen_expression_02.<locals>.make_arangec � s dd� � d�2 �I d H S )Nc � s g | z3 d H W }|�q6 S r r r� r r r
r� P s zQAsyncGenAsyncioTest.test_async_gen_expression_02.<locals>.run.<locals>.<listcomp>re r r r� r r
r� O s z=AsyncGenAsyncioTest.test_async_gen_expression_02.<locals>.runc S s g | ] }|d �qS r� r r� r r r
r� S s zDAsyncGenAsyncioTest.test_async_gen_expression_02.<locals>.<listcomp>rQ re r� r� r )r� r� r
� test_async_gen_expression_02F s
z0AsyncGenAsyncioTest.test_async_gen_expression_02c sB g ��fdd��dd� � � �fdd�}t �|� � | �g �� d S )Nc s � � |� d S r r )r{ �context)�messagesr r
�exception_handlerY s z]AsyncGenAsyncioTest.test_asyncgen_nonstarted_hooks_are_cancellable.<locals>.exception_handlerc S s dV dV d S r� r r r r r
r; \ s zYAsyncGenAsyncioTest.test_asyncgen_nonstarted_hooks_are_cancellable.<locals>.async_iteratec � s0 t �� } | ��� � � 2 z3 d H W } q,q6 d S r )r Zget_running_loopZset_exception_handler)r{ r )r; r� r r
�main` s
zPAsyncGenAsyncioTest.test_asyncgen_nonstarted_hooks_are_cancellable.<locals>.main)r r� r< )r( r� r )r; r� r� r
�.test_asyncgen_nonstarted_hooks_are_cancellableU s zBAsyncGenAsyncioTest.test_asyncgen_nonstarted_hooks_are_cancellablec s( dd� � � �fdd�}�j �|� � d S )Nc S s dV dV d S r� r r r r r
r; l s zUAsyncGenAsyncioTest.test_async_gen_await_same_anext_coro_twice.<locals>.async_iteratec � sL � � } | � � }|I d H ��td�� |I d H W 5 Q R X | �� I d H d S )Nz4cannot reuse already awaited __anext__\(\)/asend\(\))r8 r$ rU rx �r� Znxt�r; r( r r
r� p s
�zKAsyncGenAsyncioTest.test_async_gen_await_same_anext_coro_twice.<locals>.runr� r� r r� r
�*test_async_gen_await_same_anext_coro_twicek s z>AsyncGenAsyncioTest.test_async_gen_await_same_anext_coro_twicec s( dd� � � �fdd�}�j �|� � d S )Nc S s dV dV d S r� r r r r r
r; s zVAsyncGenAsyncioTest.test_async_gen_await_same_aclose_coro_twice.<locals>.async_iteratec � s> � � } | � � }|I d H ��td�� |I d H W 5 Q R X d S )Nz2cannot reuse already awaited aclose\(\)/athrow\(\))rx r$ rU r� r� r r
r� � s
�zLAsyncGenAsyncioTest.test_async_gen_await_same_aclose_coro_twice.<locals>.runr� r� r 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 r� r r r r r
r; � s z[AsyncGenAsyncioTest.test_async_gen_aclose_twice_with_different_coros.<locals>.async_iteratec � s& � � } | � � I d H | � � I d H d S r �rx r� �r; r r
r� � s zQAsyncGenAsyncioTest.test_async_gen_aclose_twice_with_different_coros.<locals>.runr� r� r r 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 r� r r r r r
r; � s zQAsyncGenAsyncioTest.test_async_gen_aclose_after_exhaustion.<locals>.async_iteratec � s, � � } | 2 z 3 d H W }q
6 | � � I d H d S r r� )r� r� r r r
r� � s zGAsyncGenAsyncioTest.test_async_gen_aclose_after_exhaustion.<locals>.runr� r� r r r
�&test_async_gen_aclose_after_exhaustion� s z:AsyncGenAsyncioTest.test_async_gen_aclose_after_exhaustionc s& dd� � � fdd�}| j �|� � d S )Nc S s t � V d S r )�objectr r r r
�async_generator� s z\AsyncGenAsyncioTest.test_async_gen_aclose_compatible_with_get_stack.<locals>.async_generatorc � s2 � � } t �| �� � t �� }|D ] }|�� q d S r )r r� rx Z all_tasksZ get_stack)ZagZtasksr� �r r r
r� � s
zPAsyncGenAsyncioTest.test_async_gen_aclose_compatible_with_get_stack.<locals>.runr� r� r r r
�/test_async_gen_aclose_compatible_with_get_stack� s zCAsyncGenAsyncioTest.test_async_gen_aclose_compatible_with_get_stackN)&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� r r r r r r r
rz l sF *## !%* "5.#"
rz �__main__)rw rr ZunittestZ test.supportr r r3 r � coroutiner r r ZTestCaser! r0 rz r r� r r r r
�<module> s,
1 S