3
�7_�� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z G dd� dej
�Z G dd� dej
�Z G dd � d ej
�Z
G d
d � d ej
�ZG d d
� d
ej
�ZG dd� dej
�Zedkr�ej� dS )� N)�support)�assert_python_failurec @ 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"dAS )B� RoundtripLegalSyntaxTestCasec C sl ||�}|j � }ytj|�}W n6 tjk
rT } z| jd||f � W Y d d }~X nX | j||j � d� d S )Nzcould not roundtrip %r: %sz!could not re-generate syntax tree)�totuple�parser� sequence2st� ParserError�fail� assertEqual)�self�f�s�st1�t�st2Zwhy� r �(/usr/lib64/python3.6/test/test_parser.py� roundtrip s $ z&RoundtripLegalSyntaxTestCase.roundtripc C s | j tj|� d S )N)r r �expr)r r
r r r �
check_expr s z'RoundtripLegalSyntaxTestCase.check_exprc C s6 t jd�}|j� }i }t|i |� | j|d t� d S )Nz/from __future__ import unicode_literals; x = ''�x)r �suite�compile�execZassertIsInstance�str)r r �codeZscoper r r �test_flags_passed s
z.RoundtripLegalSyntaxTestCase.test_flags_passedc C s | j tj|� d S )N)r r r )r r
r r r � check_suite) s z(RoundtripLegalSyntaxTestCase.check_suitec C s� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d � | j d
� | j d � | j d � | j d
� | j d� | j d� | j d� | j d� d S )Nzdef f(): yield 1zdef f(): yieldzdef f(): x += yieldzdef f(): x = yield 1zdef f(): x = y = yield 1zdef f(): x = yieldzdef f(): x = y = yieldzdef f(): 1 + (yield)*2zdef f(): (yield 1)*2zdef f(): return; yield 1zdef f(): yield 1; returnzdef f(): yield from 1zdef f(): x = yield from 1zdef f(): f((yield from 1))zdef f(): yield 1; return 1z1def f():
for x in range(30):
yield x
z)def f():
if (yield):
yield x
)r )r r r r �test_yield_statement, s"
z1RoundtripLegalSyntaxTestCase.test_yield_statementc C s@ | j d� | j d� | j d� | j d� | j d� | j d� d S )Nz async def f():
await smth()z"async def f():
foo = await smth()z'async def f():
foo, bar = await smth()z async def f():
(await smth())z#async def f():
foo((await smth()))z&async def f():
await foo(); return 42)r )r r r r �test_await_statementC s
z1RoundtripLegalSyntaxTestCase.test_await_statementc C s | j d� | j d� d S )Nz"async def f():
async with 1: passz/async def f():
async with a as b, c as d: pass)r )r r r r �test_async_with_statementK s
z6RoundtripLegalSyntaxTestCase.test_async_with_statementc C s | j d� | j d� d S )Nz'async def f():
async for i in (): passz*async def f():
async for i, b in (): pass)r )r r r r �test_async_for_statementO s
z5RoundtripLegalSyntaxTestCase.test_async_for_statementc C s | j d� | j d� d S )Nz3def f():
x = 0
def g():
nonlocal x
z:def f():
x = y = 0
def g():
nonlocal x, y
)r )r r r r �test_nonlocal_statementS s
z4RoundtripLegalSyntaxTestCase.test_nonlocal_statementc C s� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d � | j d
� | j d � | j d � | j d
� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d � | j d � | j d � | j d � | j d � | j d!� | j d"� | j d#� | j d$� | j d%� | j d&� | j d'� | j d(� d S ))Nzfoo(1)z [1, 2, 3]z[x**3 for x in range(20)]z"[x**3 for x in range(20) if x % 3]z+[x**3 for x in range(20) if x % 2 if x % 3]z list(x**3 for x in range(20))z&list(x**3 for x in range(20) if x % 3)z/list(x**3 for x in range(20) if x % 2 if x % 3)z
foo(*args)zfoo(*args, **kw)z foo(**kw)zfoo(key=value)zfoo(key=value, *args)zfoo(key=value, *args, **kw)zfoo(key=value, **kw)zfoo(a, b, c, *args)zfoo(a, b, c, *args, **kw)zfoo(a, b, c, **kw)zfoo(a, *args, keyword=23)z foo + barz foo - barz foo * barz foo / barz
foo // barz lambda: 0z lambda x: 0z lambda *y: 0zlambda *y, **z: 0z
lambda **z: 0zlambda x, y: 0zlambda foo=bar: 0z"lambda foo=bar, spaz=nifty+spit: 0zlambda foo=bar, **z: 0z#lambda foo=bar, blaz=blat+2, **z: 0z'lambda foo=bar, blaz=blat+2, *y, **z: 0zlambda x, *y, **z: 0z(x for x in range(10))zfoo(x for x in range(10))z...za[...])r )r r r r �test_expressions] sP
z-RoundtripLegalSyntaxTestCase.test_expressionsc C s | j d� d S )N�a)r )r r r r �test_simple_expression� s z3RoundtripLegalSyntaxTestCase.test_simple_expressionc C s | j d� | j d� d S )Nza = bza = b = c = d = e)r )r r r r �test_simple_assignments� s
z4RoundtripLegalSyntaxTestCase.test_simple_assignmentsc J C s� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d � | j d
� | jt�� td i i � W d Q R X | jt�� td i i � W d Q R X | jt�� td
i i � W d Q R X | jt�� tdi i � W d Q R X | jt�� tdi i � W d Q R X | jt�� tdi i � W d Q R X | jt�� tdi i � W d Q R X | jt�� tdi i � W d Q R X | jt�� tdi i � W d Q R X | jt�� tdi i � W d Q R X d S )Nz
x: int = 5z"y: List[T] = []; z: [list] = fun()zx: tuple = (1, 2)zd[f()]: int = 42zf(d[x]): str = 'abc'zx.y.z.w: complex = 42jzx: intz#def f():
x: str
y: int = 5
z#class C:
x: str
y: int = 5
zIclass C:
def __init__(self, x: int) -> None:
self.x: int = x
z2+2: intz []: int = 5zx, *y, z: int = range(5)zt: tuple = 1, 2z
u = v: intz
False: intz x.False: intz x.y,: intz[0]: intzf(): int)r � assertRaises� SyntaxErrorr )r r r r �test_var_annot� s<
z+RoundtripLegalSyntaxTestCase.test_var_annotc C s| | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d � | j d
� | j d � | j d � d S )
Nza += bza -= bza *= bza /= bza //= bza %= bza &= bza |= bza ^= bza <<= bza >>= bza **= b)r )r r r r �!test_simple_augmented_assignments� s
z>RoundtripLegalSyntaxTestCase.test_simple_augmented_assignmentsc C sl | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d � | j d
� | j d � | j d � | j d
� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d � | j d � | j d � | j d � | j d � | j d!� | j d"� | j d#� | j d$� d S )%Nz
def f(): passzdef f(*args): passzdef f(*args, **kw): passzdef f(**kw): passzdef f(foo=bar): passzdef f(foo=bar, *args): passz!def f(foo=bar, *args, **kw): passzdef f(foo=bar, **kw): passzdef f(a, b): passzdef f(a, b, *args): passz def f(a, b, *args, **kw): passzdef f(a, b, **kw): passzdef f(a, b, foo=bar): passz!def f(a, b, foo=bar, *args): passz'def f(a, b, foo=bar, *args, **kw): passz def f(a, b, foo=bar, **kw): passz@staticmethod
def f(): passz,@staticmethod
@funcattrs(x, y)
def f(): passz@funcattrs()
def f(): passzdef f(*, a): passzdef f(*, a = 5): passzdef f(*, a = 5, b): passzdef f(*, a, b = 5): passz def f(*, a, b = 5, **kwds): passzdef f(*args, a): passzdef f(*args, a = 5): passz def f(*args, a = 5, b): passz def f(*args, a, b = 5): passz$def f(*args, a, b = 5, **kwds): passzdef f(a: int): passzdef f(a: int = 5): passzdef f(*args: list): passzdef f(**kwds: dict): passzdef f(*, a: int): passzdef f(*, a: int = 5): passzdef f() -> int: pass)r )r r r r �test_function_defs� sH
z/RoundtripLegalSyntaxTestCase.test_function_defsc C s6 | j d� | j d� | j d� | j d� | j d� d S )Nzclass foo():passzclass foo(object):passz!@class_decorator
class foo():passz&@class_decorator(arg)
class foo():passz(@decorator1
@decorator2
class foo():pass)r )r r r r �test_class_defs� s
z,RoundtripLegalSyntaxTestCase.test_class_defsc C s� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d � | j d
� | j d � | j d � | j d
� | j d� | j d� | j d� | j d� d S )Nzfrom sys.path import *z from sys.path import dirnamez from sys.path import (dirname)z from sys.path import (dirname,)z*from sys.path import dirname as my_dirnamez,from sys.path import (dirname as my_dirname)z-from sys.path import (dirname as my_dirname,)z&from sys.path import dirname, basenamez(from sys.path import (dirname, basename)z)from sys.path import (dirname, basename,)z4from sys.path import dirname as my_dirname, basenamez6from sys.path import (dirname as my_dirname, basename)z7from sys.path import (dirname as my_dirname, basename,)z5from sys.path import dirname, basename as my_basenamez7from sys.path import (dirname, basename as my_basename)z8from sys.path import (dirname, basename as my_basename,)zfrom .bogus import x)r )r r r r �test_import_from_statement� s.
z7RoundtripLegalSyntaxTestCase.test_import_from_statementc C s6 | j d� | j d� | j d� | j d� | j d� d S )Nz
import syszimport sys as systemzimport sys, mathzimport sys as system, mathzimport sys, math as my_math)r )r r r r �test_basic_import_statement s
z8RoundtripLegalSyntaxTestCase.test_basic_import_statementc C sT | j d� | j d� | j d� | j d� | j d� | j d� | j d� | j d� d S ) Nzfrom . import namezfrom .. import namezfrom ... import namezfrom .... import namezfrom .pkg import namezfrom ..pkg import namezfrom ...pkg import namezfrom ....pkg import name)r )r r r r �test_relative_imports s
z2RoundtripLegalSyntaxTestCase.test_relative_importsc C s | j d� d S )Nz"# -*- coding: iso-8859-1 -*-
pass
)r )r r r r � test_pep263+ s z(RoundtripLegalSyntaxTestCase.test_pep263c C s | j d� d S )Nz assert alo < ahi and blo < bhi
)r )r r r r � test_assert/ s z(RoundtripLegalSyntaxTestCase.test_assertc C s" | j d� | j d� | j d� d S )Nzwith open('x'): pass
zwith open('x') as f: pass
z*with open('x') as f, open('y') as g: pass
)r )r r r r � test_with2 s
z&RoundtripLegalSyntaxTestCase.test_withc C s@ | j d� | j d� | j d� | j d� | j d� | j d� d S )Nztry: pass
except: pass
ztry: pass
finally: pass
z'try: pass
except A: pass
finally: pass
z4try: pass
except A: pass
except: pass
finally: pass
z"try: pass
except: pass
else: pass
z0try: pass
except: pass
else: pass
finally: pass
)r )r r r r �
test_try_stmt7 s
z*RoundtripLegalSyntaxTestCase.test_try_stmtc s@ d}t j|�}� fdd�� d&d'd(d)d*d+d,d.d/d0d1d2d3d5d7d9g}| jt� |jd d d ���|� | jt� |j� ��d d � |D �� | jt� |jd d ���d!d � |D �� | jt� |jd d"���d#d � |D �� | jt� |jd d d ���d$d � |D �� | jt� t j|d d d ���|� | jt� t j|d d d ���d%d � |D �� d S ):Nzdef f(x):
return x + 1c 3 sX | d }| d }t |ttf�rNx4| dd � D ] }x� |�D ]
}|V q:W q,W n| V d S )Nr � )�
isinstance�tuple�list)�treeZ node_type�nextZeltr )�walkr r r: G s z8RoundtripLegalSyntaxTestCase.test_position.<locals>.walkr4 �defr r � � �(� r � � �)� �:� � � �return� �+�
�1� � T)� line_info�col_infoc S s g | ]\}}}}||f�qS r r )�.0r �n�l�cr r r �
<listcomp>g s z>RoundtripLegalSyntaxTestCase.test_position.<locals>.<listcomp>)rO c S s g | ]\}}}}|||f�qS r r )rQ r rR rS rT r r r rU i s )rP c S s g | ]\}}}}|||f�qS r r )rQ r rR rS rT r r r rU k s c S s g | ] }t |��qS r )r7 )rQ r r r r rU m s c S s g | ] }t |��qS r )r7 )rQ r r r r rU s s )r4 r; r4 r )r4 r r4 r< )r= r> r4 r? )r4 r r4 r@ )rA rB r4 r= )rC rD r4 rA )r< rE r4 rF �����)r? rE rG rV )r4 rH rG r< )r4 r rG rC )rI rJ rG rK )rG rL rG rM )r< rE rG rN rV )r@ rE rG rV rV )r< rE rG rV rV )r rE rG rV )r r r
r7 r �tolistZst2tupleZst2list)r r �stZexpectedr )r: r �
test_positionA sF
z*RoundtripLegalSyntaxTestCase.test_positionc C s, | j d� | j d� | j d� | j d� d S )Nz*a = yz
x, *b, = mz [*a, *b] = yzfor [*x, b] in x: pass)r )r r r r �test_extended_unpackingu s
z4RoundtripLegalSyntaxTestCase.test_extended_unpackingc C s" | j d� | j d� | j d� d S )Nzraise
zraise e
zBtry:
suite
except Exception as e:
raise ValueError from e
)r )r r r r �test_raise_statement{ s
z1RoundtripLegalSyntaxTestCase.test_raise_statementc C s | j d� | j d� d S )Nz[]z[*{2}, 3, *[4]])r )r r r r �test_list_displays� s
z/RoundtripLegalSyntaxTestCase.test_list_displaysc C s6 | j d� | j d� | j d� | j d� | j d� d S )Nz{*{2}, 3, *[4]}z{2}z{2,}z{2, 3}z{2, 3,})r )r r r r �test_set_displays� s
z.RoundtripLegalSyntaxTestCase.test_set_displaysc C sJ | j d� | j d� | j d� | j d� | j d� | j d� | j d� d S )Nz{}z{a:b}z{a:b,}z
{a:b, c:d}z {a:b, c:d,}z{**{}}z{**{}, 3:4, **{5:6, 7:8}})r )r r r r �test_dict_displays� s
z/RoundtripLegalSyntaxTestCase.test_dict_displaysc C s6 | j d� | j d� | j d� | j d� | j d� d S )Nz
f(*a, **b)zf(a, *b, *c, *d)z f(**a, **b)zf(2, *a, *b, **b, **c, **d)z0f(*b, *() or () and (), **{} and {}, **() or {}))r )r r r r �test_argument_unpacking� s
z4RoundtripLegalSyntaxTestCase.test_argument_unpackingc C s" | j d� | j d� | j d� d S )Nz{x for x in seq}z{f(x) for x in seq}z#{f(x) for x in seq if condition(x)})r )r r r r �test_set_comprehensions� s
z4RoundtripLegalSyntaxTestCase.test_set_comprehensionsc C s" | j d� | j d� | j d� d S )Nz{x:x for x in seq}z({x**2:x[3] for x in seq if condition(x)}z4{x:x for x in seq1 for y in seq2 if condition(x, y)})r )r r r r �test_dict_comprehensions� s
z5RoundtripLegalSyntaxTestCase.test_dict_comprehensionsN)#�__name__�
__module__� __qualname__r r r r r r r r! r" r# r% r& r) r* r+ r, r- r. r/ r0 r1 r2 r3 rY rZ r[ r\ r] r^ r_ r` ra r r r r r s@
*'0
4 r c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�IllegalSyntaxTestCasec C s8 yt j|� W n t jk
r$ Y nX | jd| � d S )Nz"did not detect invalid tree for %r)r r r r )r r8 Zlabelr r r �check_bad_tree� s
z$IllegalSyntaxTestCase.check_bad_treec C s | j dd� d S )Nr4 rG � z<junk>)r4 rG rg )rf )r r r r � test_junk� s z IllegalSyntaxTestCase.test_junkc C sD d}| j |d
� d"}| j |d � d+}| j |d� d4}| j |d� d S )5N� �
� � � r4 r< rE r z too small items in terminal node� passz'non-string second item in terminal node�pass�0z'non-integer third item in terminal nodez too many items in terminal node�r4 �rm rq �rl rr �r< rE �rk rs rt �rj ru �r< rE �r rE )ri rv rw rx �r4 rn �rm ry �rl rz �r< rE �rk r{ r| �rj r} �r< rE �r rE )ri r~ r r� �r4 ro rp r �rm r� �rl r� �r< rE �rk r� r� �rj r� �r< rE �r rE )ri r� r� r� �r4 ro r r �rm r� �rl r� �r< rE �rk r� r� �rj r� �r< rE �r rE )ri r� r� r� )rf )r r8 r r r �test_illegal_terminal� sP z+IllegalSyntaxTestCase.test_illegal_terminalc $ C s dg}| j |d,� d S )hN� � � � r4 r; r � r= r> rA rB rC rD �# r< rE r? � �
� � rH �9 �$ �% �& �' �) �* �+ �, �- �. �/ �0 �1 rG rL � �yieldr@ r z def f():
return 1
yield 1�r4 r; �r4 r �r= r> �rA rB �r� r� r� �rC rD �r< rE �r? rE �r4 rH �rG rL �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� �r4 r� �rG rL �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< rE �r� r� r� �r� r� �r@ rE �r� r� r� r� �r� r� �r� r� r� r� �r� r� r� r� r� r� �r< rE �r rE �r� r� r� r� �r� r� )r� r� )rf )r r8 r r r �test_illegal_yield_1� sf z*IllegalSyntaxTestCase.test_illegal_yield_1c % C s dy}| j |d3� d S )zN� � � �
� r4 �from� �
__future__�import� �
generatorsr< rE � � r; r � r= r> rA rB rC rD �# r? � � rH �9 �$ �% �& �' �) �* �+ �, �- �. �/ �0 �1 rG rL � r� r@ r z def f():
return 1
yield 1�r4 r� �r4 r� �r� r �r4 r� �r4 r� �r� r �r� r r r
r �r� r
�r< rE �r� r r �r� r �r4 r; �r4 r �r= r> �rA rB �r� r r �rC rD �r< rE �r? rE �r4 rH �rG rL �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+ �r4 r� �rG rL �r r. �r r/ �r r0 �r r1 �r r2 �r r3 �r� r4 �r� r5 �r� r6 �r� r7 �r� r8 �r� r9 �r� r: �r� r; �r r- r< �r� r= �r� r> �r< rE �r� r? r@ �r� rA �r@ rE �r� r, rB rC �r� rD �r� r r rE �r� r r r r rF �r< rE �r rE �r� rG rH rI �r� rJ )r� r rK )rf )r r8 r r r �test_illegal_yield_2! sx z*IllegalSyntaxTestCase.test_illegal_yield_2c C s d:}| j |d� d S );N� �7 �" �# �$ �% �' �( �) �* �+ �, �- �. �/ r4 r$ � �,rT r< rE r za,,c�r4 r$ �r[ r^ �rZ r_ �rY r` �rX ra �rW rb �rV rc �rU rd �rT re �rS rf �rR rg �rQ rh �rP ri �rO rj �r\ r] �r\ r] �r4 rT �r[ rn �rZ ro �rY rp �rX rq �rW rr �rV rs �rU rt �rT ru �rS rv �rR rw �rQ rx �rP ry �rO rz �rN rk rl rm r{ �r< rE �r rE )rM r| r} r~ )rf )r r8 r r r �test_a_comma_comma_ca s. z*IllegalSyntaxTestCase.test_a_comma_comma_cc C s dD}| j |d � d S )EN� � � �
� �8 �# �$ �% �& �( �) �* �+ �, �- �. �/ �0 r4 r$ � �% �$=�br< rE r za $= b�r4 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� �r4 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< rE �r� r� r� �r� r� �r rE )r� r� r� )rf )r r8 r r r �test_illegal_operator| s: z+IllegalSyntaxTestCase.test_illegal_operatorc C s d}| j |d � d S )N� � � �
� r4 �foor< rE r zmalformed global ast�r4 r� �r� r� �r� r� �r< rE �r� r� r� �r� r� �r< rE �r rE )r� r� r� r� )rf )r r8 r r r �test_malformed_global� s z+IllegalSyntaxTestCase.test_malformed_globalc C s d }| j |d� d S ) N� � �
� � � r4 r� r� � � �fredr< rE r zfrom import fred�r4 r� �r4 r� �r4 r� �r� r� �r� r� �r� r� r� r� �r� r� �r� r� �r< rE �r� r� r� �r� r� �r< rE �r rE )r� r� r� r� )rf )r r8 r r r �test_missing_import_source� s z0IllegalSyntaxTestCase.test_missing_import_sourcec C sH d }| j |d� d}| j |d� d}| jt�� tj|� W d Q R X d S )N�S � r rE zmissed encoding�
iso-8859-1znon-string encoding� ����r rE �r� r� )r� r� �r rE �r� r� )r� r� r� �r rE �r� r� )r� r� r� )rf r' �UnicodeEncodeErrorr r )r r8 r r r �test_illegal_encoding� s z+IllegalSyntaxTestCase.test_illegal_encodingN)
rb rc rd rf rh r� r� rL r r� r� r� r� r r r r re � s .7@! re c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�CompileTestCasec C s( t jd�}t j|�}| jt|�d� d S )Nz2 + 3r? )r r � compilestr
�eval)r rX r r r r �test_compile_expr� s
z!CompileTestCase.test_compile_exprc C s6 t jd�}t j|�}i }t||� | j|d d� d S )Nzx = 2; y = x + 3�yr? )r r r� r r
)r rX r Zglobsr r r �test_compile_suite� s
z"CompileTestCase.test_compile_suitec C s t jd�}| jtt j|� d S )Nz 1 = 3 + 4)r r r' r( r� )r rX r r r �test_compile_error� s
z"CompileTestCase.test_compile_errorc C s8 t jd�}| jtt j|� t jd�}| jtt j|� d S )Nza = "\U12345678"z a = "\u1")r r r' r( r� )r rX r r r �test_compile_badunicode� s
z'CompileTestCase.test_compile_badunicodec C sB t jd�}t j|�}| jt|�d� t j|�}| jt|�d� d S )Nz-3rg �����r� )r r r� r
r� )r rX �code1�code2r r r �test_issue_9011� s
z CompileTestCase.test_issue_9011c C s t jd�}t j|�}| j|jd� |j� }| j|jd� x>dD ]6}t j||�}| j|jd� |j|�}| j|jd� q>W xvtd�td�fD ]b}| jt �� t j||�}W d Q R X | j|jd� | jt �� |j|�}W d Q R X | j|jd� q�W | j
t t j|t d�� | j
t |jt d�� d S )Nza + 5z
<syntax-tree>�file.py� file.py)r� r� )
r r r� r
� co_filenamer � bytearray�
memoryviewZ assertWarns�DeprecationWarningr' � TypeErrorr7 )r rX r �filenamer r r �test_compile_filename� s&
z%CompileTestCase.test_compile_filenameN) rb rc rd r� r� r� r� r� r r r r r r� � s
r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�ParserStackLimitTestCasec C s d| d| S )N�[�]r )r �levelr r r �_nested_expression s z+ParserStackLimitTestCase._nested_expressionc C s | j d�}tj|�}|j� d S )N�] )r r r r )r �erX r r r �test_deeply_nested_list
s
z0ParserStackLimitTestCase.test_deeply_nested_listc C s6 | j d�}td|�\}}}| jd|� | jd|� d S )N�d z-cs s_push: parser stack overflows MemoryError)r r ZassertIn)r r Zrc�out�errr r r �test_trigger_memory_error s
z2ParserStackLimitTestCase.test_trigger_memory_errorN)rb rc rd r r r r r r r r s r c @ s0 e Zd Zdd� Zdd� ZejZejdd� �ZdS )�STObjectTestCasec
C s t jd�}t jd�}t jd�}t jd�}t jd�}t jd�}| j||kd� | j||kd� | j||kd� | j||� | j||� | j||� | j||kd� | j||kd� | j||kd� | j||kd� | j||kd� | j||kd� | j||kd� | j||kd� | j||kd� | j||kd� | j||kd� | j||kd� | j||k ||k � | j||k ||k � | j||k ||k � | j||k ||k� | j||k ||k� | j||k ||k� | j||k||k� | j||k||k� | j||k||k� t|||�}t|||�}t|||g�d } | j|| k � | j||k � | j| |k � | j|| k� | j||k� | j| |k� | j||k� | j| | k� | j||k� | j|dkd� | jd|kd� | jt t
j |d � | jt t
j d|� | jt t
j
|d � | jt t
jd
|� d S ) Nz2 + 3zx = 2; y = x + 3z list(x**3 for x in range(20))TFr4 g��hҘ@zspanish armadai s waterloo)r r r r
�min�max�sortedZ
assertTruer' r �operator�ge�le�lt�gt)
r r r Zst3Zst1_copyZst2_copyZst3_copyZbottom�topZmidr r r �test_comparisons sf
z!STObjectTestCase.test_comparisonsc C s� t jd�t jd�t jd�g}x�|D ]~}tj|�}| j|j� |j� � tj|�}| j|j� |j� � x<ttj d �D ]*}tj
tj ||��}| j|j� |j� � qrW q"W d S )Nz2 + 3zx = 2; y = x + 3z list(x**3 for x in range(20))r4 ) r r r �copyr
r �deepcopy�range�pickleZHIGHEST_PROTOCOL�loads�dumps)r �stsrX Zst_copy�protor r r �test_copy_pickleX s
z!STObjectTestCase.test_copy_picklec s� dd� � t jd��tjd��� ��fdd������fdd�}|tjd �� |tjd
�� |tjd �� |tjd �� |tjd
�� |tjddd d �� d S )Nc S s0 | dkr | S | dkr | d d@ S d| d j � > S )Nr4 � rg �����)�
bit_length)rR r r r �
XXXROUNDUPk s
z0STObjectTestCase.test_sizeof.<locals>.XXXROUNDUPZPiiZhP3iP0hc s� | d kr dS d}t | �dko(t| d t�}|rB|t | d �d 7 }|rR| dd� n
| dd � }|r�|� t |��� 7 }x|D ]}|�|�7 }q|W |S )Nr r4 rV rV rV )�lenr5 r )Znode�res�hasstrZchildrenZchild)r( �nodesize�sizeofchildrenr r r- t s
z4STObjectTestCase.test_sizeof.<locals>.sizeofchildrenc s �j | � � �| j� � � d S )N)� check_sizeofr )rX )�basesizer, r r- r r �check_st_sizeof� s z5STObjectTestCase.test_sizeof.<locals>.check_st_sizeofz2 + 3z 2 + 3 + 4z x = 2 + 3rE z# -*- coding: utf-8 -*-r z2,i� r )r Z calcobjsize�structZcalcsizer r r )r r0 r )r( r/ r, r r- r � test_sizeofi s
z STObjectTestCase.test_sizeofN) rb rc rd r r$ r r. Z cpython_onlyr2 r r r r r s =r c @ s e Zd Zdd� ZdS )�OtherParserCasec
C s& | j t�� tjdd� W d Q R X d S )Nr$ r� )r' r r r )r r r r �test_two_args_to_expr� s z%OtherParserCase.test_two_args_to_exprN)rb rc rd r4 r r r r r3 � s r3 �__main__)r r r Zunittestr r1 Ztestr Ztest.support.script_helperr ZTestCaser re r� r r r3 rb �mainr r r r �<module> s( ! :x