3
2aH @ s\ d dl mZ d dlZd dlmZ G dd dejZG dd dejZedkrXej d d
dS ) )pyparseN)
namedtuplec @ s e Zd Zdd Zdd ZdS ) ParseMapTestc C s` dd dD }t j|}| j|td td | j|td td | j|d td d S )Nc S s i | ]}t |t |qS )ord).0cr r 1/opt/alt/python36/lib64/python3.6/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 jdd}| jdjt 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$ | j | jjd | j | jjd d S )Nr )r r r r )r r r r test_init" s zPyParseTest.test_initc C s~ | j }| j}|j}| jt |d W d Q R X d}xD|D ]<}| j|d& || ||j| ||jd W d Q R X q:W d S )Nr a
)stringr )r! r" )r r set_code assertRaisesAssertionErrorsubTestcode study_level)r eqpsetcodetestsr# r r r
test_set_code& s
zPyParseTest.test_set_codec C s | j }| j}|j}|j}|d | j| | j|dd d ||dd dd ||dd dd ||dd dd ||d
d dd |d ||d d dd ||d
d dd ||dd dd ||dd dd d S )Nzm"""This is a module docstring"""
class C():
def __init__(self, a,
b=True):
pass
c S s dS )NTr )indexr r r K s z8PyParseTest.test_find_good_parse_start..)Zis_char_in_stringc S s dS )NFr )r/ r r r r0 O s , c S s | dkS )Nr1 r )r/ r r r r0 S s c S s | dkS )Nr1 r )r/ r r r r0 V s ! c S s | dk S )Nr1 r )r/ r r r r0 Y s z\"""This is a module docstring"""
class C():
def __init__(self, a, b=True):
pass
c S s dS )NFr )r/ r r r r0 b s c S s | dkS )Nr1 r )r/ r r r r0 c s c S s | dkS )Nr1 r )r/ r r r r0 d s c S s | dk S )Nr1 r )r/ r r r r0 g s )r r r$ Zfind_good_parse_start assertIsNone)r r* r+ r, startr r r test_find_good_parse_start8 s z&PyParseTest.test_find_good_parse_startc C sp d}| j }|j| | jt |jd W d Q R X |jd | j|j| |jd | j|j|dd d S )Nzm"""This is a module docstring"""
class C():
def __init__(self, a,
b=True):
pass
r r1 )r r$ r% r& Zset_lor r( )r r( r+ r r r test_set_loi 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 | j | d S ) Nr6 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) r9 r: r3 )
r r* r+ r, studyNONE BACKSLASHFIRSTNEXTBRACKETr8 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 ) Nr6 r8 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, Zgettyper@ rA rB rC rD r8 r- rE r r r test_get_continuation_type s
z&PyParseTest.test_get_continuation_typec C s | j }| j}|j}|j}tdddddddg}|dd d dd d8|d
d d d d d<|dd ddd d@|dd d dd dB|dd ddd dF|dd
ddd dJ|dd
d d ddM|d d
d d ddP|d d
d!d ddU|d#d d$d d%d^|d+d dd,d dc|d.d d-d/d
dg|d0d d1d2d dm|d3d dd4d do|d5d d dd dqf}x|D ]}| j|jd6h ||j | ||jd" ||j|j ||j
|j ||j |j ||j
|j ||j|j W d Q R X q(W | j| d S )rNr8 r# r4 endlastch openbracket
bracketingr! r z1'''This is a multiline continutation docstring.
1 'r; z # Comment\
z #Comment\
za = (1 + 2) - 5 *\
*r z0
def function1(self, a,
b):
0 : . z
def function1(self, a,\
,z
def function1(self, a,
z3
def function1(self, a, # End of line comment.
3 r< z7 a = ["first item",
# Comment line
"next item",
7 & * 5 z())
)r= z)(
(z{)(]
r6 ]z:\a
z\a
)r# r r )re r r r r; rL r )rf rg rh r r r; r; rN r )ri rj rk r r )rl r r r r; rQ r )rm rn ro r; r rT r; rU r )rp rq rr r; r rT r; )rs rt r; r rT r; )ru rv r; r rT r; rV r< rY r; )rw rx ry rz r r r[ r; r\ r< rO r; r] r< r^ r; r_ r< r` r; )r{ r| r} r~ r r r r r r r r; r< r r= r )r 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$ Z_study2r r' r# r) Z
stmt_startr4 Zstmt_endrH rI ZlastopenbracketposrJ Zstmt_bracketingrK r3 )r r* r+ r, r? r8 r- rE r r r test_study2 sX
zPyParseTest.test_study2c C s | j }| j}|j}|j}tdddg}|dd|dd|dd|d d
|d d|d d|d
d|ddf}|d | jt | W d Q R X x>|D ]6}| j|jd ||j || |j W d Q R X qW d S )Nr8 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\
"""
r6 r! )r# )
r r r$ Zget_num_lines_in_stmtr r%
IndexErrorr' r# r )r r* r+ r, getlinesr8 r- rE 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 | jt | W d Q R X x$|D ] }||j || |j qW d S )Nr8 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# r )r r* r+ r, indentr8 r- rE 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* || | jt | W d Q R X W d Q R X q"W tdd }|dd |d
d |d d |d
d|dd |dd|dd|dd|dd|dd|dd|dd|dd |ddf}x>|D ]6} | j| jd || j || | j W d Q R X qW d S ) Ndef function1(self, a, b\
""" (\
a = #\
)r# r8 r# r za = (1 + 2) - 5 *\
r za = 1 + 2 - 5 *\
z a = 1 + 2 - 5 *\
r z a = "spam"\
r[ z
a = \
"a"\
z a = #\
"a"\
r6 za == \
r< za != \
z\
z \
z \
za\
r= z{}\
z(1 + 2) - 5 *\
)r r r )r# r )
r r r$ Zcompute_backslash_indentr' r% r& r r# r )
r r* r+ r, r errorsr# r8 r- rE r r r test_compute_backslash_indent@ 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 )Nr8 r# r 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# r )r r* r+ r, Z
baseindentr8 r- rE r r r test_get_base_indent_stringg 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 |j| W d Q R X qW d S )Nr8 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# r ) r yesnor+ r, openerr8 r- rE 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 |j| W d Q R X qW d S )Nr8 r# r 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# ) r r r r$ Zis_block_closerr r' r# r ) r r r r+ r, Zcloserr8 r- rE 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%|dd)|dd/|dd5|dd;|ddA|ddIf
}x@|D ]8}| j|jd ||j || |j W d Q R X qW d S )JNr8 r# bracketr! r za
z()()
r; r< r z(
)()
r= r6 z()
()
z()(
)
z(())
z(
())
z{)(]
z(((())
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; r6 r )r r r r r r= r r= r; r6 r )r r r r r r r; r< r r< r; r6 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; r6 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 r6 r= r[ r< )r r r r r r r )r r r$ Zget_last_stmt_bracketingr r' r# r )r r* r+ r, rK r8 r- rE r r r test_get_last_stmt_bracketing s(
z)PyParseTest.test_get_last_stmt_bracketingN)r r r classmethodr r r r. r5 r7 rF rG r r r r r r r r r r r r r s 12= ' r __main__r< ) verbosity)
Zidlelibr Zunittest collectionsr ZTestCaser r r mainr r r r s <