Your IP : 216.73.216.213
�
f f= � @ 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 � n d S)*� 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�xbmc @ 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 �5/opt/alt/python34/lib64/python3.4/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 @ 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 sC t d d d �| _ t | j d � � } | j � | _ Wd QXd 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 QXt | d � � } | j � } Wd QX| j t j d | � | � | j t j d t | � � | � q Wd S)Nr) r* r+ )�
TEST_FILESr � assertEqual�imghdr�whatr- r. � bytearray)r �filenameZexpectedr1 �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_jumbo3 s z1TestImghdr.test_register_test.<locals>.test_jumbos eggsr< )r6 Ztests�appendZ
addCleanup�popr5 r7 )r r@ r r r �test_register_test2 s z TestImghdr.test_register_testc C s� t t d � �0 } | j d � | j � } | j | j � Wd QXt t d � �C } | j | � | j t j | � d � | j | j � | � Wd QXd S)N�wbs
ababagalamagar+ r ) r- r �writer! r/ r$ r5 r6 r7 )r r1 �posr r r �
test_file_pos: s
zTestImghdr.test_file_posc ( C s� | j t � � t j � Wd QX| j t � � t j d � Wd QX| j t � � t j | j d � Wd QX| j t � � t j t j | j � � Wd QXt | j d � �0 } | j t � � t j | j � � Wd QXWd QXd S)N� r+ )
� assertRaises� TypeErrorr6 r7 �AttributeErrorr, �os�fsencoder- �fileno)r �fr r r �
test_bad_argsD s zTestImghdr.test_bad_argsc C s. x' d D] } | j t j d | � � q Wd S)N� �PN
� �� Y�� cutecat� 000000JFI� GIF80)rQ rR rS rT rU rV )Z assertIsNoner6 r7 )r �headerr r r �test_invalid_headersQ 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 QX| j t
� � t j t
j | � � Wd QX| j t
� � t j d | � Wd QXq$ WWd QXd S)N�ignorer) r* r+ �latin1)�warnings�catch_warnings� simplefilter� BytesWarningr4 r r- r. �decoderI rJ r6 r7 r �StringIO)r r9 �_r1 r: r r r �test_string_dataZ s
zTestImghdr.test_string_datac C s' | j t � � t j d � Wd QXd S)NZmissing)rI �FileNotFoundErrorr6 r7 )r r r r �test_missing_filef s z TestImghdr.test_missing_filec C s� t | j d � } | j � | j t � � } t j | � Wd QXt j | j � } | j � | j t � � } t j | � Wd QXd S)Nr+ )
r- r, �closerI �
ValueErrorr6 r7 r �BytesIOr/ )r r1 �cmr r r �test_closed_filej s
zTestImghdr.test_closed_filec C sj t t d � � } | j | j � Wd QXt t d � �- } | j t j � � t j | � Wd QXWd QXd S)NrD r+ )
r- r rE r/ r rI r r r6 r7 )r r1 r r r �test_unseekablet s
zTestImghdr.test_unseekablec C s^ t t d � �I } | j | j � | j d � | j t � � } t j | � Wd QXWd QXd S)NrD r ) r- r rE r/ r$ rI �OSErrorr6 r7 )r r1 rh r r r �test_output_stream{ s
z TestImghdr.test_output_streamN)r% r& r' � classmethodr2 r3 r; rC rG rP rX rb rd ri rj rl r r r r r( s
r( �__main__�r zpng�r r �r r
�r r �r
r �r r �r r �r r �r r �r r �r r ) ro rp rq rr rs rt ru rv rw rx ry )r6 r rL Zunittestr[ Z test.supportr r r r4 �FileIOr ZTestCaser( r% �mainr r r r �<module> s( e