File: //usr/local/lib/python3.10/test/__pycache__/test_fileinput.cpython-310.opt-1.pyc
o
�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)�tempfile�mkstemp�
addCleanupr
r �open�write)�selfZcontentr �fd�namer �f� r �0/usr/local/lib/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� )�nextr0 r( �append)r �size�liner r r �readlineD s zLineReader.readline���c C s> g }d} | � � }|s
|S |�|� |t|�7 }||kr |S q�Nr )r8 r5 �len)r Zhint�linesr6 r7 r r r � readlinesI s
�zLineReader.readlinesc C � d S r& r r) r r r �closeU s zLineReader.closer& )r9 )
r" r# r$ r* �propertyr, r2 r8 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 ))Nr3 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 rC )zLine %s of file 2
rD Nr rE r r r rH \ rI �
c s rC )zLine %s of file 3
rD Nr rE r r r rH ] rI � c s rC )zLine %s of file 4
rD Nr rE r r r rH ^ rI rD 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�rN �inplacer r9 �
) r! �join�range�re�compiler �printr �listr? � assertEqualr; �linenor1 r8 �
filelineno� assertFalse� isfirstline�isstdin�nextfile�sys�stdinr �stdout�upper�match�assertNotEqual�int�group)
r �t1�t2�t3�t4�pat�fir< �sZ savestdinZ
savestdoutr7 �mr r r �test_buffer_sizesY s�
�
z"BufferSizesTests.test_buffer_sizesN)r" r# r$ ry r r r r rB X s rB 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 r{ � s r{ 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 )Nr3 zThe only line there is.
r rM rD r ) r! r r8 rb rc rd r1 re r? )r rq rr rs rt rv r7 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 rM ��A
�B
�CzD
zE
�F� rT )r! r ra rb rd rc �r rq rr rv 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 rM r9 )r! r rb �filenor4 rn rh ra )r rq rr rv r7 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 r3 �U)rN r r )r� r� �C
�D) r �fail�
ValueErrorr! r Zcheck_warnings�DeprecationWarningra rb )r rv rq 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 spamrV �rb�rN r )r
Zpatchr �bufferr ra rb )r Zm_stdinrv 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� rV r� r� )ri rj r re �hasattrr ra rb )r Z
orig_stdinrv 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 ) NrD c S r> r& r )r rx r r r �<lambda>
� z7FileInputTests.test_file_opening_hook.<locals>.<lambda>)rZ r2 zGFileInput should raise if both inplace and openhook arguments are given�r2 z2FileInput should check openhook for being callablec @ rz )z=FileInputTests.test_file_opening_hook.<locals>.CustomOpenHookc S �
d| _ d S r| )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� r[ zopenhook not invoked)r r� r� r! r8 �
assertTruer~ )r rv 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�rN r2 r� r� r� zRead to end of filer3 )
r r r r � safe_unlinkr r rb r8 �UnicodeDecodeErrorr� � assertRaisesra �r r rv 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 rb r8 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 rZ s New line.) r! r r8 rb ri rk r r �read�r Z temp_fileZfobjr7 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)rZ r �errorsu Initial text �u New line r� s
s
New line ?)r! r r8 rb 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 �r1 r r r r �old_hook^ r+ zFFileInputTests.test_file_hook_backward_compatibility.<locals>.old_hookr[ r� r� )r! r r8 rb )r r� r� rv �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 rM r� r� rT r )r! r ra rb rd rc �_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 )Nr3 r rM r )r! r �OSErrorrb r� )r rq rv 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 rM )rV )r rb r� )r rv 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 rM r �line1
rD zline2
N)r! r rb )r r� rv 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)rN r r� )r! ZassertWarnsRegexr� r rb )r r� rv 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 rM rD N)z accessing lines out of order)r! r r� � RuntimeErrorrb � exceptionr� �r r� rv �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-inputr3 r rM r N)zend of input reached)r! r r� �
IndexErrorrb 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 ignoredr[ z.bakTr rY Nzos.unlink() was not invoked) �osr r{ r� r! r r� r r4 rh r� r~ )r Zos_unlink_origZos_unlink_replacementr� rv 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.r[ Tr rY N�os.fstat() was not invoked) r� �fstatr{ r� r! r r8 r� r~ )r Z
os_fstat_origZos_fstat_replacementr� rv 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.r[ Tr rY Nr� ) r� �chmodr{ r� r! r r8 r� r~ )r Z
os_chmod_origZos_chmod_replacementr� rv 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 @ rz )zQFileInputTests.test_fileno_when_ValueError_raised.<locals>.FilenoRaisesValueErrorc S � t �| t� d S r& )r{ 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� r[ r rM z _file.fileno() was not invokedr9 zfileno() should return -1)r{ r! r �_filer� r� r~ rb )r r� Z unconditionally_raise_ValueErrorr� rv 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� Zline2r3 )r% r r2 rb r, r8 �r �srcrv 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 r2 rb r, r4 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 rD )
r r! r r8 rb rc rd r1 r� �fspath)r rq rv r7 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 )rZ r z Modified %sr zModified Pathlib file.
)r r! r r8 rb r` r r� )r rq rv r7 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 testsNFr3 �r)r r2 r r� c C sJ || _ || _|| _|| _|| _|| _|| _d | _t� dd� �| _
i | _ d S )Nc S s dS r: r r r r r r� 1 r� z(MockFileInput.__init__.<locals>.<lambda>) rN rZ �backupr r2 r r� r� � collections� defaultdict�invocation_counts�
return_values)r rN rZ r� r r2 r r� r r r r* ' s
zMockFileInput.__init__c C s | j d d7 < d S )Nr? rD )r� r) r r r r? 4 s zMockFileInput.closec C � | j d d7 < | jd S )Nrh rD �r� r� r) r r r rh 7 �
zMockFileInput.nextfilec C r� )Nr1 rD r� r) r r r r1 ; r� zMockFileInput.filenamec C r� )Nrc rD r� r) r r r rc ? r� zMockFileInput.linenoc C r� )Nrd rD r� r) r r r rd C r� zMockFileInput.filelinenoc C r� )Nr� rD r� r) r r r r� G r� zMockFileInput.filenoc C r� )Nrf rD r� r) r r r rf K r� zMockFileInput.isfirstlinec C r� )Nrg rD r� r) r r r rg O r� zMockFileInput.isstdin)NFr3 )
r" r# r$ �__doc__r* r? rh r1 rc rd r� rf rg 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 )NrD )r� rb 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� �inputrb 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().)rN rZ r� r r2 r r rN rZ r� r r2 N)
r r� r r r� rN rZ r� r r2 )r rN rZ r� r r2 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� rh rb 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.rh N)r r� r� r� r� rh 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� r1 rb 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.r1 N)r r� r� r� r� r1 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� rc rb 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.rc N)r r� r� r� r� rc 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� rd rb 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.rd N)r r� r� r� r� rd 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� rb 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� rf rb 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.rf N)r r� r� r� r� rf 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� rg rb 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.rg N)r r� r� r� r� rg 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 @ rz )�InvocationRecorderc C r� r: )�invocation_countr) r r r r* z r+ zInvocationRecorder.__init__c O s" | j d7 _ ||f| _t�d�S )NrD 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 )Nr3 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� rD )r0 r� )�gzipr r' r� �hook_compressedrb 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>r0 r� r r zEx-binary string)r1 r r� r2 rb ra r3 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� rD )r8 r� )�bz2�BZ2Filer' r� r2 rb r$ r% r3 r r r �test_bz2_ext_fake� r6 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� r< r) r r r �test_gz_ext_builtin� r� z(Test_hook_compressed.test_gz_ext_builtinc C r( )Nzabcd.Bz2r� r< r) r r r �test_bz2_ext_builtin� r� z)Test_hook_compressed.test_bz2_ext_builtinc C r( )Nr- r� r< 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 )NrD �r r� ��replace_builtin_openr' r� r2 rb r$ r% �r r1 r r4 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 )NrD �localerC rD rF r r r r* � rG z2Test_hook_compressed.do_test_use_builtin_open_textc C s t j}| t _|S r& )�builtinsr )Z
new_open_funcr4 r r r rE � s z)Test_hook_compressed.replace_builtin_openN)r" r# r$ r� r r+ r. �unittestZ
skipUnlessr1 r5 r7 r9 r; r> r? r@ rA rB r= r* � staticmethodrE 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� rD r r r� ) r r� r r# rI r rb r$ r% r �popre ) r r r� r� r' r4 r1 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 rM �rN r r2 �r r r ra rb )r� �expected_linesrv 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 rS 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-7rP rQ )r rR rv r< r) r r rS s �
�z+Test_hook_encoded.test_modes.<locals>.checkr� )r� r� r� � D€ZrUr� r� )r� zB
zC
rZ ) r r r r r� Z assertWarnsr� r� r� rX r r) r �
test_modes s � � � "�z Test_hook_encoded.test_modesN)r" r# r$ r� rO rY r[ r r r r rL � s
rL c @ rA )�MiscTestc C r� r& )r Z check__all__r� r) r r r �test_all r� zMiscTest.test_allN)r" r# r$ r] r r r r r\ s r\ �__main__)5r� r� r� ri r^ r� r� rI r rJ r9 � ImportErrorr1 r r r r �pathlibr Z test.supportr Ztest.support.os_helperr r r� r
r rO r r
r r% ZTestCaserB r{ r� r� r� r r r r r r r r! r" r# r& rL r\ r" �mainr r r r �<module> sn � � $S s/? YA
�