Your IP : 216.73.216.213
U
i�fW� � @ sP d Z ddlZddlZddlZddlZddlZddlZddlZddlZz ddl Z W n e
k
rh dZ Y nX z ddl Z W n e
k
r� dZ Y nX ddl m
Z
mZ ddlmZmZ ddlmZ ddlmZmZmZ ddlmZ ddlmZ dd lmZ G d
d � d �Z G d d
� d
�Z G dd� de ej �Z G dd� d�Z G dd� de ej �Z!G dd� d�Z"G dd� dej �Z#G dd� de#�Z$G dd� de#�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,d-� d-ej �Z.G d.d/� d/ej �Z/G d0d1� d1ej �Z0e1d2k�rLe�2� dS )3z,
Tests for fileinput module.
Nick Mathewson
� N)�BytesIO�StringIO)� FileInput� hook_encoded)�Path)�verbose�TESTFN�check_warnings)�unlink)�support)�mockc @ s e Zd Zdd�dd�ZdS )� BaseTests�w��modec C s> t �� \}}| �tj|� t||��}|�|� W 5 Q R X |S �N)�tempfileZmkstemp�
addCleanupr r
�open�write)�selfZcontentr �fd�name�f� r �8/opt/alt/python38/lib64/python3.8/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 r ��
_linesread�r r r r �__init__1 s zLineReader.__init__c C s z| j d d � W �S g | _ X d S r r! r# r r r � linesread4 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 �readline? s zLineReader.readline���c C s@ g }d}| � � }|s|S |�|� |t|�7 }||kr|S qd S �Nr )r1 r. �len)r Zhint�linesr/ r0 r r r � readlinesD s
zLineReader.readlinesc C s d S r r r# r r r �closeP s zLineReader.close)N)r2 )
r r r r$ �propertyr% r+ r1 r6 r7 r r r r r / s
r c @ s 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 �}tr�td � t||||fd
�}t|�}|� � | �
t |�d� | �
|d d� | �
|d d� | �
|� � d� | �
|�
� |� t�rtd� t||||fd
�}d}|�r@|dk�r@|�� }�q$| �
|�
� |� | �
|� � d� | �
|�� d� | �|�� � | �|�� � t�r�td� |�� | �
|�� d� | �
|� � d� |� � t�r�td� t||||d fd
�}tj} zNtd �t_t|�}| �
t |�d � | �
|d d � | �
|�
� d!� |�� W 5 | t_X t�rXtd"� t||||fd
�}| �
|� � d#� | �
|�
� d � |�� | �
|� � d#� | �
|�
� d � t�r�td$� tj}
zDt||||fd
d%�}|D ] } | d d&� �� } t| � �q�|� � W 5 |
t_X t||||fd
�}|D ]N} | �
| d& d'� |�| d d&� �} | �| d � | �
t| � d
��|�� � �q*|� � d S )(Nr, c s s | ]}d |d V qdS )zLine %s of file 1
� Nr ��.0�ir r r � <genexpr>V s z5BufferSizesTests.test_buffer_sizes.<locals>.<genexpr>� c s s | ]}d |d V qdS )zLine %s of file 2
r: Nr r; r r r r> W s �
c s s | ]}d |d V qdS )zLine %s of file 3
r: Nr r; r r r r> X s � c s s | ]}d |d V qdS )zLine %s of file 4
r: Nr r; r r r r> Y s r: zLINE (\d+) OF FILE (\d+)z1. Simple iteration��files� � 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�rC �inplacer2 �
) r �join�range�re�compiler �printr �listr7 � assertEqualr4 �linenor* r1 �
filelineno� assertFalse� isfirstline�isstdin�nextfile�sys�stdinr �stdout�upper�match�assertNotEqual�int�group)
r �t1�t2�t3�t4Zpat�fir5 �sZ savestdinZ
savestdoutr0 �mr r r �test_buffer_sizesT s�
z"BufferSizesTests.test_buffer_sizesN)r r r rm r r r r r9 S s r9 c @ s e Zd Zdd� Zdd� ZdS )�UnconditionallyRaisec C s || _ d| _d S �NF)�exception_type�invoked)r rp r r r r$ � s z UnconditionallyRaise.__init__c O s d| _ | �� �d S r& )rq rp �r �args�kwargsr r r �__call__� s z UnconditionallyRaise.__call__N�r r r r$ ru r r r r rn � s rn 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� Zejed�d d � �Zd d � Zejed�d d!� �Zejed�d"d#� �Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Z d0d1� Z d2d3� Z d4S )5�FileInputTestsc C s� | � d�}| � d�}| � d�}| � d�}t||||fd�}|�� }| �|d� | �|�� d� | �|�� d� | �|�� |� |�� }| �|� | �|�� d� | �|�� d� | �|�� |� |�� d S )Nr, zThe only line there is.
rB r: r ) r r r1 rW rX rY r* rZ r7 )r rf rg rh ri rj r0 r r r �test_zero_byte_files� s
z#FileInputTests.test_zero_byte_filesc C sf | � d�}| � d�}t||fd�}t|�}| �|dddddd g� | �|�� d
� | �|�� d � d S ) N�A
B
C�D
E
FrB �A
�B
�C�D
�E
�F� rI )r r rV rW rY rX �r rf rg rj r5 r r r �%test_files_that_dont_end_with_newline� s
z4FileInputTests.test_files_that_dont_end_with_newlinec C s~ | � d�}| � d�}t||fd�}| �|�� d� t|�}| �|�� d� |�� | �|�� d� t|�}| �|�� d� d S )NzA
BzC
DrB r2 )r r rW �filenor- rc r] rV )r rf rg rj r0 r r r � test_fileno� s
zFileInputTests.test_filenoc C s� zt dd�}| �d� W n tk
r, Y nX | jddd�}tdtf�� t |dd�}W 5 Q R X tdtf�� t|�}W 5 Q R X | �|d d
d d g� d S )
Nr r z-FileInput should reject invalid mode argument� A
B
C
D�wbr, �U�rC r r{ r| �C
�D)r �fail�
ValueErrorr r �DeprecationWarningrV rW )r rj rf r5 r r r �test_opening_mode� s
z FileInputTests.test_opening_modec C sH t �d��4}td�|_tdgdd�}t|�}| �|dg� W 5 Q R X d S )Nz sys.stdin� spam, bacon, sausage, and spamrK �rbr� )r Zpatchr �bufferr rV rW )r Zm_stdinrj r5 r r r �test_stdin_binary_mode� s
z%FileInputTests.test_stdin_binary_modec C sX t j}zDtd�t _| �tt jd�� tdgdd�}t|�}| �|dg� W 5 |t _X d S )Nr� r� rK r� r� )r^ r_ r rZ �hasattrr rV rW )r Z
orig_stdinrj r5 r r r � test_detached_stdin_binary_mode� s
z.FileInputTests.test_detached_stdin_binary_modec C s� z t ddd� d�}| �d� W n tk
r2 Y nX zt dd�}| �d� W n tk
r` Y nX G dd � d �}| �d
�}|� }t |g|d��}|�� W 5 Q R X | �|jd � d S ) Nr: c S s d S r r )r rl r r r �<lambda> � z7FileInputTests.test_file_opening_hook.<locals>.<lambda>)rO r+ zGFileInput should raise if both inplace and openhook arguments are given)r+ z2FileInput should check openhook for being callablec @ s e Zd Zdd� Zdd� ZdS )z=FileInputTests.test_file_opening_hook.<locals>.CustomOpenHookc S s
d| _ d S ro )rq r# r r r r$ s zFFileInputTests.test_file_opening_hook.<locals>.CustomOpenHook.__init__c W s d| _ t|� S r& )rq r )r rs r r r ru s zFFileInputTests.test_file_opening_hook.<locals>.CustomOpenHook.__call__Nrv r r r r �CustomOpenHook s r� rP zopenhook not invoked)r r� r� r r1 �
assertTruerq )r rj r� �tZcustom_open_hookr r r �test_file_opening_hook s
z%FileInputTests.test_file_opening_hookc
C s� t td��(}|�d� |�dd � |�d� W 5 Q R X | �tt� tttd�d���}z4| �|�� d� | �|�� d � | �|�� d
� W n t k
r� | �
d � Y nX | � t �� t |� W 5 Q R X | �|�� d � | �|�� d � W 5 Q R X d S )
Nr� s A
B
C
s
123456789
i� � ��ascii�rC r+ r{ r| r� zRead to end of filer, )
r r r r � safe_unlinkr r rW r1 �UnicodeDecodeErrorr� � assertRaisesrV �r r rj r r r �
test_readline s&
� z FileInputTests.test_readlinec C s� t td��}|�d� W 5 Q R X | �tt� ttdd��V}| �|�� d� | �|�� d� | �|�� d� | �|�� d� | �|�� d� W 5 Q R X d S ) Nr� r� r� r� s A
s B
s C
Dr� )r r r r r� r rW r1 r� r r r �test_readline_binary_mode7 s z(FileInputTests.test_readline_binary_modec C sr | j ddd�}t|ddd��&}|�� }| �|d� tj�d� W 5 Q R X t|d��}| �|�� d� W 5 Q R X d S )Ns
Initial text.r� r r� T)r rO s New line.) r r r1 rW r^ r` r r �read)r Z temp_fileZfobjr0 r r r r � test_inplace_binary_write_modeD s z-FileInputTests.test_inplace_binary_write_modec C s� | � d�}| � d�}t||fd��}t|�}W 5 Q R X | �|dddddd g� | �|�� d
� | �|�� d � | �|jd � d S )
Nry rz rB r{ r| r} r~ r r� r� rI r )r r rV rW rY rX �_filesr� r r r �test_context_managerN s
z#FileInputTests.test_context_managerc C sP | � d�}z t|d��
}t�W 5 Q R X W n" tk
rJ | �|jd� Y nX d S )Nr, rB r )r r �OSErrorrW r� )r rf rj r r r �test_close_on_exceptionX s
z&FileInputTests.test_close_on_exceptionc C s( t g d��}| �|jd� W 5 Q R X d S )NrB )rK )r rW r� )r rj r r r �.test_empty_files_list_specified_to_constructor` s z=FileInputTests.test_empty_files_list_specified_to_constructor)�categoryc C sN | � d�}t|gd��.}|d }| �|d� |d }| �|d� W 5 Q R X dS )zNTests invoking FileInput.__getitem__() with the current
line number� line1
line2
rB r �line1
r: zline2
N)r r rW )r r� rj Zretval1Zretval2r r r �test__getitem__d s
z FileInputTests.test__getitem__c
C sN | � d�}| �td��. t|gd��}| �|d d� W 5 Q R X W 5 Q R X d S )Nr� z Use iterator protocol insteadrB r r� )r ZassertWarnsRegexr� r rW )r r� rj r r r �test__getitem___deprecationo s
�z*FileInputTests.test__getitem___deprecationc
C sT | � d�}t|gd��$}| �t��}|d W 5 Q R X W 5 Q R X | �|jjd� dS )zZTests invoking FileInput.__getitem__() with an index unequal to
the line numberr� rB r: )z accessing lines out of orderN)r r r� � RuntimeErrorrW � exceptionrs �r r� rj �cmr r r �test__getitem__invalid_keyv s
z)FileInputTests.test__getitem__invalid_keyc
C sT | � d�}t|gd��$}| �t��}|d W 5 Q R X W 5 Q R X | �|jjd� dS )zZTests invoking FileInput.__getitem__() with the line number but at
end-of-inputr, rB r )zend of input reachedN)r r r� �
IndexErrorrW r� rs r� r r r �test__getitem__eof� s
z!FileInputTests.test__getitem__eofc C sz t j}tt�}zP| �d�}| �tj|d � t|gdd�� }t|� |t _|� � W 5 Q R X W 5 |t _X | �
|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 ignoredrP z.bakTrN zos.unlink() was not invokedN) �osr
rn r� r r r r r- r] r� rq )r Zos_unlink_origZos_unlink_replacementr� rj r r r �%test_nextfile_oserror_deleting_backup� s
�z4FileInputTests.test_nextfile_oserror_deleting_backupc C s` t j}tt�}z6| �d�}t|gdd��}|t _|�� W 5 Q R X W 5 |t _X | �|jd� dS )z{Tests invoking FileInput.readline() when os.fstat() raises OSError.
This exception should be silently discarded.rP TrN �os.fstat() was not invokedN) r� �fstatrn r� r r r1 r� rq )r Z
os_fstat_origZos_fstat_replacementr� rj r r r �%test_readline_os_fstat_raises_OSError� s
�z4FileInputTests.test_readline_os_fstat_raises_OSErrorc C s` t j}tt�}z6| �d�}t|gdd��}|t _|�� W 5 Q R X W 5 |t _X | �|jd� dS )z{Tests invoking FileInput.readline() when os.chmod() raises OSError.
This exception should be silently discarded.rP TrN r� N) r� �chmodrn r� r r r1 r� rq )r Z
os_chmod_origZos_chmod_replacementr� rj r r r �%test_readline_os_chmod_raises_OSError� s
�z4FileInputTests.test_readline_os_chmod_raises_OSErrorc C sz G dd� dt �}|� }| �d�}t|gd��(}|j}z||_|�� }W 5 ||_X W 5 Q R X | �|jd� | �|dd� d S )Nc @ s e Zd Zdd� Zdd� ZdS )zQFileInputTests.test_fileno_when_ValueError_raised.<locals>.FilenoRaisesValueErrorc S s t �| t� d S r )rn r$ r� r# r r r r$ � s zZFileInputTests.test_fileno_when_ValueError_raised.<locals>.FilenoRaisesValueError.__init__c S s | � � d S r )ru r# r r r r� � s zXFileInputTests.test_fileno_when_ValueError_raised.<locals>.FilenoRaisesValueError.filenoN)r r r r$ r� r r r r �FilenoRaisesValueError� s r� rP rB z _file.fileno() was not invokedr2 zfileno() should return -1)rn r r �_filer� r� rq rW )r r� Z unconditionally_raise_ValueErrorr� rj Z file_backup�resultr 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 5 Q R X d S �Nz line1
line2zline3
r� r� Zline2r, )r r r+ rW r% r1 �r �srcrj 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 5 Q R X d S r� )r r r+ rW r% r- r� �
StopIterationr� r r r �test_iteration_buffering� s �z'FileInputTests.test_iteration_bufferingc C sp t | �d��}t|��P}|�� }| �|d� | �|�� d� | �|�� d� | �|�� t� |�� W 5 Q R X d S )N�
Pathlib file.r: )
r r r r1 rW rX rY r* r� �fspath)r rf rj r0 r r r �test_pathlib_file� s
z FileInputTests.test_pathlib_filec C sn t | �d��}t|dd��&}|�� }| �|d� td| � W 5 Q R X t|��}| �|�� d� W 5 Q R X d S )Nr� T)rO z Modified %szModified Pathlib file.
)r r r r1 rW rU r r� )r rf rj r0 r r r r �test_pathlib_file_inplace s
z(FileInputTests.test_pathlib_file_inplaceN) r r r rx 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 rw � s8
rw c @ sb e Zd ZdZdddd�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 testsNFr, �r)r r+ c C s> || _ || _|| _|| _|| _d | _t�dd� �| _i | _ d S )Nc S s dS r3 r r r r r r� r� z(MockFileInput.__init__.<locals>.<lambda>)
rC rO �backupr r+ r� � collections� defaultdict�invocation_counts�
return_values)r rC rO r� r r+ r r r r$ s zMockFileInput.__init__c C s | j d d7 < d S )Nr7 r: )r� r# r r r r7 s zMockFileInput.closec C s | j d d7 < | jd S )Nr] r: �r� r� r# r r r r] s zMockFileInput.nextfilec C s | j d d7 < | jd S )Nr* r: r� r# r r r r* " s zMockFileInput.filenamec C s | j d d7 < | jd S )NrX r: r� r# r r r rX & s zMockFileInput.linenoc C s | j d d7 < | jd S )NrY r: r� r# r r r rY * s zMockFileInput.filelinenoc C s | j d d7 < | jd S )Nr� r: r� r# r r r r� . s zMockFileInput.filenoc C s | j d d7 < | jd S )Nr[ r: r� r# r r r r[ 2 s zMockFileInput.isfirstlinec C s | j d d7 < | jd S )Nr\ r: r� r# r r r r\ 6 s zMockFileInput.isstdin)NFr, )
r r r �__doc__r$ r7 r] r* rX rY r� r[ r\ r r r r r�
s � r� c @ s( 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 �setUp> s z$BaseFileInputGlobalMethodsTest.setUpc C s | j t_| jt_d S r )r� r� r r� r� r# r r r �tearDownC s z'BaseFileInputGlobalMethodsTest.tearDownc C s2 |j | }| �|d|� t|j �}| �|d� d S �Nr: )r� rW r4 )r Zmock_file_inputZ method_nameZ actual_countZactual_total_countr r r �assertExactlyOneInvocationG s
z9BaseFileInputGlobalMethodsTest.assertExactlyOneInvocationN)r r r r� r� r� r� r r r r r� : s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �Test_fileinput_inputz Unit tests for fileinput.input()c C sV t � }t� |_|t_| �t��}t�� W 5 Q R X | �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.)zinput() already active�fileinput._stateN) r� �objectr� r� r� r� r� �inputrW r� rs �assertIs)r �instancer� r r r �1test_state_is_not_None_and_state_file_is_not_NoneR 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_None_ 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_Nonej s z'Test_fileinput_input.test_state_is_Nonec C s� 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().)rC rO r� r r+ r� rC rO r� r r+ N)
r� r� r� r� r� rC rO r� r r+ )r rC rO r� r r+ r� r r r r� s s
�z'Test_fileinput_input.do_test_call_inputN)r r r r� r� r� r� r� r r r r r� O s
r� c @ s 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� r7 � 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=Noner7 N)r� r� r� r7 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 @ s e Zd ZdZdd� Zdd� ZdS )�Test_fileinput_nextfilez#Unit tests for fileinput.nextfile()c C sD dt _| �t��}t �� W 5 Q R X | �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� r] rW r� rs r� �r r� r r r r� � s
z*Test_fileinput_nextfile.test_state_is_Nonec C sN 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.r] N)r� r� r� r� r� r] r� r� )r Znextfile_retvalr� �retvalr r r r� � s
z.Test_fileinput_nextfile.test_state_is_not_NoneNr� r r r r r� � s
r� c @ s e Zd ZdZdd� Zdd� ZdS )�Test_fileinput_filenamez#Unit tests for fileinput.filename()c C sD dt _| �t��}t �� W 5 Q R X | �d|jj� | �t j� dS )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� r* rW r� rs r� r� r r r r� � s
z*Test_fileinput_filename.test_state_is_Nonec C sN t � }t� }||jd< |t_t�� }| �|d� | �||� | �tj|� dS )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.r* N)r� r� r� r� r� r* r� r� )r Zfilename_retvalr� r� r r r r� � s
z.Test_fileinput_filename.test_state_is_not_NoneNr� r r r r r� � s
r� c @ s e Zd ZdZdd� Zdd� ZdS )�Test_fileinput_linenoz!Unit tests for fileinput.lineno()c C sD dt _| �t��}t �� W 5 Q R X | �d|jj� | �t j� dS )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� rX rW r� rs r� r� r r r r� � s
z(Test_fileinput_lineno.test_state_is_Nonec C sN t � }t� }||jd< |t_t�� }| �|d� | �||� | �tj|� dS )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.rX N)r� r� r� r� r� rX r� r� )r Z
lineno_retvalr� r� r r r r� � s
z,Test_fileinput_lineno.test_state_is_not_NoneNr� r r r r r� � s
r� c @ s e Zd ZdZdd� Zdd� ZdS )�Test_fileinput_filelinenoz%Unit tests for fileinput.filelineno()c C sD dt _| �t��}t �� W 5 Q R X | �d|jj� | �t j� dS )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� rY rW r� rs r� r� r r r r� � s
z,Test_fileinput_filelineno.test_state_is_Nonec C sN t � }t� }||jd< |t_t�� }| �|d� | �||� | �tj|� dS )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.rY N)r� r� r� r� r� rY r� r� )r Zfilelineno_retvalr� r� r r r r� � s
z0Test_fileinput_filelineno.test_state_is_not_NoneNr� r r r r r� � s
r� c @ s e Zd ZdZdd� Zdd� ZdS )�Test_fileinput_filenoz!Unit tests for fileinput.fileno()c C sD dt _| �t��}t �� W 5 Q R X | �d|jj� | �t j� dS )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� rW r� rs r� r� r r r r� s
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� s
z,Test_fileinput_fileno.test_state_is_not_NoneNr� r r r r r� s
r� c @ s e Zd ZdZdd� Zdd� ZdS )�Test_fileinput_isfirstlinez&Unit tests for fileinput.isfirstline()c C sD dt _| �t��}t �� W 5 Q R X | �d|jj� | �t j� dS )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� r[ rW r� rs r� r� r r r r� + s
z-Test_fileinput_isfirstline.test_state_is_Nonec C sN t � }t� }||jd< |t_t�� }| �|d� | �||� | �tj|� dS )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.r[ N)r� r� r� r� r� r[ r� r� )r Zisfirstline_retvalr� r� r r r r� 5 s
z1Test_fileinput_isfirstline.test_state_is_not_NoneNr� r r r r r� ( s
r� c @ s e Zd ZdZdd� Zdd� ZdS )�Test_fileinput_isstdinz"Unit tests for fileinput.isstdin()c C sD dt _| �t��}t �� W 5 Q R X | �d|jj� | �t j� dS )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� r\ rW r� rs r� r� r r r r� F s
z)Test_fileinput_isstdin.test_state_is_Nonec C sN t � }t� }||jd< |t_t�� }| �|d� | �||� | �tj|� dS )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.r\ N)r� r� r� r� r� r\ r� r� )r Zisstdin_retvalr� r� r r r r� P s
z-Test_fileinput_isstdin.test_state_is_not_NoneNr� r r r r r� C s
r� c @ s e Zd Zdd� Zdd� ZdS )�InvocationRecorderc C s
d| _ d S r3 )�invocation_countr# r r r r$ _ s zInvocationRecorder.__init__c O s | j d7 _ ||f| _d S r� )r� �last_invocationrr r r r ru a s zInvocationRecorder.__call__Nrv r r r r r� ^ s r� c @ s| e Zd ZdZdd� Zdd� Zdd� Ze�e d�d d
� �Z
e�e d �d d
� �Z dd� Z
dd� Zdd� Zdd� Zedd� �ZdS )�Test_hook_compressedz*Unit tests for fileinput.hook_compressed()c C s t � | _d S r )r� � fake_openr# r r r r� h s zTest_hook_compressed.setUpc C s | � dd� d S )Nr, r: ��do_test_use_builtin_openr# r r r �test_empty_stringk s z&Test_hook_compressed.test_empty_stringc C s | � dd� d S )NZabcd� r� r# r r r � test_no_extn s z Test_hook_compressed.test_no_extzRequires gzip and zlibc C sP t j}| jt _zt�dd�}W 5 |t _X | �| jjd� | �| jjdi f� d S )N�test.gzr� r: )r r� )�gzipr r� r� �hook_compressedrW r� r� �r �
original_openr� r r r �test_gz_ext_fakeq s z%Test_hook_compressed.test_gz_ext_fakez Requires bz2c C sP t j}| jt _zt�dd�}W 5 |t _X | �| jjd� | �| jjdi f� d S )N�test.bz2rE r: )r rE )�bz2ZBZ2Filer� r� r rW r� r� r r r r �test_bz2_ext_fake} s z&Test_hook_compressed.test_bz2_ext_fakec C s | � dd� d S )Nz abcd.blahrA r� r# r r r �
test_blah_ext� s z"Test_hook_compressed.test_blah_extc C s | � dd� d S )Nzabcd.GzrI r� r# r r r �test_gz_ext_builtin� s z(Test_hook_compressed.test_gz_ext_builtinc C s | � dd� d S )Nzabcd.Bz2� r� r# r r r �test_bz2_ext_builtin� s z)Test_hook_compressed.test_bz2_ext_builtinc C sV | � | j�}zt�||�}W 5 | � |� X | �| jjd� | �| jj||fi f� d S r� )�replace_builtin_openr� r� r rW r� r� )r r* r r r� r r r r� � s
�z-Test_hook_compressed.do_test_use_builtin_openc C s t j}| t _|S r )�builtinsr )Z
new_open_funcr r r r r � s z)Test_hook_compressed.replace_builtin_openN)r r r r� r� r r �unittestZ
skipUnlessr r r
r r r
r r� � staticmethodr r r r r r� e s
r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �Test_hook_encodedz'Unit tests for fileinput.hook_encoded()c C s� t � }t � }tj||d�}t� }tj}|t_zt � }t � }|||�}W 5 |t_X | �|jd� |j\} }
| � | d |� | � | d |� | � |
�
d�|� | � |
�
d�|� | � |
� d S )N��errorsr: r �encodingr ) r� r� r r� r r rW r� r� r� �poprZ ) r r r r� r� r r* r Z open_resultrs rt r r r �test� s$
zTest_hook_encoded.testc s� t td��}|�d� W 5 Q R X � �tt� � fdd�}|ddg� � �t�� |ddg� W 5 Q R X |dd g� |d
d g� d S ) Nr� s �abcc s: t tdtd| d�d��}t|�}W 5 Q R X � �||� d S )Nr� zutf-8r �rC r r+ �r r r rV rW )r �expected_linesrj r5 r# r r �check� s
�z,Test_hook_encoded.test_errors.<locals>.check�ignore�abc�strict�replaceu �abc�backslashreplacez\x80abc)r r r r r� r� r� �r r r r r# r � test_errors� s z Test_hook_encoded.test_errorsc s� t td��}|�d� W 5 Q R X � �tt� � fdd�}|ddddd g� � �t�� |d
dddd g� W 5 Q R X � �t�� |d dddd g� W 5 Q R X � �t�� |d dd
dd g� W 5 Q R X d S )Nr� s
A
B
C
D+IKw-c s6 t t| td�d��}t|�}W 5 Q R X � �||� d S )Nzutf-7r r )r r rj r5 r# r r r � s �z+Test_hook_encoded.test_modes.<locals>.checkr� r{ r| r� u D€ZrUr� r� zB
zC
) r r r r r� Z assertWarnsr� r� r� r# r r# r �
test_modes� s z Test_hook_encoded.test_modesN)r r r r� r r$ r% r r r r r � s r c @ s e Zd Zdd� ZdS )�MiscTestc C s t �| t� d S r )r Z check__all__r� r# r r r �test_all� s zMiscTest.test_allN)r r r r' r r r r r& � s r& �__main__)3r� r� r^ rS r� r� r r r r
� ImportErrorr �ior r r r �pathlibr Z test.supportr r r r
r� r r r r
r ZTestCaser9 rn rw r� r� r� r� r� r� r� r� r� r� r� r� r� r r& r �mainr r r r �<module> s^
$S a-> >A