Your IP : 216.73.216.196
o
0�i�5 � @ s� d dl mZmZ d dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
m Z m Z m
Z
d dlmZ ddddd d
d �ZG d d
� d
�ZG dd� 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 Z d dd�Zdd� Zdd� ZdS )
�TemporaryPyFile� Nc C s || _ || _d S �N)� source_code�dir)�selfr � directory� r �9/opt/alt/python310/lib64/python3.10/test/test_tabnanny.py�__init__B s
zTemporaryPyFile.__init__c C sN t jd| jddd��}|�| j� W d � n1 sw Y |j| _| jS )N�wz.pyF)�moder �suffix�delete)�tempfileZNamedTemporaryFiler �writer �name� file_path)r �fr r r � __enter__F s
� �zTemporaryPyFile.__enter__c C s t | j� d S r )r r )r �exc_type� exc_value�
exc_tracebackr r r �__exit__N s zTemporaryPyFile.__exit__)r N)�__name__�
__module__� __qualname__r r" r& r r r r r ? s
r c @ � e Zd Z dd� ZdS )�TestFormatWitnessesc C s\ g d�}|D ]$\}}| j ||d�� | �t�|�|� W d � n1 s&w Y qd 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_witnessesU s ���z)TestFormatWitnesses.test_format_witnessesN)r' r( r) r3 r r r r r+ R � r+ c @ r* )� TestErrPrintc
C s� ddgdfdgdfg d�dfg dfg}|D ];\}}| j ||d��( t� �
}tj|� W d � n1 s5w Y | �|�� |� W d � n1 sLw Y qd S ) N�first�secondz
first second
zfirst
)� � � z1 2 3
�
)Z argumentsr. )r/ r r1 Zerrprintr0 �getvalue)r r2 �argsr. �stderrr r r �
test_errprintf s
� � ���zTestErrPrint.test_errprintN)r' r( r) r? r r r r r5 c r4 r5 c @ s e Zd Zdd� ZdS )� TestNannyNagc C s& t �ddd�dddd�ft �ddd�dddd�fg}|D ]s\}}|�� }|�� }|�� }| j||d d �� | �|d |� W d � n1 sIw Y | j||d
d �� | �|d
|� W d � n1 sjw Y | j||d d
�� | �|d |� W d � n1 s�w Y q d S )Nr ZfooZbar)�lineno�msg�line� ZtestmsgZtestlinerA )� line_numberr. rB )rB r. rC )rC r. )r1 �NannyNagZ
get_linenoZget_msgZget_liner/ r0 )r r2 Znannyr. rE rB rC r r r �test_all_methodsw s2
�
��
� � � ���z TestNannyNag.test_all_methodsN)r' r( r) rG r r r r r@ v s r@ c @ sh e Zd Z dd� 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�setattrr1 rI )r r r r �setUp� s
zTestCheck.setUpr c C s| t � � }t� �
}t�|� W d � n1 sw Y W d � n1 s'w Y | �|�� |� | �|�� |� d S r )r r r1 �checkr0 r< )r Z dir_or_file�out�err�stdoutr> r r r �verify_tabnanny_check� s �� z TestCheck.verify_tabnanny_checkc C �> t td ��}| �|� W d � d S 1 sw Y d S �Nr �r � SOURCE_CODESrP �r r r r r �test_correct_file� � "�zTestCheck.test_correct_filec C s� t �� ��}|�d�g}ttd |d�}ttd |d�}|��}|�r}||fD ]
}|�|�d�� q&dt_t� � }t� �
} t� |� W d � n1 sKw Y W d � n1 sZw Y |�
� }|D ] }
| j |
d�� | � |
|� W d � n1 s~w Y qe| �
| �
� d� W d � n1 s�w Y W d � n1 s�w Y W d � d S W d � d S 1 s�w Y d S )Nz: listing directory
r �r z: Clean bill of health.
r8 )rC r )r �TemporaryDirectoryr rT �appendr1 rI r r rL r< r/ �assertInr0 ) r �tmp_dir�linesZfile1Zfile2Z
file1_pathZ
file2_pathr rO r> rC r r r � test_correct_directory_verbose� s6
�� �� �� �"�z(TestCheck.test_correct_directory_verbosec C sz t �� �.}ttd |d��
| �|� W d � n1 s w Y W d � d S W d � d S 1 s6w Y d S )Nr rX �r rY r rT rP )r r\ r r r �test_correct_directory� s
��"�z TestCheck.test_correct_directoryc C �T t td ��}d}|�d|� �}| j||d� W d � d S 1 s#w Y d S )Nr
�Iunindent does not match any outer indentation level (<tokenize>, line 3)
�: Indentation Error: �rN rS �r r rN r r r �test_when_wrong_indented� s "�z"TestCheck.test_when_wrong_indentedc C ra )Nr z(('EOF in multi-line statement', (7, 0))
z: Token Error: rd rS re r r r � test_when_tokenize_tokenerror� s "�z'TestCheck.test_when_tokenize_tokenerrorc C sb t td �� }|�d�}|d7 }|d7 }dt_| j||d� W d � d S 1 s*w 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
r8 �rM )r rT r1 rI rP �r r rM r r r � test_when_nannynag_error_verbose� s
"�z*TestCheck.test_when_nannynag_error_verbosec C sL t td ��}|� d�}| j||d� W d � d S 1 s w Y d S )Nr z 3 '\tprint("world")\n'
rh rS ri 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: r; rd )�errnoZENOENT�os�strerrorrP )r �pathrN r r r �test_when_no_file� s
��zTestCheck.test_when_no_filec
C s� t �� �U}ttd |d�}ttd |d�}|�*}|�}d}|�d|� �}| j||d� W d � n1 s6w Y W d � n1 sEw Y W d � d S W d � d S 1 s]w Y d S )Nr
rX r rb rc rd r_ )r r\ Z
error_fileZ code_fileZe_fileZc_filerN r r r �test_errored_directory� s$
�� �� �"�z TestCheck.test_errored_directoryN)r r )r' r( r) rK rP rV r^ r` rf rg rj rk rp rq r r r r rH � s
rH c @ s( e Zd Z e�d�dd� �Zdd� ZdS )�TestProcessTokensztabnanny.NannyNagc C sz t td ��,}t|��}t�t�|j�� W d � n1 s w Y | �|j � W d � d S 1 s6w Y d S rR )
r rT �openr1 �process_tokens�tokenize�generate_tokens�readlineZ assertFalseZcalled)r Z MockNannyNagr r! r r r �test_with_correct_code s
�"�z(TestProcessTokens.test_with_correct_codec C s� dD ]c}| j |d��S tt| ��=}t|��)}t�|j�}| �tj ��
t�
|� W d � n1 s4w Y W d � n1 sCw Y W d � n1 sRw Y W d � n1 saw Y qd S )N)r
r )�key) r/ r rT rs ru rv rw Z assertRaisesr1 rF rt )r ry r r! �tokensr r r � test_with_errored_codes_samples
s"
�� �� �� ���z1TestProcessTokens.test_with_errored_codes_samplesN)r' r( r) r Zpatchrx r{ r r r r rr s
rr c @ sP e Zd Z dddd�dd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Zdd� Z dS )�TestCommandLiner F)rO r> �partialc
G s� t jddg|�R � \}}}t�|�}t�|�}|rQ||f||ffD ],\}} | �� }
|�� D ] } | j| |
d�� | �| |
� W d � n1 sHw Y q.q"d S | �|�� |�� � | �|�� |�� � d S )Nz-mr1 )�std�output)r Zassert_python_okrm �fsdecode�
splitlinesr/ r[ ZassertListEqual) r rO r> r} r= �_rM rN r~ r Z_outputZ_stdr r r � validate_cmd" s
����z TestCommandLine.validate_cmdc C sT t td ��}|�d�}|d7 }| j||d� W d � d S 1 s#w Y d S )Nr
rc zHunindent does not match any outer indentation level (<tokenize>, line 3)�r> �r rT r� )r r r> r r r �test_with_errored_file3 s
"�z&TestCommandLine.test_with_errored_filec C rQ rR r� rU r r r �test_with_error_free_file; rW 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 ro r> r r r �test_command_usage@ s z"TestCommandLine.test_command_usagec C sN t td ��}|� d�}| jd||d� W d � d S 1 s w Y d S )Nr r; z-q)rO r� )r r rO r r r �test_quiet_flagF s
"�z TestCommandLine.test_quiet_flagc C �T t td ��}t�d��� }| jd||dd� W d � d S 1 s#w Y d S )Nr �$offending line: '\tprint("world")\n'z-vT�rO r} �r rT �textwrap�dedent�stripr� �r ro rO r r r �test_verbose_modeL � ��"�z!TestCommandLine.test_verbose_modec C r� )Nr r� z-vvTr� r� r� r r r �test_double_verbose_modeT r� z(TestCommandLine.test_double_verbose_modeN)
r' r( r) r� r� r� r� r� r� r� r r r r r| s r| )Zunittestr r rl rm r1 ru r r� Z test.supportr r r r Ztest.support.os_helperr rT r r+ r5 r@ rH rr r| r r r r �<module> s0 �-p