|
Server : Apache System : Linux 182.49.167.72.host.secureserver.net 4.18.0-553.92.1.el8_10.x86_64 #1 SMP Wed Jan 14 06:31:58 EST 2026 x86_64 User : kikai ( 1014) PHP Version : 8.3.30 Disable Function : NONE Directory : /opt/python38/lib/python3.8/lib2to3/tests/__pycache__/ |
Upload File : |
U
�i�_�0 � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl m
Z
m Z m Z ddl
mZ ej�ej�e�d�Zej�ed�Zej�e� ze
�d�ZW 5 ej�� X e
�d�ZG d d
� d
ej�ZdS ) z
Unit tests for refactor.py.
� N)�refactor�pygram�
fixer_base)�token�data�fixers�myfixes�
lib2to3.fixesc @ s� e Zd Zdd� Zdd� Zdd� Zdedfdd �Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zeddd fd d �Zd d � Zd!d"� Zefd#d$�Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Z d3d4� Z d5d6� Z d7d8� Z dS )9�TestRefactoringToolc C s t j�t� d S �N)�sys�path�append� FIXER_DIR��self� r �:/opt/python38/lib/python3.8/lib2to3/tests/test_refactor.py�setUp s zTestRefactoringTool.setUpc C s t j�� d S r )r r
�popr r r r �tearDown" s z TestRefactoringTool.tearDownc C s2 t ||�D ]"\}}t||�s
| �d| |� q
d S )Nz%s are not instances of %s)�zip�
isinstance�fail)r Z instances�classes�inst�clsr r r �check_instances% s
z#TestRefactoringTool.check_instancesNc C s t �|||�S r )r �RefactoringTool)r �optionsr �explicitr r r �rt* s zTestRefactoringTool.rtc C s4 | � ddi�}| �|jtj� | �|jjtj� d S )N�print_functionT)r! ZassertIsZgrammarr Z!python_grammar_no_print_statementZdriver�r r! r r r �test_print_function_option- s
�z.TestRefactoringTool.test_print_function_optionc C s2 | � � }| �|j� | � ddi�}| �|j� d S )N�write_unchanged_filesT)r! Z assertFalser% �
assertTruer# r r r �!test_write_unchanged_files_option3 s z5TestRefactoringTool.test_write_unchanged_files_optionc C sj dddddg}t �d�}t �dd�}t �d�}| �|dd � |D �� | �||� | �|d
d � |D �� d S ) Nr �firstZlastZparrotZpreorderr Fc S s g | ] }d | �qS )Zfix_r ��.0�namer r r �
<listcomp>> s zBTestRefactoringTool.test_fixer_loading_helpers.<locals>.<listcomp>c S s g | ] }d | �qS )z myfixes.fix_r r) r r r r, A s )r Zget_all_fix_names�get_fixers_from_package� assertEqual)r �contentsZ non_prefixedZprefixedZ
full_namesr r r �test_fixer_loading_helpers9 s
�z.TestRefactoringTool.test_fixer_loading_helpersc C s~ t j}t}|� }| �|d�|� | �|d�|d�� | �|d�|d�� | �|d�|d�� d}| �||�|d �� d
}| �||�|d �� d }| �||�|d�� d
}| �||�|d �� d}| �||�|d�� d}| �||�|d �� d}|D ]}| �||�|� q�d}| �||�|d�� d}| �||�|� d}| �||�|d�� d}| �||�|d�� d}| �||�|� d S )N� z%from __future__ import print_function)r" z!from __future__ import generators)�
generatorsz*from __future__ import generators, feature)r2 Zfeaturez1from __future__ import generators, print_function)r2 r" z1from __future__ import print_function, generators)r" r2 z(from __future__ import (print_function,)z3from __future__ import (generators, print_function)z2from __future__ import (generators, nested_scopes))r2 Z
nested_scopeszGfrom __future__ import generators
from __future__ import print_function)�fromzfrom 4zfrom xzfrom x 5z from x imz
from x importzfrom x import 4z1'docstring'
from __future__ import print_functionz9'docstring'
'somng'
from __future__ import print_functionz/# comment
from __future__ import print_functionz5# comment
'doc'
from __future__ import print_functionz3class x: pass
from __future__ import print_function)r Z_detect_future_features� frozensetr. )r �run�fs�emptyZinp�invalidr r r �test_detect_future_featuresC sL
�
�
�z/TestRefactoringTool.test_detect_future_featuresc C s� G dd� dt j�}G dd� dt j�}G dd� dt j�}|i g �}|i g �}|i g �}t�|||g�}|�tjj�}| �|||g� |�t j
�} | �| ||g� |� � D ]}
| �|
|g� q�d S )Nc @ s e Zd ZdS )z;TestRefactoringTool.test_get_headnode_dict.<locals>.NoneFixN)�__name__�
__module__� __qualname__r r r r �NoneFixq s r= c @ s e Zd ZdZdS )z@TestRefactoringTool.test_get_headnode_dict.<locals>.FileInputFixzfile_input< any * >N�r: r; r< ZPATTERNr r r r � FileInputFixt s r? c @ s e Zd ZdZdS )z=TestRefactoringTool.test_get_headnode_dict.<locals>.SimpleFixz'name'Nr> r r r r � SimpleFixw s r@ ) r ZBaseFixr Z_get_headnode_dictr r Zpython_symbolsZ
file_inputr. r �NAME�values) r r= r? r@ Zno_headZ with_headZsimple�dZ top_fixesZ
name_fixes�fixesr r r �test_get_headnode_dictp s
z*TestRefactoringTool.test_get_headnode_dictc C sh ddl m} ddlm} ddlm} ddlm} | �� }|� � \}}| �
||g� | �
||||g� d S )Nr )�FixFirst)�FixLast)� FixParrot)� FixPreorder) Zmyfixes.fix_firstrF Zmyfixes.fix_lastrG Zmyfixes.fix_parrotrH Zmyfixes.fix_preorderrI r! Z
get_fixersr )r rF rG rH rI r! �preZpostr r r �test_fixer_loading� s z&TestRefactoringTool.test_fixer_loadingc C sD | j t| jdgd� | j tj| jdgd� | j tj| jdgd� d S )NZnot_here�r Z no_fixer_clsZ bad_order)Z assertRaises� ImportErrorr! r Z
FixerErrorr r r r �test_naughty_fixers� s z'TestRefactoringTool.test_naughty_fixersc C sL | � � }d}|�|d�}| �t|�|� d}|�|d�}| �t|�|� d S )N�def parrot(): pass
�<test>zdef f(): pass
)r! Zrefactor_string�assertNotEqual�strr. )r r! �input�treer r r �test_refactor_string� s z(TestRefactoringTool.test_refactor_stringc sf G � fdd�dt j�}g � |t�}tj}t�d�t_z |�� W 5 |t_X ddddg}| �� |� d S )Nc s e Zd Z� fdd�ZdS )z5TestRefactoringTool.test_refactor_stdin.<locals>.MyRTc s � � ||||g� d S r )�extend)r Zold_textZnew_text�filenameZequal��resultsr r � print_output� s zBTestRefactoringTool.test_refactor_stdin.<locals>.MyRT.print_outputN)r: r; r< rZ r rX r r �MyRT� s r[ rO zdef cheese(): pass
z<stdin>F) r r �_DEFAULT_FIXERSr �stdin�io�StringIOZrefactor_stdinr. )r r[ r! �save�expectedr rX r �test_refactor_stdin� s �z'TestRefactoringTool.test_refactor_stdinTc C sv | � |�}| �|�}| j||d�}|r,||_|�|� | �|| �|�� |sPd S |�|d� | �|�}| �||� |S )N)r r T)�init_test_file� read_filer! Z log_debug�
refactor_filer. rQ ) r � test_filer r �mock_log_debug�actually_write� old_contentsr! � new_contentsr r r �check_file_refactoring� s
z*TestRefactoringTool.check_file_refactoringc C sL t jdd�}| �tj|� t�||� tj�|tj� |��}t�
|d� |S )N�2to3-test_refactor��prefixi� ) �tempfile�mkdtempZ
addCleanup�shutil�rmtree�copy�osr
�join�basename�chmod)r rf Ztmpdirr r r rc � s z"TestRefactoringTool.init_test_filec
C s* t |d��}|�� W 5 Q R � S Q R X d S )N�rb)�open�read)r rf �fpr r r rd � s z TestRefactoringTool.read_filec C s> | � |�}| �|�}| j|d�}|�|d� | �|�}||fS )NrL T)rc rd r! re )r rf r ri r! rj r r r re � s
z!TestRefactoringTool.refactor_filec C s t j�td�}| �|t� d S )N�parrot_example.py)rt r
ru r rk r\ )r rf r r r �test_refactor_file� s z&TestRefactoringTool.test_refactor_filec s� t j�td�}g � � fdd�}| j|dddi|dd� d t�t jt j�|� � }� D ] }d
|krV| � ||� q�qV| �
d |� f � d S ) Nr| c s � � | | � d S r �r )�msg�args�Zdebug_messagesr r �recording_log_debug� s zXTestRefactoringTool.test_refactor_file_write_unchanged_file.<locals>.recording_log_debugr r% TF)r r rg rh zNot writing changes to .*%szNot writing changesz%r not matched in %r) rt r
ru r rk �re�escape�seprv Z assertRegexr )r rf r� Z
message_regex�messager r� r �'test_refactor_file_write_unchanged_file� s �� z;TestRefactoringTool.test_refactor_file_write_unchanged_filec sV � fdd�}|g g � ddddddg}dg}|||� dt j�d d
�g}|||� d S ) Nc s� �fdd�}t jj}|t j_��� }g �tjdd�� zDt� tj
� � d�� | D ] }t tj
� � |�d��
� qN|�� � W 5 |t j_t�� � X ���� fdd�|D �� d S ) Nc s � � |� d S r r~ )r �fr� )�gotr r �mock_refactor_file� s zPTestRefactoringTool.test_refactor_dir.<locals>.check.<locals>.mock_refactor_filerl rm �a_dir�wbc s g | ]}t j�� |��qS r �rt r
ru )r* r
)�dirr r r, s zHTestRefactoringTool.test_refactor_dir.<locals>.check.<locals>.<listcomp>)r r re r! ro rp rq rr rt �mkdirr
ru ry �closeZ refactor_dirr. )Z structurera r� Z save_funcr! �fnr )r� r� r �check� s �z4TestRefactoringTool.test_refactor_dir.<locals>.checkZnothingzhi.pyz.dumbz .after.pyz notpy.npyZsappyr� zstuff.pyr� )r r� rT ra r r r �test_refactor_dir� s
�
�z%TestRefactoringTool.test_refactor_dirc C s t j�td�}| �|� d S )Nzdifferent_encoding.py�rt r
ru �
TEST_DATA_DIRrk )r r� r r r �test_file_encoding s z&TestRefactoringTool.test_file_encodingc C s t j�td�}| �|�}d S )Nzfalse_encoding.pyr� �r r� r r r r �test_false_file_encoding s z,TestRefactoringTool.test_false_file_encodingc C s. t j�td�}| �|�}| �|�tj�� d S )Nzbom.py) rt r
ru r� rk r& �
startswith�codecs�BOM_UTF8r� r r r �test_bom s
z TestRefactoringTool.test_bomc C sB t j}dt _z(t j�td�}t�d�}| �||� W 5 |t _X d S )Nz
�crlf.pyr )rt �linesepr
ru r� r r- rk )r Zold_sepr� rD r r r �test_crlf_newlines s
z&TestRefactoringTool.test_crlf_newlinesc C sD t j�td�}| �|�\}}| �d|� | �d|� | �d|� d S )Nr� s
s
)rt r
ru r� re ZassertInZ assertNotIn)r r� �old�newr r r �test_crlf_unchanged* s
z'TestRefactoringTool.test_crlf_unchangedc C sD | � � }d}|�|d�}| �||� d}|�|d�}| �||� d S )Nz
>>> example()
42
rP z&
>>> def parrot():
... return 43
)r! Zrefactor_docstringr. rQ )r r! �doc�outr r r �test_refactor_docstring1 s z+TestRefactoringTool.test_refactor_docstringc C sb ddl m} | jdgd�}| �t|j�d� | jdgd�}|jD ]}t||�r@ q^q@| �d� d S )Nr )� FixExplicitzmyfixes.fix_explicitrL )r zexplicit fixer not loaded)Zmyfixes.fix_explicitr� r! r. �lenZ
post_orderr r )r r� r! Zfixr r r �
test_explicitB s
z!TestRefactoringTool.test_explicit) r: r; r< r r r r\ r! r$ r' r0 r9 rE rK rN rU rb � _2TO3_FIXERSrk rc rd re r} r� r� r� r� r� r� r� r� r� r r r r r
s>
-
�
r
) �__doc__r rt r� r^ r� ro rq ZunittestZlib2to3r r r Z
lib2to3.pgen2r r
ru �dirname�__file__r� r r r r- r\ r� ZTestCaser
r r r r �<module> s$
XYZEAZ - Cyber Alpha Wolf © All Rights Reserved