Your IP : 216.73.216.52
a
XC?h�5 � @ s� d dl mZmZ d dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
m Z m Z m
Z
mZ dddddd d
�ZG d d � d �ZG d
d� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZdS )� )�TestCase�mockN)�captured_stderr�captured_stdout�
script_helper�findfile�unlinkzCfruits = [
"Apple",
"Orange",
"Banana",
print(fruits)
zLif True:
print("hello")
print("world")
else:
print("else called")
zIif True:
print("hello")
print("world")
else:
print("else called")
zNif True:
print("hello")
print("world")
else:
print("else called")
zGdef my_func():
print("hello world")
if True:
print("If called")zLdef my_func():
print("Hello world")
if True:
print("If called"))�incomplete_expression�wrong_indented�nannynag_errored�
error_free�tab_space_errored_1�tab_space_errored_2c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�TemporaryPyFile� Nc C s || _ || _d S �N)� source_code�dir)�selfr � directory� r �7/opt/alt/python39/lib64/python3.9/test/test_tabnanny.py�__init__A s zTemporaryPyFile.__init__c C sN t jd| jddd�� }|�| j� W d � n1 s60 Y |j| _| jS )N�wz.pyF)�moder �suffix�delete)�tempfileZNamedTemporaryFiler �writer �name� file_path)r �fr r r � __enter__E s
�*zTemporaryPyFile.__enter__c C s t | j� d S r )r r )r �exc_type� exc_value�
exc_tracebackr r r �__exit__M s zTemporaryPyFile.__exit__)r N)�__name__�
__module__� __qualname__r r" r& r r r r r > s
r c @ s e Zd Zdd� ZdS )�TestFormatWitnessesc C sZ g d�}|D ]H\}}| j ||d��" | �t�|�|� W d � q 1 sJ0 Y q d S )N))ZTestzat tab sizes T, e, s, t)r z at tab size )�tz
at tab size t)z t zat tab sizes , , t, , )�words�expected)�subTest� assertEqual�tabnannyZformat_witnesses)r �testsr, r- r r r �test_format_witnessesT s z)TestFormatWitnesses.test_format_witnessesN)r' r( r) r2 r r r r r* Q s r* c @ s e Zd Zdd� ZdS )� TestErrPrintc
C s� ddgdfdgdfg d�dfg dfg}|D ]v\}}| j ||d��P t� �}tj|� W d � n1 sh0 Y | �|�� |� W d � q*1 s�0 Y q*d S ) N�first�secondz
first second
zfirst
)� � � z1 2 3
�
)Z argumentsr- )r. r r0 Zerrprintr/ �getvalue)r r1 �argsr- �stderrr r r �
test_errprinte s
� (zTestErrPrint.test_errprintN)r' r( r) r= r r r r r3 b s r3 c @ s e Zd Zdd� ZdS )� TestNannyNagc C s& t �ddd�dddd�ft �ddd�dddd�fg}|D ]�\}}|�� }|�� }|�� }| j||d d �� | �|d |� W d � n1 s�0 Y | j||d
d �� | �|d
|� W d � n1 s�0 Y | j||d d
�� | �|d |� W d � q81 �s0 Y q8d S )Nr ZfooZbar)�lineno�msg�line� ZtestmsgZtestliner? )� line_numberr- r@ )r@ r- rA )rA r- )r0 �NannyNagZ
get_linenoZget_msgZget_liner. r/ )r r1 Znannyr- rC r@ rA r r r �test_all_methodsv s&
�
��
�..z TestNannyNag.test_all_methodsN)r' r( r) rE r r r r r> u s r> c @ sf e Zd Zdd� Zddd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dS )� TestCheckc C s | � ttdtj� dt_d S )N�verboser )Z
addCleanup�setattrr0 rG )r r r r �setUp� s zTestCheck.setUpr c C sz t � �@}t� �}t�|� W d � n1 s.0 Y W d � n1 sL0 Y | �|�� |� | �|�� |� d S r )r r r0 �checkr/ r: )r Z dir_or_file�out�err�stdoutr< r r r �verify_tabnanny_check� s Fz TestCheck.verify_tabnanny_checkc C s: t td ��}| �|� W d � n1 s,0 Y d S �Nr �r � SOURCE_CODESrN �r r r r r �test_correct_file� s zTestCheck.test_correct_filec C s~ t �� ��^}|�d�g}ttd |d�}ttd |d�}|��
}|��}||fD ]}|�|�d�� qNdt_t� �@}t� �} t� |� W d � n1 s�0 Y W d � n1 s�0 Y |�
� }|D ]<}
| j |
d�� | � |
|� W d � q�1 s�0 Y q�| �
| �
� d� W d � n1 �s00 Y W d � n1 �sP0 Y W d � n1 �sp0 Y d S )Nz: listing directory
r �r z: Clean bill of health.
r6 )rA r )r �TemporaryDirectoryr rQ �appendr0 rG r r rJ r: r. �assertInr/ ) r �tmp_dir�linesZfile1Zfile2Z
file1_pathZ
file2_pathr rM r< rA r r r � test_correct_directory_verbose� s F,z(TestCheck.test_correct_directory_verbosec C sf t �� �J}ttd |d�� | �|� W d � n1 s:0 Y W d � n1 sX0 Y d S )Nr rT �r rU r rQ rN )r rX r r r �test_correct_directory� s
z TestCheck.test_correct_directoryc C sP t td ��0}d}|�d|� �}| j||d� W d � n1 sB0 Y d S )Nr
�Iunindent does not match any outer indentation level (<tokenize>, line 3)
�: Indentation Error: �rL rP �r r rL r r r �test_when_wrong_indented� s z"TestCheck.test_when_wrong_indentedc C sP t td ��0}d}|�d|� �}| j||d� W d � n1 sB0 Y d S )Nr z(('EOF in multi-line statement', (7, 0))
z: Token Error: r_ rP r` r r r � test_when_tokenize_tokenerror� s z'TestCheck.test_when_tokenize_tokenerrorc C s^ t td ��>}|�d�}|d7 }|d7 }dt_| j||d� W d � n1 sP0 Y d S )Nr z': *** Line 3: trouble in tab city! ***
z%offending line: '\tprint("world")\n'
z$indent not equal e.g. at tab size 1
r6 �rK )r rQ r0 rG rN �r r rK r r r � test_when_nannynag_error_verbose� s
z*TestCheck.test_when_nannynag_error_verbosec C sH t td ��(}|� d�}| j||d� W d � n1 s:0 Y d S )Nr z 3 '\tprint("world")\n'
rc rP rd r r r �test_when_nannynag_error� s
z"TestCheck.test_when_nannynag_errorc C s< d}|�dt j� dt�t j�� d|�d�}| j||d� d S )Nz
no_file.pyz: I/O Error: [Errno z] z: r9 r_ )�errnoZENOENT�os�strerrorrN )r �pathrL r r r �test_when_no_file� s
��zTestCheck.test_when_no_filec
C s� t �� ��}ttd |d�}ttd |d�}|�T}|�0}d}|�d|� �}| j||d� W d � n1 sj0 Y W d � n1 s�0 Y W d � n1 s�0 Y d S )Nr
rT r r] r^ r_ r[ )r rX Z
error_fileZ code_fileZe_fileZc_filerL r r r �test_errored_directory� s
�� z TestCheck.test_errored_directoryN)r r )r' r( r) rI rN rS rZ r\ ra rb re rf rk rl r r r r rF � s
rF c @ s& e Zd Ze�d�dd� �Zdd� ZdS )�TestProcessTokensztabnanny.NannyNagc C sv t td ��V}t|��"}t�t�|j�� W d � n1 s>0 Y | �|j � W d � n1 sh0 Y d S rO )
r rQ �openr0 �process_tokens�tokenize�generate_tokens�readlineZ assertFalseZcalled)r Z MockNannyNagr r! r r r �test_with_correct_code s
0z(TestProcessTokens.test_with_correct_codec C s� dD ]�}| j |d��� tt| ��z}t|��R}t�|j�}| �tj �� t�
|� W d � n1 sf0 Y W d � n1 s�0 Y W d � n1 s�0 Y W d � q1 s�0 Y qd S )N)r
r )�key) r. r rQ rn rp rq rr Z assertRaisesr0 rD ro )r rt r r! �tokensr r r � test_with_errored_codes_samples s
z1TestProcessTokens.test_with_errored_codes_samplesN)r' r( r) r Zpatchrs rv r r r r rm s
rm c @ sN e Zd Zdddd�dd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Zdd� Z dS )�TestCommandLiner F)rM r< �partialc
G s� t jddg|�R � \}}}t�|�}t�|�}|r�||f||ffD ]X\}} | �� }
|�� D ]>} | j| |
d�� | �| |
� W d � qZ1 s�0 Y qZqBn(| �|�� |�� � | �|�� |�� � d S )Nz-mr0 )�std�output)r Zassert_python_okrh �fsdecode�
splitlinesr. rW ZassertListEqual) r rM r< rx r; �_rK rL ry rz Z_outputZ_stdr r r � validate_cmd! s
0z TestCommandLine.validate_cmdc C sP t td ��0}|�d�}|d7 }| j||d� W d � n1 sB0 Y d S )Nr
r^ zHunindent does not match any outer indentation level (<tokenize>, line 3)�r< �r rQ r~ )r r r< r r r �test_with_errored_file2 s
z&TestCommandLine.test_with_errored_filec C s: t td ��}| �|� W d � n1 s,0 Y d S rO r� rR r r r �test_with_error_free_file: s z)TestCommandLine.test_with_error_free_filec C s$ t d�}d|� d�}| j|d� d S )Nz tabnanny.pyzUsage: z [-v] file_or_directory ...r )r r~ )r rj r< r r r �test_command_usage? s z"TestCommandLine.test_command_usagec C sJ t td ��*}|� d�}| jd||d� W d � n1 s<0 Y d S )Nr r9 z-q)rM r� )r r rM r r r �test_quiet_flagE s
z TestCommandLine.test_quiet_flagc C sP t td ��0}t�d��� }| jd||dd� W d � n1 sB0 Y d S )Nr �$offending line: '\tprint("world")\n'z-vT�rM rx �r rQ �textwrap�dedent�stripr~ �r rj rM r r r �test_verbose_modeK s
�z!TestCommandLine.test_verbose_modec C sP t td ��0}t�d��� }| jd||dd� W d � n1 sB0 Y d S )Nr r� z-vvTr� r� r� r r r �test_double_verbose_modeS s
�z(TestCommandLine.test_double_verbose_modeN)
r' r( r) r~ r� r� r� r� r� r� r r r r rw s rw )Zunittestr r rg rh r0 rp r r� Z test.supportr r r r r rQ r r* r3 r> rF rm rw r r r r �<module> s* �-p