File: //usr/local/lib/python3.7/test/test_tools/__pycache__/test_unparse.cpython-37.pyc
B
"��g�" � @ s d Z ddlZddlZddlZddlZddlZddlZddlZddl m
Z
m Z m Z e � ej
�e d�Zej�e�� ddlZW dQ R X dd� ZdZdZd Zd
Zd Zd Zd
ZdZdZ dZ dZ dZ G dd� dej �Z!G dd� de!�Z"G dd� de!�Z#e$dkr�e�%� dS )z>Tests for the unparse.py script in the Tools/parser directory.� N)�basepath�toolsdir�skip_if_missing�parserc C sL t | d��}t�|j�d }W dQ R X t | d|d��}|�� }W dQ R X |S )znRead and return the contents of a Python source file (as a
string), taking into account the file encoding.�rbr N�r)�encoding)�open�tokenize�detect_encoding�readline�read)�filenameZpyfiler �source� r �8/usr/local/lib/python3.7/test/test_tools/test_unparse.py� read_pyfile s
r zQdef f():
for x in range(10):
break
else:
y = 2
z = 3
zIdef g():
while True:
break
else:
y = 2
z = 3
zQfrom . import fred
from .. import barney
from .australia import shrimp as prawns
zzdef f():
x = 1
def g():
nonlocal x
x = 2
y = 7
def h():
nonlocal x, y
zOtry:
1 / 0
except ZeroDivisionError as e:
raise ArithmeticError from e
z @f1(arg)
@f2
class Foo: pass
z=if cond1:
suite1
elif cond2:
suite2
else:
suite3
z,if cond1:
suite1
elif cond2:
suite2
zctry:
suite1
except ex1:
suite2
except ex2:
suite3
else:
suite4
finally:
suite5
zwith f():
suite1
zwith f() as x:
suite1
z$with f() as x, g() as y:
suite1
c @ s e Zd Zdd� Zddd�ZdS )� ASTTestCasec C s | � t�|�t�|�� d S )N)Z assertEqual�ast�dump)�self�ast1�ast2r r r �assertASTEqual{ s zASTTestCase.assertASTEqual�internalc C sL t ||dtj�}t�� }t�||� |�� }t ||dtj�}| �||� d S )N�exec) �compiler Z
PyCF_ONLY_AST�io�StringIO�unparseZUnparser�getvaluer )r �code1r r Zunparse_buffer�code2r r r r �check_roundtrip~ s zASTTestCase.check_roundtripN)r )�__name__�
__module__� __qualname__r r# r r r r r z 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?S )@�UnparseTestCasec C s | � d� | � d� d S )Nz
f'{f"{0}"*3}'z
f'{f"{y}"*3}')r# )r r r r �
test_fstrings� s
z UnparseTestCase.test_fstringsc C s | � d� d S )Nz del x, y, z)r# )r r r r �test_del_statement� s z"UnparseTestCase.test_del_statementc C s | � d� | � d� d S )Nz45 << 2z13 >> 7)r# )r r r r � test_shifts� s
zUnparseTestCase.test_shiftsc C s | � t� d S )N)r# �for_else)r r r r �
test_for_else� s z UnparseTestCase.test_for_elsec C s | � t� d S )N)r# �
while_else)r r r r �test_while_else� s z UnparseTestCase.test_while_elsec C s6 | � d� | � d� | � d� | � d� | � d� d S )Nz(-1)**7z(-1.)**8z(-1j)**6znot True or FalsezTrue or not False)r# )r r r r �test_unary_parens� s
z!UnparseTestCase.test_unary_parensc C s | � d� d S )Nz 3 .__abs__())r# )r r r r �test_integer_parens� s z#UnparseTestCase.test_integer_parensc C s, | � d� | � d� | � d� | � d� d S )NZ1e1000z-1e1000Z1e1000jz-1e1000j)r# )r r r r �test_huge_float� s
z UnparseTestCase.test_huge_floatc C s | � td�� | � td�� d S )Ni �l���� )r# �str)r r r r � test_min_int� s z UnparseTestCase.test_min_intc C s, | � d� | � d� | � d� | � d� d S )NZ7jz-7jZ0jz-0j)r# )r r r r �test_imaginary_literals� s
z'UnparseTestCase.test_imaginary_literalsc C s | � d� d S )Nz(lambda: int)())r# )r r r r �test_lambda_parentheses� s z'UnparseTestCase.test_lambda_parenthesesc C s | � d� | � d� d S )Nz
1 < 4 <= 5za is b is c is not d)r# )r r r r �test_chained_comparisons� s
z(UnparseTestCase.test_chained_comparisonsc C s| | � d� | � d� | � d� | � d� | � d� | � d� | � d� | � d� | � d � | � d
� | � d � | � d � d S )
Nz
def f(): passzdef f(a): passzdef f(b = 2): passzdef f(a, b): passzdef f(a, b = 2): passzdef f(a = 5, b = 2): passz def f(*, a = 1, b = 2): passzdef f(*, a = 1, b): passzdef f(*, a, b = 2): passz&def f(a, b = None, *, c, **kwds): passz'def f(a=2, *args, c=5, d, **kwds): passz def f(*args, **kwargs): pass)r# )r r r r �test_function_arguments� s
z'UnparseTestCase.test_function_argumentsc C s | � t� d S )N)r# �relative_import)r r r r �test_relative_import� s z$UnparseTestCase.test_relative_importc C s | � t� d S )N)r# � nonlocal_ex)r r r r �
test_nonlocal� s z UnparseTestCase.test_nonlocalc C s | � t� d S )N)r# �
raise_from)r r r r �test_raise_from� s z UnparseTestCase.test_raise_fromc C s | � d� d S )Nzb'123')r# )r r r r �
test_bytes� s zUnparseTestCase.test_bytesc C s6 | � d� | � d� | � d� | � d� | � d� d S )Nzdef f(a : int): passzdef f(a: int = 5): passzdef f(*args: [int]): passzdef f(**kwargs: dict): passzdef f() -> None: pass)r# )r r r r �test_annotations� s
z UnparseTestCase.test_annotationsc C s | � d� d S )Nz{'a', 'b', 'c'})r# )r r r r �test_set_literal� s z UnparseTestCase.test_set_literalc C s | � d� d S )Nz{x for x in range(5)})r# )r r r r �test_set_comprehension� s z&UnparseTestCase.test_set_comprehensionc C s | � d� d S )Nz{x: x*x for x in range(10)})r# )r r r r �test_dict_comprehension� s z'UnparseTestCase.test_dict_comprehensionc C s | � t� d S )N)r# �class_decorator)r r r r �test_class_decorators� s z%UnparseTestCase.test_class_decoratorsc C s | � d� d S )Nz(class A(metaclass=type, *[], **{}): pass)r# )r r r r �test_class_definition� s z%UnparseTestCase.test_class_definitionc C s | � t� | � t� d S )N)r# �elif1�elif2)r r r r �
test_elifs� s
zUnparseTestCase.test_elifsc C s | � t� d S )N)r# �try_except_finally)r r r r �test_try_except_finally� s z'UnparseTestCase.test_try_except_finallyc C s, | � d� | � d� | � d� | � d� d S )Nza, *b, c = seqza, (*b, c) = seqza, *b[0], c = seqza, *(b, c) = seq)r# )r r r r �test_starred_assignment� s
z'UnparseTestCase.test_starred_assignmentc C s | � t� d S )N)r# � with_simple)r r r r �test_with_simple� s z UnparseTestCase.test_with_simplec C s | � t� d S )N)r# �with_as)r r r r � test_with_as� s z UnparseTestCase.test_with_asc C s | � t� d S )N)r# �with_two_items)r r r r �test_with_two_items� s z#UnparseTestCase.test_with_two_itemsc C s | � d� | � d� d S )Nz{**{'y': 2}, 'x': 1}z{**{'y': 2}, **{'x': 1}})r# )r r r r �test_dict_unpacking_in_dict s
z+UnparseTestCase.test_dict_unpacking_in_dictc C s| | � d� | � d� | � d� | � d� | � d� | � d� | � d� | � d� | � d � | � d
� | � d � | � d � d S )
Nza[i]za[i,]za[i, j]za[()]za[i:j]za[:j]za[i:]za[i:j:k]za[:j:k]za[i::k]za[i:j,]z a[i:j, k])r# )r r r r �test_subscript s
z UnparseTestCase.test_subscriptN)"r$ r% r& r( r) r* r, r. r/ r0 r1 r3 r4 r5 r6 r7 r9 r; r= r> r? r@ rA rB rD rE rH rJ rK rM rO rQ rR rS r r r r r' � s> r' c @ s: e Zd ZdZdZdej�dd�fZe dd� �Z
dd� Z dS ) �DirectoryTestCasez:Test roundtrip behaviour on all files in Lib and Lib/test.NZLib�testc C s� | j d k r| j S g }xX| jD ]N}tj�t|�}x:t�|�D ],}|�d�r:|�d�s:|� tj�||�� q:W q W t
j � d�s�t
�|d�}|| _ |S )Nz.pyZbadZcpu�
)�NAMES�test_directories�os�path�joinr �listdir�endswith�
startswith�appendrU �supportZis_resource_enabled�randomZsample)�cls�names�dZtest_dir�nr r r � get_names s
zDirectoryTestCase.get_namesc
C s� | � � }xt|D ]l}tjjr&td| � tj�|�dkrPtjjrtd|� d�� q| j|d�� t |�}| �
|� W d Q R X qW d S )Nz
Testing %s)ztest_fstring.pyz Skipping z: see issue 28002)r ) rf rU r` �verbose�printrY rZ �basenameZsubTestr r# )r rc r r r r r �
test_files1 s
z DirectoryTestCase.test_files) r$ r% r& �__doc__rW rY rZ r[ rX � classmethodrf rj r r r r rT s
rT �__main__)&rk ZunittestZ test.supportrU r rY ra r
r Ztest.test_toolsr r r rZ r[ Z parser_pathr` Z
DirsOnSysPathr r r+ r- r8 r: r< rC rF rG rI rL rN rP ZTestCaser r' rT r$ �mainr r r r �<module> s>
0