File: //usr/local/lib/python3.10/test/__pycache__/test_fileinput.cpython-310.opt-2.pyc
o
�i�� � @ sr d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZzd dl Z W n e
y6 dZ Y nw zd dl Z W n e
yH dZ Y nw d dl m Z m
Z
d dlmZmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d d lmZ d d
lmZ d d lm Z G d d
� d
�Z G dd� 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 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.d/� d/ej �Z/G d0d1� d1ej �Z0G d2d3� d3ej �Z1e2d4k�r7e�3� dS dS )5� 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 sh | � d�}t|gdd�� }|d }| �|d� |d }| �|d� W d � d S 1 s-w Y d S )N� line1
line2
r rM r �line1
rD zline2
)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 s1w Y | �|jjd� d S )Nr� r rM rD )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� )Nr3 r rM r )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 s4w Y W |t _n|t _w | �
|j d� d S )Nr[ z.bakTr rY zos.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� d S �Nr[ Tr rY zos.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� r� ) 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 @ sd e Zd Z dddddd�dd�Zdd � Zd
d � Zd d
� Zdd� Zdd� Zdd� Z dd� Z
dd� Z dS )�
MockFileInputNFr3 �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$ r* r? rh r1 rc rd r� rf rg r r r r r� $ s �
r� c @ �& e Zd Z dd� Zdd� Zdd� ZdS )� BaseFileInputGlobalMethodsTestc 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� S s
r� c @ s. e Zd Z dd� Zdd� Zdd� Zdd� Zd S )
�Test_fileinput_inputc C sl t � }t� |_|t_| �t�� }t�� W d � n1 s w Y | �d|j j
� | � |tjd� d S )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_| �� d S r& )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 d t _| �� d S r& )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� d S )N)rN rZ r� r r2 r r rN rZ r� r r2 )
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 h s
r c @ � e Zd Z dd� Zdd� ZdS )�Test_fileinput_closec C s d t _t �� | �t j� d S r& )r� r� r? � assertIsNoner) r r r r � s z'Test_fileinput_close.test_state_is_Nonec C s2 t � }|t_t�� | �|d� | �tj� d S )Nr? )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 � s r c @ r
)�Test_fileinput_nextfilec C �Z d t _| �t�� }t �� W d � n1 sw Y | �d|jj� | �t j� d S �N)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 �P t � }t� }||jd< |t_t�� }| �|d� | �||� | �tj|� d S )Nrh )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_filenamec C r r ) 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 )Nr1 )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_linenoc C r r ) 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 )Nrc )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_filelinenoc C r r ) 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 )Nrd )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_filenoc C r r ) r� r� r� r� r� rb r� r� r r r r r r ) r z(Test_fileinput_fileno.test_state_is_Nonec C sV t � }t� }||jd< ||_|t_t�� }| �|d� | �||� | �tj|� d S )Nr� ) 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_isfirstlinec C r r ) 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 )Nrf )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_isstdinc C r r ) 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 )Nrg )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 Z dd� Zdd� Zdd� Ze�ed�dd � �Z e�ed�d
d � �Z
e�e d �d
d� �Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zed d � �Zd S ) �Test_hook_compressedc 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 )r/ 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>r/ r� r r zEx-binary string)r0 r r� r1 rb ra r2 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 )r7 r� )�bz2�BZ2Filer& r� r1 rb r# r$ r2 r r r �test_bz2_ext_fake� r5 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� r1 rb r# r$ �r r1 r r3 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 �localerB rC rE r r r r) � rF z2Test_hook_compressed.do_test_use_builtin_open_textc C s t j}| t _|S r& )�builtinsr )Z
new_open_funcr3 r r r rD � s z)Test_hook_compressed.replace_builtin_openN)r" r# r$ r� r* r- �unittestZ
skipUnlessr0 r4 r6 r8 r: r= r> r? r@ rA r< r) � staticmethodrD r r r r r% � s(
r% c @ r� )�Test_hook_encodedc 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" rH r rb r# r$ r �popre ) r r r� r� r& r3 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 rL �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 rR 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-7rO rP )r rQ rv r< r) r r rR s �
�z+Test_hook_encoded.test_modes.<locals>.checkr� )r� r� r� � D€ZrUr� r� )r� zB
zC
rY ) r r r r r� Z assertWarnsr� r� r� rW r r) r �
test_modes s � � � "�z Test_hook_encoded.test_modesN)r" r# r$ rN rX rZ r r r r rK � s
rK 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__)4r� r� ri r^ r� r� rH r rI r8 � ImportErrorr0 r r r r �pathlibr Z test.supportr Ztest.support.os_helperr r r� r
r rN r r
r r% ZTestCaserB r{ r� r� r� r r r r r r r r r! r" r% rK r[ r" �mainr r r r �<module> sn � � $S s/? YA
�