B
YI~dK � @ s\ d dl mZ d dlZd dlmZ G dd� dej�ZG dd� dej�ZedkrXej d d
� dS ) � )�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)�.0�cr r �C/opt/alt/python37/lib64/python3.7/idlelib/idle_test/test_pyparse.py�
s z.ParseMapTest.test_parsemap..z
� �a�xi� )r ZParseMap� assertEqualr )�selfZ keepwhite�mappingr r r �
test_parsemap
s
zParseMapTest.test_parsemapc C s$ t �dd�}| �d�t j�d� 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 )� indentwidth�tabwidth)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� assertRaises�AssertionError�subTest�code� study_level)r �eq�p�setcode�testsr" 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_study1�ranger r& r" r( r; r< r5 )
r r) r* r+ �study�NONE� BACKSLASH�FIRST�NEXT�BRACKETr: 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|jd8��h ||j� |� ||jd9� ||j|j � ||j
|j � ||j |j � ||j
|j� ||j|j� W d Q R X �q(W | �|� � d S ):Nr: r" r6 �end�lastch� 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�|dd�f}|d� | �t�� |� W d Q R X x>|D ]6}| j|jd�� ||j� ||� |j � W d Q R X q�W 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�|dd�f }|d� | �t�� |� W d Q R X x$|D ] }||j� ||� |j� q�W 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�|dd�f}x>|D ]6} | j| jd�� || j� ||� | j � W d Q R X q�W 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 q�W 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 �yes�nor* 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 q�W 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�|dd�f
}x>|D ]6}| j|jd�� ||j� ||� |j� W d Q R X q�W 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)
Zidlelibr Zunittest� collectionsr ZTestCaser r r �mainr r r r � s K