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