File: //usr/local/lib/python3.7/test/__pycache__/test_imghdr.cpython-37.opt-2.pyc
B
��g� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZm Z dZ
G dd� dej �Z G dd� dej
�Zedkr|e�� dS ) � N)�findfile�TESTFN�unlink)
)z
python.png�png)z
python.gifZgif)z
python.bmpZbmp)z
python.ppmZppm)z
python.pgmZpgm)z
python.pbmZpbm)z
python.jpgZjpeg)z
python.rasZrast)z
python.sgiZrgb)z python.tiffZtiff)z
python.xbmZxbm)z python.webpZwebp)z
python.exrZexrc @ s e Zd Zdd� Zdd� ZdS )� UnseekableIOc C s
t j�d S )N)�io�UnsupportedOperation)�self� r
�,/usr/local/lib/python3.7/test/test_imghdr.py�tell s zUnseekableIO.tellc O s
t j�d S )N)r r )r �args�kwargsr
r
r �seek s zUnseekableIO.seekN)�__name__�
__module__� __qualname__r r r
r
r
r r s r c @ sx e Zd Zedd� �Zdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� ZdS ) �
TestImghdrc C s4 t ddd�| _t| jd��}|�� | _W d Q R X d S )Nz
python.png�
imghdrdata)�subdir�rb)r �testfile�open�read�testdata)�cls�streamr
r
r �
setUpClass! s zTestImghdr.setUpClassc C s t t� d S )N)r r )r r
r
r �tearDown' s zTestImghdr.tearDownc
C s� x�t D ]�\}}t|dd�}| �t�|�|� t|d��}| �t�|�|� W d Q R X t|d��}|�� }W d Q R X | �t�d |�|� | �t�d t|��|� qW d S )Nr )r r )�
TEST_FILESr � assertEqual�imghdr�whatr r � bytearray)r �filename�expectedr �datar
r
r � test_data* s zTestImghdr.test_datac
C sR xLt D ]D\}}| j|d��* t|dd�}| �t�t�|��|� W d Q R X qW d S )N)r$ r )r )r ZsubTestr r r! r" �pathlibZPath)r r$ r% r
r
r �test_pathlike_filename5 s z!TestImghdr.test_pathlike_filenamec C s: dd� }t j�|� | �t jj� | �t �d d�d� d S )Nc S s | � d�rdS d S )Ns eggs�ham)�
startswith)�h�filer
r
r �
test_jumbo< s
z1TestImghdr.test_register_test.<locals>.test_jumbos eggsr* )r! Ztests�appendZ
addCleanup�popr r" )r r. r
r
r �test_register_test; s z TestImghdr.test_register_testc C sz t td��$}|�d� |�� }|�| j� W d Q R X t td��2}|�|� | �t�|�d� | �|�� |� W d Q R X d S )N�wbs
ababagalamagar r ) r r �writer r r r r! r" )r r �posr
r
r �
test_file_posC s
zTestImghdr.test_file_posc
C s� | � t�� t�� W d Q R X | � t�� t�d � W d Q R X | � t�� t�| jd� W d Q R X | � t�� t�t�| j�� W d Q R X t| jd��*}| � t�� t�|� � � W d Q R X W d Q R X d S )N� r )
� assertRaises� TypeErrorr! r" �AttributeErrorr �os�fsencoder �fileno)r �fr
r
r �
test_bad_argsM s zTestImghdr.test_bad_argsc C s$ x dD ]}| � t�d |�� qW d S )N)s �PN
s �s Y�s cutecats 000000JFIs GIF80)Z assertIsNoner! r" )r �headerr
r
r �test_invalid_headersZ s
z TestImghdr.test_invalid_headersc C s� t �� �� t �dt� x�tD ]�\}}t|dd�}t|d��}|�� �d�}W d Q R X | � t
�� t � t
�|�� W d Q R X | � t
�� t � d |� W d Q R X q W W d Q R X d S )N�ignorer )r r �latin1)�warnings�catch_warnings� simplefilter� BytesWarningr r r r �decoder7 r8 r! r" r �StringIO)r r$ �_r r& r
r
r �test_string_datac s
zTestImghdr.test_string_datac C s$ | � t�� t�d� W d Q R X d S )N�missing)r7 �FileNotFoundErrorr! r" )r r
r
r �test_missing_fileo s z TestImghdr.test_missing_filec C sl t | jd�}|�� | �t��}t�|� W d Q R X t�| j �}|�� | �t��}t�|� W d Q R X d S )Nr )
r r �closer7 �
ValueErrorr! r" r �BytesIOr )r r �cmr
r
r �test_closed_files s zTestImghdr.test_closed_filec
C s^ t td��}|�| j� W d Q R X ttd��(}| �tj�� t� |� W d Q R X W d Q R X d S )Nr2 r )
r r r3 r r r7 r r r! r" )r r r
r
r �test_unseekable} s
zTestImghdr.test_unseekablec
C sP t td��<}|�| j� |�d� | �t��}t�|� W d Q R X W d Q R X d S )Nr2 r ) r r r3 r r r7 �OSErrorr! r" )r r rQ r
r
r �test_output_stream� s
z TestImghdr.test_output_streamN)r r r � classmethodr r r' r) r1 r5 r> r@ rJ rM rR rS rU r
r
r
r r s
r �__main__)r! r r: r( ZunittestrC Z test.supportr r r r �FileIOr ZTestCaser r �mainr
r
r
r �<module> s k