Your IP : 216.73.216.213
3
��2a� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZm Z d/Z
G d d � d ej �Z G d d � d ej
�Zed!kr|ej� dS )0� N)�findfile�TESTFN�unlink�
python.png�png�
python.gif�gif�
python.bmp�bmp�
python.ppm�ppm�
python.pgm�pgm�
python.pbm�pbm�
python.jpg�jpeg�
python.ras�rast�
python.sgi�rgb� python.tiff�tiff�
python.xbm�xbm� python.webp�webp�
python.exr�exrc @ s e Zd Zdd� Zdd� ZdS )� UnseekableIOc C s
t j�d S )N)�io�UnsupportedOperation)�self� r# �0/opt/alt/python36/lib64/python3.6/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��}|j� | _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�}| jtj|�|� t|d��}| jtj|�|� W d Q R X t|d��}|j� }W d Q R X | jtjd |�|� | jtjd t|��|� qW d S )Nr- )r. r/ )�
TEST_FILESr � assertEqual�imghdr�whatr1 r2 � bytearray)r" �filename�expectedr5 �datar# r# r$ � test_data* s zTestImghdr.test_datac C sR xLt D ]D\}}| j|d��* t|dd�}| jtjtj|��|� W d Q R X qW d S )N)r= r- )r. )r8 ZsubTestr r9 r: r; �pathlibZPath)r" r= r> r# r# r$ �test_pathlike_filename5 s z!TestImghdr.test_pathlike_filenamec C s: dd� }t jj|� | jt jj� | jt jd d�d� d S )Nc S s | j d�rdS d S )Ns eggs�ham)�
startswith)�h�filer# r# r$ �
test_jumbo< s
z1TestImghdr.test_register_test.<locals>.test_jumbos eggsrC )r: Ztests�appendZ
addCleanup�popr9 r; )r" rG r# r# r$ �test_register_test; s z TestImghdr.test_register_testc C sz t td��$}|jd� |j� }|j| j� W d Q R X t td��2}|j|� | jtj|�d� | j|j� |� W d Q R X d S )N�wbs
ababagalamagar/ r ) r1 r �writer% r3 r( r9 r: r; )r" r5 �posr# r# r$ �
test_file_posC s
zTestImghdr.test_file_posc , C s� | j t�� tj� W d Q R X | j t�� tjd � W d Q R X | j t�� tj| jd� W d Q R X | j t�� tjtj| j�� W d Q R X t| jd��*}| j t�� tj|j � � W d Q R X W d Q R X d S )N� r/ )
� assertRaises� TypeErrorr: r; �AttributeErrorr0 �os�fsencoder1 �fileno)r" �fr# r# r$ �
test_bad_argsM s zTestImghdr.test_bad_argsc C s$ x dD ]}| j tjd |�� qW d S )N� �PN
� �� Y�� cutecat� 000000JFI� GIF80)rX rY rZ r[ r\ r] )Z assertIsNoner: r; )r" �headerr# r# r$ �test_invalid_headersZ s z TestImghdr.test_invalid_headersc C s� t j� �� t jdt� x�tD ]�\}}t|dd�}t|d��}|j� jd�}W d Q R X | j t
�� t j t
j|�� W d Q R X | j t
�� t j d |� W d Q R X q W W d Q R X d S )N�ignorer- )r. r/ �latin1)�warnings�catch_warnings� simplefilter� BytesWarningr8 r r1 r2 �decoderP rQ r: r; r �StringIO)r" r= �_r5 r? r# r# r$ �test_string_datac s
zTestImghdr.test_string_datac C s$ | j t�� tjd� W d Q R X d S )N�missing)rP �FileNotFoundErrorr: r; )r" r# r# r$ �test_missing_fileo s z TestImghdr.test_missing_filec C sl t | jd�}|j� | jt��}tj|� W d Q R X tj| j �}|j� | jt��}tj|� W d Q R X d S )Nr/ )
r1 r0 �closerP �
ValueErrorr: r; r �BytesIOr3 )r" r5 �cmr# r# r$ �test_closed_files s zTestImghdr.test_closed_filec C s^ t td��}|j| j� W d Q R X ttd��(}| jtj�� tj |� W d Q R X W d Q R X d S )NrK r/ )
r1 r rL r3 r rP r r! r: r; )r" r5 r# r# r$ �test_unseekable} s
zTestImghdr.test_unseekablec C sP t td��<}|j| j� |jd� | jt��}tj|� W d Q R X W d Q R X d S )NrK r ) r1 r rL r3 r( rP �OSErrorr: r; )r" r5 rp r# r# r$ �test_output_stream� s
z TestImghdr.test_output_streamN)r) r* r+ � classmethodr6 r7 r@ rB rJ rN rW r_ ri rl rq rr rt r# r# r# r$ r, s
r, �__main__�r r �r r �r r
�r r �r
r �r r �r r �r r �r r �r r �r r �r r �r r )
rw rx ry rz r{ r| r} r~ r r� r� r� r� )r: r rS rA Zunittestrb Z test.supportr r r r8 �FileIOr ZTestCaser, r) �mainr# r# r# r$ �<module> s. k