File: //usr/local/lib/python3.7/test/__pycache__/test_grammar.cpython-37.opt-1.pyc
B
��g� � ( @ s4 U d dl mZ d dlZd dlZd dlZd dlT d dlmZ d dlZd dlm Z d dl
m Z d dl
Z
dddd d
d d d
dddddddddddddgZ dd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBg(Z
G dCdD� dDej�ZeedE< G dFdG� dG�ZG dHdI� dIej�ZedJk�r0e�� dS )K� )�check_syntax_errorN)�*)�ChainMap)� ann_module2Z0_0_0Z4_2Z 1_0000_0000Z 0b1001_0100Z 0xffff_ffffZ0o5_7_7z 1_00_00.5z 1_00_00.5e5Z 1_00_00e5_1Z1e1_0z.1_4z.1_4e1Z0b_0Z0x_fZ0o_5Z1_00_00jz
1_00_00.5jZ 1_00_00e5_1jz.1_4jz (1_2.5+3_3j)z(.5_6j)Z0_Z42_z1.4j_Z0x_Z0b1_Z0xf_Z0o5_z
0 if 1_Else 1Z0_b0Z0_xfZ0_o5Z0_7Z09_99Z 4_______2z0.1__4z0.1__4jZ 0b1001__0100Z 0xffff__ffffZ0x___Z0o5__77Z1e1__0Z1e1__0jz1_.4z1_.4jz1._4z1._4jz._5z._5jz1.0e+_1z1.0e+_1jz1.4_jz1.4e5_jZ1_e1z1.4_e1z1.4_e1jZ1e_1z1.4e_1z1.4e_1jz
(1+1.5_j_)z (1+1.5_j)c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�
TokenTestsc C s( d}| � |dd� d}| � |dd� d S )N� z backslash for line continuationr zbackslash ending comment)� assertEqual)�self�x� r �-/usr/local/lib/python3.7/test/test_grammar.py�test_backslashg s zTokenTests.test_backslashc C sr | � td�td�� | � dd� | � dd� | � dd� | � dd� | �ttd� ddlm} |dkr�| � dd� | �ddk� | �ddk� | �ddk� x�d D ]4}y t|�}W q� tk
r� | � d
| � Y q�X q�W n�|d k�r`| � d d � | �d
dk� | �d
dk� | �ddk� xPdD ]8}y t|�}W n$ tk
�rV | � d
| � Y nX �q"W n| � d| � d S )Nr � i���� �0x)�maxsizei �l �� )Z
2147483648Z
0o40000000000Z 0x100000000Z"0b10000000000000000000000000000000z(OverflowError on huge integer literal %rl ���� l���� l ���� l ���� )Z9223372036854775808Z0o2000000000000000000000Z0x10000000000000000ZA0b100000000000000000000000000000000000000000000000000000000000000zWeird maxsize value %r)
r �type� assertRaises� SyntaxError�eval�sysr �
assertTrue�
OverflowError�fail)r r �sr
r r r �test_plain_integersq s8
z TokenTests.test_plain_integersc C s$ d}d}d}d}d}d}d}d}d S )Nr l ���� l ���? l �
~ ��l!]c l l �����r )r r
r r r �test_long_integers� s z TokenTests.test_long_integersc C s0 d}d}d}d}d}d}d}d}d}d}d}d S )Ng ��Q� @g �s@gj�t��?g ��
�Bg��t� =g �W�H�Bg F�@r )r r
r r r � test_floats� s zTokenTests.test_floatsc C s6 | � dndd� | � dr dndd� | �ttd� d S )N� r z 0 if 1Else 0)r r r r )r r r r � test_float_exponent_tokenization� s z+TokenTests.test_float_exponent_tokenizationc C sX x(t D ] }| �t|�t|�dd��� qW xtD ]}| �tt|� q0W | �ttd� d S )N�_� Z_0)�VALID_UNDERSCORE_LITERALSr r �replace�INVALID_UNDERSCORE_LITERALSr r � NameError)r Zlitr r r �test_underscore_literals� s
z#TokenTests.test_underscore_literalsc C s
d}d}| � t|�dko ||k� d}d}| � t|�dkoL||koLt|�dk� d}d}| � t|�dkoz||kozt|�dk� d}d}| � t|�d ko�||k� d
}d
}| � t|�d ko�||k� d }d }| �||� d }| �||� d }| �||� d }| �||� d S ) Nr! r �'r �' �"�" zdoesn't "shrink" does it� zdoes "shrink" doesn't itz2
The "quick"
brown fox
jumps over
the 'lazy' dog.
)r �len�ordr )r r
�yr r r �test_string_literals� s0 & & z TokenTests.test_string_literalsc C s$ d}| � |tk� | �ttd� d S )N.z.. .)r �Ellipsisr r r )r r
r r r �
test_ellipsis� s zTokenTests.test_ellipsisc
C sJ d}x@|D ]8}| � t��}t|dd� W d Q R X | �dt|j�� q
W d S )N)zdef foo(z
def foo(z def foo(
z<test>�execzunexpected EOF)r r �compileZassertIn�strZ exception)r Zsamplesr �cmr r r �test_eof_error� s
zTokenTests.test_eof_errorN) �__name__�
__module__� __qualname__r
r r r r r& r/ r1 r6 r r r r r e s
$
(r Zvar_annot_globalc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�CNSc C s
i | _ d S )N)�_dct)r r r r �__init__� s z CNS.__init__c C s || j |�� < d S )N)r; �lower)r �item�valuer r r � __setitem__� s zCNS.__setitem__c C s
| j | S )N)r; )r r> r r r � __getitem__� s zCNS.__getitem__N)r7 r8 r9 r< r@ 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 e!�"d=d>�d?d@� �Z#dAdB� Z$dCdD� Z%dEdF� Z&dGdH� Z'dIdJ� Z(dKdL� Z)dMdN� Z*dOdP� Z+dQdR� Z,dSdT� Z-dUdV� Z.dWdX� Z/dYdZ� Z0d[d\� Z1d]d^� Z2d_d`� Z3dadb� Z4dcdd� Z5dedf� Z6dgdh� Z7didj� Z8dkdl� Z9dmdn� Z:dodp� Z;dqdr� Z<dsdt� Z=duS )v� GrammarTestsc C s t d�}d S )Nz 1, 0 or 1)r )r r
r r r �test_eval_input s z GrammarTests.test_eval_inputc C s@ d}dg}dd� }t tg d d||� d < | �|dg� d S )N� �* c S s dS )Nr r r r r r �one s z/GrammarTests.test_var_annot_basics.<locals>.oner r )�int�listr )r Zvar1Zmy_lstrF r r r �test_var_annot_basics s
z"GrammarTests.test_var_annot_basicsc C s^ t | d� t | d� t | d� t | d� t | d� t | d� t | d� t | d� t | d � d S )
Nz
def f: intz x: int: strz def f():
nonlocal x: int
z [x, 0]: int
z f(): int
z (x,): intz"def f():
(x, y): int = (1, 2)
z!def f():
x: int
global x
z!def f():
global x
x: int
)r )r r r r � test_var_annot_syntax_errors s
z)GrammarTests.test_var_annot_syntax_errorsc
C s> | � t�� dd tt< W d Q R X | � t�� dtt< W d Q R X dd� }| �|ji � dd� }|� dd� }| � t�� |� W d Q R X d d
� }y
|� W n2 tk
r� } z| � t
|�t� W d d }~X Y nX G d d � d �}| �|jt t d
�� | � t�� G dd� d�}W d Q R X | � t�� G dd� d|�}W d Q R X d S )Nr r c S s d} dt _| S )NZHello)r r )�a�b)�str r r �f4 s z6GrammarTests.test_var_annot_basic_semantics.<locals>.fc S s d S )Nr r r r r �f_OK9 s z9GrammarTests.test_var_annot_basic_semantics.<locals>.f_OKc S s t | � d S )N)�print)r
r r r �fbad< s z9GrammarTests.test_var_annot_basic_semantics.<locals>.fbadc S s t t� d S )N)rP Zno_such_globalr r r r �f2badA s z:GrammarTests.test_var_annot_basic_semantics.<locals>.f2badc @ s. e Zd ZU eed< dZeed< dZdd� ZdS )z6GrammarTests.test_var_annot_basic_semantics.<locals>.C�_C__foo�attrr r c S s
|| _ d S )N)r
)r r
r r r r< N s z?GrammarTests.test_var_annot_basic_semantics.<locals>.C.__init__N) r7 r8 r9 rG �__annotations__r r4 �zr< r r r r �CJ s
rW )rS r c @ s e Zd ZU de_e dS )z9GrammarTests.test_var_annot_basic_semantics.<locals>.CBadr N)r7 r8 r9 Zno_such_name_definedrT rG r r r r �CBadR s
rX c @ s e Zd ZU eed< g e_e dS )z:GrammarTests.test_var_annot_basic_semantics.<locals>.Cbad2r
N)r7 r8 r9 rG rU r
r. rH r r r r �Cbad2U s
rY )
r �ZeroDivisionErrorZno_nameZdoes_not_existr% r rU �UnboundLocalError� ExceptionZassertIsr rG r4 ) r rN rO rQ rR �erW rX rY r r r � test_var_annot_basic_semantics+ s,
" z+GrammarTests.test_var_annot_basic_semanticsc C s8 G dd� dt �}G dd� d|d�}| �|jd d� d S )Nc @ s e Zd Zedd� �ZdS )z>GrammarTests.test_var_annot_metaclass_semantics.<locals>.CMetac [ s
dt � iS )NrU )r: )�metacls�name�bases�kwdsr r r � __prepare__[ s zJGrammarTests.test_var_annot_metaclass_semantics.<locals>.CMeta.__prepare__N)r7 r8 r9 � classmethodrc r r r r �CMetaZ s re c @ s e Zd ZU ded< dS )z;GrammarTests.test_var_annot_metaclass_semantics.<locals>.CC�ANNOTZXXN)r7 r8 r9 rU r r r r �CC^ s
rg )� metaclassZxxrf )r r rU )r re rg r r r �"test_var_annot_metaclass_semanticsY s z/GrammarTests.test_var_annot_metaclass_semanticsc C sj | � t�� ttj� W d Q R X | �tjdttt j
ttf d�� | �tj jdt d�� | �t
ji � d S )Nr )r r
r. rN �{ )Z123�o)r �AttributeErrorrP �testrU r �
ann_modulerG r4 �typingZTuple�Mr r )r r r r � test_var_annot_module_semanticsb s
z,GrammarTests.test_var_annot_module_semanticsc C sn ddl m}m}m} | �t�� |� W d Q R X | �t�� |� W d Q R X | �t�� |d� W d Q R X d S )Nr )� f_bad_ann� g_bad_ann� D_bad_annrD )Ztest.ann_module3rr rs rt r r% )r rr rs rt r r r �test_var_annot_in_modulek s z%GrammarTests.test_var_annot_in_modulec C sL i }i }t d||� | �|d dtd�� | �t�� |d W d Q R X d S )Nz.'docstring'
__annotations__[1] = 2
x: int = 5
rU r )r r
)r2 r rG r �KeyError)r ZgnsZlnsr r r �test_var_annot_simple_execv s z'GrammarTests.test_var_annot_simple_execc s� dt � i}td|� | �|d d t� | �|d d t� | �t�� |d d W d Q R X i � G � fdd�d�}tdi |� � | �� d d t� d S ) NrU z'X: int; Z: str = "Z"; (w): complex = 1jr
rV �wc s( e Zd Zdd� Z� fdd�Zdd� ZdS )z5GrammarTests.test_var_annot_custom_maps.<locals>.CNS2c S s
i | _ d S )N)r; )r r r r r< � s z>GrammarTests.test_var_annot_custom_maps.<locals>.CNS2.__init__c s || j |< |� |< d S )N)r; )r r> r? )� nonloc_nsr r r@ � s
zAGrammarTests.test_var_annot_custom_maps.<locals>.CNS2.__setitem__c S s
| j | S )N)r; )r r> r r r rA � s zAGrammarTests.test_var_annot_custom_maps.<locals>.CNS2.__getitem__N)r7 r8 r9 r< r@ rA r )ry r r �CNS2� s rz z
x: int = 1)r: r2 r rG r4 r rv )r �nsrz r )ry r �test_var_annot_custom_maps s
z'GrammarTests.test_var_annot_custom_mapsc sH t � � d� i�G � �fdd�d�}tdi |� � | ��d d t� d S )NrU c s, e Zd Z� fdd�Z�fdd�Zdd� ZdS )z1GrammarTests.test_var_annot_refleak.<locals>.CNS2c s d� i| _ d S )NrU )r; )r )�cnsr r r< � s z:GrammarTests.test_var_annot_refleak.<locals>.CNS2.__init__c s || j |< |� |< d S )N)r; )r r> r? )ry r r r@ � s
z=GrammarTests.test_var_annot_refleak.<locals>.CNS2.__setitem__c S s
| j | S )N)r; )r r> r r r rA � s z=GrammarTests.test_var_annot_refleak.<locals>.CNS2.__getitem__N)r7 r8 r9 r< r@ rA r )r} ry r r rz � s rz zX: strr
)r: r2 r r4 )r rz r )r} ry r �test_var_annot_refleak� s
z#GrammarTests.test_var_annot_refleakc s� dd� }|� |d� |di � dd� }dd� }| � |jjd� | � |jjd � d
d � }d d
� }dd� }dd� }dd� }|� |d� |d� |dd� |dd� |� |d� |d� |dd� |ddddddddd d �
|d� |d� |dd� |ddd� |ddddddddd d �
|dd� |ddd� |dddd� |ddddddddd d �
d�d d �} | � | d� | d � | g �p�dg� | d�p�di �o�d�d�p�i � | f d!di� | f d!di�p�i � d�d"d#�}
|
d� |
dd� |
d�d$di� d�d%d&�} | dd� | ddd� | d'� | d�d(�� | d�d)�� | d�d*di� d�d+d,�} | � | d� | dd� | d-� | d�d.�� | d�d$di� | f ddd/�� d�d0d1�}
|
d� |
dd� |
ddd� d�d2d3�}|dd� |ddd� |dddd� d�d4d5�}|� |d� |dd� |d6� |d � |f d!di� d�d7d8�}|d� |dd� |ddd� d�d9d:�}|dd� |ddd� |dddd� |d6� |d�d*di� d�d;d<�}|� |d� |dd� |ddd� |d�d=�� |f ddd/�� d�d>d?�}|d� |dd� |ddd� |dddd� |d6� |d�d@�� |d�d.�d*di� d�dAdB�}|dd� |ddd� |dddd� |ddddd� |d6� |d�d@�� |d�d(�dCdi� ytd�dEdi� W n tk
�r� Y n X | �dF� dGdH� }|dIdJ� dIdK�dLdM�}|dddIdN� |dddIdOdP� |dddIdOdQ� dIdN�dRdS�}|dddIdIdOdT� |dddIdOdIdU� | �ttdV� | �ttdW� | �ttdX� dYdZ� }| � |d�ddg�ddd[��d\ddd[�f� | � |d�d(d��� d6i f� | �ttd]� | � |f d^d_d`��dd^d_d`�f� | � |f dad_idbd^i��dd^d_d`�f� t | dc� t | dd� t
de�dfdZ�}| � |j dgt
i� t dh�didZ�}| � |j dDt i� tdh�djdZ�}| � |j dDti� t
dh�dkdZ�}| � |j dDt
i� ddl�dmdZ�}| � |j dndi� dddo�dpdZ�}| � |j dddo�� d�dddddq�drdZ�}| � |j dddddq�� d�ddsdt�dddddd dudvdw�dxdZ�}| � |j dddddd dudvdw�� G dydz� dz�}G d{d|� d||�}| � |jj d}di� | � |jj d}di� d~d� }|t
de�d�dZ��}| � |j dgt
i� d� � fd�dZ�}d�� fd�dZ� }dd��� fd�dZ�
}t de�� fd�dZ� }d�dZ� }d�dZ� }d�dZ� }d�dZ� }d�dZ� }d�dZ� }d�dZ� }d�dZ� }d�dZ� }d�dZ� }d�dZ� }d�dZ� }d�dZ� }d�dZ� }d�dZ� }d S )�Nc S s d S )Nr r r r r �f1� � z%GrammarTests.test_funcdef.<locals>.f1r c S s d S )Nr )� one_argumentr r r �f2� r� z%GrammarTests.test_funcdef.<locals>.f2c S s d S )Nr )�two� argumentsr r r �f3� r� z%GrammarTests.test_funcdef.<locals>.f3)r� )r� r� c S s d S )Nr )Zone_argr r r �a1� r� z%GrammarTests.test_funcdef.<locals>.a1c S s d S )Nr )r� �argsr r r �a2� r� z%GrammarTests.test_funcdef.<locals>.a2c W s d S )Nr )�restr r r �v0� r� z%GrammarTests.test_funcdef.<locals>.v0c W s d S )Nr )rK r� r r r �v1� r� z%GrammarTests.test_funcdef.<locals>.v1c W s d S )Nr )rK rL r� r r r �v2� r� z%GrammarTests.test_funcdef.<locals>.v2r r � � rD � � � r r c S s d S )Nr )rK r r r �d01� r� z&GrammarTests.test_funcdef.<locals>.d01)r rK c S s d S )Nr )rK rL r r r �d11� r� z&GrammarTests.test_funcdef.<locals>.d11rL c S s d S )Nr )rK rL �cr r r �d21� r� z&GrammarTests.test_funcdef.<locals>.d21)r r r� )r r� )r� r� c S s d S )Nr )rK rL r r r �d02� r� z&GrammarTests.test_funcdef.<locals>.d02)r r )r )rK rL c S s d S )Nr )rK rL r� r r r �d12� r� z&GrammarTests.test_funcdef.<locals>.d12c S s d S )Nr )rK rL r� �dr r r �d22� r� z&GrammarTests.test_funcdef.<locals>.d22c W s d S )Nr )rK r� r r r �d01v� r� z'GrammarTests.test_funcdef.<locals>.d01v)r r r� r� c W s d S )Nr )rK rL r� r r r �d11v� r� z'GrammarTests.test_funcdef.<locals>.d11vc W s d S )Nr )rK rL r� r� r r r �d21v r� z'GrammarTests.test_funcdef.<locals>.d21vc W s d S )Nr )rK rL r� r r r �d02v r� z'GrammarTests.test_funcdef.<locals>.d02v)r r� r� c W s d S )Nr )rK rL r� r� r r r �d12v r� z'GrammarTests.test_funcdef.<locals>.d12v)r� r� rD c W s d S )Nr )rK rL r� r� r� r r r �d22v r� z'GrammarTests.test_funcdef.<locals>.d22vr� r
s fooz/Bytes should not work as keyword argument namesc S s | S )Nr )�keyr r r �pos0key1( r� z+GrammarTests.test_funcdef.<locals>.pos0key1�d )r� )�k2c S s | |||fS )Nr )�p1�p2�k1r� r r r �pos2key2* r� z+GrammarTests.test_funcdef.<locals>.pos2key2)r� �� )r� r� )r� r� c [ s | ||||fS )Nr )r� r� r� r� Zkwargr r r � pos2key2dict. r� z/GrammarTests.test_funcdef.<locals>.pos2key2dict)r� �tokwarg1�tokwarg2)r� r� r� zdef f(*): passzdef f(*,): passzdef f(*, **kwds): passc _ s | |fS )Nr )r� �kwargsr r r rN 7 s z$GrammarTests.test_funcdef.<locals>.f)r
r. )r r� r� zf(1, x=2, *(3,4), x=5)Z scrambledZfried)�eggs�spamr� r� z
f(*g(1=2))z f(**g(1=2)))�returnc S s d S )Nr )r
r r r rN G r� r� )r
c S s d S )Nr )r
r r r rN I r� c W s d S )Nr )r
r r r rN K r� c [ s d S )Nr )r
r r r rN M r� )r. c S s d S )Nr )r
r. r r r rN O r� r. )rL r� c S s d S )Nr )rK rL r� r� r r r rN Q r� )rL r� r] �gc W s d S )Nr )rK rL r� r� r] rN r� r r r rN S r� �
)�i�j� � )rL r� r] r� �hr� �kr� c _ s d S )Nr ) rK rL r� r� r] rN r� r� r� r� r� r r r rN V s c @ s e Zd Zdd�dd�ZdS )z'GrammarTests.test_funcdef.<locals>.Spamr )� _Spam__kwc S s d S )Nr )r r� r r r rN ] s z)GrammarTests.test_funcdef.<locals>.Spam.fN)r7 r8 r9 rN r r r r �Spam\ s r� c @ s e Zd ZdS )z&GrammarTests.test_funcdef.<locals>.HamN)r7 r8 r9 r r r r �Ham_ s r� r� c S s | S )Nr )r
r r r �nullc r� z'GrammarTests.test_funcdef.<locals>.nullc S s d S )Nr )r
r r r rN d s c s � S )Nr r )�closurer r rN j r� c s � S )Nr )r
)r� r r rN k r� )r� c s � S )Nr )r� )r� r r rN l r� c s � S )Nr r )r� r r rN m r� c S s d S )Nr )rK r r r rN p r� c W s d S )Nr )r� r r r rN q r� c [ s d S )Nr )rb r r r rN r r� c W s d S )Nr )rK r� r r r rN s r� c [ s d S )Nr )rK rb r r r rN t r� c W s d S )Nr )rL r� r r r rN u r� c S s d S )Nr )rL r r r rN v r� c _ s d S )Nr )r� rb r r r rN w r� c W s d S )Nr )rK rL r� r r r rN x r� c S s d S )Nr )rK rL r r r rN y r� c _ s d S )Nr )rK r� rb r r r rN z r� c _ s d S )Nr )rL r� rb r r r rN { r� c [ s d S )Nr )rL rb r r r rN | r� c _ s d S )Nr )rK rL r� rb r r r rN } r� c [ s d S )Nr )rK rL rb r r r rN ~ 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� rD )r� rD )r )r �__code__� co_varnamesr4 � TypeErrorr r r r r rH rU rG �floatrN ) r r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rN r� r� r� r )r� r � test_funcdef� sn
zGrammarTests.test_funcdefc C sX dd� }| � |� d� dd� }dd� }| � |� dddg� d&dd�fdd�fd d�}| � |� d� d'd d�}| � |dd
�d � | � |dd
d
�d� t| d� t| d� dd�dd�}| � |dd
�d� | � |dd
dd�d� dd� }dd� }dd� } dd� }
dd� } d d� } d d� }
d d� }d d� }d d� }d!d� }d"d� }d#d� }d$d� }d%d� }d S )(Nc S s dS )Nr r r r r r �<lambda>� r� z+GrammarTests.test_lambdef.<locals>.<lambda>r c S s t t S )N)rK r� r r r r r� � r� c S s dd� dD �S )Nc S s g | ] }d |k �qS )r r )�.0r
r r r �
<listcomp>� s z?GrammarTests.test_lambdef.<locals>.<lambda>.<locals>.<listcomp>)���r� r r r r r r r� � r� r c S s | S )Nr )rV r r r r� � r� c S s | � S )Nr )r. r r r r� � r� c S s | � S )Nr )r
r r r r� � r� r c S s | | | S )Nr )r
r. rV r r r r� � r� rD r� r� zlambda x: x = 2zlambda (None,): None� )r� c S s | | | S )Nr )r
r. r� r r r r� � r� � r� �
c S s dS )Nr r )rK r r r r� � r� c W s dS )Nr r )r� r r r r� � r� c [ s dS )Nr r )rb r r r r� � r� c W s dS )Nr r )rK r� r r r r� � r� c [ s dS )Nr r )rK rb r r r r� � r� c W s dS )Nr r )rL r� r r r r� � r� c S s dS )Nr r )rL r r r r� � r� c _ s dS )Nr r )r� rb r r r r� � r� c W s dS )Nr r )rK rL r� r r r r� � r� c S s dS )Nr r )rK rL r r r r� � r� c _ s dS )Nr r )rK r� rb r r r r� � r� c _ s dS )Nr r )rL r� rb r r r r� � r� c [ s dS )Nr r )rL rb r r r r� � r� c _ s dS )Nr r )rK rL r� rb r r r r� � r� c [ s dS )Nr r )rK rL rb r r r r� � r� )r )r )r r )r �l1�l2�l3�l4�l5�l6�l10Zl11Zl12Zl13Zl14Zl15Zl16Zl17Zl18Zl19Zl20Zl21Zl22Zl23Zl24r r r � test_lambdef� s<
zGrammarTests.test_lambdefc C s d}~dd� }|� d S )Nr c S s
d} ~ d S )Nr r )r
r r r �foo� s z*GrammarTests.test_simple_stmt.<locals>.foor )r r
r� r r r �test_simple_stmt� s z GrammarTests.test_simple_stmtc C sR d}d}d } }}d\}}}d } \}}} \}}}}t | d� t | d� d S )Nr )r r r� )r r )r� r� z x + 1 = 1z
a + 1 = b + 2)r ) r r
r. rV �abcrK rL r� �xyzr r r �test_expr_stmt� s
zGrammarTests.test_expr_stmtc C s� d}ddddddg}x�|D ]�}d� |�}x�|D ]�}|� |�}| j|d ��& | �t|�� t|� W d Q R X W d Q R X |�d
d �}| j|d ��& | �td �� t|� W d Q R X W d Q R X q.W qW d S )
N)rP r2 z{} fooz {} {{1:foo}}z if 1: {} foozif 1: {} {{1:foo}}zif 1:
{} foozif 1:
{} {{1:foo}}z call to '{}')�sourcer� z(foo.)zinvalid syntax)�formatZsubTest�assertRaisesRegexr r2 r# )r �keywordsZcases�keywordZ
custom_msgZcaser� r r r �(test_former_statements_refer_to_builtins� s$
z5GrammarTests.test_former_statements_refer_to_builtinsc C s, dddg}|\}}}|||f}~~~~~d S )Nr r r� r )r r� r
r. rV r� r r r �
test_del_stmt� s
zGrammarTests.test_del_stmtc C s d S )Nr )r r r r �test_pass_stmt� s zGrammarTests.test_pass_stmtc C s xP qW d S )Nr )r r r r �test_break_stmt� s z GrammarTests.test_break_stmtc C s� d}x|rd}qqW d}x(|s@d}y
wd}W q d}Y qX qW |dkrT| � |� d}x|srd}zwZW d d}X qZW |dkr�| � |� d S )Nr r r! �okz&continue failed to continue inside tryz'continue inside try called except blockzfinally block not called)r )r r� �msgr r r �test_continue_stmt� s,
z GrammarTests.test_continue_stmtc s d� fdd� }|� d S )Nr r c sh d}xD|rH|d7 }y$| r*|dkr*| d8 } P |d8 }wW q � Y qX qW |dksZ|dkrd� � d� d S )Nr r z1continue then break in try/except in loop broken!)r )Zextra_burning_oil�countZ big_hippo)r r r �
test_inner s z9GrammarTests.test_break_continue_loop.<locals>.test_inner)r r r )r r� r )r r �test_break_continue_loop s
z%GrammarTests.test_break_continue_loopc C s* dd� }dd� }|� |� }t | d� d S )Nc S s d S )Nr r r r r �g1" r� z$GrammarTests.test_return.<locals>.g1c S s dS )Nr r r r r r �g2# r� z$GrammarTests.test_return.<locals>.g2zclass foo:return 1)r )r r� r� r
r r r � test_return s
zGrammarTests.test_returnc C s: d}x |dk r"|d7 }zW d P X qW | � |d� d}x |dk rT|d7 }zw6W d P X q6W | � |d� d}x&|dk r�|d7 }z dd W d P X qhW | � |d� x"dD ]}| � |d� zW d P X q�W | � |d� x$dD ] }| � |d� zw�W d P X q�W | � |d� x,dD ]$}| � |d� z dd W d P X �qW | � |d� d S )Nr r r )r r )r )r r� r r r �test_break_in_finally( sN
z"GrammarTests.test_break_in_finallyc C sF dd� }| � |� d� dd� }| � |� d� dd� }| � |� d � d S )
Nc S s
zW d dS )Nr r r r r r r� ] s z/GrammarTests.test_return_in_finally.<locals>.g1r c S s
zdS dS )Nr r� r r r r r r� d s z/GrammarTests.test_return_in_finally.<locals>.g2r� c S s z dd W d dS )Nr r r� r r r r r �g3k s z/GrammarTests.test_return_in_finally.<locals>.g3r� )r )r r� r� r� r r r �test_return_in_finally\ s z#GrammarTests.test_return_in_finallyc C s� dd� }dd� }dd� }dd� }dd� }dd� }t | d� t | d � d
d� }d d� }t | d � t | d
� dd� }dd� }dd� }dd� }t | d� t | d� t | d� t | d� t | d� t | d� t | d� t | d� t | d� d S )Nc s s
dV d S )Nr r r r r r r� t r� z"GrammarTests.test_yield.<locals>.gc s s dE d H d S )Nr r r r r r r� u r� c s s
dV } d S )Nr r )r
r r r r� w r� c s s dE d H } d S )Nr r )r
r r r r� x r� c s s
dV d S )N)r r r r r r r r� z r� c s s
dV } d S )N)r r r )r
r r r r� { r� zdef g(): yield from (), 1z def g(): x = yield from (), 1c s s ddV f d S )Nr r r r r r r� � r� c s s ddE d H f d S )Nr r r r r r r r� � r� zdef g(): 1, yield 1zdef g(): 1, yield from ()c s s t dV � d S )Nr )rN r r r r r� � r� c s s t dV d� d S )Nr )rN r r r r r� � r� c s s t dE d H � d S )Nr )rN r r r r r� � r� c s s t dE d H d� d S )Nr r )rN r r r r r� � r� zdef g(): f(yield 1)zdef g(): f(yield 1, 1)zdef g(): f(yield from ())z def g(): f(yield from (), 1)�yieldz
yield fromzclass foo:yield 1zclass foo:yield from ()zdef g(a:(yield)): pass)r )r r� r r r �
test_yieldr s2
zGrammarTests.test_yieldc s� dd� }dd� }� fdd�}|dd� |dd� |d d� |d
d � |d d
� |dd
� |dd� |dd� |dd� |dd� d S )Nc s s dd� dV gD � d S )Nc S s g | ]}|�qS r r )r� r
r r r r� � s zHGrammarTests.test_yield_in_comprehensions.<locals>.g.<locals>.<listcomp>r r r r r r r� � r� z4GrammarTests.test_yield_in_comprehensions.<locals>.gc s s dd� dE d H gD � d S )Nc S s g | ]}|�qS r r )r� r
r r r r� � s zHGrammarTests.test_yield_in_comprehensions.<locals>.g.<locals>.<listcomp>r r r r r r r� � r� c
sv � � t|�� t| dd� W d Q R X dd l}|�� �8 |jdtd� � �t|�� t| dd� W d Q R X W d Q R X d S )Nz
<test string>r2 r �error)�category)ZassertWarnsRegex�DeprecationWarningr3 �warnings�catch_warnings�filterwarningsr� r )�codeZwarntextr� )r r r �check� s
z8GrammarTests.test_yield_in_comprehensions.<locals>.checkz def g(): [(yield x) for x in ()]z!'yield' inside list comprehensionz)def g(): [x for x in () if not (yield x)]z-def g(): [y for x in () for y in [(yield x)]]z def g(): {(yield x) for x in ()}z 'yield' inside set comprehensionz#def g(): {(yield x): x for x in ()}z!'yield' inside dict comprehensionz#def g(): {x: (yield x) for x in ()}z def g(): ((yield x) for x in ())z#'yield' inside generator expressionz%def g(): [(yield from x) for x in ()]z class C: [(yield x) for x in ()]z[(yield x) for x in ()]r )r r� r� r )r r � test_yield_in_comprehensions� s. z)GrammarTests.test_yield_in_comprehensionsc C sD y t d��W n t k
r Y nX yt�W n tk
r> Y nX d S )Nz just testing)� RuntimeError�KeyboardInterrupt)r r r r �
test_raise� s zGrammarTests.test_raisec C sd dd l }dd l}dd l }ddlm} ddlm} ddl m}m} ddl m}m} ddl m}m} d S )Nr )�time)�path�argv)r r� r� r� )r r r� r� r� r r r � test_import� s zGrammarTests.test_importc C s d S )Nr )r r r r � test_global� s zGrammarTests.test_globalc s d� d�� �fdd�}d S )Nr c s d S )Nr r )r
r. r r rN � s z%GrammarTests.test_nonlocal.<locals>.fr )r rN r )r
r. r �
test_nonlocal� s zGrammarTests.test_nonlocalc
C sh yW n, t k
r0 } z| �d� W d d }~X Y nX yW n, t k
rb } z| �d� W d d }~X Y nX d S )Nz6'assert True' should not have raised an AssertionErrorz;'assert True, msg' should not have raised an AssertionError)�AssertionErrorr )r r] r r r � test_assert� s zGrammarTests.test_assertFz Won't work if __debug__ is Falsec
C s� yW n4 t k
r8 } z| �|jd d� W d d }~X Y n X | �d� yW n4 t k
r| } z| �t|j�d� W d d }~X Y n X | �d� d S )Nr r� z%AssertionError not raised by assert 0z+AssertionError not raised by 'assert False')r� r r� r r, )r r] r r r � testAssert2� s $
$zGrammarTests.testAssert2c C s d S )Nr )r r r r �test_if s zGrammarTests.test_ifc C s d}d}| � |d� d S )Nr r )r )r r
r r r �
test_while s zGrammarTests.test_whilec C s� x dD ]}qW xdD ]
\}}}qW G dd� d�}d}x|d�D ] }|| }q>W |dkr`| � d� g }xd D ]\}|�|� qjW | �|d
d d g� d S )
N)r r r� r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z&GrammarTests.test_for.<locals>.Squaresc S s || _ g | _d S )N)�max�sofar)r r r r r r< $ s z/GrammarTests.test_for.<locals>.Squares.__init__c S s
t | j�S )N)r, r )r r r r �__len__' r� z.GrammarTests.test_for.<locals>.Squares.__len__c S sV d| kr| j k s n t�t| j�}x$||krJ| j�|| � |d }q(W | j| S )Nr r )r �
IndexErrorr, r �append)r r� �nr r r rA ( s
z2GrammarTests.test_for.<locals>.Squares.__getitem__N)r7 r8 r9 r< r rA r r r r �Squares# s r
r r� i zfor over growing sequence))r )r )r� r r r� )r r r )r r� r� r� r
r r
�resultr r r �test_for s
zGrammarTests.test_forc
C s� y dd W n t k
r Y nX y dd W n> tk
rB Y n, tk
rb } zW d d }~X Y n Y nX y dd W n ttt fk
r� Y nX y dd W n( ttt fk
r� } zW d d }~X Y nX zW d X d S )Nr r )rZ �EOFErrorr� )r r� r r r �test_try9 s. zGrammarTests.test_tryc C s d S )Nr )r r r r �
test_suiteO s zGrammarTests.test_suitec C s dsdrdsdsdrd S )Nr Fr )r r r r � test_test^ s
zGrammarTests.test_testc C s� ddk}ddkrddkrddk r ddkr(ddkr0ddkr8ddkr@ddk rHddkrPddkrXdd k r�d kr�d kr�d kr�d kr�d kr�d kr�d kr�d kr�dk r�n n d S )Nr r r )r r
r r r �test_comparisoni s h z GrammarTests.test_comparisonc C s d}d}d}d S )Nr r r )r r
r r r �test_binary_mask_opsz s z!GrammarTests.test_binary_mask_opsc C s d}d}d}d S )Nr r r r )r r
r r r �test_shift_ops s zGrammarTests.test_shift_opsc C s d}d}d}d}d S )Nr r r� r )r r
r r r �test_additive_ops� s z GrammarTests.test_additive_opsc C s d}d}d}d}d S )Nr g �?r g r )r r
r r r �test_multiplicative_ops� s z$GrammarTests.test_multiplicative_opsc C s d}d}d}d}d}d S )Nr r� ���g �?r )r r
r r r �test_unary_ops� s
zGrammarTests.test_unary_opsc C s� dd l }dd l}|jd }|�� }|jd �� }d}|d }|d }|dd� }|d d� }|dd � }|d d � }|dd � }|d d� }|dd� }i }d |d < d
|d < d |d
< d|d< t|�}|jdd� d� | �t|�d� d S )Nr r� Z01234r� rD ���������r r )r r� )r r r� )r r r� c S s t | �j| fS )N)r r7 )r
r r r r� � r� z-GrammarTests.test_selectors.<locals>.<lambda>)r� z [1, (1,), (1, 2), (1, 2, 3)])r r� r� �modulesrH �sortr r4 ) r r r� r� r
rK r r� �Lr r r �test_selectors� s,
zGrammarTests.test_selectorsc C s� d}dpdpd}dpdpdddf}g }dg}dp6dp6dg}dpDdpDdddg}g }i }ddi}ddi}dpjddppdi}ddd�}ddd�}dddddd d
�}dh}ddh}ddd h}dddh}|}d }d
}d S )Nr r r� rF r� )rF r� r� rD r� )rF r� �threeZfourZfiveZsixr r
rj r )r r
r r r �
test_atoms� s,
zGrammarTests.test_atomsc C s| G dd� d�}G dd� d�}G dd� d|�}G dd� d|�}G d d
� d
|||�}G d d � d �}d
d� }|G dd� d��}d S )Nc @ s e Zd ZdS )z%GrammarTests.test_classdef.<locals>.BN)r7 r8 r9 r r r r �B� s r! c @ s e Zd ZdS )z&GrammarTests.test_classdef.<locals>.B2N)r7 r8 r9 r r r r �B2� s r" c @ s e Zd ZdS )z&GrammarTests.test_classdef.<locals>.C1N)r7 r8 r9 r r r r �C1� s r# c @ s e Zd ZdS )z&GrammarTests.test_classdef.<locals>.C2N)r7 r8 r9 r r r r �C2� s r$ c @ s e Zd ZdS )z%GrammarTests.test_classdef.<locals>.DN)r7 r8 r9 r r r r �D� s r% c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z%GrammarTests.test_classdef.<locals>.Cc S s d S )Nr )r r r r �meth1� r� z+GrammarTests.test_classdef.<locals>.C.meth1c S s d S )Nr )r �argr r r �meth2� r� z+GrammarTests.test_classdef.<locals>.C.meth2c S s d S )Nr )r r� r� r r r �meth3� r� z+GrammarTests.test_classdef.<locals>.C.meth3N)r7 r8 r9 r& r( r) r r r r rW � s rW c S s | S )Nr )r
r r r �class_decorator� r� z3GrammarTests.test_classdef.<locals>.class_decoratorc @ s e Zd ZdS )z%GrammarTests.test_classdef.<locals>.GN)r7 r8 r9 r r r r �G� s r+ r ) r r! r" r# r$ r% rW r* r+ r r r �
test_classdef� s zGrammarTests.test_classdefc C s, dddg}| � dd� |D �dddd�� d S )Nr r r� c S s i | ]}|d |�qS )r r )r� r� r r r �
<dictcomp>� s z/GrammarTests.test_dictcomps.<locals>.<dictcomp>r� )r r r� )r )r �numsr r r �test_dictcomps� s
zGrammarTests.test_dictcompsc s� dddddg}dddg�d d
d g}�� d d
� |D �dddg� �� dd
� |D �dddddg� �� dd
� |D �dddg� �� �fdd
�|D �ddddddd d d d d d!d"d#d$g� �� �fd%d
�|D �ddddd d d d!d#d$g
� �� d&d
� td�D �dgddgdddgdddd'gddd(d)d*gg� d+d,� }�� ||�d-d-d-g� �fd.d/�}|� t�d0� t�d1� d2d3d4g}d5d6d7g� d8d9d:d;g�� �fd<d
�|D �}�� |d=d>d?d@g� d S )ANr r r� r� rD �Apple�Banana�Coconutz Applez Banana z Coco nut c S s g | ] }|� � �qS r )�strip)r� r r r r r� � s z/GrammarTests.test_listcomps.<locals>.<listcomp>z Coco nutc S s g | ] }d | �qS )r� r )r� r
r r r r� � s r� r r� � c S s g | ]}|d kr|�qS )r r )r� r
r r r r� � s c s g | ]}� D ] }||f�q qS r r )r� r� r )�strsr r r� � s )r r0 )r r1 )r r2 )r r0 )r r1 )r r2 )r� r0 )r� r1 )r� r2 )r� r0 )r� r1 )r� r2 )rD r0 )rD r1 )rD r2 c s( g | ] }d d� � D �D ] }||f�qqS )c S s g | ]}d |kr|�qS )r r )r� rN r r r r� s z:GrammarTests.test_listcomps.<locals>.<listcomp>.<listcomp>r )r� r� r )r5 r r r� s c S s g | ]}d d� |��qS )c s � fdd�t � d �D �S )Nc s g | ] }� | �qS r r )r� r� )rK r r r� s zLGrammarTests.test_listcomps.<locals>.<listcomp>.<lambda>.<locals>.<listcomp>r )�range)rK r )rK r r� r� z8GrammarTests.test_listcomps.<locals>.<listcomp>.<lambda>r )r� r� r r r r� s � � �@ � c S s dd� | D �S )Nc S s, g | ]$}|d krd| k o"dk n �qS )r r r� r )r� r
r r r r� s zEGrammarTests.test_listcomps.<locals>.test_in_func.<locals>.<listcomp>r )�lr r r � test_in_func s z1GrammarTests.test_listcomps.<locals>.test_in_funcFc s, � � dd� dD �ddgddgdd gg� d S )
Nc S s" g | ]}d d� ||d gD ��qS )c S s g | ]}|�qS r r )r� r. r r r r� s zUGrammarTests.test_listcomps.<locals>.test_nested_front.<locals>.<listcomp>.<listcomp>r r )r� r
r r r r� s zJGrammarTests.test_listcomps.<locals>.test_nested_front.<locals>.<listcomp>)r r� rD r r r� r� rD r� )r r )r r r �test_nested_front
s z6GrammarTests.test_listcomps.<locals>.test_nested_frontz"[i, s for i in nums for s in strs]z[x if y])r �Boeing)r �Ford)r� �
Macdonalds)r� �Airliner)r� �Engine)� � Cheeseburger)r r� )r r� )r r� )r� rC c sD g | ]<\}}� D ].\}}�D ] \}}||kr ||kr ||f�q qqS r r )r� ZsnoZsnameZpnoZpnameZsp_snoZsp_pno)�parts�suppartr r r� ' s )r> rA )r> rB )r? rB )r@ rD )r r6 r )r r. Zspcsr< r= Z suppliersr
r )rE r r5 rF r �test_listcomps� sF
.
zGrammarTests.test_listcompsc s� dd� t d�D �}| �t|�dd� t d�D �� yt|� | �d� W n tk
rZ Y nX d� y(� fdd�� D �}t|� | �d � W n tk
r� Y nX | �td
d� d D ��d d� d D �� | �td
d� dD ��dd� dD �� dd� t d�D �� dd� dd� � D �D �}| �t|�tdd� t d�D ��� | �tdd� t d�D ��tdd� t d�D ��� | �tdd� t d�D ��tdd� t d�D ��� | �tdd� dd� t d�D �D ��tdd� t d�D ��� | �tdd� d d� d d� t d�D �D �D ��td d� t d�D ��� | �td d� d d� d!d� t d�D �D �D ��td"d� t d�D ��� | �td#d� d$d� d%d� t d�D �D �D ��td&d� t d�D ��� | �td'd� d(d� d)d� t d�D �D �D ��d*� t| d+� t| d,� d S )-Nc s s | ]}d d� t d�D �V qdS )c S s g | ]}|�qS r r )r� r
r r r r� 3 s z7GrammarTests.test_genexps.<locals>.<genexpr>.<listcomp>r� N)r6 )r� r
r r r � <genexpr>3 s z,GrammarTests.test_genexps.<locals>.<genexpr>r c S s g | ]}|�qS r r )r� r
r r r r� 4 s z-GrammarTests.test_genexps.<locals>.<listcomp>r� z&should produce StopIteration exceptionc 3 s | ]
}� V qd S )Nr )r� r� )rK r r rH = s zshould produce TypeErrorc s s | ]}d D ]}||fV q
qdS )�abcdNr )r� r
r. r r r rH C s rI c S s g | ]}d D ] }||f�q qS )rI r )r� r
r. r r r r� C s c s s | ]}d D ]}||fV q
qdS )�xyNr )r� r
r. r r r rH D s Zabc S s g | ]}d D ] }||f�q qS )rJ r )r� r
r. r r r r� D s c S s g | ]}|�qS r r )r� r
r r r r� F s c s s | ]
}|V qd S )Nr )r� r
r r r rH G s c s s | ]
}|V qd S )Nr )r� r. r r r rH G s c S s g | ]}|�qS r r )r� r
r r r r� H s c s s | ]}|d V qdS )r Nr )r� r
r r r rH J s c S s g | ] }|d �qS )r r )r� r
r r r r� J s c s s | ]}|d r|| V qdS )r Nr )r� r
r r r rH K s c S s g | ]}|d r|| �qS )r r )r� r
r r r r� K s c s s | ]
}|V qd S )Nr )r� r
r r r rH L s c s s | ]
}|V qd S )Nr )r� r. r r r rH L s c S s g | ]}|�qS r r )r� r
r r r r� L s c s s | ]
}|V qd S )Nr )r� r
r r r rH M s c s s | ]
}|V qd S )Nr )r� r. r r r rH M s c s s | ]
}|V qd S )Nr )r� rV r r r rH M s c S s g | ]}|�qS r r )r� r
r r r r� M s c s s | ]
}|V qd S )Nr )r� r
r r r rH N s c S s g | ]}|�qS r r )r� r. r r r r� N s c s s | ]
}|V qd S )Nr )r� rV r r r rH N s c S s g | ]}|�qS r r )r� r
r r r r� N s c s s | ]
}|V qdS )TNr )r� r
r r r rH O s c s s | ]
}|V qd S )Nr )r� r. r r r rH O s c s s | ]
}|V qdS )TNr )r� rV r r r rH O s c S s g | ]}|�qS r r )r� r
r r r r� O s c s s | ]
}|V qdS )TNr )r� r
r r r rH P s c s s | ]}d r|V qdS )FNr )r� r. r r r rH P s c s s | ]
}|V qdS )TNr )r� rV r r r rH P s r z foo(x for x in range(10), 100)z foo(100, x for x in range(10))) r6 r �nextr �
StopIterationr� rH �sumr )r r� rL r )rK r � test_genexps1 s8 $$"00:DDD2
zGrammarTests.test_genexpsc s� d�dd� t ��D �}d�| �tt|��d� d�d� � �fdd�t ��D �}d�d� | �dd � t d�D �t|�� | �d
d � t d�D �d dd g� | �td
d� t d�D ��d dd g� | �dd � dD �dddg� | �tdd� dD ��d ddg� d S )Nr� c s s | ]
}|V qd S )Nr )r� r� r r r rH V s z;GrammarTests.test_comprehension_specials.<locals>.<genexpr>rD Fc 3 s( | ] }� rt ��D ]}||fV qqd S )N)r6 )r� r� r� )�tr
r r rH Z s Tc S s" g | ]}t d �D ] }||f�qqS )rD )r6 )r� r� r� r r r r� \ s z<GrammarTests.test_comprehension_specials.<locals>.<listcomp>c S s g | ]}|d r|d r|�qS )r r� r )r� r
r r r r� ` s r r� c s s" | ]}|d r|d r|V qdS )r r� Nr )r� r
r r r rH a s c S s g | ]
\}|�qS r r )r� r
r r r r� d s ))r� )rD )r� r� r� c s s | ] \}|V qd S )Nr )r� r
r r r rH e s ))r� )r� )r r� r )r6 r r, rH )r r� r )rO r
r �test_comprehension_specialsT s $ z(GrammarTests.test_comprehension_specialsc
C s� G dd� dt �}|� � W d Q R X |� �}W d Q R X |� �
\}}W d Q R X |� � |� � W d Q R X W d Q R X |� �}|� �}W d Q R X W d Q R X |� �}|� � W d Q R X W d Q R X d S )Nc @ s e Zd Zdd� Zdd� ZdS )z1GrammarTests.test_with_statement.<locals>.managerc S s dS )N)r r r )r r r r � __enter__i s z;GrammarTests.test_with_statement.<locals>.manager.__enter__c W s d S )Nr )r r� r r r �__exit__k s z:GrammarTests.test_with_statement.<locals>.manager.__exit__N)r7 r8 r9 rQ rR r r r r �managerh s rS )�object)r rS r
r. r r r �test_with_statementg s
z GrammarTests.test_with_statementc C s� dd� }| � dd� dd� dd� fD �dg� | � d d� d
d� d d� fD �dg� | � d
n|dd�d
� | � drx|dd�nd
d
� | � dr�d
o�dnd d � | � dr�d
o�dnd d � | � d
o�dnd d� | � dr�dp�|dd�ndd� | � d � p�|dd�n|dd�d � | � d�pd
n|dd�d
� | � dnd d� | � d�r>dnd d � | � dndd� | � d
ndd
� | � dndd� | � d ndd� | � d�r�ddk ndd� d S ) Nc S s t | � |S )z@helper to check that evaluation of expressions is done correctly)rP )r� �retr r r �
_checkeval} s z2GrammarTests.test_if_else_expr.<locals>._checkevalc S s g | ]}|� r|� �qS r r )r� r
r r r r� � s z2GrammarTests.test_if_else_expr.<locals>.<listcomp>c S s dS )NTr r r r r r� � r� z0GrammarTests.test_if_else_expr.<locals>.<lambda>c S s dS )NFr r r r r r� � r� Tc S s g | ]}|d �r|d ��qS )Fr )r� r
r r r r� � s c S s | rdS dS )NFTr )r
r r r r� � r� c S s | rdS dS )NTFr )r
r r r r� � r� r rD zcheck 1r zcheck 2r� zcheck 3r r� zcheck 4zcheck 5zcheck 6Fr� r� r� g @)r )r rW r r r �test_if_else_expr{ s$ $$ " z GrammarTests.test_if_else_exprc C sj | � dd� | � dd� | � dd� | �dddkk� | �ddkdk� | �dd ko^dkn � d S )Nr� r Fr� )r r Z assertFalse)r r r r �test_paren_evaluation� s z"GrammarTests.test_paren_evaluationc C s> G dd� d�}|� }| � || d� |d }| � |jd� d S )Nc @ s e Zd Zdd� Zdd� ZdS )z'GrammarTests.test_matrix_mul.<locals>.Mc S s dS )Nr� r )r rk r r r �
__matmul__� s z2GrammarTests.test_matrix_mul.<locals>.M.__matmul__c S s
|| _ | S )N)�other)r rk r r r � __imatmul__� s z3GrammarTests.test_matrix_mul.<locals>.M.__imatmul__N)r7 r8 r9 rZ r\ r r r r rp � s rp r� rE )r r[ )r rp �mr r r �test_matrix_mul� s
z GrammarTests.test_matrix_mulc C sx dd� }| � |jd� | �t|jjtj@ �� dd� }|dd� �}| �|j� | � |jd� | �t|jjtj@ �� d S ) Nc � s dd� } t � I d H d S )Nc S s d S )Nr r r r r rM � s z8GrammarTests.test_async_await.<locals>.test.<locals>.sum)Zsomeobj)rM r r r rm � s z+GrammarTests.test_async_await.<locals>.testrm c S s t | dd� | S )N�_markedT)�setattr)�funcr r r � decorator� s z0GrammarTests.test_async_await.<locals>.decoratorc � s dS )N� r r r r r �test2� s z,GrammarTests.test_async_await.<locals>.test2rd ) r r7 r �boolr� �co_flags�inspectZ CO_COROUTINEr_ )r rm rb rd r r r �test_async_await� s z GrammarTests.test_async_awaitc sR G dd� dt ��G dd� d�� � �fdd�}| ���� |� �d � W d Q R X d S )Nc @ s e Zd ZdS )z)GrammarTests.test_async_for.<locals>.DoneN)r7 r8 r9 r r r r �Done� s ri c @ s e Zd Zdd� Zdd� ZdS )z*GrammarTests.test_async_for.<locals>.AIterc S s | S )Nr )r r r r � __aiter__� s z4GrammarTests.test_async_for.<locals>.AIter.__aiter__c � s t �d S )N)�StopAsyncIteration)r r r r � __anext__� s z4GrammarTests.test_async_for.<locals>.AIter.__anext__N)r7 r8 r9 rj rl r r r r �AIter� s rm c � s� x,� � 2 y 3 d H } W q t k
s"X q Y W x0� � 2 y3 d H \} }W q6 t k
sTX q6 Y W x,� � 2 y 3 d H } W qh t k
s�X qh Y W ��d S )N)rk )r� r� )rm ri r r r� � s $ z(GrammarTests.test_async_for.<locals>.foo)r\ r �send)r r� r )rm ri r �test_async_for� s
zGrammarTests.test_async_forc sR G dd� dt �� G dd� d��� �fdd�}| �� �� |� �d � W d Q R X d S )Nc @ s e Zd ZdS )z*GrammarTests.test_async_with.<locals>.DoneN)r7 r8 r9 r r r r ri � s ri c @ s e Zd Zdd� Zdd� ZdS )z-GrammarTests.test_async_with.<locals>.managerc � s dS )N)r r r )r r r r �
__aenter__� s z8GrammarTests.test_async_with.<locals>.manager.__aenter__c � s dS )NFr )r �excr r r � __aexit__� s z7GrammarTests.test_async_with.<locals>.manager.__aexit__N)r7 r8 r9 rp rr r r r r rS � s rS c � s, �� 4 I d H � W d Q I d H R X �� 4 I d H �} W d Q I d H R X �� 4 I d H �
\} }W d Q I d H R X �� 4 I d H �& �� 4 I d H � W d Q I d H R X W d Q I d H R X �� 4 I d H �&} �� 4 I d H �}W d Q I d H R X W d Q I d H R X �� 4 I d H �&} �� 4 I d H � W d Q I d H R X W d Q I d H R X � �d S )Nr )r
r. )ri rS r r r� � s z)GrammarTests.test_async_with.<locals>.foo)r\ r rn )r r� r )ri rS r �test_async_with� s
z GrammarTests.test_async_withN)>r7 r8 r9 rC rI rJ r^ ri rq ru rw r| r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r �unittestZ
skipUnlessr r r r r r r r r r r r r r r r, r/ rG rN rP rU rX rY r^ rh ro rs r r r r rB � st . \' 4$#
#?#rB �__main__)Z test.supportr rg rt r Ztest.ann_modulern ro � collectionsr rm r r" r$ ZTestCaser rG rU r: rB r7 �mainr r r r �<module> s�