3
�7_�� c @ 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 � | j tj|� d S �N)r r �expr�r
r r r r �
check_expr � 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 r r )r r r r r r r � check_suite) r z(RoundtripLegalSyntaxTestCase.check_suitec C � | 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 �@ | 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 42r" r# r r r �test_await_statementC s
z1RoundtripLegalSyntaxTestCase.test_await_statementc C � | 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: passr" r# r r r �test_async_with_statementK �
z6RoundtripLegalSyntaxTestCase.test_async_with_statementc C r' )Nz'async def f():
async for i in (): passz*async def f():
async for i, b in (): passr" r# r r r �test_async_for_statementO r) z5RoundtripLegalSyntaxTestCase.test_async_for_statementc C r' )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]� 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 � | j d� d S )N�ar" r# r r r �test_simple_expression� � z3RoundtripLegalSyntaxTestCase.test_simple_expressionc C r' )Nza = bza = b = c = d = er" r# r r r �test_simple_assignments� r) 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 **= br" 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: passr" r# r r r �test_function_defs� sH
z/RoundtripLegalSyntaxTestCase.test_function_defsc C �6 | 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():passr" r# r r r �test_class_defs� s
z,RoundtripLegalSyntaxTestCase.test_class_defsc C r! )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 xr" r# r r r �test_import_from_statement� s.
z7RoundtripLegalSyntaxTestCase.test_import_from_statementc C r9 )Nz
import syszimport sys as systemzimport sys, mathzimport sys as system, mathzimport sys, math as my_mathr" r# r r r �test_basic_import_statement �
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 namer" r# r r r �test_relative_imports s
z2RoundtripLegalSyntaxTestCase.test_relative_importsc C r/ )Nz"# -*- coding: iso-8859-1 -*-
pass
r" r# r r r � test_pep263+ r z(RoundtripLegalSyntaxTestCase.test_pep263c C r/ )Nz assert alo < ahi and blo < bhi
r" r# r r r � test_assert/ r z(RoundtripLegalSyntaxTestCase.test_assertc C �" | 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 �
z&RoundtripLegalSyntaxTestCase.test_withc C r% )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 � )�
isinstanceZtuple�list)�treeZ node_typeZnextZeltr ��walkr r rJ G s z8RoundtripLegalSyntaxTestCase.test_position.<locals>.walkrE �defr r � � �(� r � � �)� �:� � � �return� �+�
�1� � T)� line_info�col_infoc S s g | ]\}}}}||f�qS r r ��.0r �nZl�cr r r �
<listcomp>g � z>RoundtripLegalSyntaxTestCase.test_position.<locals>.<listcomp>)r_ c S s g | ]\}}}}|||f�qS r r ra r r r re i rf )r` c S s g | ]\}}}}|||f�qS r r ra r r r re k rf c S � g | ] }t |��qS r �rG �rb r r r r re m rf c S rg r rh ri r r r re s rf )rE rK rE r )rE r rE rL )rM rN rE rO )rE r rE rP )rQ rR rE rM )rS rT rE rQ )rL rU rE rV �����)rO rU rW rj )rE rX rW rL )rE r rW rS )rY rZ rW r[ )rW r\ rW r] )rL rU rW r^ rj )rP rU rW rj rj )rL rU rW rj rj )r rU rW rj )r r r rG r ZtolistZst2tupleZst2list)r
r �stZexpectedr rI 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: passr" r# r r r �test_extended_unpackingu s
z4RoundtripLegalSyntaxTestCase.test_extended_unpackingc C rA )Nzraise
zraise e
zBtry:
suite
except Exception as e:
raise ValueError from e
r" r# r r r �test_raise_statement{ rC z1RoundtripLegalSyntaxTestCase.test_raise_statementc C r' )Nz[]z[*{2}, 3, *[4]]r- r# r r r �test_list_displays� r) z/RoundtripLegalSyntaxTestCase.test_list_displaysc C r9 )Nz{*{2}, 3, *[4]}z{2}z{2,}z{2, 3}z{2, 3,}r- r# r r r �test_set_displays� r= 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 r9 )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� r= z4RoundtripLegalSyntaxTestCase.test_argument_unpackingc C rA )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� rC z4RoundtripLegalSyntaxTestCase.test_set_comprehensionsc C rA )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� rC z5RoundtripLegalSyntaxTestCase.test_dict_comprehensionsN)#�__name__�
__module__� __qualname__r r r r r$ r& r( r* r+ r. r1 r3 r6 r7 r8 r: r; r< r> r? r@ rB rD rl rm rn ro rp rq rr rs rt 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
rH Zlabelr r r �check_bad_tree� s
z$IllegalSyntaxTestCase.check_bad_treec C s | j dd� d S )NrE rW � z<junk>)rE rW rz �ry r# r r r � test_junk� r2 z IllegalSyntaxTestCase.test_junkc C sD d}| j |d
� d"}| j |d � d+}| j |d� d4}| j |d� d S )5N� �
� � � rE rL rU 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�rE �r� r� �r� r� �rL rU �r r� r� �r~ r� r� �r rU )r} r� r� r� �rE r� �r� r� �r� r� r� �r r� r� �r~ r� r� r� )r} r� r� r� �rE 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� r� r� �r r� r� �r~ r� r� r� )r} r� r� r� r{ �r
rH r r r �test_illegal_terminal� sP z+IllegalSyntaxTestCase.test_illegal_terminalc $ C s dg}| j |d,� d S )hNr} � � � rE rK r � rM rN rQ rR rS rT �# rL rU rO � �
� � rX �9 �$ �% �& �' �) �* �+ �, �- �. �/ �0 �1 rW r\ � �yieldrP r � def f():
return 1
yield 1�rE rK �rE r �rM rN �rQ rR �r� r� r� �rS rT r� �rO rU �rE rX �rW 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� r� r� r� r� r� r� r� r� r� r� �r� r� r� �r� r� �r� r� r� �r� r� r� �r� r� �rP rU �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 �test_illegal_yield_1� sf z*IllegalSyntaxTestCase.test_illegal_yield_1c % C s dy}| j |d3� d S )zNr} r� r� r� � rE �from� �
__future__�import� �
generatorsrL rU r� r� rK r r� rM rN rQ rR rS rT r� rO r� r� rX r� r� r� r� r� r� r� r� r� r� r� r� r� r� rW r\ r� r� rP r r� �rE r� �rE r� �r� r� �rE r� �rE 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� 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 �test_illegal_yield_2! sx z*IllegalSyntaxTestCase.test_illegal_yield_2c C s d:}| j |d� d S );N� �7 �" r� r� r� r� �( r� r� r� r� r� r� r� rE r0 � �,rd rL rU r za,,c�rE r0 �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 rd �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 �test_a_comma_comma_ca s. z*IllegalSyntaxTestCase.test_a_comma_comma_cc C s dD}| j |d � d S )ENr} r� r� r� � �8 r� r� r� r� r� r� r� r� r� r� r� r� r� rE r0 � �% �$=�brL rU r za $= br� �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� 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= r� )r} r> r� r{ r� r r r �test_illegal_operator| s: z+IllegalSyntaxTestCase.test_illegal_operatorc C s d}| j |d � d S )Nr} r� r� r� � rE �foorL rU r zmalformed global ast�rE rA �r@ rB �r� rC r� �r� rD r� �r� rE r� r� )r} rF r� r� r{ r� r r r �test_malformed_global� s z+IllegalSyntaxTestCase.test_malformed_globalc C s d }| j |d� d S ) Nr} r r~ r r@ � rE r� r� � r� �fredrL rU r zfrom import fredr� r� �rE rJ �r� rK �rI rL �rH r� r� rM �r@ rN �r rO r� �r~ rP r� �r rQ r� r� )r} rR r� r� r{ r� 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} r rU zmissed encoding�
iso-8859-1znon-string encoding� ���r� �r} r� )rT rW r� rW )rT rW rU r� rW )rT rW rV )ry r4 ZUnicodeEncodeErrorr r r� r r r �test_illegal_encoding� s z+IllegalSyntaxTestCase.test_illegal_encodingN)
ru rv rw ry r| r� r� r� r r? rG rS rX r r r r rx � s .7@! rx 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 )N�2 + 3rO �r r � compilestr Zeval)r
rk r r r r �test_compile_expr� rC z!CompileTestCase.test_compile_exprc C s6 t jd�}t j|�}i }t||� | j|d d� d S )N�x = 2; y = x + 3ZyrO )r r r\ r r )r
rk 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 r4 r5 r\ �r
rk r r r �test_compile_error� r) z"CompileTestCase.test_compile_errorc C s8 t jd�}| jtt j|� t jd�}| jtt j|� d S )Nza = "\U12345678"z a = "\u1"r` ra 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-3rz �����rd r[ )r
rk Zcode1Zcode2r 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)rf rg )
r r r\ r Z co_filenamer Z bytearrayZ
memoryviewZ assertWarnsZDeprecationWarningr4 � TypeErrorrG )r
rk r Zfilenamer r r �test_compile_filename� s&
z%CompileTestCase.test_compile_filenameN) ru rv rw r] r_ rb rc re ri r r r r rY � s
rY c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�ParserStackLimitTestCasec C s d| d| S )N�[�]r )r
Zlevelr r r �_nested_expression r z+ParserStackLimitTestCase._nested_expressionc C s | j d�}tj|�}|j� d S )Ni] )rm r r r )r
�erk r r r �test_deeply_nested_list
s
z0ParserStackLimitTestCase.test_deeply_nested_listc C s6 | j d�}td|�\}}}| jd|� | jd|� d S )Nid z-cs s_push: parser stack overflows MemoryError)rm r ZassertIn)r
rn ZrcZoutZerrr r r �test_trigger_memory_error s
z2ParserStackLimitTestCase.test_trigger_memory_errorN)ru rv rw rm ro rp r r r r rj s rj 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 ) NrZ r^ r, TFrE g��hҘ@zspanish armadai s waterloo)r r r r ZminZmaxZsortedZ
assertTruer4 rh �operatorZgeZleZltZgt)
r
r
r Zst3Zst1_copyZst2_copyZst3_copyZbottomZtopZmidr 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 )NrZ r^ r, rE ) r r r �copyr r ZdeepcopyZrange�pickleZHIGHEST_PROTOCOLZloadsZdumps)r
Zstsrk Zst_copyZprotor 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 )NrE i� rz i����)Z
bit_length)rc 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 rE rj rj rj )ZlenrF r )ZnodeZresZhasstrZchildrenZchild)rw �nodesize�sizeofchildrenr r ry t s
z4STObjectTestCase.test_sizeof.<locals>.sizeofchildrenc s �j | � � �| j� � � d S r )� check_sizeofr )rk )�basesizerx r
ry r r �check_st_sizeof� s z5STObjectTestCase.test_sizeof.<locals>.check_st_sizeofrZ z 2 + 3 + 4z x = 2 + 3rU z# -*- coding: utf-8 -*-rk z2,i� rl )r Z calcobjsize�structZcalcsizer r r )r
r| r )rw r{ rx r
ry r � test_sizeofi s
z STObjectTestCase.test_sizeofN) ru rv rw rs rv r rz Z cpython_onlyr~ r r r r rq s =rq c @ s e Zd Zdd� ZdS )�OtherParserCasec
C s& | j t�� tjdd� W d Q R X d S )Nr0 r )r4 rh r r r# r r r �test_two_args_to_expr� s z%OtherParserCase.test_two_args_to_exprN)ru rv rw r� r r r r r � s r Z__main__)rt r ru Zunittestrr r} Ztestr Ztest.support.script_helperr ZTestCaser rx rY rj rq r ru Zmainr r r r Z<module> s( ! :x