Your IP : 216.73.216.213
o
0�i� � @ s� d Z ddlZddlmZ ddlmZ ddlmZmZ ddl m
Z
G dd� de
j �Z G d d
� d
e �Z
G d d � d e �ZG d
d� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd � d e �ZG d d � d e �ZG d d � d e �ZG d!d"� d"e �ZG d#d$� d$e �ZG d%d&� d&e �ZG d'd(� d(e �Z G d)d*� d*e �Z G d+d,� d,e �Z G d-d.� d.e �Z G d/d0� d0e �Z G d1d2� d2e �Z!G d3d4� d4�Z"G d5d6� d6e e"�Z#G d7d8� d8e e"�Z$G d9d:� d:e e"�Z%G d;d<� d<e �Z&G d=d>� d>e �Z'G d?d@� d@e �Z(G dAdB� dBe �Z)G dCdD� dDe �Z*G dEdF� dFe �Z+G dGdH� dHe �Z,G dIdJ� dJe �Z-G dKdL� dLe �Z.G dMdN� dNe �Z/G dOdP� dPe �Z0G dQdR� dRe �Z1G dSdT� dTe �Z2G dUdV� dVe �Z3G dWdX� dXe �Z4G dYdZ� dZe �Z5G d[d\� d\e �Z6G d]d^� d^e �Z7G d_d`� d`e �Z8G dadb� dbe �Z9G dcdd� dde �Z:G dedf� dfe �Z;G dgdh� dhe �Z<G didj� dje �Z=G dkdl� dle �Z>G dmdn� dne �Z?G dodp� dpe �Z@G dqdr� dre �ZAG dsdt� dte �ZBdS )uz" Test suite for the fixer modules � N)�chain)�
itemgetter)�pygram�
fixer_util)�supportc @ sL e Zd Zddd�Zdd� Zddd �Zdd
d �Zd d
� Zddd�Zdd� Z dS )�
FixerTestCaseN�lib2to3c C sP |d u r| j g}t�|||�| _g | _d| _t| jj| jj�D ]}| j|_ q d S )Nz<string>)
�fixerr �get_refactorer�refactor� fixer_log�filenamer Z pre_orderZ
post_order�log)�selfZfix_listZ fixer_pkgZoptionsr � r �@/opt/alt/python310/lib64/python3.10/lib2to3/tests/test_fixers.py�setUp s �
�zFixerTestCase.setUpc C s8 t �|�}t �|�}| j�|| j�}| �|t|�� |S �N)r Zreformatr Zrefactor_stringr
� assertEqual�str)r �before�after�treer r r �_check s
zFixerTestCase._checkFc C s2 | � ||�}| �|j� |s| �| jg � d S d S r )r �
assertTrue� was_changedr r )r r r �ignore_warningsr r r r �check# s
�zFixerTestCase.checkc C s8 | � ||�}| �|d�| j�� |s| �|j� d S d S )N� )r ZassertIn�joinr r r )r r r �message� unchangedr r r r �warns) s
�zFixerTestCase.warnsc C s | j |||dd� d S )NT�r! �r" )r r r r r r �warns_unchanged/ s z FixerTestCase.warns_unchangedc C s& | � ||� |s| �| jg � d S d S r )r r r )r r r r r r r! 2 s �zFixerTestCase.unchangedc G s� | j g}|�|� t�d|�}|�� \}}d| j }|r'|d jj�|�r'd S |r6|d jj�|�r6|s6d S | �dd� dd� || D ��|f � d S )Nr Zfix_���z5Fixer run order (%s) is incorrect; %s should be last.�, c S s g | ]}|j j�qS r )� __class__�
__module__)�.0�xr r r �
<listcomp>D � z3FixerTestCase.assert_runs_after.<locals>.<listcomp>)
r �extendr r
Z
get_fixersr( r) �endswithZfailr )r �names�fixes�rZpreZpost�nr r r �assert_runs_after7 s
�z FixerTestCase.assert_runs_after)Nr N)F)
�__name__r) � __qualname__r r r r" r% r! r4 r r r r r
s
r c @ �( e Zd ZdZdd� Zdd� Zdd� ZdS ) �Test_ne�nec C � d}d}| � ||� d S )Nzif x <> y:
passzif x != y:
pass�r �r �b�ar r r �
test_basicI � zTest_ne.test_basicc C r: )Nzif x<>y:
passzif x!=y:
passr; r< r r r �test_no_spacesQ r@ zTest_ne.test_no_spacesc C r: )Nz if x<>y<>z:
passz if x!=y!=z:
passr; r< r r r � test_chainedY r@ zTest_ne.test_chainedN)r5 r) r6 r r? rA rB r r r r r8 F s
r8 c @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )� Test_has_keyZhas_keyc C r: )Nz$x = d.has_key("x") or d.has_key("y")zx = "x" in d or "y" in dr; r< r r r �test_1d � zTest_has_key.test_1c C r: )Nz x = a.b.c.d.has_key("x") ** 3zx = ("x" in a.b.c.d) ** 3r; r< r r r �test_2i rE zTest_has_key.test_2c C r: )Nz!x = a.b.has_key(1 + 2).__repr__()z x = (1 + 2 in a.b).__repr__()r; r< r r r �test_3n rE zTest_has_key.test_3c C r: )Nz,x = a.b.has_key(1 + 2).__repr__() ** -3 ** 4z(x = (1 + 2 in a.b).__repr__() ** -3 ** 4r; r< r r r �test_4s rE zTest_has_key.test_4c C r: )Nzx = a.has_key(f or g)zx = (f or g) in ar; r< r r r �test_5x rE zTest_has_key.test_5c C r: )Nzx = a + b.has_key(c)zx = a + (c in b)r; r< r r r �test_6} rE zTest_has_key.test_6c C r: )Nzx = a.has_key(lambda: 12)zx = (lambda: 12) in ar; r< r r r �test_7� rE zTest_has_key.test_7c C r: )Nzx = a.has_key(a for a in b)zx = (a for a in b) in ar; r< r r r �test_8� rE zTest_has_key.test_8c C r: )Nzif not a.has_key(b): passzif b not in a: passr; r< r r r �test_9� rE zTest_has_key.test_9c C r: )Nz$if not a.has_key(b).__repr__(): passz if not (b in a).__repr__(): passr; r< r r r �test_10� rE zTest_has_key.test_10c C r: )Nz if not a.has_key(b) ** 2: passzif not (b in a) ** 2: passr; r< r r r �test_11� rE zTest_has_key.test_11N)r5 r) r6 r rD rF rG rH rI rJ rK rL rM rN rO r r r r rC a s rC c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � 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.d/� Zd0d1� Zd2d3� Z d4d5� Z d6S )7�
Test_applyZapplyc C r: )Nzx = apply(f, g + h)z
x = f(*g + h)r; r< r r r rD � rE zTest_apply.test_1c C r: )Nzy = apply(f, g, h)zy = f(*g, **h)r; r< r r r rF � rE zTest_apply.test_2c C r: )Nz z = apply(fs[0], g or h, h or g)z z = fs[0](*g or h, **h or g)r; r< r r r rG � rE zTest_apply.test_3c C r: )Nzapply(f, (x, y) + t)zf(*(x, y) + t)r; r< r r r rH � rE zTest_apply.test_4c C r: )Nzapply(f, args,)�f(*args)r; r< r r r rI � rE zTest_apply.test_5c C r: )Nzapply(f, args, kwds,)�f(*args, **kwds)r; r< r r r rJ � rE zTest_apply.test_6c C r: )Nzx = apply(f+g, args)zx = (f+g)(*args)r; r< r r r �test_complex_1� rE zTest_apply.test_complex_1c C r: )Nzx = apply(f*g, args)zx = (f*g)(*args)r; r< r r r �test_complex_2� rE zTest_apply.test_complex_2c C r: )Nzx = apply(f**g, args)zx = (f**g)(*args)r; r< r r r �test_complex_3� rE zTest_apply.test_complex_3c C r: )Nzx = apply(f.g, args)zx = f.g(*args)r; r< r r r �test_dotted_name� rE zTest_apply.test_dotted_namec C r: )Nzx = apply(f[x], args)zx = f[x](*args)r; r< r r r �test_subscript� rE zTest_apply.test_subscriptc C r: )Nzx = apply(f(), args)zx = f()(*args)r; r< r r r � test_call� rE zTest_apply.test_callc C r: )Nz"x = apply(a.b.c.d.e.f, args, kwds)z x = a.b.c.d.e.f(*args, **kwds)r; r< r r r � test_extreme� rE zTest_apply.test_extremec C r: )Nz1apply( # foo
f, # bar
args)rQ r; r< r r r �test_weird_comments� s z Test_apply.test_weird_commentsc C � d}| � |� d S )Nzapply()r# �r �sr r r �test_unchanged_1� � zTest_apply.test_unchanged_1c C r[ )Nzapply(f)r# r\ r r r �test_unchanged_2� r_ zTest_apply.test_unchanged_2c C r[ )Nz apply(f,)r# r\ r r r �test_unchanged_3� r_ zTest_apply.test_unchanged_3c C r[ )Nz apply(f, args, kwds, extras)r# r\ r r r �test_unchanged_4� r_ zTest_apply.test_unchanged_4c C r[ )Nzapply(f, *args, **kwds)r# r\ r r r �test_unchanged_5� r_ zTest_apply.test_unchanged_5c C r[ )Nzapply(f, *args)r# r\ r r r �test_unchanged_6 r_ zTest_apply.test_unchanged_6c C r[ )Nzapply(f, **kwds)r# r\ r r r �test_unchanged_6b r_ z Test_apply.test_unchanged_6bc C r[ )Nz#apply(func=f, args=args, kwds=kwds)r# r\ r r r �test_unchanged_7
r_ zTest_apply.test_unchanged_7c C r[ )Nz apply(f, args=args, kwds=kwds)r# r\ r r r �test_unchanged_8 r_ zTest_apply.test_unchanged_8c C r[ )Nzapply(f, args, kwds=kwds)r# r\ r r r �test_unchanged_9 r_ zTest_apply.test_unchanged_9c C r: )Nzapply( f, args, kwds)rR r; �r r> r= r r r � test_space_1 rE zTest_apply.test_space_1c C r: )Nzapply( f ,args,kwds )rR r; ri r r r � test_space_2 rE zTest_apply.test_space_2N) r5 r) r6 r rD rF rG rH rI rJ rS rT rU rV rW rX rY rZ r^ r` ra rb rc rd re rf rg rh rj rk r r r r rP � s8 rP c @ �0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) � Test_reload�reloadc C r: )Nz reload(a)z$import importlib
importlib.reload(a)r; r< r r r �test# rE zTest_reload.testc C �, d}d}| � ||� d}d}| � ||� d S )Nzreload( a ) # commentz0import importlib
importlib.reload( a ) # commentzreload( a ) # commentz1import importlib
importlib.reload( a ) # commentr; r< r r r � test_comment( s zTest_reload.test_commentc C �@ d}d}| � ||� d}d}| � ||� d}d}| � ||� d S )Nz reload( a )z&import importlib
importlib.reload( a )z
reload( a)z%import importlib
importlib.reload( a)z
reload(a )z%import importlib
importlib.reload(a )r; r< r r r �
test_space2 � zTest_reload.test_spacec C s� d}| � |� d}| � |� d}| � |� d}| � |� d}| � |� d}| � |� d}| � |� d}| � |� d }| � |� d S )
Nz reload(a=1)z reload(f, g)z
reload(f, *h)zreload(f, *h, **i)zreload(f, **i)zreload(*h, **i)z
reload(*h)z reload(**i)zreload()r# r\ r r r �test_unchanged? s$
zTest_reload.test_unchangedN)r5 r) r6 r ro rq rs ru r r r r rm s
rm c @ r7 ) � Test_intern�internc C rr )Nzx = intern( a )z"import sys
x = sys.intern( a )z%y = intern("b" # test
)z4import sys
y = sys.intern("b" # test
)zz = intern(a+b+c.d, )z&import sys
z = sys.intern(a+b+c.d, )r; r< r r r �test_prefix_preservation^ s z$Test_intern.test_prefix_preservationc C rr )Nz
x = intern(a)z import sys
x = sys.intern(a)zz = intern(a+b+c.d,)z#import sys
z = sys.intern(a+b+c.d,)z"intern("y%s" % 5).replace("y", "")z1import sys
sys.intern("y%s" % 5).replace("y", "")r; r< r r r ro m rt zTest_intern.testc C �J d}| � |� d}| � |� d}| � |� d}| � |� d}| � |� d S )Nz intern(a=1)z intern(f, g)z
intern(*h)z intern(**i)zintern()r# r\ r r r ru | �
zTest_intern.test_unchangedN)r5 r) r6 r rx ro ru r r r r rv [ s
rv c @ rl ) � Test_reduce�reducec C r: )Nzreduce(a, b, c)z,from functools import reduce
reduce(a, b, c)r; r< r r r �test_simple_call� rE z Test_reduce.test_simple_callc C r: )Nzdef x(arg): reduce(sum, [])z8from functools import reduce
def x(arg): reduce(sum, [])r; r< r r r �
test_bug_7253� � zTest_reduce.test_bug_7253c C r: )Nz reduce(lambda x, y: x + y, seq)z<from functools import reduce
reduce(lambda x, y: x + y, seq)r; r< r r r �test_call_with_lambda� rE z!Test_reduce.test_call_with_lambdac C ry )Nz reduce(a)zreduce(a, b=42)zreduce(a, b, c, d)z reduce(**c)zreduce()r# r\ r r r ru � rz zTest_reduce.test_unchangedN)r5 r) r6 r r} r~ r� ru r r r r r{ � s r{ c @ � e Zd ZdZdd� Zdd� Zdd� Zdd � 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$S )%�
Test_print�printc C r: )Nzprint 1, 1+1, 1+1+1zprint(1, 1+1, 1+1+1)r; r< r r r rx � rE z#Test_print.test_prefix_preservationc C � d}| � |� d}| � |� d S )N�print()� print('')r# r\ r r r �test_idempotency� s
zTest_print.test_idempotencyc C s: t j| jj_d}| �|� d}| �|� d}| �|� d S )N�print(1, 1+1, 1+1+1)r� r� )r Z!python_grammar_no_print_statementr ZdriverZgrammarr! r\ r r r �"test_idempotency_print_as_function� s
z-Test_print.test_idempotency_print_as_functionc C r: )Nzprint 1, 1+1, 1+1+1r� r; r< r r r rD � rE zTest_print.test_1c C r: )Nz
print 1, 2z print(1, 2)r; r< r r r rF � rE zTest_print.test_2c C r: )Nr� r� r; r< r r r rG � rE zTest_print.test_3c C r: )Nzprint whatever; printzprint(whatever); print()r; r< r r r rH � r zTest_print.test_4c C r: )Nzprint; print whatever;zprint(); print(whatever);r; r< r r r rI � rE zTest_print.test_5c C r: )Nzprint (a, b, c)zprint((a, b, c))r; r< r r r �
test_tuple� rE zTest_print.test_tuplec C r: )Nzprint 1, 2, 3,zprint(1, 2, 3, end=' ')r; r< r r r �test_trailing_comma_1� rE z Test_print.test_trailing_comma_1c C r: )Nz print 1, 2,zprint(1, 2, end=' ')r; r< r r r �test_trailing_comma_2� rE z Test_print.test_trailing_comma_2c C r: )Nzprint 1,zprint(1, end=' ')r; r< r r r �test_trailing_comma_3� rE z Test_print.test_trailing_comma_3c C r: )Nzprint >>sys.stderr, 1, 2, 3z print(1, 2, 3, file=sys.stderr)r; r< r r r �!test_vargs_without_trailing_comma� rE z,Test_print.test_vargs_without_trailing_commac C r: )Nzprint >>sys.stderr, 1, 2,z%print(1, 2, end=' ', file=sys.stderr)r; r< r r r �test_with_trailing_comma rE z#Test_print.test_with_trailing_commac C r: )Nzprint >>sys.stderr, 1+1zprint(1+1, file=sys.stderr)r; r< r r r �test_no_trailing_comma rE z!Test_print.test_no_trailing_commac C r: )Nzprint >> sys.stderrzprint(file=sys.stderr)r; r< r r r �test_spaces_before_file
rE z"Test_print.test_spaces_before_filec C s& d}| � |� d}d}| �||� d S )Nz<from __future__ import print_function
print('Hai!', end=' ')zprint 'Hello, world!'zprint('Hello, world!')�r! r )r r] r= r> r r r � test_with_future_print_function s
z*Test_print.test_with_future_print_functionN)r5 r) r6 r rx r� r� rD rF rG rH rI r� r� r� r� r� r� r� r� r� r r r r r� � s& r� c @ �` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )� Test_exec�execc C r: )Nz exec code in ns1, ns2z exec(code, ns1, ns2)r; r< r r r rx rE z"Test_exec.test_prefix_preservationc C r: )Nz exec code�
exec(code)r; r< r r r r? ! rE zTest_exec.test_basicc C r: )Nzexec code in ns�exec(code, ns)r; r< r r r �test_with_globals& rE zTest_exec.test_with_globalsc C r: )Nzexec code in ns1, ns2�exec(code, ns1, ns2)r; r< r r r �test_with_globals_locals+ rE z"Test_exec.test_with_globals_localsc C r: )Nzexec (a.b()) in nszexec((a.b()), ns)r; r< r r r rS 0 rE zTest_exec.test_complex_1c C r: )Nzexec a.b() + c in nszexec(a.b() + c, ns)r; r< r r r rT 5 rE zTest_exec.test_complex_2c C r[ )Nr� r# r\ r r r r^ < r_ zTest_exec.test_unchanged_1c C r[ )Nz exec (code)r# r\ r r r r` @ r_ zTest_exec.test_unchanged_2c C r[ )Nr� r# r\ r r r ra D r_ zTest_exec.test_unchanged_3c C r[ )Nr� r# r\ r r r rb H r_ zTest_exec.test_unchanged_4N)r5 r) r6 r rx r? r� r� rS rT r^ r` ra rb r r r r r� s r� c @ �H e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )� Test_repr�reprc C r: )Nz
x = `1 + 2`zx = repr(1 + 2)r; r< r r r rx O rE z"Test_repr.test_prefix_preservationc C r: )Nz x = `1 + 2`zx = repr(1 + 2)r; r< r r r �
test_simple_1T rE zTest_repr.test_simple_1c C r: )Nzy = `x`z y = repr(x)r; r< r r r �
test_simple_2Y rE zTest_repr.test_simple_2c C r: )Nzz = `y`.__repr__()zz = repr(y).__repr__()r; r< r r r � test_complex^ rE zTest_repr.test_complexc C r: )Nz
x = `1, 2, 3`zx = repr((1, 2, 3))r; r< r r r r� c rE zTest_repr.test_tuplec C r: )Nz
x = `1 + `2``zx = repr(1 + repr(2))r; r< r r r � test_nestedh rE zTest_repr.test_nestedc C r: )Nzx = `1, 2 + `3, 4``z x = repr((1, 2 + repr((3, 4))))r; r< r r r �test_nested_tuplesm rE z Test_repr.test_nested_tuplesN) r5 r) r6 r rx r� r� r� r� r� r� r r r r r� L s r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"S )#� Test_except�exceptc C r: )Nzq
try:
pass
except (RuntimeError, ImportError), e:
passzs
try:
pass
except (RuntimeError, ImportError) as e:
passr; r< r r r rx u � z$Test_except.test_prefix_preservationc C r: )NzV
try:
pass
except Foo, e:
pass�X
try:
pass
except Foo as e:
passr; r< r r r � test_simple� r� zTest_except.test_simplec C r: )NzU
try:
pass
except Foo,e:
passr� r; r< r r r �"test_simple_no_space_before_target� r� z.Test_except.test_simple_no_space_before_targetc C r: )Nz�
def foo():
try:
pass
except Exception, (f, e):
pass
except ImportError, e:
passa
def foo():
try:
pass
except Exception as xxx_todo_changeme:
(f, e) = xxx_todo_changeme.args
pass
except ImportError as e:
passr; r< r r r �test_tuple_unpack� s z Test_except.test_tuple_unpackc C r: )Nzn
try:
pass
except (RuntimeError, ImportError), e:
passzp
try:
pass
except (RuntimeError, ImportError) as e:
passr; r< r r r �test_multi_class� s z Test_except.test_multi_classc C r: )Nza
try:
pass
except Exception, [a, b]:
passz�
try:
pass
except Exception as xxx_todo_changeme:
[a, b] = xxx_todo_changeme.args
passr; r< r r r �test_list_unpack� � z Test_except.test_list_unpackc C r: )Nz_
try:
pass
except Exception, d[5]:
passz�
try:
pass
except Exception as xxx_todo_changeme:
d[5] = xxx_todo_changeme
passr; r< r r r �test_weird_target_1� r� z Test_except.test_weird_target_1c C r: )Nz`
try:
pass
except Exception, a.foo:
passz�
try:
pass
except Exception as xxx_todo_changeme:
a.foo = xxx_todo_changeme
passr; r< r r r �test_weird_target_2� r� z Test_except.test_weird_target_2c C r: )Nzb
try:
pass
except Exception, a().foo:
passz�
try:
pass
except Exception as xxx_todo_changeme:
a().foo = xxx_todo_changeme
passr; r< r r r �test_weird_target_3� r� z Test_except.test_weird_target_3c C r: )Nz�
try:
pass
except Exception, a:
pass
except:
passz�
try:
pass
except Exception as a:
pass
except:
passr; r< r r r �test_bare_except� s z Test_except.test_bare_exceptc C r: )Nz�
try:
pass
except Exception, a:
pass
except:
pass
else:
pass
finally:
passz�
try:
pass
except Exception as a:
pass
except:
pass
else:
pass
finally:
passr; r< r r r �!test_bare_except_and_else_finally
s z-Test_except.test_bare_except_and_else_finallyc C r: )Nz�
try:
pass
except TypeError, b:
pass
except Exception, a:
pass
except:
passz�
try:
pass
except TypeError as b:
pass
except Exception as a:
pass
except:
passr; r< r r r �+test_multi_fixed_excepts_before_bare_except' s
z7Test_except.test_multi_fixed_excepts_before_bare_exceptc C �T d}d}| � ||� d}d}| � ||� d}d}| � ||� d}d}| � ||� d S ) Nzd
try: raise TypeError
except TypeError, e:
pass
zf
try: raise TypeError
except TypeError as e:
pass
zd
try:
raise TypeError
except TypeError, e: pass
zf
try:
raise TypeError
except TypeError as e: pass
zT
try: raise TypeError
except TypeError, e: pass
zV
try: raise TypeError
except TypeError as e: pass
z�
try: raise TypeError
except TypeError, e: pass
else: function()
finally: done()
z�
try: raise TypeError
except TypeError as e: pass
else: function()
finally: done()
r; r< r r r �test_one_line_suites= s z Test_except.test_one_line_suitesc C r[ )NzO
try:
pass
except:
passr# r\ r r r r^ m � z Test_except.test_unchanged_1c C r[ )NzY
try:
pass
except Exception:
passr# r\ r r r r` u r� z Test_except.test_unchanged_2c C r[ )Nzg
try:
pass
except (Exception, SystemExit):
passr# r\ r r r ra } r� z Test_except.test_unchanged_3N)r5 r) r6 r rx r� r� r� r� r� r� r� r� r� r� r� r� r^ r` ra r r r r r� r s$
0 r� c @ r� )%�
Test_raise�raisec C r: )Nzraise Exception, 5�raise Exception(5)r; r< r r r r? � rE zTest_raise.test_basicc C rp )Nzraise Exception,5r� zraise Exception, 5zraise Exception(5)r; r< r r r rx � � z#Test_raise.test_prefix_preservationc C rr )Nzraise Exception, 5 # foozraise Exception(5) # fooz raise E, (5, 6) % (a, b) # fooz raise E((5, 6) % (a, b)) # fooz:def foo():
raise Exception, 5, 6 # foozIdef foo():
raise Exception(5).with_traceback(6) # foor; r< r r r �test_with_comments� s z Test_raise.test_with_commentsc C r: )Nz raise Exception(5), None, tbz%raise Exception(5).with_traceback(tb)r; r< r r r �test_None_value� rE zTest_raise.test_None_valuec C r: )Nzraise Exception, (5, 6, 7)zraise Exception(5, 6, 7)r; r< r r r �test_tuple_value� rE zTest_raise.test_tuple_valuec C r: )Nzraise E, (5, 6) % (a, b)zraise E((5, 6) % (a, b))r; r< r r r �test_tuple_detection� rE z Test_raise.test_tuple_detectionc C r: )Nz raise (((E1, E2), E3), E4), V� raise E1(V)r; r< r r r �test_tuple_exc_1� rE zTest_raise.test_tuple_exc_1c C r: )Nzraise (E1, (E2, E3), E4), Vr� r; r< r r r �test_tuple_exc_2� rE zTest_raise.test_tuple_exc_2c C � d}| � |d� d S )Nz raise 'foo'�+Python 3 does not support string exceptions�r% r\ r r r �test_string_exc� � zTest_raise.test_string_excc C r� )Nzraise "foo", 5r� r� r\ r r r �test_string_exc_val� r� z Test_raise.test_string_exc_valc C r� )Nzraise "foo", 5, 6r� r� r\ r r r �test_string_exc_val_tb� r� z!Test_raise.test_string_exc_val_tbc C r: )Nz4def foo():
raise Exception, 5, 6�Cdef foo():
raise Exception(5).with_traceback(6)r; r< r r r � test_tb_1� � zTest_raise.test_tb_1c C r: )Nzhdef foo():
a = 5
raise Exception, 5, 6
b = 6�wdef foo():
a = 5
raise Exception(5).with_traceback(6)
b = 6r; r< r r r � test_tb_2� � zTest_raise.test_tb_2c C r: )Nz2def foo():
raise Exception,5,6r� r; r< r r r � test_tb_3� r� zTest_raise.test_tb_3c C r: )Nzfdef foo():
a = 5
raise Exception,5,6
b = 6r� r; r< r r r � test_tb_4� r� zTest_raise.test_tb_4c C r: )Nz<def foo():
raise Exception, (5, 6, 7), 6zIdef foo():
raise Exception(5, 6, 7).with_traceback(6)r; r< r r r � test_tb_5� r� zTest_raise.test_tb_5c C r: )Nzpdef foo():
a = 5
raise Exception, (5, 6, 7), 6
b = 6z}def foo():
a = 5
raise Exception(5, 6, 7).with_traceback(6)
b = 6r; r< r r r � test_tb_6� r� zTest_raise.test_tb_6N)r5 r) r6 r r? rx r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s& r� c @ � e Zd ZdZdd� Zdd� Zdd� Zdd � 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&S )'�
Test_throw�throwc C r: )Nzg.throw(Exception, 5)�g.throw(Exception(5))r; r< r r r rD rE zTest_throw.test_1c C r: )Nzg.throw(Exception,5)r� r; r< r r r rF rE zTest_throw.test_2c C r: )Nz g.throw(Exception, (5, 6, 7))zg.throw(Exception(5, 6, 7))r; r< r r r rG rE zTest_throw.test_3c C r: )Nz5 + g.throw(Exception, 5)z5 + g.throw(Exception(5))r; r< r r r rH rE zTest_throw.test_4c C r� )Nzg.throw("foo")r� r� r\ r r r � test_warn_1 r� zTest_throw.test_warn_1c C r� )Nzg.throw("foo", 5)r� r� r\ r r r � test_warn_2! r� zTest_throw.test_warn_2c C r� )Nzg.throw("foo", 5, 6)r� r� r\ r r r � test_warn_3% r� zTest_throw.test_warn_3c C r[ )Nzg.throw(Exception)r# r\ r r r �test_untouched_1+ r_ zTest_throw.test_untouched_1c C r[ )Nzg.throw(Exception(5, 6))r# r\ r r r �test_untouched_2/ r_ zTest_throw.test_untouched_2c C r[ )Nz 5 + g.throw(Exception(5, 6))r# r\ r r r �test_untouched_33 r_ zTest_throw.test_untouched_3c C r: )Nz7def foo():
g.throw(Exception, 5, 6)�Fdef foo():
g.throw(Exception(5).with_traceback(6))r; r< r r r r� 9 r� zTest_throw.test_tb_1c C r: )Nzkdef foo():
a = 5
g.throw(Exception, 5, 6)
b = 6�zdef foo():
a = 5
g.throw(Exception(5).with_traceback(6))
b = 6r; r< r r r r� @ r� zTest_throw.test_tb_2c C r: )Nz5def foo():
g.throw(Exception,5,6)r� r; r< r r r r� K r� zTest_throw.test_tb_3c C r: )Nzidef foo():
a = 5
g.throw(Exception,5,6)
b = 6r� r; r< r r r r� R r� zTest_throw.test_tb_4c C r: )Nz?def foo():
g.throw(Exception, (5, 6, 7), 6)zLdef foo():
g.throw(Exception(5, 6, 7).with_traceback(6))r; r< r r r r� ] r� zTest_throw.test_tb_5c C r: )Nzsdef foo():
a = 5
g.throw(Exception, (5, 6, 7), 6)
b = 6z�def foo():
a = 5
g.throw(Exception(5, 6, 7).with_traceback(6))
b = 6r; r< r r r r� d r� zTest_throw.test_tb_6c C r: )Nz;def foo():
a + g.throw(Exception, 5, 6)zJdef foo():
a + g.throw(Exception(5).with_traceback(6))r; r< r r r � test_tb_7o r� zTest_throw.test_tb_7c C r: )Nzodef foo():
a = 5
a + g.throw(Exception, 5, 6)
b = 6z~def foo():
a = 5
a + g.throw(Exception(5).with_traceback(6))
b = 6r; r< r r r � test_tb_8v r� zTest_throw.test_tb_8N)r5 r) r6 r rD rF rG rH r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� s( r� c @ �8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
� Test_longZlongc C r: )Nz x = long(x)z
x = int(x)r; r< r r r rD � rE zTest_long.test_1c C r: )Nzy = isinstance(x, long)zy = isinstance(x, int)r; r< r r r rF � rE zTest_long.test_2c C r: )Nzz = type(x) in (int, long)zz = type(x) in (int, int)r; r< r r r rG � rE zTest_long.test_3c C sf d}| � |� d}| � |� d}| � |� d}| � |� d}| � |� d}| � |� d}| � |� d S )Nz long = Truez
s.long = Truezdef long(): passzclass long(): passzdef f(long): passzdef f(g, long): passzdef f(x, long=True): passr# r\ r r r ru � s
zTest_long.test_unchangedc C r: )Nzx = long( x )zx = int( x )r; r< r r r rx � rE z"Test_long.test_prefix_preservationN) r5 r) r6 r rD rF rG ru rx r r r r r� � s r� c @ � e Zd ZdZdd� Zdd� ZdS )�
Test_execfileZexecfilec C s| d}d}| � ||� d}d}| � ||� d}d}| � ||� d}d}| � ||� d }d
}| � ||� d }d }| � ||� d S )
Nzexecfile("fn")z4exec(compile(open("fn", "rb").read(), "fn", 'exec'))zexecfile("fn", glob)z:exec(compile(open("fn", "rb").read(), "fn", 'exec'), glob)zexecfile("fn", glob, loc)z?exec(compile(open("fn", "rb").read(), "fn", 'exec'), glob, loc)z execfile("fn", globals=glob)zBexec(compile(open("fn", "rb").read(), "fn", 'exec'), globals=glob)zexecfile("fn", locals=loc)z@exec(compile(open("fn", "rb").read(), "fn", 'exec'), locals=loc)z(execfile("fn", globals=glob, locals=loc)zNexec(compile(open("fn", "rb").read(), "fn", 'exec'), globals=glob, locals=loc)r; r< r r r �test_conversion� �$ z Test_execfile.test_conversionc C rp )Nzexecfile( "fn" )z6exec(compile(open( "fn", "rb" ).read(), "fn", 'exec'))z execfile("fn", globals = glob)zEexec(compile(open("fn", "rb").read(), "fn", 'exec'), globals = glob)r; r< r r r � test_spacing� r� zTest_execfile.test_spacingN)r5 r) r6 r r� r� r r r r r� � s r� c @ r7 ) �Test_isinstance�
isinstancec C r� ) Nz isinstance(x, (int, int, int))zisinstance(x, int)z,isinstance(x, (int, float, int, int, float))zisinstance(x, (int, float))z1isinstance(x, (int, float, int, int, float, str))z isinstance(x, (int, float, str))z4isinstance(foo() + bar(), (x(), y(), x(), int, int))z/isinstance(foo() + bar(), (x(), y(), x(), int))r; r< r r r �test_remove_multiple_items� � z*Test_isinstance.test_remove_multiple_itemsc C r: )Nz4if isinstance( foo(), ( bar, bar, baz )) : passz/if isinstance( foo(), ( bar, baz )) : passr; r< r r r rx � rE z(Test_isinstance.test_prefix_preservationc C � | � d� d S )Nzisinstance(x, (str, int))r# �r r r r ru � � z Test_isinstance.test_unchangedN)r5 r) r6 r r� rx ru r r r r r� � s
r� c @ s( e Zd ZdZdd� Zdd� Zdd� Zdd � 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.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� Z"d@dA� Z#dBdC� Z$dDdE� Z%dFdG� Z&dHS )I� Test_dict�dictc C s| d}d}| � ||� d}d}| � ||� d}d}| � ||� d}d}| � ||� d }d
}| � ||� d }d}| � ||� d S ) Nzif d. keys ( ) : passz if list(d. keys ( )) : passzif d. items ( ) : passz!if list(d. items ( )) : passz if d. iterkeys ( ) : passz if iter(d. keys ( )) : passz"[i for i in d. iterkeys( ) ]z [i for i in d. keys( ) ]z if d. viewkeys ( ) : passzif d. keys ( ) : passz"[i for i in d. viewkeys( ) ]r; r< r r r rx � r� z"Test_dict.test_prefix_preservationc C s| d}d}| � ||� d}d}| � ||� d}d}| � ||� d}d}| � ||� d}d}| � ||� d }d}| � ||� d S )
Nzd.keys() # foozlist(d.keys()) # foozd.items() # foozlist(d.items()) # foozd.iterkeys() # fooziter(d.keys()) # fooz/[i for i in d.iterkeys() # foo
]z+[i for i in d.keys() # foo
]zd.viewitems() # foor; r< r r r �test_trailing_comment
s$ z Test_dict.test_trailing_commentc C sF t jD ] }d| }| �|� d| }| �|� d| }| �|� qd S )Nzs = %s(d.keys())zs = %s(d.values())zs = %s(d.items())�r Zconsuming_callsr! )r �wrapperr] r r r ru * �
�zTest_dict.test_unchangedc C rp )N�d.keys()�list(d.keys())za[0].foo().keys()zlist(a[0].foo().keys())r; r< r r r �test_015 r� zTest_dict.test_01c C r: )N� d.items()zlist(d.items())r; r< r r r �test_02> rE zTest_dict.test_02c C r: )N�
d.values()zlist(d.values())r; r< r r r �test_03C rE zTest_dict.test_03c C r: )Nz d.iterkeys()�iter(d.keys())r; r< r r r �test_04H rE zTest_dict.test_04c C r: )Nz
d.iteritems()ziter(d.items())r; r< r r r �test_05M rE zTest_dict.test_05c C r: )Nzd.itervalues()ziter(d.values())r; r< r r r �test_06R rE zTest_dict.test_06c C r[ )Nr� r# r\ r r r �test_07W r_ zTest_dict.test_07c C r[ )N�sorted(d.keys())r# r\ r r r �test_08[ r_ zTest_dict.test_08c C r: )Nr� ziter(list(d.keys()))r; r< r r r �test_09_ rE zTest_dict.test_09c C r: )Nz
foo(d.keys())zfoo(list(d.keys()))r; r< r r r rN d rE zTest_dict.test_10c C r: )N�for i in d.keys(): print iz for i in list(d.keys()): print ir; r< r r r rO i rE zTest_dict.test_11c C r: )Nz for i in d.iterkeys(): print ir r; r< r r r �test_12n rE zTest_dict.test_12c C r: )N�[i for i in d.keys()]z[i for i in list(d.keys())]r; r< r r r �test_13s rE zTest_dict.test_13c C r: )Nz[i for i in d.iterkeys()]r r; r< r r r �test_14x rE zTest_dict.test_14c C r: )N�(i for i in d.keys())z(i for i in list(d.keys()))r; r< r r r �test_15} rE zTest_dict.test_15c C r: )Nz(i for i in d.iterkeys())r r; r< r r r �test_16� rE zTest_dict.test_16c C r: )Nziter(d.iterkeys())r� r; r< r r r �test_17� rE zTest_dict.test_17c C r: )Nzlist(d.iterkeys())r� r; r< r r r �test_18� rE zTest_dict.test_18c C r: )Nzsorted(d.iterkeys())r r; r< r r r �test_19� rE zTest_dict.test_19c C r: )Nzfoo(d.iterkeys())zfoo(iter(d.keys()))r; r< r r r �test_20� rE zTest_dict.test_20c C r: )Nzprint h.iterkeys().next()zprint iter(h.keys()).next()r; r< r r r �test_21� rE zTest_dict.test_21c C r: )Nzprint h.keys()[0]zprint list(h.keys())[0]r; r< r r r �test_22� rE zTest_dict.test_22c C r: )Nz print list(h.iterkeys().next())z!print list(iter(h.keys()).next())r; r< r r r �test_23� rE zTest_dict.test_23c C r: )Nz for x in h.keys()[0]: print xz#for x in list(h.keys())[0]: print xr; r< r r r �test_24� rE zTest_dict.test_24c C r: )Nz d.viewkeys()r� r; r< r r r �test_25� rE zTest_dict.test_25c C r: )Nz
d.viewitems()r� r; r< r r r �test_26� rE zTest_dict.test_26c C r: )Nzd.viewvalues()r� r; r< r r r �test_27� rE zTest_dict.test_27c C r: )Nz[i for i in d.viewkeys()]r r; r< r r r �test_28� rE zTest_dict.test_28c C r: )Nz(i for i in d.viewkeys())r r; r< r r r �test_29� rE zTest_dict.test_29c C r: )Nziter(d.viewkeys())r� r; r< r r r �test_30� rE zTest_dict.test_30c C r: )Nzlist(d.viewkeys())r� r; r< r r r �test_31� rE zTest_dict.test_31c C r: )Nzsorted(d.viewkeys())r r; r< r r r �test_32� rE zTest_dict.test_32N)'r5 r) r6 r rx r� ru r� r� r� r� r� r� r r r rN rO 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� � sJ r� c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dS )� Test_xrange�xrangec C rr )Nzx = xrange( 10 )zx = range( 10 )zx = xrange( 1 , 10 )zx = range( 1 , 10 )z x = xrange( 0 , 10 , 2 )zx = range( 0 , 10 , 2 )r; r< r r r rx � rt z$Test_xrange.test_prefix_preservationc C r: )Nzx = xrange(10)z
x = range(10)r; r< r r r �test_single_arg� rE zTest_xrange.test_single_argc C r: )Nzx = xrange(1, 10)zx = range(1, 10)r; r< r r r �
test_two_args� rE zTest_xrange.test_two_argsc C r: )Nzx = xrange(0, 10, 2)zx = range(0, 10, 2)r; r< r r r �test_three_args� rE zTest_xrange.test_three_argsc C sh d}d}| � ||� d}d}| � ||� d}d}| � ||� d}d}| � ||� d }d
}| � ||� d S ) Nzx = range(10, 3, 9)zx = list(range(10, 3, 9))zx = foo(range(10, 3, 9))z x = foo(list(range(10, 3, 9)))zx = range(10, 3, 9) + [4]z x = list(range(10, 3, 9)) + [4]zx = range(10)[::-1]zx = list(range(10))[::-1]zx = range(10) [3]zx = list(range(10)) [3]r; r< r r r �test_wrap_in_list� s z Test_xrange.test_wrap_in_listc C rp )Nz for i in xrange(10):
j=izfor i in range(10):
j=iz[i for i in xrange(10)]�[i for i in range(10)]r; r< r r r �test_xrange_in_for r� z Test_xrange.test_xrange_in_forc C � | � d� | � d� d S )Nzfor i in range(10): passr" r# r� r r r �test_range_in_for �
z Test_xrange.test_range_in_forc C r� )Nzx in range(10, 3, 9)r# r� r r r �test_in_contains_test r� z!Test_xrange.test_in_contains_testc C s t jD ] }| �d| � qd S )Nza = %s(range(10))r� )r Zcallr r r �test_in_consuming_context s
�z%Test_xrange.test_in_consuming_contextN)
r5 r) r6 r rx r r r r! r# r% r' r( r r r r r � s
r c �$ e Zd Z� fdd�Zdd� Z� ZS )�Test_xrange_with_reducec s t t| ��ddg� d S )Nr r| )�superr* r r� �r( r r r ! s z Test_xrange_with_reduce.setUpc C r: )Nzreduce(x, xrange(5))z0from functools import reduce
reduce(x, range(5))r; r< r r r �test_double_transform$ � z-Test_xrange_with_reduce.test_double_transform)r5 r) r6 r r- �
__classcell__r r r, r r* s r* c @ �P e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )�Test_raw_inputZ raw_inputc C rp )Nzx = raw_input( )zx = input( )zx = raw_input( '' )�x = input( '' )r; r< r r r rx - r� z'Test_raw_input.test_prefix_preservationc C r: )Nzx = raw_input()� x = input()r; r< r r r rD 6 rE zTest_raw_input.test_1c C r: )Nzx = raw_input('')�
x = input('')r; r< r r r rF ; rE zTest_raw_input.test_2c C r: )Nzx = raw_input('prompt')�x = input('prompt')r; r< r r r rG @ rE zTest_raw_input.test_3c C r: )Nzx = raw_input(foo(a) + 6)zx = input(foo(a) + 6)r; r< r r r rH E rE zTest_raw_input.test_4c C r: )Nz x = raw_input(invite).split()zx = input(invite).split()r; r< r r r rI J rE zTest_raw_input.test_5c C r: )Nz x = raw_input(invite) . split ()z x = input(invite) . split ()r; r< r r r rJ O rE zTest_raw_input.test_6c C r: )Nzx = int(raw_input())zx = int(input())r; r< r r r rL T rE zTest_raw_input.test_8N) r5 r) r6 r rx rD rF rG rH rI rJ rL r r r r r1 * s r1 c @ �( e Zd ZdZg d�Zdd� Zdd� ZdS )�Test_funcattrsZ funcattrs)Zclosure�doc�name�defaults�code�globalsr� c C sH | j D ] }d| }d| }| �||� d| }d| }| �||� qd S )Nz a.func_%s�a.__%s__zself.foo.func_%s.foo_bar�self.foo.__%s__.foo_bar��attrsr �r �attrr= r> r r r ro ^ s
�zTest_funcattrs.testc C �F | j D ] }d| }| �|� d| }| �|� d| }| �|� qd S )Nzfoo(func_%s + 5)�
f(foo.__%s__)�f(foo.__%s__.foo)�r@ r! �r rB r] r r r ru h r� z Test_funcattrs.test_unchangedN�r5 r) r6 r r@ ro ru r r r r r7 Y s
r7 c @ r7 ) �Test_xreadlinesZ
xreadlinesc C rr )Nz for x in f.xreadlines(): passzfor x in f: passz!for x in foo().xreadlines(): passzfor x in foo(): passz'for x in (5 + foo()).xreadlines(): passzfor x in (5 + foo()): passr; r< r r r rX v rt zTest_xreadlines.test_callc C rr )Nzfoo(f.xreadlines + 5)zfoo(f.__iter__ + 5)zfoo(f().xreadlines + 5)zfoo(f().__iter__ + 5)z foo((5 + f()).xreadlines + 5)zfoo((5 + f()).__iter__ + 5)r; r< r r r �
test_attr_ref� rt z Test_xreadlines.test_attr_refc C �< d}| � |� d}| � |� d}| � |� d}| � |� d S )Nz for x in f.xreadlines(5): passz for x in f.xreadlines(k=5): passz$for x in f.xreadlines(*k, **v): passzfoo(xreadlines)r# r\ r r r ru � �
z Test_xreadlines.test_unchangedN)r5 r) r6 r rX rJ ru r r r r rI s s
rI c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�ImportsFixerTestsc C sP | j �� D ] \}}d| }d| }| �||� d| }d| }| �||� qd S )N� import %szimport foo, %s, bar��modules�itemsr �r �old�newr= r> r r r �test_import_module� � �z$ImportsFixerTests.test_import_modulec C sl | j �� D ].\}}d| }d| }| �||� d| }d| }| �||� d| }d| }| �||� qd S )Nzfrom %s import foozfrom %s import foo, barzfrom %s import (yes, no)rO rR r r r �test_import_from� s �z"ImportsFixerTests.test_import_fromc C sP | j �� D ] \}}d| }d| }| �||� d| }d| }| �||� qd S )Nzimport %s as foo_barrO rR r r r �test_import_module_as� rV z'ImportsFixerTests.test_import_module_asc C �4 | j �� D ]\}}d| }d| }| �||� qd S )Nzfrom %s import foo as barrO rR r r r �test_import_from_as� �
�z%ImportsFixerTests.test_import_from_asc C rY )N�from %s import *rO rR r r r � test_star� r[ zImportsFixerTests.test_starc C s� | j �� D ]X\}}d||f }d||f }| �||� d||f }d||f }| �||� d|f }| �|� d|||f }d|||f }| �||� d||f }d||f }| �||� qd S )NzG
import %s
foo(%s.bar)
zJ
from %s import x
%s = 23
zJ
def f():
%s.method()
zJ
import %s
%s.bar(%s.foo)
z@
import %s
x.%s
)rP rQ r r! )r rS rT r= r> r] r r r �test_import_module_usage� sD �� �� �
�� ���z*ImportsFixerTests.test_import_module_usageN) r5 r) r6 rU rW rX rZ r] r^ r r r r rM � s
rM c @ s, e Zd ZdZddlmZ dd� Zdd� ZdS ) � Test_imports�imports� ��MAPPINGc C r: )Nzimport urlparse, cStringIOzimport urllib.parse, ior; r< r r r �test_multiple_imports rE z"Test_imports.test_multiple_importsc C r: )Nzv
import copy_reg as bar, HTMLParser as foo, urlparse
s = urlparse.spam(bar.foo())
z~
import copyreg as bar, html.parser as foo, urllib.parse
s = urllib.parse.spam(bar.foo())
r; r< r r r �test_multiple_imports_as r� z%Test_imports.test_multiple_imports_asN) r5 r) r6 r �fixes.fix_importsrc rP rd re r r r r r_ s
r_ c @ s e Zd ZdZddlmZ dS )�
Test_imports2�imports2ra rb N)r5 r) r6 r �fixes.fix_imports2rc rP r r r r rg s rg c r) )�Test_imports_fixer_orderc sR t t| ��ddg� ddlm} |�� | _ddlm} dD ] }|| | j|< q d S )Nr` rh ra rb )ZdbhashZdumbdbmZdbmZgdbm)r+ rj r ri rc �copyrP rf )r Zmapping2Zmapping1�keyr, r r r s
�z Test_imports_fixer_order.setUpc C s dD ]
}|| _ | �d� qd S )N)r` rh �import)r r4 )r Zfixr r r �$test_after_local_imports_refactoring$ s �z=Test_imports_fixer_order.test_after_local_imports_refactoring)r5 r) r6 r rn r/ r r r, r rj s rj c @ s\ e Zd ZdZddlmZ dd� Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dd� ZdS )� Test_urllibZurllibra rb c C sD | j �� D ]\}}d| }dd�ttd�|�� }| �||� qd S )NrN r' r )rP rQ r �mapr r )r rS �changesr= r> r r r rU . s
�z Test_urllib.test_import_modulec
C s� | j �� D ]o\}}g }|D ]M\}}|D ]"}|�|� d||f }d||f }| �||� d| } | �| � qd|d�|�f }d|d�|�f }| �||� dd�|� } | �| � q
d|d�|�f }d�dd� |D ��}| �||� qd S )N�from %s import %s�from foo import %sr' �
c S s" g | ]
\}}d |d� |�f �qS )rr r' )r )r* rT �membersr r r r, J s �z0Test_urllib.test_import_from.<locals>.<listcomp>)rP rQ �appendr r! r )
r rS rq Z all_membersrT ru �memberr= r> r] r r r rW 4 s*
��z Test_urllib.test_import_fromc C �$ | j D ] }d| }| �|d� qd S )Nzimport %s as fooz#This module is now multiple modules�rP r% �r rS r] r r r rX N �
�z!Test_urllib.test_import_module_asc C s| | j �� D ]6\}}|D ]/\}}|D ](}d||f }d||f }| �||� d|||f }d|||f }| �||� qq qd S )N� from %s import %s as foo_barz from %s import %s as blah, %srO )r rS rq rT ru rw r= r> r r r rZ S s ���z Test_urllib.test_import_from_asc C rx )Nr\ zCannot handle star importsry rz r r r r] ^ r{ zTest_urllib.test_starc C rp )Nz6
def foo():
from urllib import urlencode, urlopen
zZ
def foo():
from urllib.parse import urlencode
from urllib.request import urlopen
zB
def foo():
other()
from urllib import urlencode, urlopen
zf
def foo():
other()
from urllib.parse import urlencode
from urllib.request import urlopen
r; r< r r r �
test_indentedc s zTest_urllib.test_indentedc C r: )Nz from urllib import getproxiesz%from urllib.request import getproxiesr; r< r r r �test_single_import| r. z Test_urllib.test_single_importc C s� | j �� D ]I\}}|D ]B\}}|D ];}d�dd� | j | D ��}d|||f }d|||f }| �||� d|||||f }d|||||f }| �||� qq qd S )Nr' c S s g | ]\}}|�qS r r )r* r3 Zmemsr r r r, � r- z8Test_urllib.test_import_module_usage.<locals>.<listcomp>z^
import %s
foo(%s.%s)
z`
import %s
%s.%s(%s.%s)
)rP rQ r r ) r rS rq rT ru rw Z
new_importr= r> r r r r^ � s.
��� � ����z$Test_urllib.test_import_module_usageN)r5 r) r6 r Zfixes.fix_urllibrc rP rU rW rX rZ r] r} r~ r^ r r r r ro * s ro c @ r� )�
Test_input�inputc C rp )Nzx = input( )zx = eval(input( ))r2 zx = eval(input( '' ))r; r< r r r rx � r� z#Test_input.test_prefix_preservationc C r: )Nzx = input() # foozx = eval(input()) # foor; r< r r r r� � rE z Test_input.test_trailing_commentc C �. d}| � |� d}| � |� d}| � |� d S )N�x = eval(input())�x = eval(input(''))�x = eval(input(foo(5) + 9))r# r\ r r r r� � �
zTest_input.test_idempotencyc C r: )Nr3 r� r; r< r r r rD � rE zTest_input.test_1c C r: )Nr4 r� r; r< r r r rF � rE zTest_input.test_2c C r: )Nr5 zx = eval(input('prompt'))r; r< r r r rG � rE zTest_input.test_3c C r: )Nzx = input(foo(5) + 9)r� r; r< r r r rH � rE zTest_input.test_4N) r5 r) r6 r rx r� r� rD rF rG rH r r r r r � s
r c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � 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*S )+�Test_tuple_paramsZ tuple_paramsc C r[ )Nzdef foo(): passr# r\ r r r r^ � r_ z"Test_tuple_params.test_unchanged_1c C r[ )Nzdef foo(a, b, c): passr# r\ r r r r` � r_ z"Test_tuple_params.test_unchanged_2c C r[ )Nz def foo(a=3, b=4, c=5): passr# r\ r r r ra � r_ z"Test_tuple_params.test_unchanged_3c C r: )Nz8
def foo(((a, b), c)):
x = 5zn
def foo(xxx_todo_changeme):
((a, b), c) = xxx_todo_changeme
x = 5r; r< r r r rD � r� zTest_tuple_params.test_1c C r: )Nz;
def foo(((a, b), c), d):
x = 5zq
def foo(xxx_todo_changeme, d):
((a, b), c) = xxx_todo_changeme
x = 5r; r< r r r rF � r� zTest_tuple_params.test_2c C r: )Nz@
def foo(((a, b), c), d) -> e:
x = 5zv
def foo(xxx_todo_changeme, d) -> e:
((a, b), c) = xxx_todo_changeme
x = 5r; r< r r r rG � r� zTest_tuple_params.test_3c C r: )Nz/
def foo(((a, b), c)): x = 5; y = 7zV
def foo(xxx_todo_changeme): ((a, b), c) = xxx_todo_changeme; x = 5; y = 7r; r< r r r �test_semicolon� r@ z Test_tuple_params.test_semicolonc C r: )NzE
def foo(((a, b), c), d, e=5) -> z:
x = 5z{
def foo(xxx_todo_changeme, d, e=5) -> z:
((a, b), c) = xxx_todo_changeme
x = 5r; r< r r r �
test_keywords r� z Test_tuple_params.test_keywordsc C r: )NzR
def foo(((a, b), c), d, *vargs, **kwargs) -> z:
x = 5z�
def foo(xxx_todo_changeme, d, *vargs, **kwargs) -> z:
((a, b), c) = xxx_todo_changeme
x = 5r; r< r r r � test_varargs r� z Test_tuple_params.test_varargsc C r: )NzH
def foo(((a, b), c), (d, e, f)) -> z:
x = 5z�
def foo(xxx_todo_changeme, xxx_todo_changeme1) -> z:
((a, b), c) = xxx_todo_changeme
(d, e, f) = xxx_todo_changeme1
x = 5r; r< r r r � test_multi_1 � z Test_tuple_params.test_multi_1c C r: )NzQ
def foo(x, ((a, b), c), d, (e, f, g), y) -> z:
x = 5z�
def foo(x, xxx_todo_changeme, d, xxx_todo_changeme1, y) -> z:
((a, b), c) = xxx_todo_changeme
(e, f, g) = xxx_todo_changeme1
x = 5r; r< r r r � test_multi_2% r� z Test_tuple_params.test_multi_2c C r: )Nzj
def foo(((a, b), c), (d, e, f)) -> z:
"foo foo foo foo"
x = 5z�
def foo(xxx_todo_changeme, xxx_todo_changeme1) -> z:
"foo foo foo foo"
((a, b), c) = xxx_todo_changeme
(d, e, f) = xxx_todo_changeme1
x = 5r; r< r r r �test_docstring1 � z Test_tuple_params.test_docstringc C r[ )N�lambda x: x + 5r# r\ r r r �test_lambda_no_change? r_ z'Test_tuple_params.test_lambda_no_changec C �T d}d}| � ||� d}d}| � ||� d}d}| � ||� d}d}| � ||� d S )Nzlambda (x): x + 5r� zlambda(x): x + 5zlambda ((((x)))): x + 5zlambda((((x)))): x + 5r; r< r r r � test_lambda_parens_single_argC r� z/Test_tuple_params.test_lambda_parens_single_argc C r� )Nzlambda (x, y): x + f(y)z lambda x_y: x_y[0] + f(x_y[1])zlambda(x, y): x + f(y)zlambda (((x, y))): x + f(y)zlambda(((x, y))): x + f(y)r; r< r r r �test_lambda_simpleT r� z$Test_tuple_params.test_lambda_simplec C �, d}d}| � ||� d}d}| � ||� d S )Nzlambda (x,): x + f(x)zlambda x1: x1[0] + f(x1[0])zlambda (((x,))): x + f(x)r; r< r r r �test_lambda_one_tuplee r� z'Test_tuple_params.test_lambda_one_tuplec C r: )Nz lambda (x, y): x + x + f(x) + xz0lambda x_y: x_y[0] + x_y[0] + f(x_y[0]) + x_y[0]r; r< r r r � test_lambda_simple_multi_usen rE z.Test_tuple_params.test_lambda_simple_multi_usec C r: )Nzlambda (x, y): y + xzlambda x_y: x_y[1] + x_y[0]r; r< r r r �test_lambda_simple_reverses rE z,Test_tuple_params.test_lambda_simple_reversec C r� )Nz lambda (x, (y, z)): x + y + zz2lambda x_y_z: x_y_z[0] + x_y_z[1][0] + x_y_z[1][1]z!lambda (((x, (y, z)))): x + y + zr; r< r r r �test_lambda_nestedx r� z$Test_tuple_params.test_lambda_nestedc C r: )Nz lambda (x, (y, z)): x + y + f(y)z5lambda x_y_z: x_y_z[0] + x_y_z[1][0] + f(x_y_z[1][0])r; r< r r r � test_lambda_nested_multi_use� rE z.Test_tuple_params.test_lambda_nested_multi_useN)r5 r) r6 r r^ r` ra rD rF rG r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s, r� c @ r6 )�Test_methodattrsZ methodattrs)�funcr �classc C sd | j D ],}d| }|dkrd}nd| }| �||� d| }|dkr%d}nd| }| �||� qd S )Nza.im_%sr� za.__self__.__class__r= zself.foo.im_%s.foo_barz#self.foo.__self__.__class__.foo_barr> r? rA r r r ro � s
�zTest_methodattrs.testc C rC )Nzfoo(im_%s + 5)rD rE rF rG r r r ru � r� z Test_methodattrs.test_unchangedNrH r r r r r� � s
r� c @ sx e Zd ZdZdd� Zdd� Zdd� Zdd � 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.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� Z"d@dA� Z#dBdC� Z$dDdE� Z%dFdG� Z&dHdI� Z'dJdK� Z(dLdM� Z)dNdO� Z*dPdQ� Z+dRdS� Z,dTdU� Z-dVdW� Z.dXdY� Z/dZd[� Z0d\S )]� Test_next�nextc C r: )Nz it.next()znext(it)r; r< r r r rD � rE zTest_next.test_1c C r: )Nza.b.c.d.next()z
next(a.b.c.d)r; r< r r r rF � rE zTest_next.test_2c C r: )Nz(a + b).next()z
next((a + b))r; r< r r r rG � rE zTest_next.test_3c C r: )Nz
a().next()z next(a())r; r< r r r rH � rE zTest_next.test_4c C r: )Nza().next() + bz
next(a()) + br; r< r r r rI � rE zTest_next.test_5c C r: )Nzc( a().next() + b)zc( next(a()) + b)r; r< r r r rJ � rE zTest_next.test_6c C r: )NzU
for a in b:
foo(a)
a.next()
zT
for a in b:
foo(a)
next(a)
r; r< r r r �test_prefix_preservation_1� r� z$Test_next.test_prefix_preservation_1c C r: )Nzq
for a in b:
foo(a) # abc
# def
a.next()
zp
for a in b:
foo(a) # abc
# def
next(a)
r; r< r r r �test_prefix_preservation_2� r� z$Test_next.test_prefix_preservation_2c C � d}d}| j ||dd� d S )Nzj
next = 5
for a in b:
foo(a)
a.next()
zn
next = 5
for a in b:
foo(a)
a.__next__()
T�r r; r< r r r �test_prefix_preservation_3� � z$Test_next.test_prefix_preservation_3c C r� )Nz�
next = 5
for a in b:
foo(a) # abc
# def
a.next()
z�
next = 5
for a in b:
foo(a) # abc
# def
a.__next__()
Tr� r; r< r r r �test_prefix_preservation_4� s z$Test_next.test_prefix_preservation_4c C r� )Nzz
next = 5
for a in b:
foo(foo(a), # abc
a.next())
z~
next = 5
for a in b:
foo(foo(a), # abc
a.__next__())
Tr� r; r< r r r �test_prefix_preservation_5 r� z$Test_next.test_prefix_preservation_5c C r: )Nze
for a in b:
foo(foo(a), # abc
a.next())
zd
for a in b:
foo(foo(a), # abc
next(a))
r; r< r r r �test_prefix_preservation_6 r� z$Test_next.test_prefix_preservation_6c C r: )Nz[
class A:
def next(self):
pass
z_
class A:
def __next__(self):
pass
r; r< r r r �
test_method_1 r� zTest_next.test_method_1c C r: )Nzc
class A(object):
def next(self):
pass
zg
class A(object):
def __next__(self):
pass
r; r< r r r �
test_method_2, r� zTest_next.test_method_2c C r: )NzX
class A:
def next(x):
pass
z\
class A:
def __next__(x):
pass
r; r< r r r �
test_method_39 r� zTest_next.test_method_3c C r: )Nz�
class A:
def __init__(self, foo):
self.foo = foo
def next(self):
pass
def __iter__(self):
return self
z�
class A:
def __init__(self, foo):
self.foo = foo
def __next__(self):
pass
def __iter__(self):
return self
r; r< r r r �
test_method_4F s zTest_next.test_method_4c C r[ )Nza
class A:
def next(self, a, b):
pass
r# r\ r r r �test_method_unchanged_ r� z Test_next.test_method_unchangedc C r� )Nzy
next = foo
class A:
def next(self, a, b):
pass
�)Calls to builtin next() possibly shadowedr� r\ r r r � test_shadowing_assign_simpleg � z&Test_next.test_shadowing_assign_simplec C r� )Nz~
(next, a) = foo
class A:
def next(self, a, b):
pass
r� r� r\ r r r � test_shadowing_assign_tuple_1q r� z'Test_next.test_shadowing_assign_tuple_1c C r� )Nz�
(a, (b, (next, c)), a) = foo
class A:
def next(self, a, b):
pass
r� r� r\ r r r � test_shadowing_assign_tuple_2{ r� z'Test_next.test_shadowing_assign_tuple_2c C r� )Nz~
[next, a] = foo
class A:
def next(self, a, b):
pass
r� r� r\ r r r � test_shadowing_assign_list_1� r� z&Test_next.test_shadowing_assign_list_1c C r� )Nz�
[a, [b, [next, c]], a] = foo
class A:
def next(self, a, b):
pass
r� r� r\ r r r � test_shadowing_assign_list_2� r� z&Test_next.test_shadowing_assign_list_2c C r� )Nz�
def foo():
__builtin__.next = foo
class A:
def next(self, a, b):
pass
r� r� r\ r r r �test_builtin_assign� � z Test_next.test_builtin_assignc C r� )Nz�
def foo():
(a, __builtin__.next) = foo
class A:
def next(self, a, b):
pass
r� r� r\ r r r � test_builtin_assign_in_tuple� r� z&Test_next.test_builtin_assign_in_tuplec C r� )Nz�
def foo():
[a, __builtin__.next] = foo
class A:
def next(self, a, b):
pass
r� r� r\ r r r �test_builtin_assign_in_list� r� z%Test_next.test_builtin_assign_in_listc C r[ )Nz�
def foo():
A.next = foo
class A:
def next(self, a, b):
pass
r# r\ r r r �test_assign_to_next� � z Test_next.test_assign_to_nextc C r[ )Nz�
def foo():
(a, A.next) = foo
class A:
def next(self, a, b):
pass
r# r\ r r r � test_assign_to_next_in_tuple� r� z&Test_next.test_assign_to_next_in_tuplec C r[ )Nz�
def foo():
[a, A.next] = foo
class A:
def next(self, a, b):
pass
r# r\ r r r �test_assign_to_next_in_list� r� z%Test_next.test_assign_to_next_in_listc C r� )Nz�
import foo.bar as next
class A:
def next(self, a, b):
pass
r� r� r\ r r r �test_shadowing_import_1� r� z!Test_next.test_shadowing_import_1c C r� )Nz�
import bar, bar.foo as next
class A:
def next(self, a, b):
pass
r� r� r\ r r r �test_shadowing_import_2� r� z!Test_next.test_shadowing_import_2c C r� )Nz�
import bar, bar.foo as next, baz
class A:
def next(self, a, b):
pass
r� r� r\ r r r �test_shadowing_import_3� r� z!Test_next.test_shadowing_import_3c C r� )Nz�
from x import next
class A:
def next(self, a, b):
pass
r� r� r\ r r r � test_shadowing_import_from_1� r� z&Test_next.test_shadowing_import_from_1c C r� )Nz�
from x.a import next
class A:
def next(self, a, b):
pass
r� r� r\ r r r � test_shadowing_import_from_2
r� z&Test_next.test_shadowing_import_from_2c C r� )Nz�
from x import a, next, b
class A:
def next(self, a, b):
pass
r� r� r\ r r r � test_shadowing_import_from_3
r� z&Test_next.test_shadowing_import_from_3c C r� )Nz�
from x.a import a, next, b
class A:
def next(self, a, b):
pass
r� r� r\ r r r � test_shadowing_import_from_4
r� z&Test_next.test_shadowing_import_from_4c C r� )Nz�
def next(a):
pass
class A:
def next(self, a, b):
pass
r� r� r\ r r r �test_shadowing_funcdef_1!
r� z"Test_next.test_shadowing_funcdef_1c C s d}d}| � ||d� d S )Nz�
def next(a):
pass
class A:
def next(self):
pass
it.next()
z�
def next(a):
pass
class A:
def __next__(self):
pass
it.__next__()
r� r$ r< r r r �test_shadowing_funcdef_2,
s
z"Test_next.test_shadowing_funcdef_2c C r� )NzW
def f():
global next
next = 5
r� r� r\ r r r �test_shadowing_global_1C
� z!Test_next.test_shadowing_global_1c C r� )Nz]
def f():
global a, next, b
next = 5
r� r� r\ r r r �test_shadowing_global_2K
r� z!Test_next.test_shadowing_global_2c C r� )Nze
for next in it():
pass
b = 5
c = 6
r� r� r\ r r r �test_shadowing_for_simpleS
r� z#Test_next.test_shadowing_for_simplec C r� )Nzh
for next, b in it():
pass
b = 5
c = 6
r� r� r\ r r r �test_shadowing_for_tuple_1]
r� z$Test_next.test_shadowing_for_tuple_1c C r� )Nzp
for a, (next, c), b in it():
pass
b = 5
c = 6
r� r� r\ r r r �test_shadowing_for_tuple_2g
r� z$Test_next.test_shadowing_for_tuple_2c C r: )Nzgnext = g.nextzgnext = g.__next__r; r< r r r �test_noncall_access_1q
rE z Test_next.test_noncall_access_1c C r: )Nz
f(g.next + 5)zf(g.__next__ + 5)r; r< r r r �test_noncall_access_2v
rE z Test_next.test_noncall_access_2c C r: )Nzf(g().next + 5)zf(g().__next__ + 5)r; r< r r r �test_noncall_access_3{
rE z Test_next.test_noncall_access_3N)1r5 r) r6 r rD rF rG rH rI rJ 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� � s^
r� c @ r� )
� Test_nonzeroZnonzeroc C r: )Nzb
class A:
def __nonzero__(self):
pass
z_
class A:
def __bool__(self):
pass
r; r< r r r rD �
r� zTest_nonzero.test_1c C r: )Nzj
class A(object):
def __nonzero__(self):
pass
�g
class A(object):
def __bool__(self):
pass
r; r< r r r rF �
r� zTest_nonzero.test_2c C r[ )Nr� r# r\ r r r r^ �
r� z Test_nonzero.test_unchanged_1c C r[ )Nzm
class A(object):
def __nonzero__(self, a):
pass
r# r\ r r r r` �
r� z Test_nonzero.test_unchanged_2c C r[ )NzE
def __nonzero__(self):
pass
r# r\ r r r �test_unchanged_func�
s z Test_nonzero.test_unchanged_funcN) r5 r) r6 r rD rF r^ r` r� r r r r r� �
s
r� c @ sx e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd S ) �Test_numliteralsZ numliteralsc C r: )NZ0755�0o755r; r< r r r � test_octal_1�
rE z Test_numliterals.test_octal_1c C r: )Nza = 12L�a = 12r; r< r r r �test_long_int_1�
rE z Test_numliterals.test_long_int_1c C r: )Nza = 12lr� r; r< r r r �test_long_int_2�
rE z Test_numliterals.test_long_int_2c C r: )Nz b = 0x12lzb = 0x12r; r< r r r �
test_long_hex�
rE z Test_numliterals.test_long_hexc C rp )Nz b = 0x12Lz
b = 0x12zb = 0755 # spamzb = 0o755 # spamr; r< r r r �test_comments_and_spacing�
r� z*Test_numliterals.test_comments_and_spacingc C r[ )N�5r# r\ r r r �test_unchanged_int�
r_ z#Test_numliterals.test_unchanged_intc C r[ )Nz5.0r# r\ r r r �test_unchanged_float�
r_ z%Test_numliterals.test_unchanged_floatc C r[ )Nr� r# r\ r r r �test_unchanged_octal�
r_ z%Test_numliterals.test_unchanged_octalc C r[ )NZ0xABCr# r\ r r r �test_unchanged_hex�
r_ z#Test_numliterals.test_unchanged_hexc C r[ )Nz5.0e10r# r\ r r r �test_unchanged_exp�
r_ z#Test_numliterals.test_unchanged_expc C r[ )Nz5 + 4jr# r\ r r r �test_unchanged_complex_int�
r_ z+Test_numliterals.test_unchanged_complex_intc C r[ )Nz
5.4 + 4.9jr# r\ r r r � test_unchanged_complex_float�
r_ z-Test_numliterals.test_unchanged_complex_floatc C r� )NZ4jz4.4jr# r\ r r r �test_unchanged_complex_bare�
s
z,Test_numliterals.test_unchanged_complex_bareN)r5 r) r6 r r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� �
s r� c @ s8 e Zd ZdZddiZdd� Zdd� Zdd � Zd
d � Zd S )
� Test_renames�renames�sys)Zmaxint�maxsizec C sV t | j�� �D ]!\}\}}d||f }d||f }| �||� d| }| �|� qd S )Nrr rs )�listrP rQ r r! )r �modrS rT r= r> r] r r r rW �
s �z Test_renames.test_import_fromc C sD t | j�� �D ]\}\}}d||f }d||f }| �||� qd S )Nr| �r� rP rQ r �r r� rS rT r= r> r r r rZ s
�z Test_renames.test_import_from_asc C sL t | j�� �D ] \}\}}d||||f }d||||f }| �||� qd S )NzJ
import %s
foo(%s, %s.%s)
r� r� r r r r^ s
�
��z%Test_renames.test_import_module_usagec C sL t | j�� �D ] \}\}}d||||f }d||||f }| �||� qd S )NzO
from %s import %s
foo(%s, %s)
r� r� r r r �XXX_test_from_import_usage s
�
��z'Test_renames.XXX_test_from_import_usageN) r5 r) r6 r rP rW rZ r^ r� r r r r r� �
s r� c @ r� )� Test_unicodeZunicodec C rr )Nz unicode( x)zstr( x)z unicode(x )z str(x )z u'h'z 'h'r; r< r r r �test_whitespace' rt z Test_unicode.test_whitespacec C r: )Nzunicode(x, y, z)z str(x, y, z)r; r< r r r �test_unicode_call4 rE z Test_unicode.test_unicode_callc C r: )Nz unichr(u'h')zchr('h')r; r< r r r � test_unichr9 rE zTest_unicode.test_unichrc C r: )Nzu"x"z"x"r; r< r r r �test_unicode_literal_1> rE z#Test_unicode.test_unicode_literal_1c C r: )Nzur'x'zr'x'r; r< r r r �test_unicode_literal_2C rE z#Test_unicode.test_unicode_literal_2c C r: )Nz
UR'''x''' z R'''x''' r; r< r r r �test_unicode_literal_3H rE z#Test_unicode.test_unicode_literal_3c C s, d}d}| � ||� d}d}| � ||� d S )N�'\\\u20ac\U0001d121\\u20ac'z '\\\\u20ac\\U0001d121\\u20ac'� r'\\\u20ac\U0001d121\\u20ac'r; r< r r r � test_native_literal_escape_uM r� z)Test_unicode.test_native_literal_escape_uc C s, d}d}| � ||� d}d}| � ||� d S )Nz b'\\\u20ac\U0001d121\\u20ac'z br'\\\u20ac\U0001d121\\u20ac'r; r< r r r �test_bytes_literal_escape_uV r� z(Test_unicode.test_bytes_literal_escape_uc C rp )Nz u'\\\u20ac\U0001d121\\u20ac'r� z ur'\\\u20ac\U0001d121\\u20ac'r� r; r< r r r � test_unicode_literal_escape_u_ r� z*Test_unicode.test_unicode_literal_escape_uc C s@ d}|d }|d }| � ||� |d }|d }| � ||� d S )Nz(from __future__ import unicode_literals
r� r� r; )r �fr= r> r r r �$test_native_unicode_literal_escape_uh s z1Test_unicode.test_native_unicode_literal_escape_uN)r5 r) r6 r r� r� r� r� r� r� r� r� r� r� r r r r r� $ s
r� c @ r� )
� Test_filter�filterc C rp )Nz#x = filter( foo, 'abc' )z)x = list(filter( foo, 'abc' ))z x = filter( None , 'abc' )z x = [_f for _f in 'abc' if _f]r; r< r r r rx v r� z$Test_filter.test_prefix_preservationc C sh d}d}| � ||� d}d}| � ||� d}d}| � ||� d}d}| � ||� d}d }| � ||� d S )
Nzx = filter(None, 'abc')z x = [_f for _f in 'abc' if _f]zx = len(filter(f, 'abc'))z x = len(list(filter(f, 'abc')))z)x = filter(lambda x: x%2 == 0, range(10))z&x = [x for x in range(10) if x%2 == 0]z+x = filter(lambda (x): x%2 == 0, range(10))z5filter(lambda x: True if x > 2 else False, [1, 2, 3])z4[x for x in [1, 2, 3] if (True if x > 2 else False)]r; r< r r r �test_filter_basic s z Test_filter.test_filter_basicc C sT d}d}| � ||� d}d}| � ||� d}d}| � ||� d}d}| � ||� d S )Nzx = filter(None, 'abc')[0]z!x = [_f for _f in 'abc' if _f][0]z x = len(filter(f, 'abc')[0])z"x = len(list(filter(f, 'abc'))[0])z,x = filter(lambda x: x%2 == 0, range(10))[0]z)x = [x for x in range(10) if x%2 == 0][0]z.x = filter(lambda (x): x%2 == 0, range(10))[0]r; r< r r r �test_filter_trailers� s z Test_filter.test_filter_trailersc C � d}| � |� d}| � |� d}| � |� d}| � |� d}| � |� d}| � |� d}| � |� d}| � |� d }| � |� d
}| � |� d }| � |� d }| � |� d
}| � |� d}| � |� d}| � |� d}| � |� d}| � |� d}| � |� d}| � |� d S )Nzb.join(filter(f, 'abc'))z#(a + foo(5)).join(filter(f, 'abc'))ziter(filter(f, 'abc'))zlist(filter(f, 'abc'))zlist(filter(f, 'abc'))[0]zset(filter(f, 'abc'))zset(filter(f, 'abc')).pop()ztuple(filter(f, 'abc'))zany(filter(f, 'abc'))zall(filter(f, 'abc'))zsum(filter(f, 'abc'))zsorted(filter(f, 'abc'))z"sorted(filter(f, 'abc'), key=blah)z%sorted(filter(f, 'abc'), key=blah)[0]zenumerate(filter(f, 'abc'))z$enumerate(filter(f, 'abc'), start=1)z for i in filter(f, 'abc'): passz [x for x in filter(f, 'abc')]z (x for x in filter(f, 'abc'))r# �r r> r r r �test_filter_nochange� �L
z Test_filter.test_filter_nochangec C �4 d}| � |� d}d}| �||� d}| � |� d S )Nz:from future_builtins import spam, filter; filter(f, 'ham')z6from future_builtins import spam; x = filter(f, 'abc')z<from future_builtins import spam; x = list(filter(f, 'abc'))z/from future_builtins import *; filter(f, 'ham')r� ri r r r �test_future_builtins� �
z Test_filter.test_future_builtinsN) r5 r) r6 r rx r� r� r� r r r r r r� s s (r� c sX e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z � Z S )�Test_maprp c �& | � d| |� tt| ��||� d S )Nz!from future_builtins import map; )r! r+ r r r< r, r r r � � zTest_map.checkc C r: )Nz x = map( f, 'abc' )z#x = list(map( f, 'abc' ))r; r< r r r rx � rE z!Test_map.test_prefix_preservationc C r� ) Nzx = map(f, 'abc')[0]zx = list(map(f, 'abc'))[0]zx = map(None, l)[0]zx = list(l)[0]zx = map(lambda x:x, l)[0]zx = [x for x in l][0]zx = map(f, 'abc')[0][1]z x = list(map(f, 'abc'))[0][1]r; r< r r r �test_map_trailers� r� zTest_map.test_map_trailersc C r: )Nzx = map(f, 'abc') # fooz!x = list(map(f, 'abc')) # foor; r< r r r r� � rE z Test_map.test_trailing_commentc C r� )Nzx = map(None, a, b, c)z5cannot convert map(None, ...) with multiple argumentsr� r\ r r r �!test_None_with_multiple_arguments� r� z*Test_map.test_None_with_multiple_argumentsc C s~ d}d}| � ||� d}d}| � ||� d}d}| � ||� d}d}| � ||� d }d}| � ||� d
}d }| �||d � d S )
Nzx = map(f, 'abc')zx = list(map(f, 'abc'))z x = len(map(f, 'abc', 'def'))z#x = len(list(map(f, 'abc', 'def')))zx = map(None, 'abc')zx = list('abc')z x = map(lambda x: x+1, range(4))zx = [x+1 for x in range(4)]z"x = map(lambda (x): x+1, range(4))zG
foo()
# foo
map(f, x)
zM
foo()
# foo
list(map(f, x))
z You should use a for loop here)r r" r< r r r �test_map_basic s$ zTest_map.test_map_basicc C r� )Nzb.join(map(f, 'abc'))z (a + foo(5)).join(map(f, 'abc'))ziter(map(f, 'abc'))zlist(map(f, 'abc'))zlist(map(f, 'abc'))[0]zset(map(f, 'abc'))zset(map(f, 'abc')).pop()ztuple(map(f, 'abc'))zany(map(f, 'abc'))zall(map(f, 'abc'))zsum(map(f, 'abc'))zsorted(map(f, 'abc'))z sorted(map(f, 'abc'), key=blah)z"sorted(map(f, 'abc'), key=blah)[0]zenumerate(map(f, 'abc'))z!enumerate(map(f, 'abc'), start=1)z for i in map(f, 'abc'): passz[x for x in map(f, 'abc')]z(x for x in map(f, 'abc'))r# r� r r r �test_map_nochange$ r zTest_map.test_map_nochangec C r )Nz:from future_builtins import spam, map, eggs; map(f, 'ham')z9from future_builtins import spam, eggs; x = map(f, 'abc')z?from future_builtins import spam, eggs; x = list(map(f, 'abc'))z,from future_builtins import *; map(f, 'ham')r� ri r r r r L r z Test_map.test_future_builtins)
r5 r) r6 r r rx r r� r r r
r r/ r r r, r r � s "(r c s@ e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Z� Z S ) �Test_zip�zipc r )Nz!from future_builtins import zip; )r! r+ r r r< r, r r r Z r zTest_zip.checkc C rr )Nz x = zip()zx = list(zip())zx = zip(a, b, c)zx = list(zip(a, b, c))zx = len(zip(a, b))zx = len(list(zip(a, b)))r; r< r r r �test_zip_basic^ rt zTest_zip.test_zip_basicc C rp )Nzx = zip(a, b, c)[0]zx = list(zip(a, b, c))[0]zx = zip(a, b, c)[0][1]z x = list(zip(a, b, c))[0][1]r; r< r r r �test_zip_trailersk r� zTest_zip.test_zip_trailersc C r� )Nzb.join(zip(a, b))z (a + foo(5)).join(zip(a, b))ziter(zip(a, b))zlist(zip(a, b))zlist(zip(a, b))[0]zset(zip(a, b))zset(zip(a, b)).pop()ztuple(zip(a, b))zany(zip(a, b))zall(zip(a, b))zsum(zip(a, b))zsorted(zip(a, b))zsorted(zip(a, b), key=blah)z sorted(zip(a, b), key=blah)[0]zenumerate(zip(a, b))z enumerate(zip(a, b), start=1)zfor i in zip(a, b): passz[x for x in zip(a, b)]z(x for x in zip(a, b))r# r� r r r �test_zip_nochanget r zTest_zip.test_zip_nochangec C r )Nz6from future_builtins import spam, zip, eggs; zip(a, b)z5from future_builtins import spam, eggs; x = zip(a, b)z;from future_builtins import spam, eggs; x = list(zip(a, b))z(from future_builtins import *; zip(a, b)r� ri r r r r � r z Test_zip.test_future_builtins)
r5 r) r6 r r r
r r r r/ r r r, r r W s
(r c @ � e Zd ZdZdd� ZdS )�Test_standarderrorZ
standarderrorc C rr )Nzx = StandardError()zx = Exception()zx = StandardError(a, b, c)zx = Exception(a, b, c)z f(2 + StandardError(a, b, c))zf(2 + Exception(a, b, c))r; r< r r r ro � rt zTest_standarderror.testN)r5 r) r6 r ro r r r r r � � r c @ r )�
Test_types�typesc C s� d}d}| � ||� d}d}| � ||� d}d}| � ||� d}d}| � ||� d }d}| � ||� d
}d }| � ||� d }d
}| � ||� d S )Nztypes.StringType�bytesztypes.DictTyper� ztypes . IntType�intztypes.ListTyper� ztypes.LongTypeztypes.NoneTypez
type(None)ztypes.StringTypesz(str,)r; r< r r r �test_basic_types_convert� s* z#Test_types.test_basic_types_convertN)r5 r) r6 r r r r r r r � r r c @ r� )'� Test_idiomsZidiomsc C rr )Nzwhile 1: foo()zwhile True: foo()zwhile 1: foo()zwhile True: foo()z8
while 1:
foo()
z;
while True:
foo()
r; r< r r r �
test_while� s zTest_idioms.test_whilec C rK )Nzwhile 11: foo()zwhile 0: foo()zwhile foo(): foo()zwhile []: foo()r# r\ r r r �test_while_unchanged� rL z Test_idioms.test_while_unchangedc C rp )Nz type(x) == T�isinstance(x, T)zif type(x) == T: pass�if isinstance(x, T): passr; r< r r r �test_eq_simple� r� zTest_idioms.test_eq_simplec C rp )Nz T == type(x)r zif T == type(x): passr r; r< r r r �test_eq_reverse
r� zTest_idioms.test_eq_reversec C rp )Nztype(x+y) == d.get('T')�isinstance(x+y, d.get('T'))z type( x + y) == d.get('T')� isinstance(x + y, d.get('T'))r; r< r r r �test_eq_expression
r� z Test_idioms.test_eq_expressionc C rp )Nz type(x) is Tr zif type(x) is T: passr r; r< r r r �test_is_simple
r� zTest_idioms.test_is_simplec C rp )Nz T is type(x)r zif T is type(x): passr r; r< r r r �test_is_reverse
r� zTest_idioms.test_is_reversec C rp )Nztype(x+y) is d.get('T')r z type( x + y) is d.get('T')r r; r< r r r �test_is_expression'
r� z Test_idioms.test_is_expressionc C rp )Nztype(x) is not T�not isinstance(x, T)zif type(x) is not T: pass� if not isinstance(x, T): passr; r< r r r �test_is_not_simple0
r� z Test_idioms.test_is_not_simplec C rp )NzT is not type(x)r% zif T is not type(x): passr& r; r< r r r �test_is_not_reverse9
r� z Test_idioms.test_is_not_reversec C rp )Nztype(x+y) is not d.get('T')� not isinstance(x+y, d.get('T'))z"type( x + y) is not d.get('T')�#not isinstance(x + y, d.get('T'))r; r< r r r �test_is_not_expressionB
r� z"Test_idioms.test_is_not_expressionc C rp )Nz type(x) != Tr% zif type(x) != T: passr& r; r< r r r �test_ne_simpleK
r� zTest_idioms.test_ne_simplec C rp )Nz T != type(x)r% zif T != type(x): passr& r; r< r r r �test_ne_reverseT
r� zTest_idioms.test_ne_reversec C rp )Nztype(x+y) != d.get('T')r) z type( x + y) != d.get('T')r* r; r< r r r �test_ne_expression]
r� z Test_idioms.test_ne_expressionc C r[ )Nztype(x).__name__r# r� r r r �test_type_unchangedf
r_ z Test_idioms.test_type_unchangedc C s� d}d}| � ||� d}d}| � ||� d}d}| � ||� d}d}| � ||� d }d
}| � ||� d }d }| � ||� d
}d}| � ||� d}d}| � ||� d S )NzM
v = list(t)
v.sort()
foo(v)
�:
v = sorted(t)
foo(v)
zV
v = list(foo(b) + d)
v.sort()
foo(v)
�C
v = sorted(foo(b) + d)
foo(v)
zn
while x:
v = list(t)
v.sort()
foo(v)
�W
while x:
v = sorted(t)
foo(v)
z_
v = list(t)
# foo
v.sort()
foo(v)
�L
v = sorted(t)
# foo
foo(v)
zP
v = list( t)
v.sort()
foo(v)
z=
v = sorted( t)
foo(v)
zl
try:
m = list(s)
m.sort()
except: pass
zU
try:
m = sorted(s)
except: pass
z�
try:
m = list(s)
# foo
m.sort()
except: pass
zk
try:
m = sorted(s)
# foo
except: pass
zI
m = list(s)
# more comments
m.sort()z6
m = sorted(s)
# more commentsr; r< r r r �test_sort_list_callj
s0 z Test_idioms.test_sort_list_callc C � d}d}| � ||� d}d}| � ||� d}d}| � ||� d}d}| � ||� d }d
}| � ||� d }d }| � ||� d
}d}| � ||� d S )NzG
v = t
v.sort()
foo(v)
r0 zL
v = foo(b)
v.sort()
foo(v)
z?
v = sorted(foo(b))
foo(v)
zN
v = b.keys()
v.sort()
foo(v)
zA
v = sorted(b.keys())
foo(v)
zP
v = foo(b) + d
v.sort()
foo(v)
r1 zh
while x:
v = t
v.sort()
foo(v)
r2 zY
v = t
# foo
v.sort()
foo(v)
r3 zI
v = t
v.sort()
foo(v)
z<
v = sorted(t)
foo(v)
r; r< r r r �test_sort_simple_expr�
s* z!Test_idioms.test_sort_simple_exprc C r� )NzM
v = list(t)
w.sort()
foo(w)
zN
v = list(t)
v.sort(u)
foo(v)
r# r\ r r r �test_sort_unchanged s
z Test_idioms.test_sort_unchangedN)r5 r) r6 r r r r r r! r" r# r$ r' r( r+ r, r- r. r/ r4 r6 r7 r r r r r � s(
d Rr c @ r )�Test_basestringZ
basestringc C r: )Nzisinstance(x, basestring)zisinstance(x, str)r; r< r r r �test_basestring2 rE z Test_basestring.test_basestringN)r5 r) r6 r r9 r r r r r8 / r r8 c @ r� )� Test_buffer�bufferc C r: )Nz
x = buffer(y)zx = memoryview(y)r; r< r r r � test_buffer: rE zTest_buffer.test_bufferc C r: )Nzbuffer(y)[4:5]zmemoryview(y)[4:5]r; r< r r r � test_slicing? rE zTest_buffer.test_slicingN)r5 r) r6 r r< r= r r r r r: 7 s r: c @ r� )� Test_futureZfuturec C rr )Nz from __future__ import bracesr z'# comment
from __future__ import bracesz
# comment
z'from __future__ import braces
# commentz
# commentr; r< r r r � test_futureG rt zTest_future.test_futurec C r� )Nr� �r4 r� r r r �test_run_orderT r� zTest_future.test_run_orderN)r5 r) r6 r r? rA r r r r r> D s
r> c @ r0 )�Test_itertools� itertoolsc C s< dD ]}dD ]}||d | }|| }| � ||� qqd S )N)z
itertools.r )rp r� r �ir; )r r r rD r� r= r> r r r �checkallZ s ��zTest_itertools.checkallc C r: )Nzitertools.izip(a, b)z zip(a, b)r; r< r r r �test_0d s zTest_itertools.test_0c C � d}d}| � ||� d S )Nz%s(f, a)�rE r< r r r rD k rE zTest_itertools.test_1c C rp )Nz itertools.ifilterfalse(a, b)zitertools.filterfalse(a, b)z itertools.izip_longest(a, b)zitertools.zip_longest(a, b)r; r< r r r �test_qualifiedp r� z Test_itertools.test_qualifiedc C rp )Nzifilterfalse(a, b)zfilterfalse(a, b)zizip_longest(a, b)zzip_longest(a, b)r; r< r r r rF y r� zTest_itertools.test_2c C rG )Nz %s(f, a)rH r< r r r rj � rE zTest_itertools.test_space_1c C rp )Nz itertools.ifilterfalse(a, b)z itertools.filterfalse(a, b)z itertools.izip_longest(a, b)z itertools.zip_longest(a, b)r; r< r r r rk � r� zTest_itertools.test_space_2c C s | � ddd� d S )Nrp r r� r@ r� r r r rA � s z Test_itertools.test_run_orderN) r5 r) r6 r rE rF rD rI rF rj rk rA r r r r rB W s
rB c @ r� )�Test_itertools_importsZitertools_importsc C rr )Nz%from itertools import imap, izip, foo�from itertools import fooz*from itertools import bar, imap, izip, fooz from itertools import bar, fooz'from itertools import chain, imap, izipzfrom itertools import chainr; r< r r r � test_reduced� rt z#Test_itertools_imports.test_reducedc C r: )Nz%#foo
from itertools import imap, izipz#foo
r; r< r r r �
test_comments� rE z$Test_itertools_imports.test_commentsc C r� )Nz from itertools import imap, izipr zfrom itertools import izipr; r< r r r � test_none� r� z Test_itertools_imports.test_nonec C sb d}d}| � ||� d}d}| � ||� d}d}| � ||� d}d}| � ||� d}| �|� d S )Nz-from itertools import izip, bar as bang, imapz!from itertools import bar as bangz-from itertools import izip as _zip, imap, barzfrom itertools import barz"from itertools import imap as _mapr z0from itertools import imap as _map, izip as _zip)r r! )r r= r> r] r r r �test_import_as� s z%Test_itertools_imports.test_import_asc C sn dD ]2}d|f }d|f }| � ||� d|f }d|f }| � ||� d|f }d|f }| � ||� qd S )N)� filterfalse� zip_longestzfrom itertools import i%szfrom itertools import %sz$from itertools import imap, i%s, fooz from itertools import %s, fooz#from itertools import bar, i%s, fooz"from itertools import bar, %s, foor; )r r9 r= r> r r r � test_ifilter_and_zip_longest� s
�z3Test_itertools_imports.test_ifilter_and_zip_longestc C r[ )Nzfrom itertools import *r# r\ r r r �test_import_star� r_ z'Test_itertools_imports.test_import_starc C r[ )NrK r# r\ r r r ru � r_ z%Test_itertools_imports.test_unchangedN) r5 r) r6 r rL rM rN rO rR rS ru r r r r rJ � s
rJ c s� e Zd ZdZ� fdd�Zdd� Z� fdd�Zdd � 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'� Z� ZS )(� Test_importrm c sD t t� ��� g � _t� � _d� _� fdd�}ddlm} ||_ d S )NTc s � j �| � � jp
| � jv S r )�
files_checkedrv �
always_exists�
present_files�r9 r� r r � fake_exists� s z&Test_import.setUp.<locals>.fake_existsr ��
fix_import)
r+ rT r rU �setrW rV �
lib2to3.fixesr[ �exists)r rY r[ r, r� r r � s
zTest_import.setUpc C s ddl m} tjj|_d S )Nr rZ )r] r[ �os�pathr^ )r r[ r r r �tearDown� s zTest_import.tearDownc s2 d| _ tt| ��||� d| _ tt| ��|� d S )NTF)rV r+ rT r r! r< r, r r �
check_both� s zTest_import.check_bothc s� dd� }d| _ tdg�| _dtjjdddd f}|d
�d |d �f}|D ]7� g | _� | _| �d
� tj� � �r>tj� � �d � nd� t� fdd�|D ��}|�
d� | � t| j�|� q"d S )Nc S s t jj�| �d��S )N�/)r_ r` �pathsepr �split)r` r r r �p� s z)Test_import.test_files_checked.<locals>.pF� __init__.pyz.pyz.pycz.soz.slz.pydz
/spam/eggs.pyzni.pyz../../shrubbery.pyz
import jamz/jamZjamc 3 s � | ]}� | V qd S r r )r* ZextrX r r � <genexpr> s � z1Test_import.test_files_checked.<locals>.<genexpr>) rV r\ rW r_ r` �seprU r
r! �dirname�addr )r rf Zexpected_extensionsZ
names_to_testZexpected_checksr rX r �test_files_checked� s
�z Test_import.test_files_checkedc C s$ d}d| _ tdg�| _| �|� d S )N�
import barF�bar.py�rV r\ rW r! r\ r r r �test_not_in_package s z Test_import.test_not_in_packagec C s& d}d| _ tddg�| _| �|� d S )Nz1from __future__ import absolute_import
import barFrg rn ro r\ r r r �!test_with_absolute_import_enabled s z-Test_import.test_with_absolute_import_enabledc C s, d}d}d| _ tddg�| _| �||� d S )Nrm �from . import barFrg rn )rV r\ rW r r< r r r �test_in_package s
zTest_import.test_in_packagec C s4 d}d}d| _ tddtjj g�| _| �||� d S )Nrm rr Frg Zbar)rV r\ r_ r` ri rW r r< r r r �test_import_from_package$ s
z$Test_import.test_import_from_packagec C r[ )Nrr r# r\ r r r � test_already_relative_import+ r_ z(Test_import.test_already_relative_importc C r: )Nzimport bar # Foozfrom . import bar # Foor; r< r r r �test_comments_and_indent/ rE z$Test_import.test_comments_and_indentc C rr )Nzfrom foo import bar, bazzfrom .foo import bar, bazzfrom foo import barzfrom .foo import barzfrom foo import (bar, baz)zfrom .foo import (bar, baz)�rb r< r r r � test_from4 rt zTest_import.test_fromc C r: )Nzfrom green.eggs import hamzfrom .green.eggs import hamrw r< r r r �test_dotted_fromA rE z Test_import.test_dotted_fromc C r: )Nz"from green.eggs import ham as spamz#from .green.eggs import ham as spamrw r< r r r � test_from_asF rE zTest_import.test_from_asc C r� ) Nz
import foozfrom . import foo�import foo, barzfrom . import foo, barzimport foo, bar, xzfrom . import foo, bar, xzimport x, y, zzfrom . import x, y, zrw r< r r r � test_importK r� zTest_import.test_importc C rp )Nzimport foo as xzfrom . import foo as xz import a as b, b as c, c as dz$from . import a as b, b as c, c as drw r< r r r rO \ r� zTest_import.test_import_asc C s( d| _ tddg�| _d}| �|d� d S )NFzfoo.pyrg r{ z#absolute and local imports together)rV r\ rW r% r\ r r r �test_local_and_absolutee s z#Test_import.test_local_and_absolutec C r: )Nzimport foo.barzfrom . import foo.barrw r< r r r �test_dotted_importl rE z Test_import.test_dotted_importc C r: )Nzimport foo.bar as bangz from . import foo.bar as bangrw r< r r r �test_dotted_import_asq rE z!Test_import.test_dotted_import_asc C r: )Nz1
# prefix
import foo.bar
z8
# prefix
from . import foo.bar
rw r< r r r � test_prefixv r� zTest_import.test_prefix)r5 r) r6 r r ra rb rl rp rq rs rt ru rv rx ry rz r| rO r} r~ r r� r/ r r r, r rT � s*
rT c @ r� )
�Test_set_literalZ set_literalc C s� d}d}| � ||� d}d}| � ||� d}d}| � ||� d}| � ||� d}d}| � ||� d }| � ||� d
}d }| � ||� d }d
}| � ||� d}d}| � ||� d S )Nzset([1, 2, 3])z {1, 2, 3}zset((1, 2, 3))z set((1,))�{1}zset([1])z set((a, b))z{a, b}z set([a, b])zset((a*234, f(args=23)))z{a*234, f(args=23)}zset([a*23, f(23)])z
{a*23, f(23)}zset([a-234**23])z {a-234**23}r; r< r r r r? � s2 zTest_set_literal.test_basicc C r� ) Nzset([x for x in y])z{x for x in y}z set([x for x in y if x == m])z{x for x in y if x == m}z set([x for x in y for a in b])z{x for x in y for a in b}zset([f(x) - 23 for x in y])z{f(x) - 23 for x in y}r; r< r r r �test_listcomps� r� z Test_set_literal.test_listcompsc C r5 )Nz set( [1, 2])z{1, 2}z
set([1 , 2])z{1 , 2}z
set([ 1 ])z{ 1 }z
set( [1] )r� zset([ 1, 2 ])z { 1, 2 }zset([x for x in y ])z{x for x in y }z<set(
[1, 2]
)
z{1, 2}
r; r< r r r r� � s* z Test_set_literal.test_whitespacec C rp )Nzset((1, 2)) # Hiz {1, 2} # HizZ
# Foo
set( # Bar
(1, 2)
)
z2
# Foo
{1, 2}
r; r< r r r rM � s z Test_set_literal.test_commentsc C sX d}| � |� d}| � |� d}| � |� d}| � |� d}| � |� d}| � |� d S )Nzset()zset(a)z set(a, b, c)zset(x for x in y)zset(x for x in y if z)zset(a*823-23**2 + f(23))r# r\ r r r ru � s
z Test_set_literal.test_unchangedN) r5 r) r6 r r? r� r� rM ru r r r r r� � s # r� c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )� Test_sys_excZsys_excc C r: )Nz sys.exc_typezsys.exc_info()[0]r; r< r r r rF rE zTest_sys_exc.test_0c C r: )Nz
sys.exc_valuezsys.exc_info()[1]r; r< r r r rD rE zTest_sys_exc.test_1c C r: )Nzsys.exc_tracebackzsys.exc_info()[2]r; r< r r r rF rE zTest_sys_exc.test_2c C r: )Nzsys.exc_type # Foozsys.exc_info()[0] # Foor; r< r r r rG rE zTest_sys_exc.test_3c C r: )Nzsys. exc_typezsys. exc_info()[0]r; r< r r r rH rE zTest_sys_exc.test_4c C r: )Nzsys .exc_typezsys .exc_info()[0]r; r< r r r rI rE zTest_sys_exc.test_5N)
r5 r) r6 r rF rD rF rG rH rI r r r r r� s r� c @ sp e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS )�
Test_parenZparenc C r: )Nz[i for i in 1, 2 ]z[i for i in (1, 2) ]r; r< r r r rF & rE zTest_paren.test_0c C r: )Nz[i for i in 1, 2, ]z[i for i in (1, 2,) ]r; r< r r r rD + rE zTest_paren.test_1c C r: )Nz[i for i in 1, 2 ]z[i for i in (1, 2) ]r; r< r r r rF 0 rE zTest_paren.test_2c C r: )Nz[i for i in 1, 2 if i]z[i for i in (1, 2) if i]r; r< r r r rG 5 rE zTest_paren.test_3c C r: )Nz[i for i in 1, 2 ]z[i for i in (1, 2) ]r; r< r r r rH : rE zTest_paren.test_4c C r: )Nz(i for i in 1, 2)�(i for i in (1, 2))r; r< r r r rI ? rE zTest_paren.test_5c C r: )Nz(i for i in 1 ,2 if i)z (i for i in (1 ,2) if i)r; r< r r r rJ D rE zTest_paren.test_6c C r[ )Nz[i for i in (1, 2)]r# r\ r r r �test_unchanged_0I r_ zTest_paren.test_unchanged_0c C r[ )Nz[i for i in foo()]r# r\ r r r r^ M r_ zTest_paren.test_unchanged_1c C r[ )Nz [i for i in (1, 2) if nothing]r# r\ r r r r` Q r_ zTest_paren.test_unchanged_2c C r[ )Nr� r# r\ r r r ra U r_ zTest_paren.test_unchanged_3c C r[ )Nz[i for i in m]r# r\ r r r rb Y r_ zTest_paren.test_unchanged_4N)r5 r) r6 r rF rD rF rG rH rI rJ r� r^ r` ra rb r r r r r� # s r� c @ r7 ) �Test_metaclass� metaclassc C sf | � d� | � d� | � d� | � d� | � d� | � d� | � d� d}| � |� d }| � |� d S )
Nzclass X(): passzclass X(object): passz class X(object1, object2): passz(class X(object1, object2, object3): passz class X(metaclass=Meta): passz'class X(b, arg=23, metclass=Meta): passz2class X(b, arg=23, metaclass=Meta, other=42): passzD
class X:
def __metaclass__(self): pass
z1
class X:
a[23] = 74
r# r\ r r r ru a s
z Test_metaclass.test_unchangedc C rp )NzQ
class X:
# hi
__metaclass__ = AppleMeta
zQ
class X(metaclass=AppleMeta):
# hi
pass
zR
class X:
__metaclass__ = Meta
# Bedtime!
zR
class X(metaclass=Meta):
pass
# Bedtime!
r; r< r r r rM v s z Test_metaclass.test_commentsc C s� d}d}| � ||� d}d}| � ||� d}d}| � ||� d}d}| � ||� d }d}| � ||� d
}d }| � ||� d }d
}| � ||� d}d}| � ||� d}d}| � ||� d}d}| � ||� d}d}| � ||� d S )NzK
class X():
__metaclass__ = Q
pass
z8
class X(metaclass=Q):
pass
z"class X(object): __metaclass__ = Qz"class X(object, metaclass=Q): passzW
class X(object):
__metaclass__ = Meta
bar = 7
zF
class X(object, metaclass=Meta):
bar = 7
zJ
class X:
__metaclass__ = Meta; x = 4; g = 23
zD
class X(metaclass=Meta):
x = 4; g = 23
zW
class X(object):
bar = 7
__metaclass__ = Meta
zl
class X():
__metaclass__ = A
__metaclass__ = B
bar = 7
z;
class X(metaclass=B):
bar = 7
z[
class X(clsA, clsB):
__metaclass__ = Meta
bar = 7
zJ
class X(clsA, clsB, metaclass=Meta):
bar = 7
z(class m(a, arg=23): __metaclass__ = Metaz(class m(a, arg=23, metaclass=Meta): passzN
class X(expression(2 + 4)):
__metaclass__ = Meta
zN
class X(expression(2 + 4), metaclass=Meta):
pass
zT
class X(expression(2 + 4), x**4):
__metaclass__ = Meta
zT
class X(expression(2 + 4), x**4, metaclass=Meta):
pass
zT
class X:
__metaclass__ = Meta
save.py = 23
zC
class X(metaclass=Meta):
save.py = 23
r; r< r r r � test_meta� sB zTest_metaclass.test_metaN)r5 r) r6 r ru rM r� r r r r r� ] s
r� c @ r� )
� Test_getcwduZgetcwduc C r� ) Nz
os.getcwduz os.getcwdz os.getcwdu()� os.getcwd()zmeth = os.getcwduzmeth = os.getcwdzos.getcwdu(args)zos.getcwd(args)r; r< r r r r? r� zTest_getcwdu.test_basicc C r: )Nzos.getcwdu() # Foozos.getcwd() # Foor; r< r r r rq rE zTest_getcwdu.test_commentc C r� )Nr� z getcwdu()z os.getcwdb()r# r\ r r r ru r� zTest_getcwdu.test_unchangedc C r: )Nz<
if 1:
os.getcwdu()
z;
if 1:
os.getcwd()
r; r< r r r �test_indentation& r� z Test_getcwdu.test_indentationc C rr )Nz
os .getcwdu()z os .getcwd()z os. getcwduz os. getcwdzos.getcwdu ( )zos.getcwd ( )r; r< r r r �test_multilation1 rt z Test_getcwdu.test_multilationN) r5 r) r6 r r? rq ru r� r� r r r r r� s
r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd S ) �
Test_operator�operatorc C r: )Nzoperator.isCallable(x)z callable(x)r; r< r r r �test_operator_isCallableC rE z&Test_operator.test_operator_isCallablec C rr )Nz operator.sequenceIncludes(x, y)zoperator.contains(x, y)z operator .sequenceIncludes(x, y)zoperator .contains(x, y)z!operator. sequenceIncludes(x, y)zoperator. contains(x, y)r; r< r r r � test_operator_sequenceIncludesH rt z,Test_operator.test_operator_sequenceIncludesc C r: )Nzoperator.isSequenceType(x)z>import collections.abc
isinstance(x, collections.abc.Sequence)r; r< r r r � test_operator_isSequenceTypeU rE z*Test_operator.test_operator_isSequenceTypec C r: )Nzoperator.isMappingType(x)z=import collections.abc
isinstance(x, collections.abc.Mapping)r; r< r r r �test_operator_isMappingTypeZ rE z)Test_operator.test_operator_isMappingTypec C r: )Nzoperator.isNumberType(x)z,import numbers
isinstance(x, numbers.Number)r; r< r r r �test_operator_isNumberType_ rE z(Test_operator.test_operator_isNumberTypec C rr )Nzoperator.repeat(x, n)zoperator.mul(x, n)zoperator .repeat(x, n)zoperator .mul(x, n)zoperator. repeat(x, n)zoperator. mul(x, n)r; r< r r r �test_operator_repeatd rt z"Test_operator.test_operator_repeatc C rr )Nzoperator.irepeat(x, n)zoperator.imul(x, n)zoperator .irepeat(x, n)zoperator .imul(x, n)zoperator. irepeat(x, n)zoperator. imul(x, n)r; r< r r r �test_operator_irepeatq rt z#Test_operator.test_operator_irepeatc C r: )Nz
isCallable(x)z"You should use 'callable(x)' here.r� �r r] �tr r r �test_bare_isCallable~ rE z"Test_operator.test_bare_isCallablec C r: )NzsequenceIncludes(x, y)z.You should use 'operator.contains(x, y)' here.r� r� r r r �test_bare_sequenceIncludes� rE z(Test_operator.test_bare_sequenceIncludesc C r: )NzisSequenceType(z)z>You should use 'isinstance(z, collections.abc.Sequence)' here.r� r� r r r �!test_bare_operator_isSequenceType� rE z/Test_operator.test_bare_operator_isSequenceTypec C r: )NzisMappingType(x)z=You should use 'isinstance(x, collections.abc.Mapping)' here.r� r� r r r � test_bare_operator_isMappingType� rE z.Test_operator.test_bare_operator_isMappingTypec C r: )NzisNumberType(y)z4You should use 'isinstance(y, numbers.Number)' here.r� r� r r r � test_bare_operator_isNumberType� rE z-Test_operator.test_bare_operator_isNumberTypec C r: )Nz repeat(x, n)z)You should use 'operator.mul(x, n)' here.r� r� r r r �test_bare_operator_repeat� rE z'Test_operator.test_bare_operator_repeatc C r: )Nzirepeat(y, 187)z,You should use 'operator.imul(y, 187)' here.r� r� r r r �test_bare_operator_irepeat� rE z(Test_operator.test_bare_operator_irepeatN)r5 r) r6 r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� ? s
r� c @ r� )�
Test_exitfuncZexitfuncc C r: )NzI
import sys
sys.exitfunc = my_atexit
ze
import sys
import atexit
atexit.register(my_atexit)
r; r< r r r r� � r� zTest_exitfunc.test_simplec C r: )NzO
import sys, crumbs
sys.exitfunc = my_func
zY
import sys, crumbs, atexit
atexit.register(my_func)
r; r< r r r �test_names_import� r� z Test_exitfunc.test_names_importc C r: )Nzh
import sys
sys.exitfunc = do(d)/a()+complex(f=23, g=23)*expression
z�
import sys
import atexit
atexit.register(do(d)/a()+complex(f=23, g=23)*expression)
r; r< r r r �test_complex_expression� r� z%Test_exitfunc.test_complex_expressionc C rp )NzN
import sys # Foo
sys.exitfunc = f # Blah
zj
import sys
import atexit # Foo
atexit.register(f) # Blah
zo
import apples, sys, crumbs, larry # Pleasant comments
sys.exitfunc = func
zy
import apples, sys, crumbs, larry, atexit # Pleasant comments
atexit.register(func)
r; r< r r r rM � s zTest_exitfunc.test_commentsc C r: )Nz]
import sys
def f():
sys.exitfunc = func
zz
import sys
import atexit
def f():
atexit.register(func)
r; r< r r r �test_in_a_function� r� z Test_exitfunc.test_in_a_functionc C s d}d}d}| � |||� d S )Nzsys.exitfunc = fzatexit.register(f)zKCan't find sys import; Please add an atexit import at the top of your file.r$ )r r= r> �msgr r r �test_no_sys_import� s z Test_exitfunc.test_no_sys_importc C r[ )Nzf(sys.exitfunc)r# r\ r r r ru � r_ z Test_exitfunc.test_unchangedN) r5 r) r6 r r� r� r� rM r� r� ru r r r r r� � s r� c @ r7 ) � Test_assertsZassertsc C s& g d�}|D ]
\}}| � ||� qd S )N) )zself.assert_(True)�self.assertTrue(True))zself.assertEquals(2, 2)�self.assertEqual(2, 2))zself.assertNotEquals(2, 3)�self.assertNotEqual(2, 3))z self.assertAlmostEquals(2, 3)� self.assertAlmostEqual(2, 3))z self.assertNotAlmostEquals(2, 8)� self.assertNotAlmostEqual(2, 8))zself.failUnlessEqual(2, 2)r� )zself.failIfEqual(2, 3)r� )z self.failUnlessAlmostEqual(2, 3)r� )z self.failIfAlmostEqual(2, 8)r� )zself.failUnless(True)r� )zself.failUnlessRaises(foo)zself.assertRaises(foo))zself.failIf(False)zself.assertFalse(False)r; )r Ztestsr= r> r r r �test_deprecated_names� s �z"Test_asserts.test_deprecated_namesc C r5 )Nzeq = self.assertEqualszeq = self.assertEqualz#self.assertEquals(2, 3, msg="fail")z"self.assertEqual(2, 3, msg="fail")z)self.assertEquals(2, 3, msg="fail") # fooz(self.assertEqual(2, 3, msg="fail") # foozself.assertEquals (2, 3)zself.assertEqual (2, 3)z self.assertEquals (2, 3)z self.assertEqual (2, 3)z0with self.failUnlessRaises(Explosion): explode()z,with self.assertRaises(Explosion): explode()z6with self.failUnlessRaises(Explosion) as cm: explode()z2with self.assertRaises(Explosion) as cm: explode()r; r< r r r �
test_variants s* zTest_asserts.test_variantsc C r$ )Nzself.assertEqualsOnSaturdayz!self.assertEqualsOnSaturday(3, 5)r# r� r r r ru ' r& zTest_asserts.test_unchangedN)r5 r) r6 r r� r� ru r r r r r� � s
r� )C�__doc__r_ rC r r� r r r r Z
lib2to3.testsr ZTestCaser r8 rC rP rm rv r{ r� r� r� r� r� r� r� r� r� r� r r* r1 r7 rI rM r_ rg rj ro r r� r� r� r� r� r� r� r� r r r r r r8 r: r> rB rJ rT r� r� r� r� r� r� r� r� r r r r �<module> s� 9: ;1#j3& }.& gH /+br/ < ]4B.Oh|P Z
=J %": &=cX