B
fK @ s` d Z ddlmZ ddlZddlmZ G dd dejZG dd dejZe d kr\ej
d
d dS ) zTest pyparse, coverage 96%. )pyparseN)
namedtuplec @ s e Zd Zdd Zdd ZdS ) ParseMapTestc C s` dd dD }t |}| |td td | |td td | |d td d S )Nc S s i | ]}t |t |qS )ord).0cr r C/opt/alt/python37/lib64/python3.7/idlelib/idle_test/test_pyparse.py
s z.ParseMapTest.test_parsemap..z
axi )r ZParseMap assertEqualr )selfZ keepwhitemappingr r r
test_parsemap
s
zParseMapTest.test_parsemapc C s$ t dd}| dt jd d S )N z a([{b}])b"c'd
zxxx(((x)))x"x'x
)r Parserr translateZtrans)r parserr r r
test_trans s zParseMapTest.test_transN)__name__
__module__ __qualname__r r r r r r r s r c @ s e Zd Zedd Zedd Zdd Zdd Zd d
Zd d Z d
d Z
dd Z dd Z dd Z
dd Zdd Zdd Zdd Zd d Zd d Zd!S )" PyParseTestc C s t jddd| _d S )Nr ) indentwidthtabwidth)r r r )clsr r r
setUpClass s zPyParseTest.setUpClassc C s | ` d S )N)r )r r r r
tearDownClass s zPyParseTest.tearDownClassc C s$ | | jjd | | jjd d S )Nr )r r r r )r r r r test_init" s zPyParseTest.test_initc
C s~ | j }| j}|j}| t |d W d Q R X d}xD|D ]<}| j|d& || ||j| ||jd W d Q R X q:W d S )Nr ) za
)stringr )r r set_code assertRaisesAssertionErrorsubTestcode study_level)r eqpsetcodetestsr" r r r
test_set_code& s
zPyParseTest.test_set_codec C sB | j }| j}|j}|j}dd }|d |||d |d |||d |d | t | W d Q R X | t |d W d Q R X | |dd d
|||d ||d d d
d ||d
d d
d ||dd d
d |d |||d ||dd d
d ||dd d
d ||dd d
d d S )Nc S s dS )NFr )indexr r r char_in_string_false= zDPyParseTest.test_find_good_parse_start..char_in_string_falsez def spam():
r zclass spam(
zm"""This is a module docstring"""
class C():
def __init__(self, a,
b=True):
pass
Fc S s dS )NTr )r. r r r Z r0 z8PyParseTest.test_find_good_parse_start..)Zis_char_in_string, c S s | dkS )Nr2 r )r. r r r r1 b r0 c S s | dkS )Nr2 r )r. r r r r1 e r0 ! c S s | dk S )Nr2 r )r. r r r r1 h r0 z\"""This is a module docstring"""
class C():
def __init__(self, a, b=True):
pass
c S s | dkS )Nr2 r )r. r r r r1 r r0 c S s | dkS )Nr2 r )r. r r r r1 s r0 c S s | dk S )Nr2 r )r. r r r r1 v r0 )r r r# Zfind_good_parse_startr$ TypeError assertIsNone)r r) r* r+ startr/ r r r test_find_good_parse_start8 s0 z&PyParseTest.test_find_good_parse_startc C sp d}| j }|| | t |d W d Q R X |d | |j| |d | |j|dd d S )Nzm"""This is a module docstring"""
class C():
def __init__(self, a,
b=True):
pass
r r2 )r r# r$ r% Zset_lor r' )r r' r* r r r test_set_lox s
zPyParseTest.test_set_loc
C s | j }| j}|j}|j}td\}}}}} tddddg}
|
ddg||
ddd g||
d
dd g||
d dd g||
d dd g||
d
dd g||
ddd g||
dddg||
ddd g||
ddd g||
ddd dg||
ddd g||
ddd g| |
ddd g||
ddd g||
ddd dg||
ddd dg| |
ddd dg| |
d dd g||
d dd g| |
d dd g|f} x`| D ]X} | j| jd > || j | ||jd ||j | j ||j
| j
W d Q R X qpW | | d S ) Nr8 TestInfor" goodlines continuationr! r z$"""This is a complete docstring."""
z$'''This is a complete docstring.'''
z""""This is a continued docstring.
z"'''This is a continued docstring.
z""""Closing quote does not match."
z"""Bracket in docstring [
z#'''Incomplete two line docstring.
z "This is a complete string."
z "This is an incomplete string.
z'This is more incomplete.
z # Comment\
z"("""Complete string in bracket"""
z("""Open string in bracket
za = (1 + 2) - 5 *\
z0
def function1(self, a,
b):
z
def function1(self, a,\
z
def function1(self, a,
z())
z)(
z{)(]
)r" ) r r r# Z_study1ranger r& r" r( r; r< r5 )
r r) r* r+ studyNONE BACKSLASHFIRSTNEXTBRACKETr: r, testr r r test_study1 sJ
zPyParseTest.test_study1c
C s | j }| j}|j}|j}td\}}}}} tdddg}
|
d||
d||
d||
d||
d | f} x>| D ]6} | j| jd
|| j || | j W d Q R X qjW d S ) Nr8 r: r" r< r! z%"""This is a continuation docstring.
z-'''This is a multiline-continued docstring.
za = (1 + 2) - 5 *\
z
def function1(self, a,\
)r" ) r r r# Zget_continuation_typer@ r r& r" r< )
r r) r* r+ ZgettyperB rC rD rE rF r: r, rG r r r test_get_continuation_type s
z&PyParseTest.test_get_continuation_typec C s | j }| j}|j}|j}tdddddddg}|dd d dd d
|d d d d
d d|dd ddd d|dd d dd d
|dd ddd d|ddd dd d|ddd d d d |d dd!d d d |d"dd#d d d$|d%d d&d d'd(|d)d d*d+d d,|d-d d.d/dd0|d1d d2d3d d4|d5d d*d6d d
|d7d d dd d
f}x|D ]}| j|jd8h ||j | ||jd9 ||j|j ||j
|j ||j |j ||j
|j ||j|j W d Q R X q(W | | d S ):Nr: r" r6 endlastch openbracket
bracketingr! r ))r r z0'''This is a multiline continuation docstring.
0 '))r r )r r= )rN r z # Comment\
))r r )r= r= )rP r z #Comment\
za = (1 + 2) - 5 *\
*))r r )r r= ) r z0
def function1(self, a,
b):
r= :))r= r ) r= ). r z
def function1(self, a,\
,rU ))r= r )rU r= z
def function1(self, a,
z3
def function1(self, a, # End of line comment.
3 ))r= r )rU r= )rW r> )rZ r= z7 a = ["first item",
# Comment line
"next item",
7 ))r r )r\ r= ) r> )rQ r= ) r> )& r= )* r> )5 r= z())
r )))r r )r r= )r> r )r? r z)(
r? ())r r )r= r )r= r= z{)(]
r8 ]))r r )r r= )r> r )r> r= )r r z:\a
z\a
)r" r> )r r r# Z_study2r r& r" r( Z
stmt_startr6 Zstmt_endrJ rK ZlastopenbracketposrL Zstmt_bracketingrM r5 )r r) r* r+ rA r: r, rG r r r test_study2 sV
zPyParseTest.test_study2c
C s | j }| j}|j}|j}tdddg}|dd|dd|dd|d d
|d d|d d|d
d|ddf}|d | t | W d Q R X x>|D ]6}| j|jd ||j || |j W d Q R X qW d S )Nr: r" linesz[x for x in a]
r= z[x
for x in a
r> z[x\
for x in a\
z[x
for x in a
]
r? z%
"""Docstring comment L1"""
L2
L3
L4
z%
"""Docstring comment L1
L2"""
L3
L4
z&
"""Docstring comment L1\
L2\
L3\
L4\
r z+
"""Docstring comment L1\
L2\
L3\
L4\
"""
r8 r! )r" )
r r r# Zget_num_lines_in_stmtr r$
IndexErrorr& r" rg )r r) r* r+ getlinesr: r, rG r r r test_get_num_lines_in_stmt s(
z&PyParseTest.test_get_num_lines_in_stmtc
C s | j }| j}|j}|j}tdddg}|dd|dd|dd|d d
|d d
|d d
|d
d|dd|dd|dd|ddf }|d | t | W d Q R X x$|D ] }||j || |j qW d S )Nr: r" spaceszdef function1(self, a,
z
def function1(self, a,
z
def function1(self, a,
z
def function1(
z
def function1(
z
def function1(
z2[
"first item",
# Comment line
"next item",
r z4[
"first item",
# Comment line
"next item",
r> z1["first item",
# Comment line
"next item",
r= z(
r z(a
zdef function1(self, a, b):
) r r r# Zcompute_bracket_indentr r$ r% r" rk )r r) r* r+ indentr: r, rG r r r test_compute_bracket_indent/ s,
z'PyParseTest.test_compute_bracket_indentc
C s( | j }| j}|j}|j}d}xH|D ]@}| j|d* || | t | W d Q R X W d Q R X q"W tdd}|dd|dd|dd |d
d |d d|d
d|dd|dd|dd|dd |dd |dd|dd|ddf}x>|D ]6} | j| jd || j || | j W d Q R X qW d S )N)zdef function1(self, a, b\
z """ (\
za = #\
)r" r: )r" rk za = (1 + 2) - 5 *\
r za = 1 + 2 - 5 *\
z a = 1 + 2 - 5 *\
rn z a = "spam"\
r\ z
a = \
"a"\
z a = #\
"a"\
r8 za == \
r> za != \
z\
z \
z \
za\
r? z{}\
z(1 + 2) - 5 *\
)
r r r# Zcompute_backslash_indentr& r$ r% r r" rk )
r r) r* r+ ro errorsr" r: r, rG r r r test_compute_backslash_indentO s:
z)PyParseTest.test_compute_backslash_indentc
C s | j }| j}|j}|j}tdddg}|dd|dd|dd|dd |d
d |d d |d d f}x>|D ]6}| j|jd
||j || |j W d Q R X qhW d S )Nr: r" ro r! z def a():
z
def a():
r z
def a():
z z def a(
z
def a(
z
# Comment.
)r" )r r r# Zget_base_indent_stringr r& r" ro )r r) r* r+ Z
baseindentr: r, rG r r r test_get_base_indent_stringv s
z'PyParseTest.test_get_base_indent_stringc C s | j }| j}| j}|j}|j}tdddg}|d||d||d||d||d||d ||d
||d ||d ||d
||d||d|f }x<|D ]4}| j|jd ||j || W d Q R X qW d S )Nr: r" assert_z def a():
z0
def function1(self, a,
b):
z:
za:
z):
z(:
z":
z
def function1(self, a,
z!def function1(self, a):
pass
z
# A comment:
z"""A docstring:
)r" )
assertTrue assertFalser r# Zis_block_openerr r& r" rt ) r yesnor* r+ openerr: r, rG r r r test_is_block_opener s,
z PyParseTest.test_is_block_openerc C s | j }| j}| j}|j}|j}tdddg}|d||d||d||d||d||d ||d
||d ||d ||d
||d||d||d|f
}x<|D ]4}| j|jd ||j || W d Q R X qW d S )Nr: r" rt zreturn
z break
z continue
z raise
z pass
zpass
z return #
zraised
z
returning
z # return
z """break
z
"continue
z!def function1(self, a):
pass
)r" ) ru rv r r# Zis_block_closerr r& r" rt ) r rw rx r* r+ Zcloserr: r, rG r r r test_is_block_closer s.
z PyParseTest.test_is_block_closerc C s | j }| j}|j}|j}tdddg}|dd|dd|dd|d d
|d d |d
d|dd|dd|dd|ddf
}x>|D ]6}| j|jd ||j || |j W d Q R X qW d S )Nr: r" bracketr! ))r r za
z()()
))r r )r r= )r> r )r> r= )r r z(
)()
))r r )r r= )r? r )r? r= )r8 r z()
()
))r? r )r? r= )r8 r z()(
)
))r r )r r= )r> r )r> r= )r8 r z(())
))r r )r r= )r= r> )r? r= )r r z(
())
))r r )r r= )r> r> )r r= )r8 r z{)(]
z(((())
))r r )r r= )r= r> )r> r? )r? r )r8 r? )r\ r> )r" )r r r# Zget_last_stmt_bracketingr r& r" r| )r r) r* r+ rM r: r, rG r r r test_get_last_stmt_bracketing s(
z)PyParseTest.test_get_last_stmt_bracketingN)r r r classmethodr r r r- r7 r9 rH rI rf rj rp rr rs rz r{ r} r r r r r s @2= ' r __main__r> ) verbosity) __doc__Zidlelibr Zunittest collectionsr ZTestCaser r r mainr r r r s K