Your IP : 216.73.217.32
���]v � @ s� d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z m Z m Z d/ Z Gd d � d e j
� Z Gd d � d e j � Z
e d! k r� e j � d S)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 Z d Z d d � Z d d � Z d S)� UnseekableIOc C s
t j � d S)N)�io�UnsupportedOperation)�self� r# �0/opt/alt/python35/lib64/python3.5/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 @ s� e Z d Z e d d � � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d S)�
TestImghdrc C sD t d d d �| _ t | j d � � } | j � | _ Wd QRXd S)Nz
python.png�subdir�
imghdrdata�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 | d d �} | j t j | � | � t | d � � } | j t j | � | � Wd QRXt | d � � } | j � } Wd QRX| j t j d | � | � | j t j d t | � � | � q Wd S)Nr- r. r/ )�
TEST_FILESr � assertEqual�imghdr�whatr1 r2 � bytearray)r" �filenameZexpectedr5 �datar# r# r$ � test_data) s zTestImghdr.test_datac C sO d d � } t j j | � | j t j j � | j t j d d � d � d S)Nc S s | j d � r d Sd S)Ns eggs�ham)�
startswith)�h�filer# r# r$ �
test_jumbo5 s z1TestImghdr.test_register_test.<locals>.test_jumbos eggsr@ )r: Ztests�appendZ
addCleanup�popr9 r; )r" rD r# r# r$ �test_register_test4 s z TestImghdr.test_register_testc C s� t t d � �0 } | j d � | j � } | j | j � Wd QRXt t d � �C } | j | � | j t j | � d � | j | j � | � Wd QRXd S)N�wbs
ababagalamagar/ r ) r1 r �writer% r3 r( r9 r: r; )r" r5 �posr# r# r$ �
test_file_pos< s
zTestImghdr.test_file_posc , C s� | j t � � t j � Wd QRX| j t � � t j d � Wd QRX| j t � � t j | j d � Wd QRX| j t � � t j t j | j � � Wd QRXt | j d � �1 } | j t � � t j | j � � Wd QRXWd QRXd S)N� r/ )
� assertRaises� TypeErrorr: r; �AttributeErrorr0 �os�fsencoder1 �fileno)r" �fr# r# r$ �
test_bad_argsF s zTestImghdr.test_bad_argsc C s. x' d D] } | j t j d | � � q Wd S)N� �PN
� �� Y�� cutecat� 000000JFI� GIF80)rU rV rW rX rY rZ )Z assertIsNoner: r; )r" �headerr# r# r$ �test_invalid_headersS s
z TestImghdr.test_invalid_headersc C s� t j � �� t j d t � x� t D]� \ } } t | d d �} t | d � � } | j � j d � } Wd QRX| j t
� � t j t
j | � � Wd QRX| j t
� � t j d | � Wd QRXq$ WWd QRXd S)N�ignorer- r. r/ �latin1)�warnings�catch_warnings� simplefilter� BytesWarningr8 r r1 r2 �decoderM rN r: r; r �StringIO)r" r= �_r5 r> r# r# r$ �test_string_data\ s
zTestImghdr.test_string_datac C s( | j t � � t j d � Wd QRXd S)N�missing)rM �FileNotFoundErrorr: r; )r" r# r# r$ �test_missing_fileh s z TestImghdr.test_missing_filec C s� t | j d � } | j � | j t � � } t j | � Wd QRXt j | j � } | j � | j t � � } t j | � Wd QRXd S)Nr/ )
r1 r0 �closerM �
ValueErrorr: r; r �BytesIOr3 )r" r5 �cmr# r# r$ �test_closed_filel s
zTestImghdr.test_closed_filec C sm t t d � � } | j | j � Wd QRXt t d � �. } | j t j � � t j | � Wd QRXWd QRXd S)NrH r/ )
r1 r rI r3 r rM r r! r: r; )r" r5 r# r# r$ �test_unseekablev s
zTestImghdr.test_unseekablec C s` t t d � �J } | j | j � | j d � | j t � � } t j | � Wd QRXWd QRXd S)NrH r ) r1 r rI r3 r( rM �OSErrorr: r; )r" r5 rm r# r# r$ �test_output_stream} s
z TestImghdr.test_output_streamN)r) r* r+ � classmethodr6 r7 r? rG rK rT r\ rf ri rn ro rq 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 )
rt ru rv rw rx ry rz r{ r| r} r~ r r� )r: r rP Zunittestr_ Z test.supportr r r r8 �FileIOr ZTestCaser, r) �mainr# r# r# r$ �<module> s, e