Your IP : 216.73.216.213
o
0�i�� � @ st d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z zddl
Z
W n e y7 dZ
Y nw zddl Z W n e yI dZ Y nw ddlm
Z
mZ ddlmZmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lm Z dd l m Z G d
d� d�Z G dd� d�Z G dd� de e j �Z!G dd� d�Z"G dd� de e j �Z#G dd� d�Z$G dd� de j �Z%G dd � d e%�Z&G d d � d e%�Z'G d d � d e%�Z(G d!d"� d"e%�Z)G d#d$� d$e%�Z*G d%d&� d&e%�Z+G d'd(� d(e%�Z,G d)d*� d*e%�Z-G d+d,� d,e%�Z.G d-d.� d.�Z/G d/d0� d0e j �Z0G d1d2� d2e j �Z1G d3d4� d4e j �Z2e3d5k�r8e �4� dS dS )6z,
Tests for fileinput module.
Nick Mathewson
� N)�BytesIO�StringIO)� FileInput� hook_encoded)�Path)�verbose)�TESTFN)�unlink)� os_helper)�warnings_helper)�support)�mockc @ s e Zd Zdd�dd�ZdS )� BaseTests�w��modec C sh t �� \}}| �tj|� d|v rd nd}t|||d��}|�|� W d � |S 1 s-w Y |S )N�b�utf-8��encoding)�tempfileZmkstemp�
addCleanupr
r �open�write)�selfZcontentr �fd�namer �f� r �:/opt/alt/python310/lib64/python3.10/test/test_fileinput.py�writeTmp, s
��zBaseTests.writeTmpN)�__name__�
__module__� __qualname__r r r r r r ) s r c @ sD e Zd Zdd� Zedd� �Zdd� Zddd �Zdd d �Zd
d� Z dS )�
LineReaderc C s
g | _ d S �N��
_linesread�r r r r �__init__6 �
zLineReader.__init__c C s z | j d d � W g | _ S g | _ w r% r& r( r r r � linesread9 s zLineReader.linesreadc C s t |�d��| _| S �NT)�iter�
splitlines�it)r �filenamer r r r �openhook@ s zLineReader.openhookNc C s t | jd�}| j�|� |S )N� )�nextr/ r' �append)r �size�liner r r �readlineD s zLineReader.readline���c C s> g }d} | � � }|s
|S |�|� |t|�7 }||kr |S q�Nr )r7 r4 �len)r Zhint�linesr5 r6 r r r � readlinesI s
�zLineReader.readlinesc C � d S r% r r( r r r �closeU s zLineReader.closer% )r8 )
r! r" r# r) �propertyr+ r1 r7 r< r> r r r r r$ 4 s
r$ c @ � e Zd Zdd� ZdS )�BufferSizesTestsc
C s� | � d�dd� td�D ���}| � d�dd� td�D ���}| � d�dd� td�D ���}| � d�d d� td
�D ���}t�d �}trGtd � t||||fd
d�}t|�}|� � | �
t |�d� | �
|d d� | �
|d d� | �
|� � d� | �
|�
� |� tr�td� t||||fd
d�}d}|r�|dkr�|�� }|r�|dks�| �
|�
� |� | �
|� � d� | �
|�� d� | �|�� � | �|�� � tr�td� |�� | �
|�� d� | �
|� � d� |� � tr�td � t||||d fd
d�}tj} z*td �t_t|�}| �
t |�d � | �
|d d!� | �
|�
� d"� |�� W | t_n| t_w t�r1td#� t||||fd
d�}| �
|� � d$� | �
|�
� d � |�� | �
|� � d$� | �
|�
� d � t�rftd%� tj}
z&t||||fd
d
d&�}|D ]} | d d'� �� } t| � �qw|� � W |
t_n|
t_w t||||fd
d�}|D ]'} | �
| d' d(� |�| d d'� �} | �| d � | �
t| � d
��|�� � �q�|� � d S ))Nr2 c s � � | ] }d |d V qdS )zLine %s of file 1
� Nr ��.0�ir r r � <genexpr>[ � � z5BufferSizesTests.test_buffer_sizes.<locals>.<genexpr>� c s rB )zLine %s of file 2
rC Nr rD r r r rG \ rH �
c s rB )zLine %s of file 3
rC Nr rD r r r rG ] rH � c s rB )zLine %s of file 4
rC Nr rD r r r rG ^ rH rC zLINE (\d+) OF FILE (\d+)z1. Simple iterationr ��filesr � � zLine 5 of file 1
� zLine 1 of file 4
z2. Status variables�xzLine 6 of file 2
� � z 3. NextfilezLine 1 of file 3
� z4. Stdin�-z Line 1 of stdin
Line 2 of stdin
�! � zLine 2 of stdin
z<stdin>z5. Boundary conditionsr z
6. Inplace�rM �inplacer r8 �
) r �join�range�re�compiler �printr �listr> � assertEqualr: �linenor0 r7 �
filelineno� assertFalse� isfirstline�isstdin�nextfile�sys�stdinr �stdout�upper�match�assertNotEqual�int�group)
r �t1�t2�t3�t4Zpat�fir; �sZ savestdinZ
savestdoutr6 �mr r r �test_buffer_sizesY s�
�
z"BufferSizesTests.test_buffer_sizesN)r! r" r# rw r r r r rA X s rA c @ � e Zd Zdd� Zdd� ZdS )�UnconditionallyRaisec C s || _ d| _d S �NF)�exception_type�invoked)r r{ r r r r) � s
z UnconditionallyRaise.__init__c O s d| _ | �� �r, )r| r{ �r �args�kwargsr r r �__call__� s z UnconditionallyRaise.__call__N�r! r" r# r) r� r r r r ry � s ry c @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zejed �d d!� �Zd"d#� Zejed �d$d%� �Zejed �d&d'� �Zd(d)� Zd*d+� Zd,d-� Zd.d/� Z d0d1� Z d2d3� Z d4d5� Z d6d7� Z d8S )9�FileInputTestsc C s� | � d�}| � d�}| � d�}| � d�}t||||fdd�}|�� }| �|d� | �|�� d� | �|�� d� | �|�� |� |�� }| �|� | �|�� d� | �|�� d� | �|�� |� |�� d S )Nr2 zThe only line there is.
r rL rC r ) r r r7 ra rb rc r0 rd r> )r rp rq rr rs rt r6 r r r �test_zero_byte_files� s
z#FileInputTests.test_zero_byte_filesc C s` | � d�}| � d�}t||fdd�}t|�}| �|g d�� | �|�� d� | �|�� d� d S )N�A
B
C�D
E
Fr rL ��A
�B
�CzD
zE
�F� rS )r r r` ra rc rb �r rp rq rt r; r r r �%test_files_that_dont_end_with_newline� s
z4FileInputTests.test_files_that_dont_end_with_newlinec C s� | � d�}| � d�}t||fdd�}| �|�� d� t|�}| �|�� d� |�� | �|�� d� t|�}| �|�� d� d S )NzA
BzC
Dr rL r8 )r r ra �filenor3 rm rg r` )r rp rq rt r6 r r r � test_fileno� s
zFileInputTests.test_filenoc C s� z
t ddd�}| �d� W n ty Y nw | jddd�}t�dtf�� t |d dd
�}W d � n1 s7w Y t�dtf�� t|�}W d � n1 sRw Y | �|g d �� d S ) Nr r )r r z-FileInput should reject invalid mode argument� A
B
C
D�wbr r2 �U)rM r r )r� r� �C
�D) r �fail�
ValueErrorr r Zcheck_warnings�DeprecationWarningr` ra )r rt rp r; r r r �test_opening_mode� s � �
�z FileInputTests.test_opening_modec C s^ t �d�� }td�|_tdgdd�}t|�}| �|dg� W d � d S 1 s(w Y d S )Nz sys.stdin� spam, bacon, sausage, and spamrU �rb�rM r )r
Zpatchr �bufferr r` ra )r Zm_stdinrt r; r r r �test_stdin_binary_mode� s
"�z%FileInputTests.test_stdin_binary_modec C s\ t j}z&td�t _| �tt jd�� tdgdd�}t|�}| �|dg� W |t _d S |t _w )Nr� r� rU r� r� )rh ri r rd �hasattrr r` ra )r Z
orig_stdinrt r; r r r � test_detached_stdin_binary_mode� s
z.FileInputTests.test_detached_stdin_binary_modec C s� zt ddd� d�}| �d� W n ty Y nw z t dd�}| �d� W n ty. Y nw G dd � d �}| �d
�}|� }t |g|d�� }|�� W d � n1 sTw Y | �|jd � d S ) NrC c S r= r% r )r rv r r r �<lambda>
� z7FileInputTests.test_file_opening_hook.<locals>.<lambda>)rY r1 zGFileInput should raise if both inplace and openhook arguments are given�r1 z2FileInput should check openhook for being callablec @ rx )z=FileInputTests.test_file_opening_hook.<locals>.CustomOpenHookc S �
d| _ d S rz )r| r( r r r r) r* zFFileInputTests.test_file_opening_hook.<locals>.CustomOpenHook.__init__c _ s d| _ t|ddi�S )NTr r )r| r )r r~ Zkargsr r r r� s zFFileInputTests.test_file_opening_hook.<locals>.CustomOpenHook.__call__Nr� r r r r �CustomOpenHook � r� rZ zopenhook not invoked)r r� r� r r7 �
assertTruer| )r rt r� �tZcustom_open_hookr r r �test_file_opening_hook
s& �
�
�z%FileInputTests.test_file_opening_hookc C s2 t td��}|�d� |�dd � |�d� W d � n1 s!w Y | �tt� tttd�d��[}z| �|�� d� | �|�� d � | �|�� d
� W n t y] | �
d � Y nw | � t �� t |� W d � n1 srw Y | �|�� d � | �|�� d � W d � d S 1 s�w Y d S )
Nr� s A
B
C
s
123456789
i� � ��ascii�rM r1 r� r� r� zRead to end of filer2 )
r r r r � safe_unlinkr r ra r7 �UnicodeDecodeErrorr� � assertRaisesr` �r r rt r r r �
test_readline% s.
� � �
�"�z FileInputTests.test_readlinec C s� t td��
}|�d� W d � n1 sw Y | �tt� ttdd��1}| �|�� d� | �|�� d� | �|�� d� | �|�� d� | �|�� d� W d � d S 1 sZw Y d S ) Nr� r� r� r� s A
s B
s C
D� )r r r r r� r ra r7 r� r r r �test_readline_binary_mode< s � "�z(FileInputTests.test_readline_binary_modec C s� | j ddd�}t|ddd��}|�� }| �|d� tj�d� W d � n1 s)w Y t|d��}| �|�� d� W d � d S 1 sGw Y d S )Ns
Initial text.r� r r� T)r rY s New line.) r r r7 ra rh rj r r �read�r Z temp_fileZfobjr6 r r r r � test_inplace_binary_write_modeI s � "�z-FileInputTests.test_inplace_binary_write_modec C s� | j ddd�}t|dddd��}|�� }| �|d� td � W d � n1 s(w Y t|d
��}| �|�� �d �d � W d � d S 1 sIw Y d S )
Ns Initial text �r� r Tr� �replace)rY r �errorsu Initial text �u New line r� s
s
New line ?)r r r7 ra r_ r r� �rstripr� r r r � test_inplace_encoding_errorsS s �
� "�z+FileInputTests.test_inplace_encoding_errorsc C sX dd� }| � d�}t|g|d�� }|�� }W d � n1 s w Y | �|d� d S )Nc S �
t �d�S )N�(I used to receive only filename and mode)�ior �r0 r r r r �old_hook^ r* zFFileInputTests.test_file_hook_backward_compatibility.<locals>.old_hookrZ r� r� )r r r7 ra )r r� r� rt �resultr r r �%test_file_hook_backward_compatibility] s
�z4FileInputTests.test_file_hook_backward_compatibilityc C s� | � d�}| � d�}t||fdd�� }t|�}W d � n1 s!w Y | �|g d�� | �|�� d� | �|�� d� | �|jd� d S ) Nr� r� r rL r� r� rS r )r r r` ra rc rb �_filesr� r r r �test_context_managere s
�z#FileInputTests.test_context_managerc C sV | � d�}zt|dd��}t�1 sw Y W d S ty* | �|jd� Y d S w )Nr2 r rL r )r r �OSErrorra r� )r rp rt r r r �test_close_on_exceptiono s
� �z&FileInputTests.test_close_on_exceptionc C s@ t g dd��}| �|jd� W d � d S 1 sw Y d S )Nr rL )rU )r ra r� )r rt r r r �.test_empty_files_list_specified_to_constructorw s "�z=FileInputTests.test_empty_files_list_specified_to_constructor)�categoryc C sf | � d�}t|gdd�� }|d }| �|d� |d }| �|d� W d � dS 1 s,w Y dS ) zNTests invoking FileInput.__getitem__() with the current
line number� line1
line2
r rL r �line1
rC zline2
N)r r ra )r r� rt Zretval1Zretval2r r r �test__getitem__{ s
"�z FileInputTests.test__getitem__c C s� | � d�}| �td��/ t|gd��}| �|d d� W d � n1 s%w Y W d � d S W d � d S 1 s=w Y d S )Nr� z Use iterator protocol instead)rM r r� )r ZassertWarnsRegexr� r ra )r r� rt r r r �test__getitem___deprecation� s
� ��"�z*FileInputTests.test__getitem___deprecationc C �~ | � d�}t|gdd��!}| �t�� }|d W d � n1 s!w Y W d � n1 s0w Y | �|jjd� dS )zZTests invoking FileInput.__getitem__() with an index unequal to
the line numberr� r rL rC N)z accessing lines out of order)r r r� � RuntimeErrorra � exceptionr~ �r r� rt �cmr r r �test__getitem__invalid_key� �
�� �z)FileInputTests.test__getitem__invalid_keyc C r� )zZTests invoking FileInput.__getitem__() with the line number but at
end-of-inputr2 r rL r N)zend of input reached)r r r� �
IndexErrorra r� r~ r� r r r �test__getitem__eof� r� z!FileInputTests.test__getitem__eofc C s� t j}tt�}z5| �d�}| �t|d � t|gddd��}t|� |t _|� � W d � n1 s3w Y W |t _n|t _w | �
|j d� dS )z�Tests invoking FileInput.nextfile() when the attempt to delete
the backup file would raise OSError. This error is expected to be
silently ignoredrZ z.bakTr rX Nzos.unlink() was not invoked) �osr ry r� r r r� r r3 rg r� r| )r Zos_unlink_origZos_unlink_replacementr� rt r r r �%test_nextfile_oserror_deleting_backup� s
���z4FileInputTests.test_nextfile_oserror_deleting_backupc C �| t j}tt�}z)| �d�}t|gddd��}|t _|�� W d � n1 s'w Y W |t _n|t _w | �|jd� dS )z{Tests invoking FileInput.readline() when os.fstat() raises OSError.
This exception should be silently discarded.rZ Tr rX N�os.fstat() was not invoked) r� �fstatry r� r r r7 r� r| )r Z
os_fstat_origZos_fstat_replacementr� rt r r r �%test_readline_os_fstat_raises_OSError� �
���z4FileInputTests.test_readline_os_fstat_raises_OSErrorc C r� )z{Tests invoking FileInput.readline() when os.chmod() raises OSError.
This exception should be silently discarded.rZ Tr rX Nr� ) r� �chmodry r� r r r7 r� r| )r Z
os_chmod_origZos_chmod_replacementr� rt r r r �%test_readline_os_chmod_raises_OSError� r� z4FileInputTests.test_readline_os_chmod_raises_OSErrorc C s� G dd� dt �}|� }| �d�}t|gdd�� }|j}z ||_|�� }W ||_n||_w W d � n1 s6w Y | �|jd� | �|dd� d S ) Nc @ rx )zQFileInputTests.test_fileno_when_ValueError_raised.<locals>.FilenoRaisesValueErrorc S � t �| t� d S r% )ry r) r� r( r r r r) � � zZFileInputTests.test_fileno_when_ValueError_raised.<locals>.FilenoRaisesValueError.__init__c S s | � � d S r% )r� r( r r r r� � � zXFileInputTests.test_fileno_when_ValueError_raised.<locals>.FilenoRaisesValueError.filenoN)r! r" r# r) r� r r r r �FilenoRaisesValueError� r� r� rZ r rL z _file.fileno() was not invokedr8 zfileno() should return -1)ry r r �_filer� r� r| ra )r r� Z unconditionally_raise_ValueErrorr� rt Z file_backupr� r r r �"test_fileno_when_ValueError_raised� s
� �z1FileInputTests.test_fileno_when_ValueError_raisedc C s� t � }tddg|jd��`}| �|jg � | �|�� d� | �|jdg� | �|�� d� | �|jdg� | �|�� d� | �|jddg� | �|�� d� | �|jdg� | �|�� d� | �|jg � W d � d S 1 sow Y d S �Nz line1
line2zline3
r� r� Zline2r2 )r$ r r1 ra r+ r7 �r �srcrt r r r �test_readline_buffering� s" �"�z&FileInputTests.test_readline_bufferingc C s� t � }tddg|jd��^}| �|jg � | �t|�d� | �|jdg� | �t|�d� | �|jdg� | �t|�d� | �|jddg� | �tt|� | �|jdg� | �tt|� | �|jg � W d � d S 1 smw Y d S r� )r$ r r1 ra r+ r3 r� �
StopIterationr� r r r �test_iteration_buffering s" �"�z'FileInputTests.test_iteration_bufferingc C s� t | �d��}t|dd��.}|�� }| �|d� | �|�� d� | �|�� d� | �|�� t� |�� W d � d S 1 s>w Y d S )N�
Pathlib file.r r rC )
r r r r7 ra rb rc r0 r� �fspath)r rp rt r6 r r r �test_pathlib_file s "�z FileInputTests.test_pathlib_filec C s� t | �d��}t|ddd��}|�� }| �|d� td| � W d � n1 s)w Y t|dd��}| �|�� d� W d � d S 1 sHw Y d S )Nr� Tr )rY r z Modified %sr zModified Pathlib file.
)r r r r7 ra r_ r r� )r rp rt r6 r r r r �test_pathlib_file_inplace s �"�z(FileInputTests.test_pathlib_file_inplaceN)!r! r" r# r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r Zignore_warningsr� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s>
r� c @ sf e Zd ZdZdddddd�dd�Zd d
� Zd d � Zd
d� Zdd� Zdd� Z dd� Z
dd� Z dd� Z dS )�
MockFileInputzDA class that mocks out fileinput.FileInput for use during unit testsNFr2 �r)r r1 r r� c C sJ || _ || _|| _|| _|| _|| _|| _d | _t� dd� �| _
i | _ d S )Nc S s dS r9 r r r r r r� 1 r� z(MockFileInput.__init__.<locals>.<lambda>) rM rY �backupr r1 r r� r� � collections� defaultdict�invocation_counts�
return_values)r rM rY r� r r1 r r� r r r r) ' s
zMockFileInput.__init__c C s | j d d7 < d S )Nr> rC )r� r( r r r r> 4 s zMockFileInput.closec C � | j d d7 < | jd S )Nrg rC �r� r� r( r r r rg 7 �
zMockFileInput.nextfilec C r� )Nr0 rC r� r( r r r r0 ; r� zMockFileInput.filenamec C r� )Nrb rC r� r( r r r rb ? r� zMockFileInput.linenoc C r� )Nrc rC r� r( r r r rc C r� zMockFileInput.filelinenoc C r� )Nr� rC r� r( r r r r� G r� zMockFileInput.filenoc C r� )Nre rC r� r( r r r re K r� zMockFileInput.isfirstlinec C r� )Nrf rC r� r( r r r rf O r� zMockFileInput.isstdin)NFr2 )
r! r" r# �__doc__r) r> rg r0 rb rc r� re rf r r r r r� $ s �
r� c @ �( e Zd ZdZdd� Zdd� Zdd� ZdS ) � BaseFileInputGlobalMethodsTestzQBase class for unit tests for the global function of
the fileinput module.c C s t j| _t j| _tt _d S r% )� fileinput�_state� _orig_stater �_orig_FileInputr� r( r r r �setUpW s
z$BaseFileInputGlobalMethodsTest.setUpc C s | j t_| jt_d S r% )r� r� r r� r� r( r r r �tearDown\ s z'BaseFileInputGlobalMethodsTest.tearDownc C s2 |j | }| �|d|� t|j �}| �|d� d S )NrC )r� ra r: )r Zmock_file_inputZ method_nameZ actual_countZactual_total_countr r r �assertExactlyOneInvocation` s
z9BaseFileInputGlobalMethodsTest.assertExactlyOneInvocationN)r! r" r# r� r� r� r r r r r r� S s
r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �Test_fileinput_inputz Unit tests for fileinput.input()c C sj t � }t� |_|t_| �t�� }t�� W d � n1 s w Y | �d|j j
� | � |tjd� dS )z�Tests invoking fileinput.input() when fileinput._state is not None
and its _file attribute is also not None. Expect RuntimeError to
be raised with a meaningful error message and for fileinput._state
to *not* be modified.N)zinput() already active�fileinput._state) r� �objectr� r� r� r� r� �inputra r� r~ �assertIs)r �instancer� r r r �1test_state_is_not_None_and_state_file_is_not_Nonek s
�zFTest_fileinput_input.test_state_is_not_None_and_state_file_is_not_Nonec C s t � }d|_|t_| �� dS )a\ Tests invoking fileinput.input() when fileinput._state is not None
but its _file attribute *is* None. Expect it to create and return
a new fileinput.FileInput object with all method parameters passed
explicitly to the __init__() method; also ensure that
fileinput._state is set to the returned instance.N)r� r� r� r� �do_test_call_input�r r r r r �-test_state_is_not_None_and_state_file_is_Nonex s zBTest_fileinput_input.test_state_is_not_None_and_state_file_is_Nonec C s dt _| �� dS )a4 Tests invoking fileinput.input() when fileinput._state is None
Expect it to create and return a new fileinput.FileInput object
with all method parameters passed explicitly to the __init__()
method; also ensure that fileinput._state is set to the returned
instance.N)r� r� r r( r r r �test_state_is_None� s z'Test_fileinput_input.test_state_is_Nonec C s� t � }t � }t � }t � }t � }t � }tj||||||d�}| �|tjd� | �||jd� | �||jd� | �||jd� | �||jd� | �||j d� dS ) a Tests that fileinput.input() creates a new fileinput.FileInput
object, passing the given parameters unmodified to
fileinput.FileInput.__init__(). Note that this test depends on the
monkey patching of fileinput.FileInput done by setUp().)rM rY r� r r1 r r rM rY r� r r1 N)
r r� r r r� rM rY r� r r1 )r rM rY r� r r1 r r� r r r r � s
�z'Test_fileinput_input.do_test_call_inputN)r! r" r# r� r r
r r r r r r r h s
r c @ � e Zd ZdZdd� Zdd� ZdS )�Test_fileinput_closez Unit tests for fileinput.close()c C s dt _t �� | �t j� dS )zPTests that fileinput.close() does nothing if fileinput._state
is NoneN)r� r� r> � assertIsNoner( r r r r � s z'Test_fileinput_close.test_state_is_Nonec C s0 t � }|t_t�� | �|d� | �tj� dS )z`Tests that fileinput.close() invokes close() on fileinput._state
and sets _state=Noner> N)r� r� r� r> r r r r r r �test_state_is_not_None� s
z+Test_fileinput_close.test_state_is_not_NoneN�r! r" r# r� r r r r r r r
� s r
c @ r )�Test_fileinput_nextfilez#Unit tests for fileinput.nextfile()c C �X dt _| �t�� }t �� W d � n1 sw Y | �d|jj� | �t j� dS )z�Tests fileinput.nextfile() when fileinput._state is None.
Ensure that it raises RuntimeError with a meaningful error message
and does not modify fileinput._stateN�zno active input()) r� r� r� r� rg ra r� r~ r �r r� r r r r � �
�z*Test_fileinput_nextfile.test_state_is_Nonec C �N t � }t� }||jd< |t_t�� }| �|d� | �||� | �tj|� dS )a Tests fileinput.nextfile() when fileinput._state is not None.
Ensure that it invokes fileinput._state.nextfile() exactly once,
returns whatever it returns, and does not modify fileinput._state
to point to a different object.rg N)r r� r� r� r� rg r r )r Znextfile_retvalr �retvalr r r r � �
z.Test_fileinput_nextfile.test_state_is_not_NoneNr r r r r r � �
r c @ r )�Test_fileinput_filenamez#Unit tests for fileinput.filename()c C r )z�Tests fileinput.filename() when fileinput._state is None.
Ensure that it raises RuntimeError with a meaningful error message
and does not modify fileinput._stateNr ) r� r� r� r� r0 ra r� r~ r r r r r r � r z*Test_fileinput_filename.test_state_is_Nonec C r )a Tests fileinput.filename() when fileinput._state is not None.
Ensure that it invokes fileinput._state.filename() exactly once,
returns whatever it returns, and does not modify fileinput._state
to point to a different object.r0 N)r r� r� r� r� r0 r r )r Zfilename_retvalr r r r r r � r z.Test_fileinput_filename.test_state_is_not_NoneNr r r r r r � r r c @ r )�Test_fileinput_linenoz!Unit tests for fileinput.lineno()c C r )z�Tests fileinput.lineno() when fileinput._state is None.
Ensure that it raises RuntimeError with a meaningful error message
and does not modify fileinput._stateNr ) r� r� r� r� rb ra r� r~ r r r r r r � r z(Test_fileinput_lineno.test_state_is_Nonec C r )z�Tests fileinput.lineno() when fileinput._state is not None.
Ensure that it invokes fileinput._state.lineno() exactly once,
returns whatever it returns, and does not modify fileinput._state
to point to a different object.rb N)r r� r� r� r� rb r r )r Z
lineno_retvalr r r r r r � r z,Test_fileinput_lineno.test_state_is_not_NoneNr r r r r r � r r c @ r )�Test_fileinput_filelinenoz%Unit tests for fileinput.filelineno()c C r )z�Tests fileinput.filelineno() when fileinput._state is None.
Ensure that it raises RuntimeError with a meaningful error message
and does not modify fileinput._stateNr ) r� r� r� r� rc ra r� r~ r r r r r r r z,Test_fileinput_filelineno.test_state_is_Nonec C r )a Tests fileinput.filelineno() when fileinput._state is not None.
Ensure that it invokes fileinput._state.filelineno() exactly once,
returns whatever it returns, and does not modify fileinput._state
to point to a different object.rc N)r r� r� r� r� rc r r )r Zfilelineno_retvalr r r r r r r z0Test_fileinput_filelineno.test_state_is_not_NoneNr r r r r r r r c @ r )�Test_fileinput_filenoz!Unit tests for fileinput.fileno()c C r )z�Tests fileinput.fileno() when fileinput._state is None.
Ensure that it raises RuntimeError with a meaningful error message
and does not modify fileinput._stateNr ) r� r� r� r� r� ra r� r~ r r r r r r ) r z(Test_fileinput_fileno.test_state_is_Nonec C sT t � }t� }||jd< ||_|t_t�� }| �|d� | �||� | �tj|� dS )z�Tests fileinput.fileno() when fileinput._state is not None.
Ensure that it invokes fileinput._state.fileno() exactly once,
returns whatever it returns, and does not modify fileinput._state
to point to a different object.r� N) r r� r� �
fileno_retvalr� r� r� r r )r r r r r r r r 3 s
z,Test_fileinput_fileno.test_state_is_not_NoneNr r r r r r & r r c @ r )�Test_fileinput_isfirstlinez&Unit tests for fileinput.isfirstline()c C r )z�Tests fileinput.isfirstline() when fileinput._state is None.
Ensure that it raises RuntimeError with a meaningful error message
and does not modify fileinput._stateNr ) r� r� r� r� re ra r� r~ r r r r r r E r z-Test_fileinput_isfirstline.test_state_is_Nonec C r )a Tests fileinput.isfirstline() when fileinput._state is not None.
Ensure that it invokes fileinput._state.isfirstline() exactly once,
returns whatever it returns, and does not modify fileinput._state
to point to a different object.re N)r r� r� r� r� re r r )r Zisfirstline_retvalr r r r r r O r z1Test_fileinput_isfirstline.test_state_is_not_NoneNr r r r r r B r r c @ r )�Test_fileinput_isstdinz"Unit tests for fileinput.isstdin()c C r )z�Tests fileinput.isstdin() when fileinput._state is None.
Ensure that it raises RuntimeError with a meaningful error message
and does not modify fileinput._stateNr ) r� r� r� r� rf ra r� r~ r r r r r r ` r z)Test_fileinput_isstdin.test_state_is_Nonec C r )z�Tests fileinput.isstdin() when fileinput._state is not None.
Ensure that it invokes fileinput._state.isstdin() exactly once,
returns whatever it returns, and does not modify fileinput._state
to point to a different object.rf N)r r� r� r� r� rf r r )r Zisstdin_retvalr r r r r r j r z-Test_fileinput_isstdin.test_state_is_not_NoneNr r r r r r ] r r c @ rx )�InvocationRecorderc C r� r9 )�invocation_countr( r r r r) z r* zInvocationRecorder.__init__c O s" | j d7 _ ||f| _t�d�S )NrC s
some bytes)r" �last_invocationr� r r} r r r r� } s
zInvocationRecorder.__call__Nr� r r r r r! x s r! c @ s� e Zd ZdZdd� Zdd� Zdd� Ze�e d�d d
� �Z
e�e d�d d � �Z e�e d
�dd� �Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zed d � �Zd S )!�Test_hook_compressedz*Unit tests for fileinput.hook_compressed()c C s t � | _d S r% )r! � fake_openr( r r r r� � r� zTest_hook_compressed.setUpc C � | � dd� d S )Nr2 r� �� do_test_use_builtin_open_textr( r r r �test_empty_string� r� z&Test_hook_compressed.test_empty_stringc C r& �N�abcdr� r' r( r r r � test_no_ext� r� z Test_hook_compressed.test_no_extzRequires gzip and zlibc C �V t j}| jt _z t�dd�}W |t _n|t _w | �| jjd� | �| jjdi f� d S )N�test.gzr� rC )r. r� )�gzipr r% r� �hook_compressedra r" r# �r �
original_openr� r r r �test_gz_ext_fake� � z%Test_hook_compressed.test_gz_ext_fakec C sJ t j}dd� t _z
tjdddd�}W |t _n|t _w | �t|�dg� d S )Nc S r� )Ns Ex-binary string)r� r r� r r r r� � s
zATest_hook_compressed.test_gz_with_encoding_fake.<locals>.<lambda>r. r� r r zEx-binary string)r/ r r� r0 ra r` r1 r r r �test_gz_with_encoding_fake� s
z/Test_hook_compressed.test_gz_with_encoding_fakez Requires bz2c C r- )N�test.bz2r� rC )r6 r� )�bz2ZBZ2Filer% r� r0 ra r" r# r1 r r r �test_bz2_ext_fake� r4 z&Test_hook_compressed.test_bz2_ext_fakec C r& )Nz abcd.blahr� �� do_test_use_builtin_open_binaryr( r r r �
test_blah_ext� r� z"Test_hook_compressed.test_blah_extc C r& )Nzabcd.Gzr� r9 r( r r r �test_gz_ext_builtin� r� z(Test_hook_compressed.test_gz_ext_builtinc C r& )Nzabcd.Bz2r� r9 r( r r r �test_bz2_ext_builtin� r� z)Test_hook_compressed.test_bz2_ext_builtinc C r& )Nr+ r� r9 r( r r r �test_binary_mode_encoding� r� z.Test_hook_compressed.test_binary_mode_encodingc C r& r* r' r( r r r �test_text_mode_encoding� r� z,Test_hook_compressed.test_text_mode_encodingc C sf | � | j�}z
t�||�}W | � |� n| � |� w | �| jjd� | �| jj||fd d d�f� d S )NrC �r r� ��replace_builtin_openr% r� r0 ra r" r# �r r0 r r2 r� r r r r: � �
�z4Test_hook_compressed.do_test_use_builtin_open_binaryc C sf | � | j�}z
t�||�}W | � |� n| � |� w | �| jjd� | �| jj||fdd d�f� d S )NrC �localer@ rA rC r r r r( � rD z2Test_hook_compressed.do_test_use_builtin_open_textc C s t j}| t _|S r% )�builtinsr )Z
new_open_funcr2 r r r rB � s z)Test_hook_compressed.replace_builtin_openN)r! r" r# r� r� r) r, �unittestZ
skipUnlessr/ r3 r5 r7 r8 r; r< r= r> r? r: r( � staticmethodrB r r r r r$ � s(
r$ c @ r� ) �Test_hook_encodedz'Unit tests for fileinput.hook_encoded()c C s� t � }t � }tj||d�}t� }tj}|t_zt � }t � }|||�}W |t_n|t_w | �|jd� |j\} }
| � | d |� | � | d |� | � |
�
d�|� | � |
�
d�|� | � |
� d S )N�r� rC r r r� ) r r� r r! rF r ra r" r# r �poprd ) r r r� r� r% r2 r0 r Z open_resultr~ r r r r �test� s$
zTest_hook_encoded.testc s� t td��
}|�d� W d � n1 sw Y � �tt� � fdd�}|ddg� � �t�� |ddg� W d � n1 sBw Y |dd g� |d
d g� d S ) Nr� s �abcc sN t tdtd| d�d�� }t|�}W d � n1 sw Y � �||� d S )Nr� r rJ �rM r r1 �r r r r` ra )r� �expected_linesrt r; r( r r �check� s
�
�z,Test_hook_encoded.test_errors.<locals>.check�ignore�abc�strictr� u �abc�backslashreplacez\x80abc)r r r r r� r� r� �r r rP r r( r � test_errors� s � � z Test_hook_encoded.test_errorsc s t td��
}|�d� W d � n1 sw Y � �tt� � fdd�}|dg d�� � �t�� |dg d�� W d � n1 sDw Y � �t�� |dg d�� W d � n1 s`w Y � �t�� |d g d
�� W d � d S 1 s}w Y d S ) Nr� s
A
B
C
D+IKw-c sJ t t| td�d�� }t|�}W d � n1 sw Y � �||� d S )Nzutf-7rM rN )r rO rt r; r( r r rP s �
�z+Test_hook_encoded.test_modes.<locals>.checkr� )r� r� r� � D€ZrUr� r� )r� zB
zC
rW ) r r r r r� Z assertWarnsr� r� r� rU r r( r �
test_modes s � � � "�z Test_hook_encoded.test_modesN)r! r" r# r� rL rV rX r r r r rI � s
rI c @ r@ )�MiscTestc C r� r% )r Z check__all__r� r( r r r �test_all r� zMiscTest.test_allN)r! r" r# rZ r r r r rY s rY �__main__)5r� r� r� rh r] r� r� rF r rG r7 � ImportErrorr/ r r r r Zpathlibr Z test.supportr Ztest.support.os_helperr r r� r
r rL r r
r r$ ZTestCaserA ry r� r� r� r r
r r r r r r r r! r$ rI rY r! �mainr r r r �<module> sn � � $S s/? YA
�