B
� f<\ � @ s@ d Z ddlmZ ddlZddlmZ ddlmZ ddlm Z m
Z
ddl m Z ddl
mZ G d d
� d
ej�ZG d d � d ej�ZG d
d� dej�ZG dd� dej�ZG dd� dej�ZG dd� d�ZG dd� d�ZG dd� dej�ZG dd� d�ZG dd � d ej�ZG d d � d ej�ZG d d � d ej�Ze d!k�r b:
return a
elif a < b:
return b
else:
return None
c C sr | j j}| j}| j}|�ddd� ddddg}||� ddd�|�|f� |�ddd � ||� d
d dddgf� d S ) Nrk z7.0z10.0r
z def compare(self):z if a > b:�
z1.0r% z15.0z16.0)rc �
get_regionr r rl �join�
tag_remove)r
rg r �eqZexpected_linesr r r �test_get_region� s z FormatRegionTest.test_get_regionc
C s& | j j}| j}| j}|j}t�� |_| j�� d }| j�� d }|� ddd� | j �
� \}}} }
|||| |
� |j� � ||� dd�| � ||� dd�| � |�
ddd � d
} | �d �} |d d
| | � ||� dd�| � ||� d d�| � ||� dd�|� ||� dd�|� |�
ddd � ||_d S )Nr< � rk z6.0z11.0z sel.firstzsel.lastz1.0r% zadded line 1
r{ z7.0z10.0z7.0-1cz12.0-1c)rc �
set_regionr r Zbellr rv rx �
splitlinesrl r| Zassert_called_oncerg r~ r* )
r
Zset_r r Z save_bellZline6Zline10�head�tail�chars�linesZ newstring�newlinesr r r �test_set_region� s.
z FormatRegionTest.test_set_regionc C s> | j j}| j}| j}|�ddd� |� ||�dd�d� d S )Nrk z7.0z10.0z2
def compare(self):
if a > b:
)rc Zindent_region_eventr r rl rg )r
�indentr r r r r �test_indent_region_event� s z)FormatRegionTest.test_indent_region_eventc C s> | j j}| j}| j}|�ddd� |� ||�dd�d� d S )Nrk z7.0z10.0z"
def compare(self):
if a > b:
)rc Zdedent_region_eventr r rl rg )r
�dedentr r r r r �test_dedent_region_event� s z)FormatRegionTest.test_dedent_region_eventc C s> | j j}| j}| j}|�ddd� |� ||�dd�d� d S )Nrk z7.0z10.0z0##
## def compare(self):
## if a > b:
)rc �comment_region_eventr r rl rg )r
r: r r r r r �test_comment_region_event� s z*FormatRegionTest.test_comment_region_eventc C s� | j j}| j j}| j}| j}|�ddd� |� |� ||�dd�d� |�ddd� |�ddd� |� ||�dd �d
� | j �ddd d d g� |� ||�dd �d
� d S )Nrk z7.0z10.0z*
def compare(self):
if a > b:
z1.0r% z3.0z4.0z3.endz # Class comment.r
z# Class comment.z Class comment.) rc r� Zuncomment_region_eventr r rl rg r~ r� )r
r: Z uncommentr r r r r �test_uncomment_region_event s z,FormatRegionTest.test_uncomment_region_event� _asktabwidthc C s\ | j j}| j}| j}|�ddd� d |_| �|� � d|_| �|� � ||�dd�d� d S )Nrk z7.0z10.0r0 z$
def compare(self):
if a > b:
) rc �tabify_region_eventr r rl � return_value� assertIsNone�assertIsNotNonerg )r
r� Ztabifyr r r r r �test_tabify_region_event s z)FormatRegionTest.test_tabify_region_eventc C sl | j j}| j}| j}|�ddd� d |_| �|� � d|_| j �� d|_| �|� � ||� dd�d� d S )Nrk z7.0z10.0r$ r0 z0
def compare(self):
if a > b:
)
rc Zuntabify_region_eventr r rl r� r� r� r� rg )r
r� Zuntabifyr r r r r �test_untabify_region_event) s
z+FormatRegionTest.test_untabify_region_event�
askintegerc C s | j j}d|_| �|� d� d S )N�
)rc r� r� r )r
r� Zaskr r r �test_ask_tabwidth: s z"FormatRegionTest.test_ask_tabwidthN)r r r rB r! rf ry rz rx r� r� r� r� r� r� r �patch�objectr rw r� r� r� r r r r ru � s #
ru c @ s8 e Zd Zej�ed�dd� �Zej�ed�dd� �ZdS )� IndentsTest�askyesnoc C sj t d d �}t�|�}d|_|�d � | �|jd� | �|jd� |�d � | �|jd� | �|jd� d S )NT� F)rr r �Indentsr� Ztoggle_tabs_eventr rt rs )r
r� rP �indentsr r r �test_toggle_tabsC s
z IndentsTest.test_toggle_tabsr� c C sx t d d �}t�|�}d |_|�d � | �|jd� d|_|�d � | �|jd� d|_d|_|�d � | �|jd� d S )Nr2 r0 r< T)rr r r� r� Zchange_indentwidth_eventr rs rt )r
r� rP r� r r r �test_change_indentwidthQ s
z#IndentsTest.test_change_indentwidthN) r r r r r� r� r r� r� r r r r r� A s r� c @ s< e Zd Zedd� �Zedd� �Zdd� Zdd� Zd d
� Zd S ) �
RstripTestc C sH t d� t� | _| j�� t| j�| _t| jd�| _t� | j�j
| _
d S )Nra )r ) r r r^ rb r r �
MockEditorrP r ZRstrip� do_rstrip)r r r r r! f s
zRstripTest.setUpClassc C s( | ` | `| `| j�� | j�� | `d S )N)r r� rP r^ rd re )r r r r rf o s
zRstripTest.tearDownClassc C s | j �dd� d S )Nz1.0zend-1c)r r+ )r
r r r rz v s zRstripTest.tearDownc C s8 d}d}| j �d|� | �� | �| j �dd�|� d S )Nz�Line with an ending tab
Line ending in 5 spaces
Linewithnospaces
indented line
indented line with trailing space
zyLine with an ending tab
Line ending in 5 spaces
Linewithnospaces
indented line
indented line with trailing space
z1.0r& )r r& r� r rg )r
Zoriginal�strippedr r r �test_rstrip_linesy s
z RstripTest.test_rstrip_linesc
C s� | j }xLdD ]D}| j|d��. |�d|� | �� | �|�dd�d� W d Q R X q W xXdD ]P}| j|d��: |�dd� |�d|� | �� | �|�dd�d� W d Q R X qZW d S )N)r
r{ z
)�codez1.0zend-1cr
)za
za
za
za
)r ZsubTestr& r� r rg r+ )r
r r� r r r �test_rstrip_end� s
"
zRstripTest.test_rstrip_endN) r r r rB r! rf rz r� r� r r r r r� d s
r� �__main__r$ )� verbosity�exit) r Zidlelibr r Zunittestr Z test.supportr Ztkinterr r Zidlelib.editorr Zidlelib.idle_test.mock_idler r� ZTestCaser r rC rK rO rU r` rr ru r� r� r �mainr r r r � s. w
9#7