a
ze! @ st d Z ddlZddlZddlZddlmZ ddlmZmZ ddl Z ej
rNdd Z G dd dej Z
ed krpe dS )
z/
Test cases for codeop.py
Nick Mathewson
N)support)compile_commandPyCF_DONT_IMPLY_DEDENTc C s( | D ]\}}t|drd| |< q| S )N__call__T)itemshasattr)dnv r &/usr/lib/python3.9/test/test_codeop.pyunify_callables s
r
c @ sZ e Zd ZdddZdddZdddZd d
Zd d Zd
d Zdd Z dd Z
dd Z dS ) CodeopTestssinglec C s t jrt|d|}| | |dkrpi i }}tj}t t_z&t|| tt |dd| W |t_q|t_0 n,|dkrddi}dt
||i}dt
||i}| t |t | n"t |d|t
}| t|d|| dS )z(succeed iff str is a valid piece of coder evala valueN)r is_jythonr
assertTruesysstdoutioStringIOexeccompiler assertEqualr
r ) selfstrsymbolcoder rZ saved_stdoutctxZexpectedr r r assertValid s$
zCodeopTests.assertValidc C s | t||dd dS )z5succeed iff str is the start of a valid piece of coder N)r r )r r r r r r assertIncomplete. s z CodeopTests.assertIncomplete c C sX zt ||d | d W n8 ty6 | | Y n tyR | | Y n0 dS )z8succeed iff str is the start of an invalid piece of coder% z$No exception raised for invalid codeN)r Zfail SyntaxErrorr
OverflowError)r r r Z is_syntaxr r r
assertInvalid2 s zCodeopTests.assertInvalidc C s | j }tjsB| tdtdddt | tdtdddt n|d |d |d |d |d |d |d
|d |d |d
|d |d |d |d |d |d |d |d |d |d |d |d |dd |d d |d d |d d |d d |d d |d!d |d
|d" |d |d# |d d |d d |d!d |d$ |d% |d& d S )'N passr r
a = 1z
a = 1a = 1
za = 1
z
a = 1
zdef x():
pass
z if 1:
pass
z
if 1: pass
z
if 1: pass
zdef x():
pass
zdef x():
pass
zdef x():
pass
zpass
z3**3
z if 9==3:
pass
else:
pass
z(if 1:
pass
if 1:
pass
else:
pass
z #a
#b
a = 3
z #a
a=3
za=3
z
a = 9+ \
3z3**3r z(lambda z:
z**3)z9+ \
3z9+ \
3
z
a**3z
a**3z
#a
#b
a**3z
if 1: a=1
z
#a
a=3
z1def f():
try: pass
finally: [x for x in (1,2)]
zdef f():
pass
#foo
z@a.b.c
def f():
pass
)r$ r r r r r r )r avr r r
test_valid< sf
zCodeopTests.test_validc C sH | j }|d |d |d |d |d |d |d |d |d |d |d
|d |d |d
|d |d |d |d |d |d |d |d |d |d |d |dd |dd |d d |d d |d d |d d |d d |d! |d" |d# |d$ |d% |d& |d' |d( |d) |d* |d+ |d, |d- |d. |d/ |d0 |d1 |d2 |d3 |d4 |d5 |d6 |d7 |d8 |d9 |d: |d; |d< |d= |d> |d? |d@ |dA |dB |dC |dD |dE |dF |dG |dH |dI |dJ |d |dK |dL |dM |d |dN |dO |dP |dQ |dR |dS |dT |dU |dV |dW |dX |dY |dZ |d[ |d\ |d] |d^ |d_ |d` |da |db |dc |dd d S )eNz(a **z(a,b,z(a,b,(a = (za = {zb + {zif 9==3:
pass
else:zif 9==3:
pass
else:
z if 9==3:
pass
else:
passzif 1:zif 1:
z if 1:
pass
if 1:
pass
else:z!if 1:
pass
if 1:
pass
else:
z'if 1:
pass
if 1:
pass
else:
passzdef x():z def x():
z
def x():
zdef x():
passzdef x():
pass
zdef x():
pass
z
def x():
passza = 9+ \za = 'a\z a = '''xyr+ r r- (z(
z(9+z9+ \z lambda z: \z!if True:
if True:
if True:
z@a(z@a(bz@a(b,z@a(b,cz@a(b,c,zfrom a import (zfrom a import (bzfrom a import (b,zfrom a import (b,czfrom a import (b,c,[z[az[a,z[a,bz[a,b,{z{az{a:z{a:bz{a:b,z{a:b,cz{a:b,c:z{a:b,c:dz {a:b,c:d,za(za(bza(b,za(b,cza(b,c,za[za[bza[b,za[b:za[b:cza[b:c:za[b:c:dzdef a(zdef a(bzdef a(b,z def a(b,cz
def a(b,c,z(az(a,z(a,bzif a:
pass
elif b:z if a:
pass
elif b:
pass
else:zwhile a:zwhile a:
pass
else:z for a in b:zfor a in b:
pass
else:ztry:ztry:
pass
except:ztry:
pass
finally:z!try:
pass
except:
pass
finally:zwith a:z with a as b:zclass a:zclass a(z class a(bz
class a(b,z
class a():z[x forz [x for x inz
[x for x in (z(x forz (x for x inz
(x for x in ()r& r Zair r r test_incomplete~ s
zCodeopTests.test_incompletec C s | j }|d |d |d |d |d |d |d |d |d |d
|d |d d
|dd
|dd
|dd
|dd
|dd
|dd
|dd
|d |d |d |d |d |d |d d S )Nza bza @za b @za ** @za = za = 9 +zdef x():
pass
z
if 1: pass
passz a = 9+ \
za = 'a\ za = 'a\
r. r r2 ]z())z[}z9+z lambda z:z
return 2.3zif (a == 1 and b = 2): passzdel 1zdel (1,)zdel [1]zdel '1'z"[i for i in range(10)] = (1, 2, 3)r* r6 r r r test_invalid s6
zCodeopTests.test_invalidc C sR | j }|ddd |ddd |ddd |ddd |ddd |ddd d S ) Nz raise = 4r r% zdef a-bzawait?z=!=za await raise bza await raise b?+1r9 r6 r r r test_invalid_exec$ s z CodeopTests.test_invalid_execc C s@ | tddjtdddj | tddjtdddj d S )Nr/ abcr def)r r co_filenamer ZassertNotEqual)r r r r
test_filename- s zCodeopTests.test_filenamec C s t dtfdtf*}td | t|jd W d n1 sD0 Y t R | t
( t dt tddd W d n1 s0 Y W d n1 s0 Y d S ) Nz .*literalz .*invalidz '\e' is 0r errorz1 is 1r r% ) r Zcheck_warnings
SyntaxWarningDeprecationWarningr r lenwarningscatch_warningsZ assertRaisesr( simplefilter)r wr r r test_warning3 s 0 zCodeopTests.test_warningN)r )r )r r' ) __name__
__module__ __qualname__r$ r& r* r1 r7 r: r; r? rH r r r r r s
B & r __main__)__doc__r ZunittestrD testr Zcodeopr r r r r
ZTestCaser rI mainr r r r s /