Your IP : 216.73.216.213
3
��2ax� � @ s� d dl Z d dlmZ d dlmZ d dlmZmZ d dlm Z G dd� de j
�Z G dd � d e �Z G d
d � d e �Z
G d d
� d
e �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �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 �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 d0d1� d1e �Z G d2d3� d3�Z!G d4d5� d5e e!�Z"G d6d7� d7e e!�Z#G d8d9� d9e e!�Z$G d:d;� d;e �Z%G d<d=� d=e �Z&G d>d?� d?e �Z'G d@dA� dAe �Z(G dBdC� dCe �Z)G dDdE� dEe �Z*G dFdG� dGe �Z+G dHdI� dIe �Z,G dJdK� dKe �Z-G dLdM� dMe �Z.G dNdO� dOe �Z/G dPdQ� dQe �Z0G dRdS� dSe �Z1G dTdU� dUe �Z2G dVdW� dWe �Z3G dXdY� dYe �Z4G dZd[� d[e �Z5G d\d]� d]e �Z6G d^d_� d_e �Z7G d`da� dae �Z8G dbdc� dce �Z9G ddde� dee �Z:G dfdg� dge �Z;G dhdi� die �Z<G djdk� dke �Z=G dldm� dme �Z>G dndo� doe �Z?G dpdq� dqe �Z@G drds� dse �ZAdS )t� N)�chain)�
itemgetter)�pygram�
fixer_util)�supportc @ sL e Zd Zddd�Zdd� Zddd �Zdd
d �Zd d
� Zddd�Zdd� Z dS )�
FixerTestCaseN�lib2to3c C sT |d kr| j g}tj|||�| _g | _d| _x"t| jj| jj�D ] }| j|_ q@W d S )Nz<string>)
�fixerr �get_refactorer�refactor� fixer_log�filenamer Z pre_orderZ
post_order�log)�selfZfix_listZ fixer_pkg�optionsr � r �0/opt/alt/python36/lib64/python3.6/test_fixers.py�setUp s
zFixerTestCase.setUpc C s8 t j|�}t j|�}| jj|| j�}| j|t|�� |S )N)r Zreformatr Zrefactor_stringr
� assertEqual�str)r �before�after�treer r r �_check s
zFixerTestCase._checkFc C s. | j ||�}| j|j� |s*| j| jg � d S )N)r �
assertTrue� was_changedr r )r r r �ignore_warningsr r r r �check# s zFixerTestCase.checkc C s4 | j ||�}| j|dj| j�� |s0| j|j� 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" | j ||� |s | j| jg � d S )N)r r r )r r r r r r r! 2 s zFixerTestCase.unchangedc G s� | j g}|j|� tjd|�}|j� \}}d| j }|rN|d jjj|�rNd S |rn|d jjj|�rn| rnd S | jddj dd� || D ��|f � d S )
Nr Zfix_� z5Fixer run order (%s) is incorrect; %s should be last.z, c S s g | ] }|j j�qS r )� __class__�
__module__)�.0�xr r r �
<listcomp>D s z3FixerTestCase.assert_runs_after.<locals>.<listcomp>���r* )
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)F)F)
�__name__r& � __qualname__r r r r" r# r! r1 r r r r r
s
r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �Test_ne�nec C s d}d}| j ||� d S )Nzif x <> y:
passzif x != y:
pass)r )r �b�ar r r �
test_basicI s zTest_ne.test_basicc C s d}d}| j ||� d S )Nzif x<>y:
passzif x!=y:
pass)r )r r6 r7 r r r �test_no_spacesQ s zTest_ne.test_no_spacesc C s d}d}| j ||� d S )Nz if x<>y<>z:
passz if x!=y!=z:
pass)r )r r6 r7 r r r � test_chainedY s zTest_ne.test_chainedN)r2 r& r3 r r8 r9 r: r r r r r4 F s r4 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 s d}d}| j ||� d S )Nz$x = d.has_key("x") or d.has_key("y")zx = "x" in d or "y" in d)r )r r6 r7 r r r �test_1d s zTest_has_key.test_1c C s d}d}| j ||� d S )Nz x = a.b.c.d.has_key("x") ** 3zx = ("x" in a.b.c.d) ** 3)r )r r6 r7 r r r �test_2i s zTest_has_key.test_2c C s d}d}| j ||� d S )Nz!x = a.b.has_key(1 + 2).__repr__()z x = (1 + 2 in a.b).__repr__())r )r r6 r7 r r r �test_3n s zTest_has_key.test_3c C s d}d}| j ||� d S )Nz,x = a.b.has_key(1 + 2).__repr__() ** -3 ** 4z(x = (1 + 2 in a.b).__repr__() ** -3 ** 4)r )r r6 r7 r r r �test_4s s zTest_has_key.test_4c C s d}d}| j ||� d S )Nzx = a.has_key(f or g)zx = (f or g) in a)r )r r6 r7 r r r �test_5x s zTest_has_key.test_5c C s d}d}| j ||� d S )Nzx = a + b.has_key(c)zx = a + (c in b))r )r r6 r7 r r r �test_6} s zTest_has_key.test_6c C s d}d}| j ||� d S )Nzx = a.has_key(lambda: 12)zx = (lambda: 12) in a)r )r r6 r7 r r r �test_7� s zTest_has_key.test_7c C s d}d}| j ||� d S )Nzx = a.has_key(a for a in b)zx = (a for a in b) in a)r )r r6 r7 r r r �test_8� s zTest_has_key.test_8c C s d}d}| j ||� d S )Nzif not a.has_key(b): passzif b not in a: pass)r )r r6 r7 r r r �test_9� s zTest_has_key.test_9c C s d}d}| j ||� d S )Nz$if not a.has_key(b).__repr__(): passz if not (b in a).__repr__(): pass)r )r r6 r7 r r r �test_10� s zTest_has_key.test_10c C s d}d}| j ||� d S )Nz if not a.has_key(b) ** 2: passzif not (b in a) ** 2: pass)r )r r6 r7 r r r �test_11� s zTest_has_key.test_11N)r2 r& r3 r r< r= r> r? r@ rA rB rC rD rE rF r r r r r; a 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 d6S )7�
Test_applyZapplyc C s d}d}| j ||� d S )Nzx = apply(f, g + h)z
x = f(*g + h))r )r r6 r7 r r r r< � s zTest_apply.test_1c C s d}d}| j ||� d S )Nzy = apply(f, g, h)zy = f(*g, **h))r )r r6 r7 r r r r= � s zTest_apply.test_2c C s d}d}| j ||� d S )Nz z = apply(fs[0], g or h, h or g)z z = fs[0](*g or h, **h or g))r )r r6 r7 r r r r> � s zTest_apply.test_3c C s d}d}| j ||� d S )Nzapply(f, (x, y) + t)zf(*(x, y) + t))r )r r6 r7 r r r r? � s zTest_apply.test_4c C s d}d}| j ||� d S )Nzapply(f, args,)zf(*args))r )r r6 r7 r r r r@ � s zTest_apply.test_5c C s d}d}| j ||� d S )Nzapply(f, args, kwds,)zf(*args, **kwds))r )r r6 r7 r r r rA � s zTest_apply.test_6c C s d}d}| j ||� d S )Nzx = apply(f+g, args)zx = (f+g)(*args))r )r r6 r7 r r r �test_complex_1� s zTest_apply.test_complex_1c C s d}d}| j ||� d S )Nzx = apply(f*g, args)zx = (f*g)(*args))r )r r6 r7 r r r �test_complex_2� s zTest_apply.test_complex_2c C s d}d}| j ||� d S )Nzx = apply(f**g, args)zx = (f**g)(*args))r )r r6 r7 r r r �test_complex_3� s zTest_apply.test_complex_3c C s d}d}| j ||� d S )Nzx = apply(f.g, args)zx = f.g(*args))r )r r6 r7 r r r �test_dotted_name� s zTest_apply.test_dotted_namec C s d}d}| j ||� d S )Nzx = apply(f[x], args)zx = f[x](*args))r )r r6 r7 r r r �test_subscript� s zTest_apply.test_subscriptc C s d}d}| j ||� d S )Nzx = apply(f(), args)zx = f()(*args))r )r r6 r7 r r r � test_call� s zTest_apply.test_callc C s d}d}| j ||� d S )Nz"x = apply(a.b.c.d.e.f, args, kwds)z x = a.b.c.d.e.f(*args, **kwds))r )r r6 r7 r r r � test_extreme� s zTest_apply.test_extremec C s d}d}| j ||� d S )Nz1apply( # foo
f, # bar
args)zf(*args))r )r r6 r7 r r r �test_weird_comments� s z Test_apply.test_weird_commentsc C s d}| j |� d S )Nzapply())r! )r �sr r r �test_unchanged_1� s zTest_apply.test_unchanged_1c C s d}| j |� d S )Nzapply(f))r! )r rP r r r �test_unchanged_2� s zTest_apply.test_unchanged_2c C s d}| j |� d S )Nz apply(f,))r! )r rP r r r �test_unchanged_3� s zTest_apply.test_unchanged_3c C s d}| j |� d S )Nz apply(f, args, kwds, extras))r! )r rP r r r �test_unchanged_4� s zTest_apply.test_unchanged_4c C s d}| j |� d S )Nzapply(f, *args, **kwds))r! )r rP r r r �test_unchanged_5� s zTest_apply.test_unchanged_5c C s d}| j |� d S )Nzapply(f, *args))r! )r rP r r r �test_unchanged_6 s zTest_apply.test_unchanged_6c C s d}| j |� d S )Nzapply(f, **kwds))r! )r rP r r r �test_unchanged_6b s z Test_apply.test_unchanged_6bc C s d}| j |� d S )Nz#apply(func=f, args=args, kwds=kwds))r! )r rP r r r �test_unchanged_7
s zTest_apply.test_unchanged_7c C s d}| j |� d S )Nz apply(f, args=args, kwds=kwds))r! )r rP r r r �test_unchanged_8 s zTest_apply.test_unchanged_8c C s d}| j |� d S )Nzapply(f, args, kwds=kwds))r! )r rP r r r �test_unchanged_9 s zTest_apply.test_unchanged_9c C s d}d}| j ||� d S )Nzapply( f, args, kwds)zf(*args, **kwds))r )r r7 r6 r r r � test_space_1 s zTest_apply.test_space_1c C s d}d}| j ||� d S )Nzapply( f ,args,kwds )zf(*args, **kwds))r )r r7 r6 r r r � test_space_2 s zTest_apply.test_space_2N) r2 r& r3 r r< r= r> r? r@ rA rH rI rJ rK rL rM rN rO rQ rR rS rT rU rV rW rX rY rZ r[ r\ r r r r rG � s6 rG c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) � Test_reload�reloadc C s d}d}| j ||� d S )Nz reload(a)zimport imp
imp.reload(a))r )r r6 r7 r r r �test# s zTest_reload.testc C s, d}d}| j ||� d}d}| j ||� d S )Nzreload( a ) # commentz$import imp
imp.reload( a ) # commentzreload( a ) # commentz%import imp
imp.reload( a ) # comment)r )r r6 r7 r r r � test_comment( s zTest_reload.test_commentc C s@ d}d}| j ||� d}d}| j ||� d}d}| j ||� d S )Nz reload( a )zimport imp
imp.reload( a )z
reload( a)zimport imp
imp.reload( a)z
reload(a )zimport imp
imp.reload(a ))r )r r6 r7 r r r �
test_space2 s zTest_reload.test_spacec C s� d}| j |� d}| j |� d}| j |� d}| j |� d}| j |� d}| j |� d}| j |� d}| j |� d }| j |� 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 rP r r r �test_unchanged? s$
zTest_reload.test_unchangedN)r2 r& r3 r r_ r` ra rb r r r r r] s
r] c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � Test_intern�internc C s@ d}d}| j ||� d}d}| j ||� d}d}| j ||� d S )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 r6 r7 r r r �test_prefix_preservation^ s z$Test_intern.test_prefix_preservationc C s@ d}d}| j ||� d}d}| j ||� d}d}| j ||� d S )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 r6 r7 r r r r_ m s zTest_intern.testc C sJ d}| j |� d}| j |� d}| j |� d}| j |� d}| j |� d S )Nz intern(a=1)z intern(f, g)z
intern(*h)z intern(**i)zintern())r! )r rP r r r rb | s
zTest_intern.test_unchangedN)r2 r& r3 r re r_ rb r r r r rc [ s rc c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) � Test_reduce�reducec C s d}d}| j ||� d S )Nzreduce(a, b, c)z,from functools import reduce
reduce(a, b, c))r )r r6 r7 r r r �test_simple_call� s z Test_reduce.test_simple_callc C s d}d}| j ||� d S )Nzdef x(arg): reduce(sum, [])z8from functools import reduce
def x(arg): reduce(sum, []))r )r r6 r7 r r r �
test_bug_7253� s zTest_reduce.test_bug_7253c C s d}d}| j ||� d S )Nz reduce(lambda x, y: x + y, seq)z<from functools import reduce
reduce(lambda x, y: x + y, seq))r )r r6 r7 r r r �test_call_with_lambda� s z!Test_reduce.test_call_with_lambdac C sJ d}| j |� d}| j |� d}| j |� d}| j |� d}| j |� d S )Nz reduce(a)zreduce(a, b=42)zreduce(a, b, c, d)z reduce(**c)zreduce())r! )r rP r r r rb � s
zTest_reduce.test_unchangedN)r2 r& r3 r rh ri rj rb r r r r rf � s
rf 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$S )%�
Test_print�printc C s d}d}| j ||� d S )Nzprint 1, 1+1, 1+1+1zprint(1, 1+1, 1+1+1))r )r r6 r7 r r r re � s z#Test_print.test_prefix_preservationc C s d}| j |� d}| j |� d S )Nzprint()z print(''))r! )r rP r r r �test_idempotency� s
zTest_print.test_idempotencyc C s: t j| jj_d}| j|� d}| j|� d}| j|� d S )Nzprint(1, 1+1, 1+1+1)zprint()z print(''))r Z!python_grammar_no_print_statementr ZdriverZgrammarr! )r rP r r r �"test_idempotency_print_as_function� s
z-Test_print.test_idempotency_print_as_functionc C s d}d}| j ||� d S )Nzprint 1, 1+1, 1+1+1zprint(1, 1+1, 1+1+1))r )r r6 r7 r r r r< � s zTest_print.test_1c C s d}d}| j ||� d S )Nz
print 1, 2z print(1, 2))r )r r6 r7 r r r r= � s zTest_print.test_2c C s d}d}| j ||� d S )Nrl zprint())r )r r6 r7 r r r r> � s zTest_print.test_3c C s d}d}| j ||� d S )Nzprint whatever; printzprint(whatever); print())r )r r6 r7 r r r r? � s zTest_print.test_4c C s d}d}| j ||� d S )Nzprint; print whatever;zprint(); print(whatever);)r )r r6 r7 r r r r@ � s zTest_print.test_5c C s d}d}| j ||� d S )Nzprint (a, b, c)zprint((a, b, c)))r )r r6 r7 r r r �
test_tuple� s zTest_print.test_tuplec C s d}d}| j ||� d S )Nzprint 1, 2, 3,zprint(1, 2, 3, end=' '))r )r r6 r7 r r r �test_trailing_comma_1� s z Test_print.test_trailing_comma_1c C s d}d}| j ||� d S )Nz print 1, 2,zprint(1, 2, end=' '))r )r r6 r7 r r r �test_trailing_comma_2� s z Test_print.test_trailing_comma_2c C s d}d}| j ||� d S )Nzprint 1,zprint(1, end=' '))r )r r6 r7 r r r �test_trailing_comma_3� s z Test_print.test_trailing_comma_3c C s d}d}| j ||� d S )Nzprint >>sys.stderr, 1, 2, 3z print(1, 2, 3, file=sys.stderr))r )r r6 r7 r r r �!test_vargs_without_trailing_comma� s z,Test_print.test_vargs_without_trailing_commac C s d}d}| j ||� d S )Nzprint >>sys.stderr, 1, 2,z%print(1, 2, end=' ', file=sys.stderr))r )r r6 r7 r r r �test_with_trailing_comma s z#Test_print.test_with_trailing_commac C s d}d}| j ||� d S )Nzprint >>sys.stderr, 1+1zprint(1+1, file=sys.stderr))r )r r6 r7 r r r �test_no_trailing_comma s z!Test_print.test_no_trailing_commac C s d}d}| j ||� d S )Nzprint >> sys.stderrzprint(file=sys.stderr))r )r r6 r7 r r r �test_spaces_before_file
s z"Test_print.test_spaces_before_filec C s& d}| j |� d}d}| j||� d S )Nz<from __future__ import print_function
print('Hai!', end=' ')zprint 'Hello, world!'zprint('Hello, world!'))r! r )r rP r6 r7 r r r � test_with_future_print_function s
z*Test_print.test_with_future_print_functionN)r2 r& r3 r re rm rn r< r= r> r? r@ ro rp rq rr rs rt ru rv rw r r r r rk � s$ rk c @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )� Test_exec�execc C s d}d}| j ||� d S )Nz exec code in ns1, ns2z exec(code, ns1, ns2))r )r r6 r7 r r r re s z"Test_exec.test_prefix_preservationc C s d}d}| j ||� d S )Nz exec codez
exec(code))r )r r6 r7 r r r r8 ! s zTest_exec.test_basicc C s d}d}| j ||� d S )Nzexec code in nszexec(code, ns))r )r r6 r7 r r r �test_with_globals& s zTest_exec.test_with_globalsc C s d}d}| j ||� d S )Nzexec code in ns1, ns2zexec(code, ns1, ns2))r )r r6 r7 r r r �test_with_globals_locals+ s z"Test_exec.test_with_globals_localsc C s d}d}| j ||� d S )Nzexec (a.b()) in nszexec((a.b()), ns))r )r r6 r7 r r r rH 0 s zTest_exec.test_complex_1c C s d}d}| j ||� d S )Nzexec a.b() + c in nszexec(a.b() + c, ns))r )r r6 r7 r r r rI 5 s zTest_exec.test_complex_2c C s d}| j |� d S )Nz
exec(code))r! )r rP r r r rQ < s zTest_exec.test_unchanged_1c C s d}| j |� d S )Nz exec (code))r! )r rP r r r rR @ s zTest_exec.test_unchanged_2c C s d}| j |� d S )Nzexec(code, ns))r! )r rP r r r rS D s zTest_exec.test_unchanged_3c C s d}| j |� d S )Nzexec(code, ns1, ns2))r! )r rP r r r rT H s zTest_exec.test_unchanged_4N)r2 r& r3 r re r8 rz r{ rH rI rQ rR rS rT r r r r rx s rx c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )� Test_repr�reprc C s d}d}| j ||� d S )Nz
x = `1 + 2`zx = repr(1 + 2))r )r r6 r7 r r r re O s z"Test_repr.test_prefix_preservationc C s d}d}| j ||� d S )Nz x = `1 + 2`zx = repr(1 + 2))r )r r6 r7 r r r �
test_simple_1T s zTest_repr.test_simple_1c C s d}d}| j ||� d S )Nzy = `x`z y = repr(x))r )r r6 r7 r r r �
test_simple_2Y s zTest_repr.test_simple_2c C s d}d}| j ||� d S )Nzz = `y`.__repr__()zz = repr(y).__repr__())r )r r6 r7 r r r � test_complex^ s zTest_repr.test_complexc C s d}d}| j ||� d S )Nz
x = `1, 2, 3`zx = repr((1, 2, 3)))r )r r6 r7 r r r ro c s zTest_repr.test_tuplec C s d}d}| j ||� d S )Nz
x = `1 + `2``zx = repr(1 + repr(2)))r )r r6 r7 r r r � test_nestedh s zTest_repr.test_nestedc C s d}d}| j ||� d S )Nzx = `1, 2 + `3, 4``z x = repr((1, 2 + repr((3, 4)))))r )r r6 r7 r r r �test_nested_tuplesm s z Test_repr.test_nested_tuplesN) r2 r& r3 r re r~ r r� ro 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 s d}d}| j ||� d S )Nzq
try:
pass
except (RuntimeError, ImportError), e:
passzs
try:
pass
except (RuntimeError, ImportError) as e:
pass)r )r r6 r7 r r r re u s z$Test_except.test_prefix_preservationc C s d}d}| j ||� d S )NzV
try:
pass
except Foo, e:
passzX
try:
pass
except Foo as e:
pass)r )r r6 r7 r r r � test_simple� s zTest_except.test_simplec C s d}d}| j ||� d S )NzU
try:
pass
except Foo,e:
passzX
try:
pass
except Foo as e:
pass)r )r r6 r7 r r r �"test_simple_no_space_before_target� s z.Test_except.test_simple_no_space_before_targetc C s d}d}| j ||� d S )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:
pass)r )r r6 r7 r r r �test_tuple_unpack� s
z Test_except.test_tuple_unpackc C s d}d}| j ||� d S )Nzn
try:
pass
except (RuntimeError, ImportError), e:
passzp
try:
pass
except (RuntimeError, ImportError) as e:
pass)r )r r6 r7 r r r �test_multi_class� s z Test_except.test_multi_classc C s d}d}| j ||� d S )Nza
try:
pass
except Exception, [a, b]:
passz�
try:
pass
except Exception as xxx_todo_changeme:
[a, b] = xxx_todo_changeme.args
pass)r )r r6 r7 r r r �test_list_unpack� s z Test_except.test_list_unpackc C s d}d}| j ||� d S )Nz_
try:
pass
except Exception, d[5]:
passz�
try:
pass
except Exception as xxx_todo_changeme:
d[5] = xxx_todo_changeme
pass)r )r r6 r7 r r r �test_weird_target_1� s z Test_except.test_weird_target_1c C s d}d}| j ||� d S )Nz`
try:
pass
except Exception, a.foo:
passz�
try:
pass
except Exception as xxx_todo_changeme:
a.foo = xxx_todo_changeme
pass)r )r r6 r7 r r r �test_weird_target_2� s z Test_except.test_weird_target_2c C s d}d}| j ||� d S )Nzb
try:
pass
except Exception, a().foo:
passz�
try:
pass
except Exception as xxx_todo_changeme:
a().foo = xxx_todo_changeme
pass)r )r r6 r7 r r r �test_weird_target_3� s z Test_except.test_weird_target_3c C s d}d}| j ||� d S )Nz�
try:
pass
except Exception, a:
pass
except:
passz�
try:
pass
except Exception as a:
pass
except:
pass)r )r r6 r7 r r r �test_bare_except� s z Test_except.test_bare_exceptc C s d}d}| j ||� d S )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:
pass)r )r r6 r7 r r r �!test_bare_except_and_else_finally
s z-Test_except.test_bare_except_and_else_finallyc C s d}d}| j ||� d S )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:
pass)r )r r6 r7 r r r �+test_multi_fixed_excepts_before_bare_except' s
z7Test_except.test_multi_fixed_excepts_before_bare_exceptc C sT d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� 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 r6 r7 r r r �test_one_line_suites= s z Test_except.test_one_line_suitesc C s d}| j |� d S )NzO
try:
pass
except:
pass)r! )r rP r r r rQ m s z Test_except.test_unchanged_1c C s d}| j |� d S )NzY
try:
pass
except Exception:
pass)r! )r rP r r r rR u s z Test_except.test_unchanged_2c C s d}| j |� d S )Nzg
try:
pass
except (Exception, SystemExit):
pass)r! )r rP r r r rS } s z Test_except.test_unchanged_3N)r2 r& r3 r re r� r� r� r� r� r� r� r� r� r� r� r� rQ rR rS r r r r r� r s"
0r� 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$S )%�
Test_raise�raisec C s d}d}| j ||� d S )Nzraise Exception, 5zraise Exception(5))r )r r6 r7 r r r r8 � s zTest_raise.test_basicc C s, d}d}| j ||� d}d}| j ||� d S )Nzraise Exception,5zraise Exception(5)zraise Exception, 5zraise Exception(5))r )r r6 r7 r r r re � s z#Test_raise.test_prefix_preservationc C s@ d}d}| j ||� d}d}| j ||� d}d}| j ||� d S )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) # foo)r )r r6 r7 r r r �test_with_comments� s z Test_raise.test_with_commentsc C s d}d}| j ||� d S )Nz raise Exception(5), None, tbz%raise Exception(5).with_traceback(tb))r )r r6 r7 r r r �test_None_value� s zTest_raise.test_None_valuec C s d}d}| j ||� d S )Nzraise Exception, (5, 6, 7)zraise Exception(5, 6, 7))r )r r6 r7 r r r �test_tuple_value� s zTest_raise.test_tuple_valuec C s d}d}| j ||� d S )Nzraise E, (5, 6) % (a, b)zraise E((5, 6) % (a, b)))r )r r6 r7 r r r �test_tuple_detection� s z Test_raise.test_tuple_detectionc C s d}d}| j ||� d S )Nz raise (((E1, E2), E3), E4), Vz raise E1(V))r )r r6 r7 r r r �test_tuple_exc_1� s zTest_raise.test_tuple_exc_1c C s d}d}| j ||� d S )Nzraise (E1, (E2, E3), E4), Vz raise E1(V))r )r r6 r7 r r r �test_tuple_exc_2� s zTest_raise.test_tuple_exc_2c C s d}| j |d� d S )Nz raise 'foo'z+Python 3 does not support string exceptions)r# )r rP r r r �test_string_exc� s zTest_raise.test_string_excc C s d}| j |d� d S )Nzraise "foo", 5z+Python 3 does not support string exceptions)r# )r rP r r r �test_string_exc_val� s z Test_raise.test_string_exc_valc C s d}| j |d� d S )Nzraise "foo", 5, 6z+Python 3 does not support string exceptions)r# )r rP r r r �test_string_exc_val_tb� s z!Test_raise.test_string_exc_val_tbc C s d}d}| j ||� d S )Nz4def foo():
raise Exception, 5, 6zCdef foo():
raise Exception(5).with_traceback(6))r )r r6 r7 r r r � test_tb_1� s zTest_raise.test_tb_1c C s d}d}| j ||� d S )Nzhdef foo():
a = 5
raise Exception, 5, 6
b = 6zwdef foo():
a = 5
raise Exception(5).with_traceback(6)
b = 6)r )r r6 r7 r r r � test_tb_2� s zTest_raise.test_tb_2c C s d}d}| j ||� d S )Nz2def foo():
raise Exception,5,6zCdef foo():
raise Exception(5).with_traceback(6))r )r r6 r7 r r r � test_tb_3� s zTest_raise.test_tb_3c C s d}d}| j ||� d S )Nzfdef foo():
a = 5
raise Exception,5,6
b = 6zwdef foo():
a = 5
raise Exception(5).with_traceback(6)
b = 6)r )r r6 r7 r r r � test_tb_4� s zTest_raise.test_tb_4c C s d}d}| j ||� d S )Nz<def foo():
raise Exception, (5, 6, 7), 6zIdef foo():
raise Exception(5, 6, 7).with_traceback(6))r )r r6 r7 r r r � test_tb_5� s zTest_raise.test_tb_5c C s d}d}| j ||� d S )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 = 6)r )r r6 r7 r r r � test_tb_6� s zTest_raise.test_tb_6N)r2 r& r3 r r8 re r� r� r� r� r� r� r� r� r� r� r� r� r� 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 d � Zd d!� Zd"d#� Zd$d%� Zd&S )'�
Test_throw�throwc C s d}d}| j ||� d S )Nzg.throw(Exception, 5)zg.throw(Exception(5)))r )r r6 r7 r r r r< s zTest_throw.test_1c C s d}d}| j ||� d S )Nzg.throw(Exception,5)zg.throw(Exception(5)))r )r r6 r7 r r r r= s zTest_throw.test_2c C s d}d}| j ||� d S )Nz g.throw(Exception, (5, 6, 7))zg.throw(Exception(5, 6, 7)))r )r r6 r7 r r r r> s zTest_throw.test_3c C s d}d}| j ||� d S )Nz5 + g.throw(Exception, 5)z5 + g.throw(Exception(5)))r )r r6 r7 r r r r? s zTest_throw.test_4c C s d}| j |d� d S )Nzg.throw("foo")z+Python 3 does not support string exceptions)r# )r rP r r r � test_warn_1 s zTest_throw.test_warn_1c C s d}| j |d� d S )Nzg.throw("foo", 5)z+Python 3 does not support string exceptions)r# )r rP r r r � test_warn_2! s zTest_throw.test_warn_2c C s d}| j |d� d S )Nzg.throw("foo", 5, 6)z+Python 3 does not support string exceptions)r# )r rP r r r � test_warn_3% s zTest_throw.test_warn_3c C s d}| j |� d S )Nzg.throw(Exception))r! )r rP r r r �test_untouched_1+ s zTest_throw.test_untouched_1c C s d}| j |� d S )Nzg.throw(Exception(5, 6)))r! )r rP r r r �test_untouched_2/ s zTest_throw.test_untouched_2c C s d}| j |� d S )Nz 5 + g.throw(Exception(5, 6)))r! )r rP r r r �test_untouched_33 s zTest_throw.test_untouched_3c C s d}d}| j ||� d S )Nz7def foo():
g.throw(Exception, 5, 6)zFdef foo():
g.throw(Exception(5).with_traceback(6)))r )r r6 r7 r r r r� 9 s zTest_throw.test_tb_1c C s d}d}| j ||� d S )Nzkdef foo():
a = 5
g.throw(Exception, 5, 6)
b = 6zzdef foo():
a = 5
g.throw(Exception(5).with_traceback(6))
b = 6)r )r r6 r7 r r r r� @ s zTest_throw.test_tb_2c C s d}d}| j ||� d S )Nz5def foo():
g.throw(Exception,5,6)zFdef foo():
g.throw(Exception(5).with_traceback(6)))r )r r6 r7 r r r r� K s zTest_throw.test_tb_3c C s d}d}| j ||� d S )Nzidef foo():
a = 5
g.throw(Exception,5,6)
b = 6zzdef foo():
a = 5
g.throw(Exception(5).with_traceback(6))
b = 6)r )r r6 r7 r r r r� R s zTest_throw.test_tb_4c C s d}d}| j ||� d S )Nz?def foo():
g.throw(Exception, (5, 6, 7), 6)zLdef foo():
g.throw(Exception(5, 6, 7).with_traceback(6)))r )r r6 r7 r r r r� ] s zTest_throw.test_tb_5c C s d}d}| j ||� d S )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 = 6)r )r r6 r7 r r r r� d s zTest_throw.test_tb_6c C s d}d}| j ||� d S )Nz;def foo():
a + g.throw(Exception, 5, 6)zJdef foo():
a + g.throw(Exception(5).with_traceback(6)))r )r r6 r7 r r r � test_tb_7o s zTest_throw.test_tb_7c C s d}d}| j ||� d S )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 = 6)r )r r6 r7 r r r � test_tb_8v s zTest_throw.test_tb_8N)r2 r& r3 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 @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
� Test_longZlongc C s d}d}| j ||� d S )Nz x = long(x)z
x = int(x))r )r r6 r7 r r r r< � s zTest_long.test_1c C s d}d}| j ||� d S )Nzy = isinstance(x, long)zy = isinstance(x, int))r )r r6 r7 r r r r= � s zTest_long.test_2c C s d}d}| j ||� d S )Nzz = type(x) in (int, long)zz = type(x) in (int, int))r )r r6 r7 r r r r> � s zTest_long.test_3c C sf d}| j |� d}| j |� d}| j |� d}| j |� d}| j |� d}| j |� d}| j |� d S )Nz long = Truez
s.long = Truezdef long(): passzclass long(): passzdef f(long): passzdef f(g, long): passzdef f(x, long=True): pass)r! )r rP r r r rb � s
zTest_long.test_unchangedc C s d}d}| j ||� d S )Nzx = long( x )zx = int( x ))r )r r6 r7 r r r re � s z"Test_long.test_prefix_preservationN) r2 r& r3 r r< r= r> rb re r r r r r� � s r� c @ s e Zd ZdZdd� Zdd� ZdS )�
Test_execfileZexecfilec C s| d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� d }d
}| j ||� d }d }| j ||� 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 r6 r7 r r r �test_conversion� s$ z Test_execfile.test_conversionc C s, d}d}| j ||� d}d}| j ||� d S )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 r6 r7 r r r � test_spacing� s zTest_execfile.test_spacingN)r2 r& r3 r r� r� r r r r r� � s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �Test_isinstance�
isinstancec C sT d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� d S ) 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 r6 r7 r r r �test_remove_multiple_items� s z*Test_isinstance.test_remove_multiple_itemsc C s d}d}| j ||� d S )Nz4if isinstance( foo(), ( bar, bar, baz )) : passz/if isinstance( foo(), ( bar, baz )) : pass)r )r r6 r7 r r r re � s z(Test_isinstance.test_prefix_preservationc C s | j d� d S )Nzisinstance(x, (str, int)))r! )r r r r rb � s z Test_isinstance.test_unchangedN)r2 r& r3 r r� re rb 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}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� d }d
}| j ||� d }d}| j ||� 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 r6 r7 r r r re � s$ z"Test_dict.test_prefix_preservationc C s| d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� d }d}| j ||� 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() # foo)r )r r6 r7 r r r �test_trailing_comment
s$ z Test_dict.test_trailing_commentc C sJ xDt jD ]:}d| }| j|� d| }| j|� d| }| j|� qW d S )Nzs = %s(d.keys())zs = %s(d.values())zs = %s(d.items()))r �consuming_callsr! )r �wrapperrP r r r rb * s
zTest_dict.test_unchangedc C s, d}d}| j ||� d}d}| j ||� d S )Nzd.keys()zlist(d.keys())za[0].foo().keys()zlist(a[0].foo().keys()))r )r r6 r7 r r r �test_015 s zTest_dict.test_01c C s d}d}| j ||� d S )Nz d.items()zlist(d.items()))r )r r6 r7 r r r �test_02> s zTest_dict.test_02c C s d}d}| j ||� d S )Nz
d.values()zlist(d.values()))r )r r6 r7 r r r �test_03C s zTest_dict.test_03c C s d}d}| j ||� d S )Nz d.iterkeys()ziter(d.keys()))r )r r6 r7 r r r �test_04H s zTest_dict.test_04c C s d}d}| j ||� d S )Nz
d.iteritems()ziter(d.items()))r )r r6 r7 r r r �test_05M s zTest_dict.test_05c C s d}d}| j ||� d S )Nzd.itervalues()ziter(d.values()))r )r r6 r7 r r r �test_06R s zTest_dict.test_06c C s d}| j |� d S )Nzlist(d.keys()))r! )r rP r r r �test_07W s zTest_dict.test_07c C s d}| j |� d S )Nzsorted(d.keys()))r! )r rP r r r �test_08[ s zTest_dict.test_08c C s d}d}| j ||� d S )Nziter(d.keys())ziter(list(d.keys())))r )r r6 r7 r r r �test_09_ s zTest_dict.test_09c C s d}d}| j ||� d S )Nz
foo(d.keys())zfoo(list(d.keys())))r )r r6 r7 r r r rE d s zTest_dict.test_10c C s d}d}| j ||� d S )Nzfor i in d.keys(): print iz for i in list(d.keys()): print i)r )r r6 r7 r r r rF i s zTest_dict.test_11c C s d}d}| j ||� d S )Nz for i in d.iterkeys(): print izfor i in d.keys(): print i)r )r r6 r7 r r r �test_12n s zTest_dict.test_12c C s d}d}| j ||� d S )Nz[i for i in d.keys()]z[i for i in list(d.keys())])r )r r6 r7 r r r �test_13s s zTest_dict.test_13c C s d}d}| j ||� d S )Nz[i for i in d.iterkeys()]z[i for i in d.keys()])r )r r6 r7 r r r �test_14x s zTest_dict.test_14c C s d}d}| j ||� d S )Nz(i for i in d.keys())z(i for i in list(d.keys())))r )r r6 r7 r r r �test_15} s zTest_dict.test_15c C s d}d}| j ||� d S )Nz(i for i in d.iterkeys())z(i for i in d.keys()))r )r r6 r7 r r r �test_16� s zTest_dict.test_16c C s d}d}| j ||� d S )Nziter(d.iterkeys())ziter(d.keys()))r )r r6 r7 r r r �test_17� s zTest_dict.test_17c C s d}d}| j ||� d S )Nzlist(d.iterkeys())zlist(d.keys()))r )r r6 r7 r r r �test_18� s zTest_dict.test_18c C s d}d}| j ||� d S )Nzsorted(d.iterkeys())zsorted(d.keys()))r )r r6 r7 r r r �test_19� s zTest_dict.test_19c C s d}d}| j ||� d S )Nzfoo(d.iterkeys())zfoo(iter(d.keys())))r )r r6 r7 r r r �test_20� s zTest_dict.test_20c C s d}d}| j ||� d S )Nzprint h.iterkeys().next()zprint iter(h.keys()).next())r )r r6 r7 r r r �test_21� s zTest_dict.test_21c C s d}d}| j ||� d S )Nzprint h.keys()[0]zprint list(h.keys())[0])r )r r6 r7 r r r �test_22� s zTest_dict.test_22c C s d}d}| j ||� d S )Nz print list(h.iterkeys().next())z!print list(iter(h.keys()).next()))r )r r6 r7 r r r �test_23� s zTest_dict.test_23c C s d}d}| j ||� d S )Nz for x in h.keys()[0]: print xz#for x in list(h.keys())[0]: print x)r )r r6 r7 r r r �test_24� s zTest_dict.test_24c C s d}d}| j ||� d S )Nz d.viewkeys()zd.keys())r )r r6 r7 r r r �test_25� s zTest_dict.test_25c C s d}d}| j ||� d S )Nz
d.viewitems()z d.items())r )r r6 r7 r r r �test_26� s zTest_dict.test_26c C s d}d}| j ||� d S )Nzd.viewvalues()z
d.values())r )r r6 r7 r r r �test_27� s zTest_dict.test_27c C s d}d}| j ||� d S )Nz[i for i in d.viewkeys()]z[i for i in d.keys()])r )r r6 r7 r r r �test_28� s zTest_dict.test_28c C s d}d}| j ||� d S )Nz(i for i in d.viewkeys())z(i for i in d.keys()))r )r r6 r7 r r r �test_29� s zTest_dict.test_29c C s d}d}| j ||� d S )Nziter(d.viewkeys())ziter(d.keys()))r )r r6 r7 r r r �test_30� s zTest_dict.test_30c C s d}d}| j ||� d S )Nzlist(d.viewkeys())zlist(d.keys()))r )r r6 r7 r r r �test_31� s zTest_dict.test_31c C s d}d}| j ||� d S )Nzsorted(d.viewkeys())zsorted(d.keys()))r )r r6 r7 r r r �test_32� s zTest_dict.test_32N)'r2 r& r3 r re r� rb r� r� r� r� r� r� r� r� r� rE rF 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� � sH 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 s@ d}d}| j ||� d}d}| j ||� d}d}| j ||� d S )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 r6 r7 r r r re � s z$Test_xrange.test_prefix_preservationc C s d}d}| j ||� d S )Nzx = xrange(10)z
x = range(10))r )r r6 r7 r r r �test_single_arg� s zTest_xrange.test_single_argc C s d}d}| j ||� d S )Nzx = xrange(1, 10)zx = range(1, 10))r )r r6 r7 r r r �
test_two_args� s zTest_xrange.test_two_argsc C s d}d}| j ||� d S )Nzx = xrange(0, 10, 2)zx = range(0, 10, 2))r )r r6 r7 r r r �test_three_args� s zTest_xrange.test_three_argsc C sh d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� d }d
}| j ||� 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 r6 r7 r r r �test_wrap_in_list� s z Test_xrange.test_wrap_in_listc C s, d}d}| j ||� d}d}| j ||� d S )Nz for i in xrange(10):
j=izfor i in range(10):
j=iz[i for i in xrange(10)]z[i for i in range(10)])r )r r6 r7 r r r �test_xrange_in_for s z Test_xrange.test_xrange_in_forc C s | j d� | j d� d S )Nzfor i in range(10): passz[i for i in range(10)])r! )r r r r �test_range_in_for s
z Test_xrange.test_range_in_forc C s | j d� d S )Nzx in range(10, 3, 9))r! )r r r r �test_in_contains_test s z!Test_xrange.test_in_contains_testc C s" x t jD ]}| jd| � qW d S )Nza = %s(range(10)))r r� r! )r Zcallr r r �test_in_consuming_context s z%Test_xrange.test_in_consuming_contextN)
r2 r& r3 r re r� r� r� r� r� r� r� r� r r r r r� � s
r� c s$ e Zd Z� fdd�Zdd� Z� ZS )�Test_xrange_with_reducec s t t| �jddg� d S )Nr� rg )�superr� r )r )r% r r r ! s z Test_xrange_with_reduce.setUpc C s d}d}| j ||� d S )Nzreduce(x, xrange(5))z0from functools import reduce
reduce(x, range(5)))r )r r6 r7 r r r �test_double_transform$ s z-Test_xrange_with_reduce.test_double_transform)r2 r& r3 r r� �
__classcell__r r )r% r r� s r� c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )�Test_raw_inputZ raw_inputc C s, d}d}| j ||� d}d}| j ||� d S )Nzx = raw_input( )zx = input( )zx = raw_input( '' )zx = input( '' ))r )r r6 r7 r r r re - s z'Test_raw_input.test_prefix_preservationc C s d}d}| j ||� d S )Nzx = raw_input()z x = input())r )r r6 r7 r r r r< 6 s zTest_raw_input.test_1c C s d}d}| j ||� d S )Nzx = raw_input('')z
x = input(''))r )r r6 r7 r r r r= ; s zTest_raw_input.test_2c C s d}d}| j ||� d S )Nzx = raw_input('prompt')zx = input('prompt'))r )r r6 r7 r r r r> @ s zTest_raw_input.test_3c C s d}d}| j ||� d S )Nzx = raw_input(foo(a) + 6)zx = input(foo(a) + 6))r )r r6 r7 r r r r? E s zTest_raw_input.test_4c C s d}d}| j ||� d S )Nz x = raw_input(invite).split()zx = input(invite).split())r )r r6 r7 r r r r@ J s zTest_raw_input.test_5c C s d}d}| j ||� d S )Nz x = raw_input(invite) . split ()z x = input(invite) . split ())r )r r6 r7 r r r rA O s zTest_raw_input.test_6c C s d}d}| j ||� d S )Nzx = int(raw_input())zx = int(input()))r )r r6 r7 r r r rC T s zTest_raw_input.test_8N) r2 r& r3 r re r< r= r> r? r@ rA rC r r r r r� * s r� c @ s2 e Zd ZdZdddddddgZd d
� Zd d � Zd
S )�Test_funcattrsZ funcattrsZclosure�doc�name�defaults�code�globalsr� c C sL xF| j D ]<}d| }d| }| j||� d| }d| }| j||� qW d S )Nz a.func_%sza.__%s__zself.foo.func_%s.foo_barzself.foo.__%s__.foo_bar)�attrsr )r �attrr6 r7 r r r r_ ^ s zTest_funcattrs.testc C sJ xD| j D ]:}d| }| j|� d| }| j|� d| }| j|� qW d S )Nzfoo(func_%s + 5)z
f(foo.__%s__)zf(foo.__%s__.foo))r� r! )r r� rP r r r rb h s
z Test_funcattrs.test_unchangedN)r2 r& r3 r r� r_ rb r r r r r� Y s
r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �Test_xreadlinesZ
xreadlinesc C s@ d}d}| j ||� d}d}| j ||� d}d}| j ||� d S )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()): pass)r )r r6 r7 r r r rM v s zTest_xreadlines.test_callc C s@ d}d}| j ||� d}d}| j ||� d}d}| j ||� d S )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 r6 r7 r r r �
test_attr_ref� s z Test_xreadlines.test_attr_refc C s< d}| j |� d}| j |� d}| j |� d}| j |� 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 rP r r r rb � s
z Test_xreadlines.test_unchangedN)r2 r& r3 r rM r� rb r r r r r� s s
r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�ImportsFixerTestsc C sT xN| j j� D ]@\}}d| }d| }| j||� d| }d| }| j||� q W d S )Nz import %szimport foo, %s, bar)�modules�itemsr )r �old�newr6 r7 r r r �test_import_module� s z$ImportsFixerTests.test_import_modulec C sp xj| j j� D ]\\}}d| }d| }| j||� d| }d| }| j||� d| }d| }| j||� q W d S )Nzfrom %s import foozfrom %s import foo, barzfrom %s import (yes, no))r� r� r )r r� r� r6 r7 r r r �test_import_from� s z"ImportsFixerTests.test_import_fromc C sT xN| j j� D ]@\}}d| }d| }| j||� d| }d| }| j||� q W d S )Nzimport %s as foo_bar)r� r� r )r r� r� r6 r7 r r r �test_import_module_as� s z'ImportsFixerTests.test_import_module_asc C s8 x2| j j� D ]$\}}d| }d| }| j||� q W d S )Nzfrom %s import foo as bar)r� r� r )r r� r� r6 r7 r r r �test_import_from_as� s z%ImportsFixerTests.test_import_from_asc C s8 x2| j j� D ]$\}}d| }d| }| j||� q W d S )Nzfrom %s import *)r� r� r )r r� r� r6 r7 r r r � test_star� s zImportsFixerTests.test_starc C s� x�| j j� D ]�\}}d||f }d||f }| j||� d||f }d||f }| j||� d|f }| j|� d|||f }d|||f }| j||� d||f }d||f }| j||� q W d 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
)r� r� r r! )r r� r� r6 r7 rP r r r �test_import_module_usage� s
z*ImportsFixerTests.test_import_module_usageN) r2 r& r3 r� r� r� r� r� r� r r r r r� � s
r� c @ s, e Zd ZdZddlmZ dd� Zdd� ZdS ) � Test_imports�imports� )�MAPPINGc C s d}d}| j ||� d S )Nzimport urlparse, cStringIOzimport urllib.parse, io)r )r r6 r7 r r r �test_multiple_imports s z"Test_imports.test_multiple_importsc C s d}d}| j ||� d S )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 r6 r7 r r r �test_multiple_imports_as s z%Test_imports.test_multiple_imports_asN) r2 r& r3 r �fixes.fix_importsr� r� r� r� r r r r r� s r� c @ s e Zd ZdZddlmZ dS )�
Test_imports2�imports2r� )r� N)r2 r& r3 r �fixes.fix_imports2r� r� r r r r r s r c s$ e Zd Z� fdd�Zdd� Z� ZS )�Test_imports_fixer_orderc sV t t| �jddg� ddlm} |j� | _ddlm} xd D ]}|| | j|< q<W d S )
Nr� r r� )r� �dbhash�dumbdbm�dbm�gdbm)r r r r )r� r r r r� �copyr� r )r Zmapping2Zmapping1�key)r% r r r s
z Test_imports_fixer_order.setUpc C s" x dD ]}|| _ | jd� qW d S )Nr� r �import)r� r )r r1 )r Zfixr r r �$test_after_local_imports_refactoring$ s
z=Test_imports_fixer_order.test_after_local_imports_refactoring)r2 r& r3 r r r� r r )r% r r s r c @ sT e Zd ZdZddlmZ dd� Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dS )� Test_urllibZurllibr� )r� c C sH xB| j j� D ]4\}}d| }ddjttd�|�� }| j||� q W d S )Nz import %sz, r )r� r� r �mapr r )r r� �changesr6 r7 r r r r� . s z Test_urllib.test_import_modulec
C s� x�| j j� D ]�\}}g }x�|D ]�\}}xL|D ]D}|j|� d||f }d||f }| j||� d| } | j| � q,W d|dj|�f }d|dj|�f }| j||� ddj|� } | j| � q W d|dj|�f }djdd� |D ��}| j||� q W d S )Nzfrom %s import %szfrom foo import %sz, �
c S s" g | ]\}}d |dj |�f �qS )zfrom %s import %sz, )r )r' r� �membersr r r r) J s z0Test_urllib.test_import_from.<locals>.<listcomp>)r� r� �appendr r! r )
r r� r Z all_membersr� r �memberr6 r7 rP r r r r� 4 s&
z Test_urllib.test_import_fromc C s( x"| j D ]}d| }| j|d� qW d S )Nzimport %s as fooz#This module is now multiple modules)r� r# )r r� rP r r r r� N s z!Test_urllib.test_import_module_asc C s� x�| j j� D ]t\}}xj|D ]b\}}xX|D ]P}d||f }d||f }| j||� d|||f }d|||f }| j||� q(W qW q W d S )Nz from %s import %s as foo_barz from %s import %s as blah, %s)r� r� r )r r� r r� r r r6 r7 r r r r� S s
z Test_urllib.test_import_from_asc C s( x"| j D ]}d| }| j|d� qW d S )Nzfrom %s import *zCannot handle star imports)r� r# )r r� rP r r r r� ^ s zTest_urllib.test_starc C s, d}d}| j ||� d}d}| j ||� d S )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 r6 r7 r r r �
test_indentedc s zTest_urllib.test_indentedc C s� x�| j j� D ]�\}}x�|D ]�\}}x~|D ]v}djdd� | j | D ��}d|||f }d|||f }| j||� d|||||f }d|||||f }| j||� q(W qW q W d S )Nz, c S s g | ] \}}|�qS r r )r' r0 Zmemsr r r r) � s z8Test_urllib.test_import_module_usage.<locals>.<listcomp>z^
import %s
foo(%s.%s)
z`
import %s
%s.%s(%s.%s)
)r� r� r r ) r r� r r� r r Z
new_importr6 r7 r r r r� ~ s
z$Test_urllib.test_import_module_usageN)r2 r& r3 r Zfixes.fix_urllibr� r� r� r� r� r� r� r r� r r r r r
* s r
c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�
Test_input�inputc C s, d}d}| j ||� d}d}| j ||� d S )Nzx = input( )zx = eval(input( ))zx = input( '' )zx = eval(input( '' )))r )r r6 r7 r r r re � s z#Test_input.test_prefix_preservationc C s d}d}| j ||� d S )Nzx = input() # foozx = eval(input()) # foo)r )r r6 r7 r r r r� � s z Test_input.test_trailing_commentc C s. d}| j |� d}| j |� d}| j |� d S )Nzx = eval(input())zx = eval(input(''))zx = eval(input(foo(5) + 9)))r! )r rP r r r rm � s
zTest_input.test_idempotencyc C s d}d}| j ||� d S )Nz x = input()zx = eval(input()))r )r r6 r7 r r r r< � s zTest_input.test_1c C s d}d}| j ||� d S )Nz
x = input('')zx = eval(input('')))r )r r6 r7 r r r r= � s zTest_input.test_2c C s d}d}| j ||� d S )Nzx = input('prompt')zx = eval(input('prompt')))r )r r6 r7 r r r r> � s zTest_input.test_3c C s d}d}| j ||� d S )Nzx = input(foo(5) + 9)zx = eval(input(foo(5) + 9)))r )r r6 r7 r r r r? � s zTest_input.test_4N) r2 r& r3 r re r� rm r< r= 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 d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*S )+�Test_tuple_paramsZ tuple_paramsc C s d}| j |� d S )Nzdef foo(): pass)r! )r rP r r r rQ � s z"Test_tuple_params.test_unchanged_1c C s d}| j |� d S )Nzdef foo(a, b, c): pass)r! )r rP r r r rR � s z"Test_tuple_params.test_unchanged_2c C s d}| j |� d S )Nz def foo(a=3, b=4, c=5): pass)r! )r rP r r r rS � s z"Test_tuple_params.test_unchanged_3c C s d}d}| j ||� d S )Nz8
def foo(((a, b), c)):
x = 5zn
def foo(xxx_todo_changeme):
((a, b), c) = xxx_todo_changeme
x = 5)r )r r6 r7 r r r r< � s zTest_tuple_params.test_1c C s d}d}| j ||� d S )Nz;
def foo(((a, b), c), d):
x = 5zq
def foo(xxx_todo_changeme, d):
((a, b), c) = xxx_todo_changeme
x = 5)r )r r6 r7 r r r r= � s zTest_tuple_params.test_2c C s d}d}| j ||� d S )Nz@
def foo(((a, b), c), d) -> e:
x = 5zv
def foo(xxx_todo_changeme, d) -> e:
((a, b), c) = xxx_todo_changeme
x = 5)r )r r6 r7 r r r r> � s zTest_tuple_params.test_3c C s d}d}| j ||� d S )Nz/
def foo(((a, b), c)): x = 5; y = 7zV
def foo(xxx_todo_changeme): ((a, b), c) = xxx_todo_changeme; x = 5; y = 7)r )r r6 r7 r r r �test_semicolon� s z Test_tuple_params.test_semicolonc C s d}d}| j ||� d S )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 = 5)r )r r6 r7 r r r �
test_keywords� s z Test_tuple_params.test_keywordsc C s d}d}| j ||� d S )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 = 5)r )r r6 r7 r r r � test_varargs
s z Test_tuple_params.test_varargsc C s d}d}| j ||� d S )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 = 5)r )r r6 r7 r r r � test_multi_1 s z Test_tuple_params.test_multi_1c C s d}d}| j ||� d S )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 = 5)r )r r6 r7 r r r � test_multi_2! s z Test_tuple_params.test_multi_2c C s d}d}| j ||� d S )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 = 5)r )r r6 r7 r r r �test_docstring- s z Test_tuple_params.test_docstringc C s d}| j |� d S )Nzlambda x: x + 5)r! )r rP r r r �test_lambda_no_change; s z'Test_tuple_params.test_lambda_no_changec C sT d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� d S )Nzlambda (x): x + 5zlambda x: x + 5zlambda(x): x + 5zlambda ((((x)))): x + 5zlambda((((x)))): x + 5)r )r r6 r7 r r r � test_lambda_parens_single_arg? s z/Test_tuple_params.test_lambda_parens_single_argc C sT d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� d S )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 r6 r7 r r r �test_lambda_simpleP s z$Test_tuple_params.test_lambda_simplec C s, d}d}| j ||� d}d}| j ||� d S )Nzlambda (x,): x + f(x)zlambda x1: x1[0] + f(x1[0])zlambda (((x,))): x + f(x))r )r r6 r7 r r r �test_lambda_one_tuplea s z'Test_tuple_params.test_lambda_one_tuplec C s d}d}| j ||� d S )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 r6 r7 r r r � test_lambda_simple_multi_usej s z.Test_tuple_params.test_lambda_simple_multi_usec C s d}d}| j ||� d S )Nzlambda (x, y): y + xzlambda x_y: x_y[1] + x_y[0])r )r r6 r7 r r r �test_lambda_simple_reverseo s z,Test_tuple_params.test_lambda_simple_reversec C s, d}d}| j ||� d}d}| j ||� d S )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 + z)r )r r6 r7 r r r �test_lambda_nestedt s z$Test_tuple_params.test_lambda_nestedc C s d}d}| j ||� d S )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 r6 r7 r r r � test_lambda_nested_multi_use} s z.Test_tuple_params.test_lambda_nested_multi_useN)r2 r& r3 r rQ rR rS 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 @ s* e Zd ZdZdddgZdd� Zdd� Zd S )
�Test_methodattrsZ methodattrs�funcr �classc C sh xb| j D ]X}d| }|dkr"d}nd| }| j||� d| }|dkrLd}nd| }| j||� qW d S )Nza.im_%sr( za.__self__.__class__za.__%s__zself.foo.im_%s.foo_barz#self.foo.__self__.__class__.foo_barzself.foo.__%s__.foo_bar)r� r )r r� r6 r7 r r r r_ � s zTest_methodattrs.testc C sJ xD| j D ]:}d| }| j|� d| }| j|� d| }| j|� qW d S )Nzfoo(im_%s + 5)z
f(foo.__%s__)zf(foo.__%s__.foo))r� r! )r r� rP r r r rb � s
z Test_methodattrs.test_unchangedN)r2 r& r3 r r� r_ rb 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 s d}d}| j ||� d S )Nz it.next()znext(it))r )r r6 r7 r r r r< � s zTest_next.test_1c C s d}d}| j ||� d S )Nza.b.c.d.next()z
next(a.b.c.d))r )r r6 r7 r r r r= � s zTest_next.test_2c C s d}d}| j ||� d S )Nz(a + b).next()z
next((a + b)))r )r r6 r7 r r r r> � s zTest_next.test_3c C s d}d}| j ||� d S )Nz
a().next()z next(a()))r )r r6 r7 r r r r? � s zTest_next.test_4c C s d}d}| j ||� d S )Nza().next() + bz
next(a()) + b)r )r r6 r7 r r r r@ � s zTest_next.test_5c C s d}d}| j ||� d S )Nzc( a().next() + b)zc( next(a()) + b))r )r r6 r7 r r r rA � s zTest_next.test_6c C s d}d}| j ||� d S )NzU
for a in b:
foo(a)
a.next()
zT
for a in b:
foo(a)
next(a)
)r )r r6 r7 r r r �test_prefix_preservation_1� s z$Test_next.test_prefix_preservation_1c C s d}d}| j ||� d S )Nzq
for a in b:
foo(a) # abc
# def
a.next()
zp
for a in b:
foo(a) # abc
# def
next(a)
)r )r r6 r7 r r r �test_prefix_preservation_2� s z$Test_next.test_prefix_preservation_2c C s 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 r6 r7 r r r �test_prefix_preservation_3� s z$Test_next.test_prefix_preservation_3c C s d}d}| j ||dd� d S )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__()
T)r )r )r r6 r7 r r r �test_prefix_preservation_4� s z$Test_next.test_prefix_preservation_4c C s d}d}| j ||dd� d S )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__())
T)r )r )r r6 r7 r r r �test_prefix_preservation_5� s z$Test_next.test_prefix_preservation_5c C s d}d}| j ||� d S )Nze
for a in b:
foo(foo(a), # abc
a.next())
zd
for a in b:
foo(foo(a), # abc
next(a))
)r )r r6 r7 r r r �test_prefix_preservation_6 s z$Test_next.test_prefix_preservation_6c C s d}d}| j ||� d S )Nz[
class A:
def next(self):
pass
z_
class A:
def __next__(self):
pass
)r )r r6 r7 r r r �
test_method_1 s zTest_next.test_method_1c C s d}d}| j ||� d S )Nzc
class A(object):
def next(self):
pass
zg
class A(object):
def __next__(self):
pass
)r )r r6 r7 r r r �
test_method_2( s zTest_next.test_method_2c C s d}d}| j ||� d S )NzX
class A:
def next(x):
pass
z\
class A:
def __next__(x):
pass
)r )r r6 r7 r r r �
test_method_35 s zTest_next.test_method_3c C s d}d}| j ||� d S )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 r6 r7 r r r �
test_method_4B s zTest_next.test_method_4c C s d}| j |� d S )Nza
class A:
def next(self, a, b):
pass
)r! )r rP r r r �test_method_unchanged[ s z Test_next.test_method_unchangedc C s d}| j |d� d S )Nzy
next = foo
class A:
def next(self, a, b):
pass
z)Calls to builtin next() possibly shadowed)r# )r rP r r r � test_shadowing_assign_simplec s z&Test_next.test_shadowing_assign_simplec C s d}| j |d� d S )Nz~
(next, a) = foo
class A:
def next(self, a, b):
pass
z)Calls to builtin next() possibly shadowed)r# )r rP r r r � test_shadowing_assign_tuple_1m s z'Test_next.test_shadowing_assign_tuple_1c C s d}| j |d� d S )Nz�
(a, (b, (next, c)), a) = foo
class A:
def next(self, a, b):
pass
z)Calls to builtin next() possibly shadowed)r# )r rP r r r � test_shadowing_assign_tuple_2w s z'Test_next.test_shadowing_assign_tuple_2c C s d}| j |d� d S )Nz~
[next, a] = foo
class A:
def next(self, a, b):
pass
z)Calls to builtin next() possibly shadowed)r# )r rP r r r � test_shadowing_assign_list_1� s z&Test_next.test_shadowing_assign_list_1c C s d}| j |d� d S )Nz�
[a, [b, [next, c]], a] = foo
class A:
def next(self, a, b):
pass
z)Calls to builtin next() possibly shadowed)r# )r rP r r r � test_shadowing_assign_list_2� s z&Test_next.test_shadowing_assign_list_2c C s d}| j |d� d S )Nz�
def foo():
__builtin__.next = foo
class A:
def next(self, a, b):
pass
z)Calls to builtin next() possibly shadowed)r# )r rP r r r �test_builtin_assign� s z Test_next.test_builtin_assignc C s d}| j |d� d S )Nz�
def foo():
(a, __builtin__.next) = foo
class A:
def next(self, a, b):
pass
z)Calls to builtin next() possibly shadowed)r# )r rP r r r � test_builtin_assign_in_tuple� s z&Test_next.test_builtin_assign_in_tuplec C s d}| j |d� d S )Nz�
def foo():
[a, __builtin__.next] = foo
class A:
def next(self, a, b):
pass
z)Calls to builtin next() possibly shadowed)r# )r rP r r r �test_builtin_assign_in_list� s z%Test_next.test_builtin_assign_in_listc C s d}| j |� d S )Nz�
def foo():
A.next = foo
class A:
def next(self, a, b):
pass
)r! )r rP r r r �test_assign_to_next� s z Test_next.test_assign_to_nextc C s d}| j |� d S )Nz�
def foo():
(a, A.next) = foo
class A:
def next(self, a, b):
pass
)r! )r rP r r r � test_assign_to_next_in_tuple� s z&Test_next.test_assign_to_next_in_tuplec C s d}| j |� d S )Nz�
def foo():
[a, A.next] = foo
class A:
def next(self, a, b):
pass
)r! )r rP r r r �test_assign_to_next_in_list� s z%Test_next.test_assign_to_next_in_listc C s d}| j |d� d S )Nz�
import foo.bar as next
class A:
def next(self, a, b):
pass
z)Calls to builtin next() possibly shadowed)r# )r rP r r r �test_shadowing_import_1� s z!Test_next.test_shadowing_import_1c C s d}| j |d� d S )Nz�
import bar, bar.foo as next
class A:
def next(self, a, b):
pass
z)Calls to builtin next() possibly shadowed)r# )r rP r r r �test_shadowing_import_2� s z!Test_next.test_shadowing_import_2c C s d}| j |d� d S )Nz�
import bar, bar.foo as next, baz
class A:
def next(self, a, b):
pass
z)Calls to builtin next() possibly shadowed)r# )r rP r r r �test_shadowing_import_3� s z!Test_next.test_shadowing_import_3c C s d}| j |d� d S )Nz�
from x import next
class A:
def next(self, a, b):
pass
z)Calls to builtin next() possibly shadowed)r# )r rP r r r � test_shadowing_import_from_1� s z&Test_next.test_shadowing_import_from_1c C s d}| j |d� d S )Nz�
from x.a import next
class A:
def next(self, a, b):
pass
z)Calls to builtin next() possibly shadowed)r# )r rP r r r � test_shadowing_import_from_2� s z&Test_next.test_shadowing_import_from_2c C s d}| j |d� d S )Nz�
from x import a, next, b
class A:
def next(self, a, b):
pass
z)Calls to builtin next() possibly shadowed)r# )r rP r r r � test_shadowing_import_from_3
s z&Test_next.test_shadowing_import_from_3c C s d}| j |d� d S )Nz�
from x.a import a, next, b
class A:
def next(self, a, b):
pass
z)Calls to builtin next() possibly shadowed)r# )r rP r r r � test_shadowing_import_from_4
s z&Test_next.test_shadowing_import_from_4c C s d}| j |d� d S )Nz�
def next(a):
pass
class A:
def next(self, a, b):
pass
z)Calls to builtin next() possibly shadowed)r# )r rP r r r �test_shadowing_funcdef_1
s z"Test_next.test_shadowing_funcdef_1c C s d}d}| j ||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__()
z)Calls to builtin next() possibly shadowed)r" )r r6 r7 r r r �test_shadowing_funcdef_2(
s
z"Test_next.test_shadowing_funcdef_2c C s d}| j |d� d S )NzW
def f():
global next
next = 5
z)Calls to builtin next() possibly shadowed)r# )r rP r r r �test_shadowing_global_1?
s z!Test_next.test_shadowing_global_1c C s d}| j |d� d S )Nz]
def f():
global a, next, b
next = 5
z)Calls to builtin next() possibly shadowed)r# )r rP r r r �test_shadowing_global_2G
s z!Test_next.test_shadowing_global_2c C s d}| j |d� d S )Nze
for next in it():
pass
b = 5
c = 6
z)Calls to builtin next() possibly shadowed)r# )r rP r r r �test_shadowing_for_simpleO
s z#Test_next.test_shadowing_for_simplec C s d}| j |d� d S )Nzh
for next, b in it():
pass
b = 5
c = 6
z)Calls to builtin next() possibly shadowed)r# )r rP r r r �test_shadowing_for_tuple_1Y
s z$Test_next.test_shadowing_for_tuple_1c C s d}| j |d� d S )Nzp
for a, (next, c), b in it():
pass
b = 5
c = 6
z)Calls to builtin next() possibly shadowed)r# )r rP r r r �test_shadowing_for_tuple_2c
s z$Test_next.test_shadowing_for_tuple_2c C s d}d}| j ||� d S )Nzgnext = g.nextzgnext = g.__next__)r )r r6 r7 r r r �test_noncall_access_1m
s z Test_next.test_noncall_access_1c C s d}d}| j ||� d S )Nz
f(g.next + 5)zf(g.__next__ + 5))r )r r6 r7 r r r �test_noncall_access_2r
s z Test_next.test_noncall_access_2c C s d}d}| j ||� d S )Nzf(g().next + 5)zf(g().__next__ + 5))r )r r6 r7 r r r �test_noncall_access_3w
s z Test_next.test_noncall_access_3N)1r2 r& r3 r r< r= r> r? r@ rA r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ rA rB rC rD rE rF rG rH rI rJ rK rL rM rN rO rP rQ r r r r r) � s\
r) c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
� Test_nonzeroZnonzeroc C s d}d}| j ||� d S )Nzb
class A:
def __nonzero__(self):
pass
z_
class A:
def __bool__(self):
pass
)r )r r6 r7 r r r r<
s zTest_nonzero.test_1c C s d}d}| j ||� d S )Nzj
class A(object):
def __nonzero__(self):
pass
zg
class A(object):
def __bool__(self):
pass
)r )r r6 r7 r r r r= �
s zTest_nonzero.test_2c C s d}| j |� d S )Nzg
class A(object):
def __bool__(self):
pass
)r! )r rP r r r rQ �
s z Test_nonzero.test_unchanged_1c C s d}| j |� d S )Nzm
class A(object):
def __nonzero__(self, a):
pass
)r! )r rP r r r rR �
s z Test_nonzero.test_unchanged_2c C s d}| j |� d S )NzE
def __nonzero__(self):
pass
)r! )r rP r r r �test_unchanged_func�
s z Test_nonzero.test_unchanged_funcN) r2 r& r3 r r< r= rQ rR rS r r r r rR |
s
rR 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 s d}d}| j ||� d S )NZ0755�0o755)r )r r6 r7 r r r � test_octal_1�
s z Test_numliterals.test_octal_1c C s d}d}| j ||� d S )Nza = 12Lza = 12)r )r r6 r7 r r r �test_long_int_1�
s z Test_numliterals.test_long_int_1c C s d}d}| j ||� d S )Nza = 12lza = 12)r )r r6 r7 r r r �test_long_int_2�
s z Test_numliterals.test_long_int_2c C s d}d}| j ||� d S )Nz b = 0x12lzb = 0x12)r )r r6 r7 r r r �
test_long_hex�
s z Test_numliterals.test_long_hexc C s, d}d}| j ||� d}d}| j ||� d S )Nz b = 0x12Lz
b = 0x12zb = 0755 # spamzb = 0o755 # spam)r )r r6 r7 r r r �test_comments_and_spacing�
s z*Test_numliterals.test_comments_and_spacingc C s d}| j |� d S )N�5)r! )r rP r r r �test_unchanged_int�
s z#Test_numliterals.test_unchanged_intc C s d}| j |� d S )Nz5.0)r! )r rP r r r �test_unchanged_float�
s z%Test_numliterals.test_unchanged_floatc C s d}| j |� d S )NrU )r! )r rP r r r �test_unchanged_octal�
s z%Test_numliterals.test_unchanged_octalc C s d}| j |� d S )NZ0xABC)r! )r rP r r r �test_unchanged_hex�
s z#Test_numliterals.test_unchanged_hexc C s d}| j |� d S )Nz5.0e10)r! )r rP r r r �test_unchanged_exp�
s z#Test_numliterals.test_unchanged_expc C s d}| j |� d S )Nz5 + 4j)r! )r rP r r r �test_unchanged_complex_int�
s z+Test_numliterals.test_unchanged_complex_intc C s d}| j |� d S )Nz
5.4 + 4.9j)r! )r rP r r r � test_unchanged_complex_float�
s z-Test_numliterals.test_unchanged_complex_floatc C s d}| j |� d}| j |� d S )NZ4jz4.4j)r! )r rP r r r �test_unchanged_complex_bare�
s
z,Test_numliterals.test_unchanged_complex_bareN)r2 r& r3 r rV rW rX rY rZ r\ r] r^ r_ r` ra rb rc r r r r rT �
s rT c @ s8 e Zd ZdZddiZdd� Zdd� Zd d
� Zd d � Zd
S )� Test_renames�renames�sys�maxint�maxsizec C sZ xTt | jj� �D ]B\}\}}d||f }d||f }| j||� d| }| j|� qW d S )Nzfrom %s import %szfrom foo import %s)�listr� r� r r! )r �modr� r� r6 r7 rP r r r r� �
s z Test_renames.test_import_fromc C sH xBt | jj� �D ]0\}\}}d||f }d||f }| j||� qW d S )Nz from %s import %s as foo_bar)ri r� r� r )r rj r� r� r6 r7 r r r r� s z Test_renames.test_import_from_asc C sP xJt | jj� �D ]8\}\}}d||||f }d||||f }| j||� qW d S )NzJ
import %s
foo(%s, %s.%s)
)ri r� r� r )r rj r� r� r6 r7 r r r r� s z%Test_renames.test_import_module_usagec C sP xJt | jj� �D ]8\}\}}d||||f }d||||f }| j||� qW d S )NzO
from %s import %s
foo(%s, %s)
)ri r� r� r )r rj r� r� r6 r7 r r r �XXX_test_from_import_usage s z'Test_renames.XXX_test_from_import_usageN)rg rh ) r2 r& r3 r r� r� r� r� rk r r r r rd �
s rd c @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )� Test_unicodeZunicodec C s@ d}d}| j ||� d}d}| j ||� d}d}| j ||� d S )Nz unicode( x)zstr( x)z unicode(x )z str(x )z u'h'z 'h')r )r r6 r7 r r r �test_whitespace# s z Test_unicode.test_whitespacec C s d}d}| j ||� d S )Nzunicode(x, y, z)z str(x, y, z))r )r r6 r7 r r r �test_unicode_call0 s z Test_unicode.test_unicode_callc C s d}d}| j ||� d S )Nz unichr(u'h')zchr('h'))r )r r6 r7 r r r � test_unichr5 s zTest_unicode.test_unichrc C s d}d}| j ||� d S )Nzu"x"z"x")r )r r6 r7 r r r �test_unicode_literal_1: s z#Test_unicode.test_unicode_literal_1c C s d}d}| j ||� d S )Nzur'x'zr'x')r )r r6 r7 r r r �test_unicode_literal_2? s z#Test_unicode.test_unicode_literal_2c C s d}d}| j ||� d S )Nz
UR'''x''' z R'''x''' )r )r r6 r7 r r r �test_unicode_literal_3D s z#Test_unicode.test_unicode_literal_3c C s, d}d}| j ||� d}d}| j ||� d S )Nz'\\\u20ac\U0001d121\\u20ac'z '\\\\u20ac\\U0001d121\\u20ac'z r'\\\u20ac\U0001d121\\u20ac')r )r r6 r7 r r r � test_native_literal_escape_uI s z)Test_unicode.test_native_literal_escape_uc C s, d}d}| j ||� d}d}| j ||� d S )Nz b'\\\u20ac\U0001d121\\u20ac'z br'\\\u20ac\U0001d121\\u20ac')r )r r6 r7 r r r �test_bytes_literal_escape_uR s z(Test_unicode.test_bytes_literal_escape_uc C s, d}d}| j ||� d}d}| j ||� d S )Nz u'\\\u20ac\U0001d121\\u20ac'z'\\\u20ac\U0001d121\\u20ac'z ur'\\\u20ac\U0001d121\\u20ac'z r'\\\u20ac\U0001d121\\u20ac')r )r r6 r7 r r r � test_unicode_literal_escape_u[ s z*Test_unicode.test_unicode_literal_escape_uc C s@ d}|d }|d }| j ||� |d }|d }| j ||� d S )Nz(from __future__ import unicode_literals
z'\\\u20ac\U0001d121\\u20ac'z r'\\\u20ac\U0001d121\\u20ac')r )r �fr6 r7 r r r �$test_native_unicode_literal_escape_ud s z1Test_unicode.test_native_unicode_literal_escape_uN)r2 r& r3 r rm rn ro rp rq rr rs rt ru rw r r r r rl s
rl c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
� Test_filter�filterc C s, d}d}| j ||� d}d}| j ||� d S )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 r6 r7 r r r re r s z$Test_filter.test_prefix_preservationc C sT d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� 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)))r )r r6 r7 r r r �test_filter_basic{ s z Test_filter.test_filter_basicc C sT d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� 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 r6 r7 r r r �test_filter_trailers� s z Test_filter.test_filter_trailersc C s 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 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 r7 r r r �test_filter_nochange� sL
z Test_filter.test_filter_nochangec C s4 d}| j |� d}d}| j||� d}| j |� 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! r )r r7 r6 r r r �test_future_builtins� s
z Test_filter.test_future_builtinsN) r2 r& r3 r re rz r{ r| r} r r r r rx o s (rx c sX e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z � Z S )�Test_mapr c s& | j d| |� tt| �j||� d S )Nz!from future_builtins import map; )r! r� r~ r )r r6 r7 )r% r r r � s zTest_map.checkc C s d}d}| j ||� d S )Nz x = map( f, 'abc' )z#x = list(map( f, 'abc' )))r )r r6 r7 r r r re � s z!Test_map.test_prefix_preservationc C sT d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� d S ) 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 r6 r7 r r r �test_map_trailers� s zTest_map.test_map_trailersc C s d}d}| j ||� d S )Nzx = map(f, 'abc') # fooz!x = list(map(f, 'abc')) # foo)r )r r6 r7 r r r r� � s z Test_map.test_trailing_commentc C s d}| j |d� d S )Nzx = map(None, a, b, c)z5cannot convert map(None, ...) with multiple arguments)r# )r rP r r r �!test_None_with_multiple_arguments� s z*Test_map.test_None_with_multiple_argumentsc C s~ d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� d }d}| j ||� d
}d }| j||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 r6 r7 r r r �test_map_basic� s$ zTest_map.test_map_basicc C s 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 S )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 r7 r r r �test_map_nochange sL
zTest_map.test_map_nochangec C s4 d}| j |� d}d}| j||� d}| j |� d S )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! r )r r7 r6 r r r r} C s
z Test_map.test_future_builtins)
r2 r& r3 r r re 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 s& | j d| |� tt| �j||� d S )Nz!from future_builtins import zip; )r! r� r� r )r r6 r7 )r% r r r Q s zTest_zip.checkc C s@ d}d}| j ||� d}d}| j ||� d}d}| j ||� d S )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 r6 r7 r r r �test_zip_basicU s zTest_zip.test_zip_basicc C s, d}d}| j ||� d}d}| j ||� d S )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 r6 r7 r r r �test_zip_trailersb s zTest_zip.test_zip_trailersc C s 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 S )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 r7 r r r �test_zip_nochangek sL
zTest_zip.test_zip_nochangec C s4 d}| j |� d}d}| j||� d}| j |� d S )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! r )r r7 r6 r r r r} � s
z Test_zip.test_future_builtins)
r2 r& r3 r r r� r� r� r} r� r r )r% r r� N s
(r� c @ s e Zd ZdZdd� ZdS )�Test_standarderrorZ
standarderrorc C s@ d}d}| j ||� d}d}| j ||� d}d}| j ||� d S )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 r6 r7 r r r r_ � s zTest_standarderror.testN)r2 r& r3 r r_ r r r r r� � s r� c @ s e Zd ZdZdd� ZdS )�
Test_types�typesc C s� d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� d }d}| j ||� d
}d }| j ||� d }d
}| j ||� d S )Nztypes.StringType�bytesztypes.DictTyper� ztypes . IntType�intztypes.ListTyperi ztypes.LongTypeztypes.NoneTypez
type(None)ztypes.StringTypesz(str,))r )r r6 r7 r r r �test_basic_types_convert� s* z#Test_types.test_basic_types_convertN)r2 r& r3 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 d � Zd d!� Zd"d#� Zd$d%� Zd&S )'� Test_idiomsZidiomsc C s@ d}d}| j ||� d}d}| j ||� d}d}| j ||� d S )Nzwhile 1: foo()zwhile True: foo()zwhile 1: foo()zwhile True: foo()z8
while 1:
foo()
z;
while True:
foo()
)r )r r6 r7 r r r �
test_while� s zTest_idioms.test_whilec C s< d}| j |� d}| j |� d}| j |� d}| j |� d S )Nzwhile 11: foo()zwhile 0: foo()zwhile foo(): foo()zwhile []: foo())r! )r rP r r r �test_while_unchanged� s
z Test_idioms.test_while_unchangedc C s, d}d}| j ||� d}d}| j ||� d S )Nz type(x) == Tzisinstance(x, T)zif type(x) == T: passzif isinstance(x, T): pass)r )r r6 r7 r r r �test_eq_simple� s zTest_idioms.test_eq_simplec C s, d}d}| j ||� d}d}| j ||� d S )Nz T == type(x)zisinstance(x, T)zif T == type(x): passzif isinstance(x, T): pass)r )r r6 r7 r r r �test_eq_reverse� s zTest_idioms.test_eq_reversec C s, d}d}| j ||� d}d}| j ||� d S )Nztype(x+y) == d.get('T')zisinstance(x+y, d.get('T'))z type( x + y) == d.get('T')z isinstance(x + y, d.get('T')))r )r r6 r7 r r r �test_eq_expression
s z Test_idioms.test_eq_expressionc C s, d}d}| j ||� d}d}| j ||� d S )Nz type(x) is Tzisinstance(x, T)zif type(x) is T: passzif isinstance(x, T): pass)r )r r6 r7 r r r �test_is_simple
s zTest_idioms.test_is_simplec C s, d}d}| j ||� d}d}| j ||� d S )Nz T is type(x)zisinstance(x, T)zif T is type(x): passzif isinstance(x, T): pass)r )r r6 r7 r r r �test_is_reverse
s zTest_idioms.test_is_reversec C s, d}d}| j ||� d}d}| j ||� d S )Nztype(x+y) is d.get('T')zisinstance(x+y, d.get('T'))z type( x + y) is d.get('T')z isinstance(x + y, d.get('T')))r )r r6 r7 r r r �test_is_expression
s z Test_idioms.test_is_expressionc C s, d}d}| j ||� d}d}| j ||� d S )Nztype(x) is not Tznot isinstance(x, T)zif type(x) is not T: passz if not isinstance(x, T): pass)r )r r6 r7 r r r �test_is_not_simple'
s z Test_idioms.test_is_not_simplec C s, d}d}| j ||� d}d}| j ||� d S )NzT is not type(x)znot isinstance(x, T)zif T is not type(x): passz if not isinstance(x, T): pass)r )r r6 r7 r r r �test_is_not_reverse0
s z Test_idioms.test_is_not_reversec C s, d}d}| j ||� d}d}| j ||� d S )Nztype(x+y) is not d.get('T')z not isinstance(x+y, d.get('T'))z"type( x + y) is not d.get('T')z#not isinstance(x + y, d.get('T')))r )r r6 r7 r r r �test_is_not_expression9
s z"Test_idioms.test_is_not_expressionc C s, d}d}| j ||� d}d}| j ||� d S )Nz type(x) != Tznot isinstance(x, T)zif type(x) != T: passz if not isinstance(x, T): pass)r )r r6 r7 r r r �test_ne_simpleB
s zTest_idioms.test_ne_simplec C s, d}d}| j ||� d}d}| j ||� d S )Nz T != type(x)znot isinstance(x, T)zif T != type(x): passz if not isinstance(x, T): pass)r )r r6 r7 r r r �test_ne_reverseK
s zTest_idioms.test_ne_reversec C s, d}d}| j ||� d}d}| j ||� d S )Nztype(x+y) != d.get('T')z not isinstance(x+y, d.get('T'))z type( x + y) != d.get('T')z#not isinstance(x + y, d.get('T')))r )r r6 r7 r r r �test_ne_expressionT
s z Test_idioms.test_ne_expressionc C s d}| j |� d S )Nztype(x).__name__)r! )r r7 r r r �test_type_unchanged]
s z Test_idioms.test_type_unchangedc C s� d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� d }d
}| j ||� d }d }| j ||� d
}d}| j ||� d}d}| j ||� d S )NzM
v = list(t)
v.sort()
foo(v)
z:
v = sorted(t)
foo(v)
zV
v = list(foo(b) + d)
v.sort()
foo(v)
zC
v = sorted(foo(b) + d)
foo(v)
zn
while x:
v = list(t)
v.sort()
foo(v)
zW
while x:
v = sorted(t)
foo(v)
z_
v = list(t)
# foo
v.sort()
foo(v)
zL
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 comments)r )r r6 r7 r r r �test_sort_list_calla
s0 z Test_idioms.test_sort_list_callc C s� d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� d }d
}| j ||� d }d }| j ||� d
}d}| j ||� d S )NzG
v = t
v.sort()
foo(v)
z:
v = sorted(t)
foo(v)
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)
zC
v = sorted(foo(b) + d)
foo(v)
zh
while x:
v = t
v.sort()
foo(v)
zW
while x:
v = sorted(t)
foo(v)
zY
v = t
# foo
v.sort()
foo(v)
zL
v = sorted(t)
# foo
foo(v)
zI
v = t
v.sort()
foo(v)
z<
v = sorted(t)
foo(v)
)r )r r6 r7 r r r �test_sort_simple_expr�
s* z!Test_idioms.test_sort_simple_exprc C s d}| j |� d}| j |� d S )NzM
v = list(t)
w.sort()
foo(w)
zN
v = list(t)
v.sort(u)
foo(v)
)r! )r rP r r r �test_sort_unchanged s
z Test_idioms.test_sort_unchangedN)r2 r& r3 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&
dRr� c @ s e Zd ZdZdd� ZdS )�Test_basestringZ
basestringc C s d}d}| j ||� d S )Nzisinstance(x, basestring)zisinstance(x, str))r )r r6 r7 r r r �test_basestring) s z Test_basestring.test_basestringN)r2 r& r3 r r� r r r r r� & s r� c @ s e Zd ZdZdd� Zdd� ZdS )� Test_buffer�bufferc C s d}d}| j ||� d S )Nz
x = buffer(y)zx = memoryview(y))r )r r6 r7 r r r � test_buffer1 s zTest_buffer.test_bufferc C s d}d}| j ||� d S )Nzbuffer(y)[4:5]zmemoryview(y)[4:5])r )r r6 r7 r r r � test_slicing6 s zTest_buffer.test_slicingN)r2 r& r3 r r� r� r r r r r� . s r� c @ s e Zd ZdZdd� Zdd� ZdS )� Test_futureZfuturec C s@ d}d}| j ||� d}d}| j ||� d}d}| j ||� d S )Nz from __future__ import bracesr z'# comment
from __future__ import bracesz
# comment
z'from __future__ import braces
# commentz
# comment)r )r r6 r7 r r r � test_future> s zTest_future.test_futurec C s | j d� d S )Nrl )r1 )r r r r �test_run_orderK s zTest_future.test_run_orderN)r2 r& r3 r r� r� r r r r r� ; s
r� c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )�Test_itertools� itertoolsc C sD x>dD ]6}x0dD ](}||d | }|| }| j ||� qW qW d S ) N�
itertools.r r ry r� �i)r� r )r ry r� )r )r r r r� rv r6 r7 r r r �checkallQ s
zTest_itertools.checkallc C s d}d}| j ||� d S )Nzitertools.izip(a, b)z zip(a, b))r )r r6 r7 r r r �test_0[ s zTest_itertools.test_0c C s d}d}| j ||� d S )Nz%s(f, a))r� )r r6 r7 r r r r< b s zTest_itertools.test_1c C s, d}d}| j ||� d}d}| j ||� d S )Nz itertools.ifilterfalse(a, b)zitertools.filterfalse(a, b)z itertools.izip_longest(a, b)zitertools.zip_longest(a, b))r )r r6 r7 r r r �test_qualifiedg s z Test_itertools.test_qualifiedc C s, d}d}| j ||� d}d}| j ||� d S )Nzifilterfalse(a, b)zfilterfalse(a, b)zizip_longest(a, b)zzip_longest(a, b))r )r r6 r7 r r r r= p s zTest_itertools.test_2c C s d}d}| j ||� d S )Nz %s(f, a))r� )r r6 r7 r r r r[ y s zTest_itertools.test_space_1c C s, d}d}| j ||� d}d}| j ||� d S )Nz itertools.ifilterfalse(a, b)z itertools.filterfalse(a, b)z itertools.izip_longest(a, b)z itertools.zip_longest(a, b))r )r r6 r7 r r r r\ ~ s zTest_itertools.test_space_2c C s | j ddd� d S )Nr r� ry )r1 )r r r r r� � s z Test_itertools.test_run_orderN) r2 r& r3 r r� r� r< r� r= r[ r\ r� r r r r r� N s
r� c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�Test_itertools_importsZitertools_importsc C s@ d}d}| j ||� d}d}| j ||� d}d}| j ||� d S )Nz%from itertools import imap, izip, foozfrom itertools import fooz*from itertools import bar, imap, izip, fooz from itertools import bar, fooz'from itertools import chain, imap, izipzfrom itertools import chain)r )r r6 r7 r r r � test_reduced� s z#Test_itertools_imports.test_reducedc C s d}d}| j ||� d S )Nz%#foo
from itertools import imap, izipz#foo
)r )r r6 r7 r r r �
test_comments� s z$Test_itertools_imports.test_commentsc C s, d}d}| j ||� d}d}| j ||� d S )Nz from itertools import imap, izipr zfrom itertools import izip)r )r r6 r7 r r r � test_none� s z Test_itertools_imports.test_nonec C sb d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� d}| j|� 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 r6 r7 rP r r r �test_import_as� s z%Test_itertools_imports.test_import_asc C sr xld D ]d}d|f }d|f }| j ||� d|f }d|f }| j ||� d|f }d|f }| j ||� qW d 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, foo)r� r� )r )r r� r6 r7 r r r � test_ifilter_and_zip_longest� s
z3Test_itertools_imports.test_ifilter_and_zip_longestc C s d}| j |� d S )Nzfrom itertools import *)r! )r rP r r r �test_import_star� s z'Test_itertools_imports.test_import_starc C s d}| j |� d S )Nzfrom itertools import foo)r! )r rP r r r rb � s z%Test_itertools_imports.test_unchangedN) r2 r& r3 r r� r� r� r� r� r� rb r r r r r� � s
r� 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_importr c sD t t� �j� g � _t� � _d� _� fdd�}ddlm} ||_ d S )NTc s � j j| � � jp| � jkS )N)�
files_checkedr �
always_exists�
present_files)r� )r r r � fake_exists� s z&Test_import.setUp.<locals>.fake_existsr )�
fix_import)
r� r� r r� �setr� r� �
lib2to3.fixesr� �exists)r r� r� )r% )r r r � s zTest_import.setUpc C s ddl m} tjj|_d S )Nr )r� )r� r� �os�pathr� )r r� r r r �tearDown� s zTest_import.tearDownc s2 d| _ tt| �j||� d| _ tt| �j|� d S )NTF)r� r� r� r r! )r r6 r7 )r% r r �
check_both� s zTest_import.check_bothc s� dd� }d| _ tdg�| _dtjjdddd f}|d
�d |d �f}xv|D ]n� g | _� | _| jd
� tjj � �r~tjj � �d � nd� t� fdd�|D ��}|j
d� | j t| j�|� qFW d S )Nc S s t jjj| jd��S )N�/)r� r� �pathsepr �split)r� r r r �p� s z)Test_import.test_files_checked.<locals>.pFz __init__.pyz.pyz.pycz.soz.slz.pydz
/spam/eggs.pyzni.pyz../../shrubbery.pyz
import jamz/jamZjamc 3 s | ]}� | V qd S )Nr )r' Zext)r� r r � <genexpr> s z1Test_import.test_files_checked.<locals>.<genexpr>) r� r� r� r� r� �sepr� r
r! �dirname�addr )r r� Zexpected_extensionsZ
names_to_testZexpected_checksr )r� r �test_files_checked� s
z Test_import.test_files_checkedc C s$ d}d| _ tdg�| _| j|� d S )Nz
import barFzbar.py)r� r� r� r! )r rP r r r �test_not_in_package s z Test_import.test_not_in_packagec C s& d}d| _ tddg�| _| j|� d S )Nz1from __future__ import absolute_import
import barFz __init__.pyzbar.py)r� r� r� r! )r rP r r r �!test_with_absolute_import_enabled s z-Test_import.test_with_absolute_import_enabledc C s, d}d}d| _ tddg�| _| j||� d S )Nz
import barzfrom . import barFz __init__.pyzbar.py)r� r� r� r )r r6 r7 r r r �test_in_package s
zTest_import.test_in_packagec C s4 d}d}d| _ tddtjj g�| _| j||� d S )Nz
import barzfrom . import barFz __init__.pyZbar)r� r� r� r� r� r� r )r r6 r7 r r r �test_import_from_package s
z$Test_import.test_import_from_packagec C s d}| j |� d S )Nzfrom . import bar)r! )r rP r r r � test_already_relative_import" s z(Test_import.test_already_relative_importc C s d}d}| j ||� d S )Nzimport bar # Foozfrom . import bar # Foo)r )r r6 r7 r r r �test_comments_and_indent& s z$Test_import.test_comments_and_indentc C s@ d}d}| j ||� d}d}| j ||� d}d}| j ||� d S )Nzfrom foo import bar, bazzfrom .foo import bar, bazzfrom foo import barzfrom .foo import barzfrom foo import (bar, baz)zfrom .foo import (bar, baz))r� )r r6 r7 r r r � test_from+ s zTest_import.test_fromc C s d}d}| j ||� d S )Nzfrom green.eggs import hamzfrom .green.eggs import ham)r� )r r6 r7 r r r �test_dotted_from8 s z Test_import.test_dotted_fromc C s d}d}| j ||� d S )Nz"from green.eggs import ham as spamz#from .green.eggs import ham as spam)r� )r r6 r7 r r r � test_from_as= s zTest_import.test_from_asc C sT d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� d S ) Nz
import foozfrom . import foozimport foo, barzfrom . import foo, barzimport foo, bar, xzfrom . import foo, bar, xzimport x, y, zzfrom . import x, y, z)r� )r r6 r7 r r r � test_importB s zTest_import.test_importc C s, d}d}| j ||� d}d}| j ||� d S )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 d)r� )r r6 r7 r r r r� S s zTest_import.test_import_asc C s( d| _ tddg�| _d}| j|d� d S )NFzfoo.pyz __init__.pyzimport foo, barz#absolute and local imports together)r� r� r� r# )r rP r r r �test_local_and_absolute\ s z#Test_import.test_local_and_absolutec C s d}d}| j ||� d S )Nzimport foo.barzfrom . import foo.bar)r� )r r6 r7 r r r �test_dotted_importc s z Test_import.test_dotted_importc C s d}d}| j ||� d S )Nzimport foo.bar as bangz from . import foo.bar as bang)r� )r r6 r7 r r r �test_dotted_import_ash s z!Test_import.test_dotted_import_asc C s d}d}| j ||� d S )Nz1
# prefix
import foo.bar
z8
# prefix
from . import foo.bar
)r� )r r6 r7 r r r � test_prefixm s zTest_import.test_prefix)r2 r& r3 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 @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�Test_set_literalZ set_literalc C s� d}d}| j ||� d}d}| j ||� d}d}| j ||� d}| j ||� d}d}| j ||� d }| j ||� d
}d }| j ||� d }d
}| j ||� d}d}| j ||� d S )Nzset([1, 2, 3])z {1, 2, 3}zset((1, 2, 3))z set((1,))z{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 r6 r7 r r r r8 } s2 zTest_set_literal.test_basicc C sT d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� d S ) 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 r6 r7 r r r �test_listcomps� s z Test_set_literal.test_listcompsc C s� d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� d }d
}| j ||� d }d }| j ||� d
}d}| j ||� d S )Nz set( [1, 2])z{1, 2}z
set([1 , 2])z{1 , 2}z
set([ 1 ])z{ 1 }z
set( [1] )z{1}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 r6 r7 r r r rm � s* z Test_set_literal.test_whitespacec C s, d}d}| j ||� d}d}| j ||� d S )Nzset((1, 2)) # Hiz {1, 2} # HizZ
# Foo
set( # Bar
(1, 2)
)
z2
# Foo
{1, 2}
)r )r r6 r7 r r r r� � s z Test_set_literal.test_commentsc C sX d}| j |� d}| j |� d}| j |� d}| j |� d}| j |� d}| j |� 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 rP r r r rb � s
z Test_set_literal.test_unchangedN) r2 r& r3 r r8 r� rm r� rb r r r r r� y s # r� c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )� Test_sys_excZsys_excc C s d}d}| j ||� d S )Nz sys.exc_typezsys.exc_info()[0])r )r r6 r7 r r r r� � s zTest_sys_exc.test_0c C s d}d}| j ||� d S )Nz
sys.exc_valuezsys.exc_info()[1])r )r r6 r7 r r r r< s zTest_sys_exc.test_1c C s d}d}| j ||� d S )Nzsys.exc_tracebackzsys.exc_info()[2])r )r r6 r7 r r r r= s zTest_sys_exc.test_2c C s d}d}| j ||� d S )Nzsys.exc_type # Foozsys.exc_info()[0] # Foo)r )r r6 r7 r r r r>
s zTest_sys_exc.test_3c C s d}d}| j ||� d S )Nzsys. exc_typezsys. exc_info()[0])r )r r6 r7 r r r r? s zTest_sys_exc.test_4c C s d}d}| j ||� d S )Nzsys .exc_typezsys .exc_info()[0])r )r r6 r7 r r r r@ s zTest_sys_exc.test_5N)
r2 r& r3 r r� r< r= r> r? r@ 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 s d}d}| j ||� d S )Nz[i for i in 1, 2 ]z[i for i in (1, 2) ])r )r r6 r7 r r r r� s zTest_paren.test_0c C s d}d}| j ||� d S )Nz[i for i in 1, 2, ]z[i for i in (1, 2,) ])r )r r6 r7 r r r r< " s zTest_paren.test_1c C s d}d}| j ||� d S )Nz[i for i in 1, 2 ]z[i for i in (1, 2) ])r )r r6 r7 r r r r= ' s zTest_paren.test_2c C s d}d}| j ||� d S )Nz[i for i in 1, 2 if i]z[i for i in (1, 2) if i])r )r r6 r7 r r r r> , s zTest_paren.test_3c C s d}d}| j ||� d S )Nz[i for i in 1, 2 ]z[i for i in (1, 2) ])r )r r6 r7 r r r r? 1 s zTest_paren.test_4c C s d}d}| j ||� d S )Nz(i for i in 1, 2)z(i for i in (1, 2)))r )r r6 r7 r r r r@ 6 s zTest_paren.test_5c C s d}d}| j ||� d S )Nz(i for i in 1 ,2 if i)z (i for i in (1 ,2) if i))r )r r6 r7 r r r rA ; s zTest_paren.test_6c C s d}| j |� d S )Nz[i for i in (1, 2)])r! )r rP r r r �test_unchanged_0@ s zTest_paren.test_unchanged_0c C s d}| j |� d S )Nz[i for i in foo()])r! )r rP r r r rQ D s zTest_paren.test_unchanged_1c C s d}| j |� d S )Nz [i for i in (1, 2) if nothing])r! )r rP r r r rR H s zTest_paren.test_unchanged_2c C s d}| j |� d S )Nz(i for i in (1, 2)))r! )r rP r r r rS L s zTest_paren.test_unchanged_3c C s d}| j |� d S )Nz[i for i in m])r! )r rP r r r rT P s zTest_paren.test_unchanged_4N)r2 r& r3 r r� r< r= r> r? r@ rA r� rQ rR rS rT r r r r r� s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �Test_metaclass� metaclassc C sf | j d� | j d� | j d� | j d� | j d� | j d� | j d� d}| j |� d }| j |� 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 rP r r r rb X s
z Test_metaclass.test_unchangedc C s, d}d}| j ||� d}d}| j ||� d S )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 r6 r7 r r r r� m s z Test_metaclass.test_commentsc C s� d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� d }d}| j ||� d
}d }| j ||� d }d
}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� 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 r6 r7 r r r � test_meta� sB zTest_metaclass.test_metaN)r2 r& r3 r rb r� r� r r r r r� T s r� c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
� Test_getcwduZgetcwduc C sT d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� d S ) Nz
os.getcwduz os.getcwdz os.getcwdu()z os.getcwd()zmeth = os.getcwduzmeth = os.getcwdzos.getcwdu(args)zos.getcwd(args))r )r r6 r7 r r r r8 � s zTest_getcwdu.test_basicc C s d}d}| j ||� d S )Nzos.getcwdu() # Foozos.getcwd() # Foo)r )r r6 r7 r r r r` s zTest_getcwdu.test_commentc C s. d}| j |� d}| j |� d}| j |� d S )Nz os.getcwd()z getcwdu()z os.getcwdb())r! )r rP r r r rb s
zTest_getcwdu.test_unchangedc C s d}d}| j ||� d S )Nz<
if 1:
os.getcwdu()
z;
if 1:
os.getcwd()
)r )r r6 r7 r r r �test_indentation s z Test_getcwdu.test_indentationc C s@ d}d}| j ||� d}d}| j ||� d}d}| j ||� d S )Nz
os .getcwdu()z os .getcwd()z os. getcwduz os. getcwdzos.getcwdu ( )zos.getcwd ( ))r )r r6 r7 r r r �test_multilation( s z Test_getcwdu.test_multilationN) r2 r& r3 r r8 r` rb 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 s d}d}| j ||� d S )Nzoperator.isCallable(x)zhasattr(x, '__call__'))r )r r6 r7 r r r �test_operator_isCallable: s z&Test_operator.test_operator_isCallablec C s@ d}d}| j ||� d}d}| j ||� d}d}| j ||� d S )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 r6 r7 r r r � test_operator_sequenceIncludes? s z,Test_operator.test_operator_sequenceIncludesc C s d}d}| j ||� d S )Nzoperator.isSequenceType(x)z6import collections
isinstance(x, collections.Sequence))r )r r6 r7 r r r � test_operator_isSequenceTypeL s z*Test_operator.test_operator_isSequenceTypec C s d}d}| j ||� d S )Nzoperator.isMappingType(x)z5import collections
isinstance(x, collections.Mapping))r )r r6 r7 r r r �test_operator_isMappingTypeQ s z)Test_operator.test_operator_isMappingTypec C s d}d}| j ||� d S )Nzoperator.isNumberType(x)z,import numbers
isinstance(x, numbers.Number))r )r r6 r7 r r r �test_operator_isNumberTypeV s z(Test_operator.test_operator_isNumberTypec C s@ d}d}| j ||� d}d}| j ||� d}d}| j ||� d S )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 r6 r7 r r r �test_operator_repeat[ s z"Test_operator.test_operator_repeatc C s@ d}d}| j ||� d}d}| j ||� d}d}| j ||� d S )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 r6 r7 r r r �test_operator_irepeath s z#Test_operator.test_operator_irepeatc C s d}d}| j ||� d S )Nz
isCallable(x)z-You should use 'hasattr(x, '__call__')' here.)r# )r rP �tr r r �test_bare_isCallableu s z"Test_operator.test_bare_isCallablec C s d}d}| j ||� d S )NzsequenceIncludes(x, y)z.You should use 'operator.contains(x, y)' here.)r# )r rP r� r r r �test_bare_sequenceIncludesz s z(Test_operator.test_bare_sequenceIncludesc C s d}d}| j ||� d S )NzisSequenceType(z)z:You should use 'isinstance(z, collections.Sequence)' here.)r# )r rP r� r r r �!test_bare_operator_isSequenceType s z/Test_operator.test_bare_operator_isSequenceTypec C s d}d}| j ||� d S )NzisMappingType(x)z9You should use 'isinstance(x, collections.Mapping)' here.)r# )r rP r� r r r � test_bare_operator_isMappingType� s z.Test_operator.test_bare_operator_isMappingTypec C s d}d}| j ||� d S )NzisNumberType(y)z4You should use 'isinstance(y, numbers.Number)' here.)r# )r rP r� r r r � test_bare_operator_isNumberType� s z-Test_operator.test_bare_operator_isNumberTypec C s d}d}| j ||� d S )Nz repeat(x, n)z)You should use 'operator.mul(x, n)' here.)r# )r rP r� r r r �test_bare_operator_repeat� s z'Test_operator.test_bare_operator_repeatc C s d}d}| j ||� d S )Nzirepeat(y, 187)z,You should use 'operator.imul(y, 187)' here.)r# )r rP r� r r r �test_bare_operator_irepeat� s z(Test_operator.test_bare_operator_irepeatN)r2 r& r3 r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� 6 s
r� c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�
Test_exitfuncZexitfuncc C s d}d}| j ||� d S )NzI
import sys
sys.exitfunc = my_atexit
ze
import sys
import atexit
atexit.register(my_atexit)
)r )r r6 r7 r r r r� � s zTest_exitfunc.test_simplec C s d}d}| j ||� d S )NzO
import sys, crumbs
sys.exitfunc = my_func
zY
import sys, crumbs, atexit
atexit.register(my_func)
)r )r r6 r7 r r r �test_names_import� s z Test_exitfunc.test_names_importc C s d}d}| j ||� d S )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 r6 r7 r r r �test_complex_expression� s z%Test_exitfunc.test_complex_expressionc C s, d}d}| j ||� d}d}| j ||� d S )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 r6 r7 r r r r� � s zTest_exitfunc.test_commentsc C s d}d}| j ||� d S )Nz]
import sys
def f():
sys.exitfunc = func
zz
import sys
import atexit
def f():
atexit.register(func)
)r )r r6 r7 r r r �test_in_a_function� s z Test_exitfunc.test_in_a_functionc C s d}d}d}| j |||� 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 r6 r7 �msgr r r �test_no_sys_import� s z Test_exitfunc.test_no_sys_importc C s d}| j |� d S )Nzf(sys.exitfunc))r! )r rP r r r rb � s z Test_exitfunc.test_unchangedN) r2 r& r3 r r� r� r� r� r� r� rb r r r r r� � s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � Test_assertsZassertsc
C s> ddddddddd d d d g }x |D ]\}}| j ||� q"W d S ) N�self.assert_(True)�self.assertTrue(True)�self.assertEquals(2, 2)�self.assertEqual(2, 2)�self.assertNotEquals(2, 3)�self.assertNotEqual(2, 3)� self.assertAlmostEquals(2, 3)� self.assertAlmostEqual(2, 3)� self.assertNotAlmostEquals(2, 8)� self.assertNotAlmostEqual(2, 8)�self.failUnlessEqual(2, 2)�self.failIfEqual(2, 3)� self.failUnlessAlmostEqual(2, 3)� self.failIfAlmostEqual(2, 8)�self.failUnless(True)�self.failUnlessRaises(foo)�self.assertRaises(foo)�self.failIf(False)�self.assertFalse(False))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 Ztestsr6 r7 r r r �test_deprecated_names� s z"Test_asserts.test_deprecated_namesc C s� d}d}| j ||� d}d}| j ||� d}d}| j ||� d}d}| j ||� d }d
}| j ||� d }d }| j ||� d
}d}| j ||� d S )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 r6 r7 r r r �
test_variants s* zTest_asserts.test_variantsc C s | j d� | j d� d S )Nzself.assertEqualsOnSaturdayz!self.assertEqualsOnSaturday(3, 5))r! )r r r r rb s
zTest_asserts.test_unchangedN)r2 r& r3 r r r rb r r r r r � s r )Br� r� r r� r r r r Z
lib2to3.testsr ZTestCaser r4 r; rG r] rc rf rk rx r| r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r
r r r& r) rR rT rd rl rx r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r �<module> s� 9: ;1#j3& }.& gH /+bn/ < ]4B.Oc|P Z
=J %": &=cX