| Server IP : 118.27.122.248 / Your IP : 216.73.216.109 Web Server : Apache System : Linux web0264.sh.tyo1 4.18.0-553.79.1.lve.el7h.x86_64 #1 SMP Wed Oct 15 16:34:46 UTC 2025 x86_64 User : c9415830 ( 11735) PHP Version : 8.4.17 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : ON | Sudo : OFF | Pkexec : OFF Directory : /opt/alt/python36/lib64/python3.6/test/__pycache__/ |
Upload File : |
3
�w2_H� � @ sD d Z ddlZddlZddlZddlZddlZddlZddlZddlZy ddl Z W n e
k
rh dZ Y nX y ddl Z W n e
k
r� dZ Y nX ddl m
Z
mZ ddlmZmZ ddlmZmZmZ ddlmZ ddlmZ ddlmZ G d d
� d
�ZG d d � d �ZG d
d� deej �Z G dd� d�Z G dd� deej �Z G dd� d�Z G dd� dej �Z!G dd� de!�Z"G dd� de!�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*�Z+G d+d,� d,ej �Z,G d-d.� d.ej �Z-G d/d0� d0ej �Z.e/d1k�r@ej0� dS )2z,
Tests for fileinput module.
Nick Mathewson
� N)�BytesIO�StringIO)� FileInput� hook_encoded)�verbose�TESTFN�check_warnings)�unlink)�support)�mockc @ s e Zd Zdd�dd�ZdS )� BaseTests�w)�modec C s> t j� \}}| jtj|� t||��}|j|� W d Q R X |S )N)�tempfileZmkstemp�
addCleanupr
r �open�write)�selfZcontentr �fd�name�f� r �8/opt/alt/python36/lib64/python3.6/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__0 s zLineReader.__init__c C s z| j d d � S g | _ X d S )N)r )r r r r � linesread3 s zLineReader.linesreadc C s t |jd��| _| S )NT)�iter�
splitlines�it)r �filenamer r r r �openhook: s zLineReader.openhookNc C s t | jd�}| jj|� |S )N� )�nextr# r �append)r �size�liner r r �readline> s zLineReader.readline� c C sD g }d}x6| j � }|s|S |j|� |t|�7 }||kr
|S q
W d S )Nr )r+ r( �len)r Zhint�linesr) r* r r r � readlinesC s
zLineReader.readlinesc C s d S )Nr )r r r r �closeO s zLineReader.close)N���)r1 )
r r r r �propertyr r% r+ r/ r0 r r r r r . s
r c @ s e Zd Zdd� Zddd�ZdS )�BufferSizesTestsc C s� x�dD ]�\}}| j djdd� td�D ���}| j djdd� td �D ���}| j djd
d� td �D ���}| j djd d� td�D ���}|r�| jt�� | j||||||� W d Q R X q| j||||||� qW d S )Nr r, � r& c s s | ]}d |d V qdS )zLine %s of file 1
r, 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 )r5 r6 r r r r7 W s �
c s s | ]}d |d V qdS )zLine %s of file 3
r, Nr )r5 r6 r r r r7 X s � c s s | ]}d |d V qdS )zLine %s of file 4
r, Nr )r5 r6 r r r r7 Y s �r r �r, r4 )r; r< )r �join�range� assertWarns�DeprecationWarning�buffer_size_test)r �round�bs�t1�t2�t3�t4r r r �test_buffer_sizesS s z"BufferSizesTests.test_buffer_sizesr c C sr t jd�}d|d }tr.td|d |f � t||||f|d�} t| �}
| j� | jt|
�d� | j|
d d � | j|
d
d � | j| j � d� | j| j
� |� tr�td |d |f � t||||f|d�} d
} x| r�| dkr�| j � } q�W | j| j
� |� | j| j � d� | j| j � d� | j
| j� � | j
| j� � t�rRtd|d |f � | j� | j| j � d� | j| j � d� | j� t�r�td|d |f � t||||df|d�} tj} zNtd�t_t| �}
| jt|
�d� | j|
d d� | j| j
� d� | j� W d | t_X t�r*td |d |f � t||||f|d�} | j| j � d� | j| j
� d � | j� | j| j � d� | j| j
� d � t�r�td |d |f � tj}
zJt||||fd|d �} x&| D ] }|d d!� j� }t|� �q�W | j� W d |
t_X t||||f|d�} xV| D ]N}| j|d" d � |j|d d#� �}| j|d � | jt|jd��| j � � �qW | j� d S )$NzLINE (\d+) OF FILE (\d+)r, � z %s. Simple iteration (bs=%s)r )�files�bufsize� � zLine 5 of file 1
r4 zLine 1 of file 4
z %s. Status variables (bs=%s)�xzLine 6 of file 2
� z%s. Nextfile (bs=%s)� zLine 1 of file 3
� z%s. Stdin (bs=%s)� �-z Line 1 of stdin
Line 2 of stdin
�! � zLine 2 of stdin
z<stdin>z %s. Boundary conditions (bs=%s)z%s. Inplace (bs=%s)r: )rJ �inplacerK �
r1 r1 r1 )�re�compiler �printr �listr0 � assertEqualr- �linenor$ r+ �
filelineno� assertFalse� isfirstline�isstdin�nextfile�sys�stdinr �stdout�upper�match�assertNotEqual�int�group)r rD rE rF rG rC rB Zpat�start�fir. �sZ savestdinZ
savestdoutr* �mr r r rA ` s�
z!BufferSizesTests.buffer_size_testN)r r )r r r rH rA r r r r r3 R s
r3 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| _ | j� �d S )NT)rq rp )r �args�kwargsr r r �__call__� s z UnconditionallyRaise.__call__N)r r r r rt r r r r ro � s ro 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 � Zd d � Zd!d"� Zejeed#� d$�d%d&� �Zd'd(� Zd)d*� Zd+d,� Zd-S ).�FileInputTestsc C s� | j d�}| j d�}| j d�}| j d�}t||||fd�}|j� }| j|d� | j|j� d� | j|j� d� | j|j� |� |j� }| j|� | j|j� d� | j|j� d� | j|j� |� |j� d S )Nr& zThe only line there is.
)rJ r, r ) r r r+ r\ r] r^ r$ r_ r0 )r rD rE rF rG rl r* r r r �test_zero_byte_files� s
z#FileInputTests.test_zero_byte_filesc C sf | j d�}| j d�}t||fd�}t|�}| j|dddddd g� | j|j� d
� | j|j� d � d S ) NzA
B
CzD
E
F)rJ zA
zB
�CzD
zE
�FrR rI )r r r[ r\ r^ r] )r rD rE rl r. r r r �%test_files_that_dont_end_with_newline� s
z4FileInputTests.test_files_that_dont_end_with_newlinec C s~ | j d�}| j d�}t||fd�}| j|j� d� t|�}| j|j� d� |j� | j|j� d� t|�}| j|j� d� d S ) NzA
BzC
D)rJ r, r1 r1 r1 r1 )r r r\ �filenor' rh rb r[ )r rD rE rl r* r r r � test_fileno� s
zFileInputTests.test_filenoc C s� yt dd�}| jd� W n tk
r, Y nX | jddd�}tdtf�� t |dd�}W d Q R X tdtf�� t|�}W d Q R X | j|d d
d d g� d S )
Nr
)r z-FileInput should reject invalid mode arguments A
B
C
D�wbr& �U)rJ r zA
zB
zC
�D)r �fail�
ValueErrorr r r@ r[ r\ )r rl rD r. r r r �test_opening_mode� s
z FileInputTests.test_opening_modec C sH t jd��4}td�|_tdgdd�}t|�}| j|dg� W d Q R X d S )Nz sys.stdins spam, bacon, sausage, and spamrS �rb)rJ r )r Zpatchr �bufferr r[ r\ )r Zm_stdinrl r. r r r �test_stdin_binary_mode� s
z%FileInputTests.test_stdin_binary_modec C sX t j}zDtd�t _| jtt jd�� tdgdd�}t|�}| j|dg� W d |t _X d S )Ns spam, bacon, sausage, and spamr� rS r� )rJ r )rc rd r r_ �hasattrr r[ r\ )r Z
orig_stdinrl r. r r r � test_detached_stdin_binary_mode s
z.FileInputTests.test_detached_stdin_binary_modec C s� y t ddd� d�}| jd� W n tk
r2 Y nX yt dd�}| jd� W n tk
r` Y nX G dd � d �}| jd
�}|� }t |g|d��}|j� W d Q R X | j|jd � d S ) Nr, c S s d S )Nr )r rn r r r �<lambda> s z7FileInputTests.test_file_opening_hook.<locals>.<lambda>)rV 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 )NF)rq )r r r r r s zFFileInputTests.test_file_opening_hook.<locals>.CustomOpenHook.__init__c W s d| _ t|� S )NT)rq r )r rr r r r rt s zFFileInputTests.test_file_opening_hook.<locals>.CustomOpenHook.__call__N)r r r r rt r r r r �CustomOpenHook s r� rW zopenhook not invoked)r r r� r r+ �
assertTruerq )r rl r� �tZcustom_open_hookr r r �test_file_opening_hook s
z%FileInputTests.test_file_opening_hookc C s� t td��(}|jd� |jdd � |jd� W d Q R X | jtt� tttd�d���}y4| j|j� d� | j|j� d � | j|j� d
� W n t k
r� | j
d � Y nX | j t �� t |� W d Q R X | j|j� d � | j|j� d � W d Q R X d S )
Nr| s A
B
C
s
123456789
i� � ��ascii)rJ r% zA
zB
zC
zRead to end of filer& )
r r r r � safe_unlinkr r r\ r+ �UnicodeDecodeErrorr � assertRaisesr[ )r r rl r r r �
test_readline' s"
z FileInputTests.test_readlinec C s� t td��}|jd� W d Q R X | jtt� ttdd��V}| j|j� d� | j|j� d� | j|j� d� | j|j� d� | j|j� d� W d Q R X d S ) Nr| s A
B
C
Dr� )rJ r s A
s B
s C
D� )r r r r r� r r\ r+ )r r rl r r r �test_readline_binary_mode> s z(FileInputTests.test_readline_binary_modec C s� | j d�}| j d�}t||fd��}t|�}W d Q R X | j|dddddd g� | j|j� d
� | j|j� d � | j|jf � d S ) NzA
B
CzD
E
F)rJ zA
zB
rw zD
zE
rx rR rI )r r r[ r\ r^ r] �_files)r rD rE rl r. r r r �test_context_managerK s
z#FileInputTests.test_context_managerc C sP | j d�}y t|d��
}t�W d Q R X W n" tk
rJ | j|jf � Y nX d S )Nr& )rJ )r r �OSErrorr\ r� )r rD rl r r r �test_close_on_exceptionU s
z&FileInputTests.test_close_on_exceptionc
C s( t g d��}| j|jd� W d Q R X d S )N)rJ rS )rS )r r\ r� )r rl r r r �.test_empty_files_list_specified_to_constructor] s z=FileInputTests.test_empty_files_list_specified_to_constructorc
C sN | j d�}t|gd��.}|d }| j|d� |d }| j|d� W dQ R X dS )zNTests invoking FileInput.__getitem__() with the current
line numberz line1
line2
)rJ r zline1
r, zline2
N)r r r\ )r r� rl Zretval1Zretval2r r r �test__getitem__a s
z FileInputTests.test__getitem__c C sT | j d�}t|gd��$}| jt��}|d W dQ R X W dQ R X | j|jjd� dS )zZTests invoking FileInput.__getitem__() with an index unequal to
the line numberz line1
line2
)rJ r, N� accessing lines out of order)r� )r r r� � RuntimeErrorr\ � exceptionrr )r r� rl �cmr r r �test__getitem__invalid_keyk s
z)FileInputTests.test__getitem__invalid_keyc C sT | j d�}t|gd��$}| jt��}|d W dQ R X W dQ R X | j|jjd� dS )zZTests invoking FileInput.__getitem__() with the line number but at
end-of-inputr& )rJ r N�end of input reached)r� )r r r� �
IndexErrorr\ r� rr )r r� rl r� r r r �test__getitem__eoft s
z!FileInputTests.test__getitem__eofc C sz t j}tt�}zP| jd�}| jtj|d � t|gdd�� }t|� |t _|j � W dQ R X W d|t _X | j
|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 ignoredrW z.bakT)rJ rV Nzos.unlink() was not invoked) �osr ro r� r r r
r r' rb r� rq )r Zos_unlink_origZos_unlink_replacementr� rl r r r �%test_nextfile_oserror_deleting_backup} s
z4FileInputTests.test_nextfile_oserror_deleting_backupc C s` t j}tt�}z6| jd�}t|gdd��}|t _|j� W dQ R X W d|t _X | j|jd� dS )z{Tests invoking FileInput.readline() when os.fstat() raises OSError.
This exception should be silently discarded.rW T)rJ rV Nzos.fstat() was not invoked) r� �fstatro r� r r r+ r� rq )r Z
os_fstat_origZos_fstat_replacementr� rl r r r �%test_readline_os_fstat_raises_OSError� s
z4FileInputTests.test_readline_os_fstat_raises_OSError�chmodzos.chmod does not existc C s` t j}tt�}z6| jd�}t|gdd��}|t _|j� W dQ R X W d|t _X | j|jd� dS )z{Tests invoking FileInput.readline() when os.chmod() raises OSError.
This exception should be silently discarded.rW T)rJ rV Nzos.fstat() was not invoked) r� r� ro r� r r r+ r� rq )r Z
os_chmod_origZos_chmod_replacementr� rl r r r �%test_readline_os_chmod_raises_OSError� s
z4FileInputTests.test_readline_os_chmod_raises_OSErrorc C sz G dd� dt �}|� }| jd�}t|gd��(}|j}z||_|j� }W d ||_X W d Q R X | j|jd� | j|dd� d S ) Nc @ s e Zd Zdd� Zdd� ZdS )zQFileInputTests.test_fileno_when_ValueError_raised.<locals>.FilenoRaisesValueErrorc S s t j| t� d S )N)ro r r� )r r r r r � s zZFileInputTests.test_fileno_when_ValueError_raised.<locals>.FilenoRaisesValueError.__init__c S s | j � d S )N)rt )r r r r rz � s zXFileInputTests.test_fileno_when_ValueError_raised.<locals>.FilenoRaisesValueError.filenoN)r r r r rz r r r r �FilenoRaisesValueError� s r� rW )rJ z _file.fileno() was not invokedr, zfileno() should return -1r1 )ro r r �_filerz r� rq r\ )r r� Z unconditionally_raise_ValueErrorr� rl Z file_backup�resultr r r �"test_fileno_when_ValueError_raised� s
z1FileInputTests.test_fileno_when_ValueError_raisedc C s� t � }tddg|jd���}| j|jg � | j|j� d� | j|jdg� | j|j� d� | j|jdg� | j|j� d� | j|jddg� | j|j� d� | j|jdg� | j|j� d� | j|jg � W d Q R X d S )Nz line1
line2zline3
)rJ r% zline1
�line2r& )r r r% r\ r r+ )r �srcrl r r r �test_readline_buffering� s z&FileInputTests.test_readline_bufferingc C s� t � }tddg|jd���}| j|jg � | jt|�d� | j|jdg� | jt|�d� | j|jdg� | jt|�d� | j|jddg� | jtt|� | j|jdg� | jtt|� | j|jg � W d Q R X d S )Nz line1
line2zline3
)rJ r% zline1
r� r& )r r r% r\ r r' r� �
StopIteration)r r� rl r r r �test_iteration_buffering� s z'FileInputTests.test_iteration_bufferingN) r r r rv ry r{ r� r� r� r� r� r� r� r� r� r� r� r� r� r� �unittestZskipIfr� r� r� r� r� r� r r r r ru � s*
ru c @ sZ e Zd ZdZddd�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 �rc C sD || _ || _|| _|| _|| _|| _d | _tjdd� �| _ i | _
d S )Nc S s dS )Nr r r r r r r� � s z(MockFileInput.__init__.<locals>.<lambda>) rJ rV �backuprK r r% r� � collections� defaultdict�invocation_counts�
return_values)r rJ rV r� rK r r% r r r r � s zMockFileInput.__init__c C s | j d d7 < d S )Nr0 r, )r� )r r r r r0 � s zMockFileInput.closec C s | j d d7 < | jd S )Nrb r, )r� r� )r r r r rb s zMockFileInput.nextfilec C s | j d d7 < | jd S )Nr$ r, )r� r� )r r r r r$ s zMockFileInput.filenamec C s | j d d7 < | jd S )Nr] r, )r� r� )r r r r r] s zMockFileInput.linenoc C s | j d d7 < | jd S )Nr^ r, )r� r� )r r r r r^
s zMockFileInput.filelinenoc C s | j d d7 < | jd S )Nrz r, )r� r� )r r r r rz s zMockFileInput.filenoc C s | j d d7 < | jd S )Nr` r, )r� r� )r r r r r` s zMockFileInput.isfirstlinec C s | j d d7 < | jd S )Nra r, )r� r� )r r r r ra s zMockFileInput.isstdin)NFr& r r� N)
r r r �__doc__r r0 rb r$ r] r^ rz r` ra 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 )N)� fileinput�_state� _orig_stater �_orig_FileInputr� )r r r r �setUp! s z$BaseFileInputGlobalMethodsTest.setUpc C s | j t_| jt_d S )N)r� r� r r� r� )r r r r �tearDown& s z'BaseFileInputGlobalMethodsTest.tearDownc C s2 |j | }| j|d|� t|j �}| j|d� d S )Nr, )r� r\ 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 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_| jt��}tj� W dQ R X | jd|j 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�input() already activezfileinput._state)r� ) r� �objectr� r� r� r� r� �inputr\ r� rr �assertIs)r �instancer� r r r �1test_state_is_not_None_and_state_file_is_not_None5 s zFTest_fileinput_input.test_state_is_not_None_and_state_file_is_not_Nonec C s t � }d|_|t_| j� 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_NoneB s zBTest_fileinput_input.test_state_is_not_None_and_state_file_is_Nonec C s dt _| j� 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_NoneM s z'Test_fileinput_input.test_state_is_Nonec C s� t � }t � }t � }t � }t � }t � }tj||||||d�}| j|tjd� | j||jd� | j||jd� | j||jd� | j||jd� | j||j d� | j||j
d� d S )
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().)rJ rV r� rK r r% zfileinput._staterJ rV r� rK r r% N) r� r� r� r� r� rJ rV r� rK r r% )r rJ rV r� rK r r% r� r r r r� V s
z'Test_fileinput_input.do_test_call_inputN)r r r r� r� r� r� r� r r r r r� 2 s
r� c @ s e Zd ZdZdd� Zdd� ZdS )�Test_fileinput_closez Unit tests for fileinput.close()c C s dt _t j� | jt j� dS )zPTests that fileinput.close() does nothing if fileinput._state
is NoneN)r� r� r0 � assertIsNone)r r r r r� v s z'Test_fileinput_close.test_state_is_Nonec C s0 t � }|t_tj� | j|d� | jtj� dS )z`Tests that fileinput.close() invokes close() on fileinput._state
and sets _state=Noner0 N)r� r� r� r0 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 s r� c @ s e Zd ZdZdd� Zdd� ZdS )�Test_fileinput_nextfilez#Unit tests for fileinput.nextfile()c
C sD dt _| jt��}t j� W dQ R X | jd|jj� | jt 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�no active input())r� ) r� r� r� r� rb r\ r� rr r� )r r� r r r r� � s
z*Test_fileinput_nextfile.test_state_is_Nonec C sN t � }t� }||jd< |t_tj� }| j|d� | j||� | jtj|� 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.rb N)r� r� r� r� r� rb r� r� )r Znextfile_retvalr� �retvalr r r r� � s
z.Test_fileinput_nextfile.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_filenamez#Unit tests for fileinput.filename()c
C sD dt _| jt��}t j� W dQ R X | jd|jj� | jt 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._stateN�no active input())r� ) r� r� r� r� r$ r\ r� rr r� )r r� r r r r� � s
z*Test_fileinput_filename.test_state_is_Nonec C sN t � }t� }||jd< |t_tj� }| j|d� | j||� | jtj|� 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_NoneN)r r r r� r� r� 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 _| jt��}t j� W dQ R X | jd|jj� | jt 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._stateN�no active input())r� ) r� r� r� r� r] r\ r� rr r� )r r� r r r r� � s
z(Test_fileinput_lineno.test_state_is_Nonec C sN t � }t� }||jd< |t_tj� }| j|d� | j||� | jtj|� 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.r] N)r� r� r� r� r� r] r� r� )r Z
lineno_retvalr� r� r r r r� � s
z,Test_fileinput_lineno.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_filelinenoz%Unit tests for fileinput.filelineno()c
C sD dt _| jt��}t j� W dQ R X | jd|jj� | jt 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._stateN�no active input())r� ) r� r� r� r� r^ r\ r� rr r� )r r� r r r r� � s
z,Test_fileinput_filelineno.test_state_is_Nonec C sN t � }t� }||jd< |t_tj� }| j|d� | j||� | jtj|� 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.r^ N)r� r� r� r� r� r^ r� r� )r Zfilelineno_retvalr� r� r r r r� � s
z0Test_fileinput_filelineno.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_filenoz!Unit tests for fileinput.fileno()c
C sD dt _| jt��}t j� W dQ R X | jd|jj� | jt 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._stateN�no active input())r� ) r� r� r� r� rz r\ r� rr r� )r r� r r r r� � s
z(Test_fileinput_fileno.test_state_is_Nonec C sT t � }t� }||jd< ||_|t_tj� }| j|d� | j||� | jtj|� 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.rz N) r� r� r� �
fileno_retvalr� r� rz r� r� )r r� r� r� r r r r� � s
z,Test_fileinput_fileno.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_isfirstlinez&Unit tests for fileinput.isfirstline()c
C sD dt _| jt��}t j� W dQ R X | jd|jj� | jt 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._stateN�no active input())r� ) r� r� r� r� r` r\ r� rr r� )r r� r r r r� s
z-Test_fileinput_isfirstline.test_state_is_Nonec C sN t � }t� }||jd< |t_tj� }| j|d� | j||� | jtj|� 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� s
z1Test_fileinput_isfirstline.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_isstdinz"Unit tests for fileinput.isstdin()c
C sD dt _| jt��}t j� W dQ R X | jd|jj� | jt 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._stateN�no active input())r� ) r� r� r� r� ra r\ r� rr r� )r r� r r r r� , s
z)Test_fileinput_isstdin.test_state_is_Nonec C sN t � }t� }||jd< |t_tj� }| j|d� | j||� | jtj|� 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.ra N)r� r� r� r� r� ra r� r� )r Zisstdin_retvalr� r� r r r r� 6 s
z-Test_fileinput_isstdin.test_state_is_not_NoneN)r r r r� r� r� r r r r r� ) s
r� c @ s e Zd Zdd� Zdd� ZdS )�InvocationRecorderc C s
d| _ d S )Nr )�invocation_count)r r r r r E s zInvocationRecorder.__init__c O s | j d7 _ ||f| _d S )Nr, )r� �last_invocation)r rr rs r r r rt G s zInvocationRecorder.__call__N)r r r r rt r r r r r� D s r� c @ s| e Zd ZdZdd� Zdd� Zdd� Zeje d�d d
� �Z
eje 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 )N)r� � fake_open)r r r r r� N s zTest_hook_compressed.setUpc C s | j dd� d S )Nr& r, )�do_test_use_builtin_open)r r r r �test_empty_stringQ s z&Test_hook_compressed.test_empty_stringc C s | j dd� d S )NZabcdrP )r� )r r r r � test_no_extT s z Test_hook_compressed.test_no_extzRequires gzip and zlibc C sP t j}| jt _ztjdd�}W d |t _X | j| jjd� | j| jjdi f� d S )N�test.gzrR r, )r� rR )�gzipr r� r� �hook_compressedr\ r� r� )r �
original_openr� r r r �test_gz_ext_fakeW s z%Test_hook_compressed.test_gz_ext_fakez Requires bz2c C sP t j}| jt _ztjdd�}W d |t _X | j| jjd� | j| jjdi f� d S )N�test.bz2rM r, )r� rM )�bz2ZBZ2Filer� r� r� r\ r� r� )r r� r� r r r �test_bz2_ext_fakec s z&Test_hook_compressed.test_bz2_ext_fakec C s | j dd� d S )Nz abcd.blahr: )r� )r r r r �
test_blah_exto s z"Test_hook_compressed.test_blah_extc C s | j dd� d S )Nzabcd.GzrI )r� )r r r r �test_gz_ext_builtinr s z(Test_hook_compressed.test_gz_ext_builtinc C s | j dd� d S )Nzabcd.Bz2� )r� )r r r r �test_bz2_ext_builtinu s z)Test_hook_compressed.test_bz2_ext_builtinc C sV | j | j�}ztj||�}W d | j |� X | j| jjd� | j| jj||fi f� d S )Nr, )�replace_builtin_openr� r� r� r\ r� r� )r r$ r r� r� r r r r� x s
z-Test_hook_compressed.do_test_use_builtin_openc C s t j}| t _|S )N)�builtinsr )Z
new_open_funcr� r r r r� � s z)Test_hook_compressed.replace_builtin_openN)r r r r� r� r� r� r� Z
skipUnlessr� r� r� r� r� r� r� r� � staticmethodr� r r r r r� K 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 d |t_X | j|jd� |j\} }
| j | d |� | j | d |� | j |
j
d�|� | j |
j
d�|� | j |
� d S )N)�errorsr, r �encodingr� ) r� r� r r� r� r r\ r� r� r� �popr_ ) r r� r� r� r� r� r$ r Z open_resultrr rs r r r �test� s$
zTest_hook_encoded.testc s� t td��}|jd� W d Q R X � jtt� � fdd�}|ddg� � jt�� |ddg� W d Q R X |dd g� |d
d g� d S ) Nr| s �abcc
s: t tdtd| d�d��}t|�}W d Q R X � j||� d S )Nr� zutf-8)r� )rJ r r% )r r r r[ r\ )r� �expected_linesrl r. )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��}|jd� W d Q R X � jtt� � fdd�}|ddddd g� � jt�� |d
dddd g� W d Q R X � jt�� |d dddd g� W d Q R X � jt�� |d dd
dd g� W d Q R X d S )Nr| s
A
B
C
D+IKw-c
s6 t t| td�d��}t|�}W d Q R X � j||� d S )Nzutf-7)rJ r r% )r r r r[ r\ )r r� rl r. )r r r r� � s z+Test_hook_encoded.test_modes.<locals>.checkr� zA
zB
zC
u D€ZrUr} r� zB
zC
) r r r r r� r? r@ r� r� )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 j| t� d S )N)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__)1r� r� rc rX r� r� r� r r� r� � ImportErrorr� �ior r r r Z test.supportr r r r r� r� r
r r r ZTestCaser3 ro ru r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r �mainr r r r �<module> s\
$[ <.A >A