Your IP : 216.73.216.213
�
f f�S � @ s� d Z d d l Z d d l m Z d d l Z d d l Z d d l Z e j d � Z d Z d Z
Gd d � d e j � Z Gd d
� d
e j
� Z Gd d � d e � Z Gd
d � d e � Z d d d � Z e d k r� e d d � n d S)z!Test script for the gzip module.
� N)�support�gzipsy int length=DEFAULTALLOC, err = Z_OK;
PyObject *RetVal;
int flushmode = Z_FINISH;
unsigned long start_total_out;
s� /* zlibmodule.c -- gzip-compatible data compression */
/* See http://www.gzip.org/zlib/
/* See http://www.winimage.com/zLibDll for Windows */
c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)� UnseekableIOc C s d S)NF� )�selfr r �3/opt/alt/python34/lib64/python3.4/test/test_gzip.py�seekable s zUnseekableIO.seekablec C s
t j � d S)N)�io�UnsupportedOperation)r r r r �tell s zUnseekableIO.tellc G s
t j � d S)N)r r
)r �argsr r r �seek s zUnseekableIO.seekN)�__name__�
__module__� __qualname__r r r
r r r r r s r c @ s1 e Z d Z e j Z d d � Z d d � Z d S)�BaseTestc C s t j | j � d S)N)r �unlink�filename)r r r r �setUp& s zBaseTest.setUpc C s t j | j � d S)N)r r r )r r r r �tearDown) s zBaseTest.tearDownN)r r r r ZTESTFNr r r r r r r r # s r c @ s� e Z d Z d 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 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 d0 d1 � Z d2 d3 � Z d4 d5 � Z d6 d7 � Z d8 d9 � Z d: d; � Z d< d= � Z d> d? � Z! d@ dA � Z" dB dC � Z# dD dE � Z$ dF S)G�TestGzip�bc C s� t | � } t j | j d | � � } | j | � } Wd QX| j | t | � � t j | j d | � � } | j | j � | � Wd QXd S)N�w�r)�bytesr �GzipFiler �write� assertEqual�len�read)r �data�modeZb_data�f�lr r r �write_and_read_back. s z TestGzip.write_and_read_backc
C s� t j | j d � �[ } | j t d � | j � | j � t t d � rb t j | j � � n | j
� Wd QX| j
� d S)N�wb�2 �fsync) r r r r �data1�flush�fileno�hasattr�osr' �close)r r" r r r �
test_write6 s
zTestGzip.test_writec C s^ | j t t d � � t t t d � � � } | j d d d d d g �} | j | � d S)Nr& � �B�shape� � )r$ �
memoryviewr( r �range�cast)r �mr r r r �test_write_memoryviewG s z TestGzip.test_write_memoryviewc C s | j t t d � � d S)Nr& )r$ � bytearrayr( )r r r r �test_write_bytearrayM s z TestGzip.test_write_bytearrayc C s� t j | j d � �] } | j t � � | j d � Wd QX| j t � � | j d g � Wd QX| j t � Wd QXt j | j d � � } | j | j � t � Wd QXd S)Nr% �a� �rb) r r r � assertRaises� TypeErrorr r( r r )r r" r r r � test_write_incompatible_typeP s z%TestGzip.test_write_incompatible_typec
C sL | j � t j | j d � � } | j � } Wd QX| j | t d � d S)Nr r& )r. r r r r r r( )r r" �dr r r � test_read] s
zTestGzip.test_readc
C s� | j � g } d } t j | j d � �W } xM | j � } | sG Pn | j | � | t | � 7} | j | j � | � q1 WWd QX| j d j | � t
d � d S)Nr r � r& ) r. r r r �read1�appendr r r �joinr( )r Zblocks�nreadr" rA r r r �
test_read1d s
zTestGzip.test_read1c C s� | j � t j | j d � } | j � | j t � � | j d � Wd QX| j t � � | j d � Wd QX| j t � � | j � Wd QXt j | j d � } | j � | j t � � | j
d � Wd QX| j t � � | j � Wd QXd S)Nr r< r r rC ) r. r r r r- r> �
ValueErrorr r
r r r) )r r" r r r �test_io_on_closed_objects s
z!TestGzip.test_io_on_closed_objectc C s� | j � t j | j d � � } | j t d � Wd QXt j | j d � � } | j � } Wd QX| j | t d t d � d S)N�ab� r= r& ) r. r r r r �data2r r r( )r r" rA r r r � test_append� s
zTestGzip.test_appendc C s� t j | j d d � � } | j d � Wd QXxE t d d � D]4 } t j | j d d � � } | j d � Wd QXq> Wt j | j d � �7 } d } x' | j d � } | | 7} | s� Pq� q� WWd QX| j | d d
� d S) Nr% � � ar �� rK r= rC i �� )r r r r r5 r r )r r" �iZzgfile�contentsZztxtr r r �test_many_append� s
zTestGzip.test_many_appendc C s� t j | j d � � } | j t d � Wd QXt j | j d � �! } | j | j � t d � Wd QX| j t � � t j | j d � Wd QXd S)N�xbr& r= ) r r r r r( r r r> �FileExistsError)r r" r r r �test_exclusive_write� s z TestGzip.test_exclusive_writec C sw | j � t j | j d � �2 } t j | � � } d d � | D� } Wd QXWd QX| j | d t j d d � � d S)Nr= c S s g | ] } | � q Sr r )�.0�liner r r �
<listcomp>� s z1TestGzip.test_buffered_reader.<locals>.<listcomp>r& �keependsT) r. r r r r �BufferedReaderr r( �
splitlines)r r" r �linesr r r �test_buffered_reader� s
z TestGzip.test_buffered_readerc
C s� | j � t j | j d � �a } d } xQ | j | � } | rQ | d k rQ Pn | j t | � | k � | d d } q+ WWd QXd S)Nr= r r< r& )r. r r r �readline�
assertTruer )r r" Z line_length�Lr r r �
test_readline� s
zTestGzip.test_readlinec C s| | j � t j | j d � � } | j � } Wd QXt j | j d � �- } x# | j d � } | g k rO PqO qO WWd QXd S)Nr= � )r. r r r � readlines)r r" rc r r r �test_readlines� s
zTestGzip.test_readlinesc C s� | j � t j | j � �� } x� | j � } | j � } | sD Pn | j � } | j | � t | � d k rx d } n t | � } | j | � } | j | d | � | � | j | � q" WWd QXd S)N�
)
r. r r r r ra r
r r r )r r" ZoldposZline1ZnewposZamountZline2r r r �test_seek_read� s
zTestGzip.test_seek_readc C sr | j � t j | j � �6 } | j d � | j d d d �| j d � } Wd QX| j | t d d � � d S)Nrh �whencer< � � )r. r r r r r
r r( )r r" �yr r r �test_seek_whence� s
zTestGzip.test_seek_whencec C sY t j | j d � �> } x4 t d d d � D] } | j | � | j d � q+ WWd QXd S)Nr r r/ � s GZ
)r r r r5 r
r )r r" �posr r r �test_seek_write� s
zTestGzip.test_seek_writec C s� | j � t j | j d � � } | j | j j d � Wd QXt j | j � t j | j d � � } | j | j j d � Wd QXd S)Nr r= �xrV ) r. r r r r Z myfileobjr! r r )r r" r r r � test_mode� s
zTestGzip.test_modec C s_ xX d D]P } t j | j | � �3 } | j t | d � � | j | j | j � Wd QXq Wd S)Nr% r= �name)zwbzrb)r r r rb r+ r rt )r r! r" r r r � test_1647484� s
zTestGzip.test_1647484c C s^ | j � t j | j d � �9 } | j t | j d � � | j | j j | j � Wd QXd S)Nr= rt ) r. r r r rb r+ �fileobjr rt )r r" r r r �test_paddedfile_getattr s
z TestGzip.test_paddedfile_getattrc C s� d } t j | j d d | �� } | j t � Wd QXt j | j � �L } | j � } | j | t � | j t | d � � | j | j | � Wd QXd S)Ni�[r �mtime)
r r r r r( r r rb r+ rx )r rx �fWrite�fReadZdataReadr r r �
test_mtime s zTestGzip.test_mtimec C s� d } t j | j d d | �� } | j t � Wd QXt | j d � �x} | j d � } | j | d � | j d � } | j | d � | j d � } | j | d � | j d � } | j | t j d
| � � | j d � } | j | d � | j d � } | j | d � | j j
d
� d }
| j t |
� � } | j | |
� | j t
j | j � j d � | j d � } | j | d � | j d � }
| j |
t j d
t t � � � Wd QXd S)Ni�[r rx r= � s �r< s r3 z<is s �zLatin-1s r2 s ��d�)r r r r r( �openr r �structZpack�encoder r
r, �stat�st_size)r rx ry rz ZidBytesZcmByteZ flagsByteZ
mtimeBytesZxflByteZosByteZexpectedZ nameBytesZ
crc32BytesZ
isizeBytesr r r �
test_metadata s0 zTestGzip.test_metadatac ! C s� t j | j d � � } | j d � Wd QXt j | j d � } | j � y | � Wd QXWn t k
ro Yn X| j d � y* t j | j d � � } d d Wd QXWn t k
r� Yn X| j d � d S)Nr% s xxxr= z4__enter__ on a closed file didn't raise an exceptionr< r z 1/0 didn't raise an exception)r r r r r- rI Zfail�ZeroDivisionError)r r" r r r �test_with_open@ s
zTestGzip.test_with_openc C s� t j | j d � � } | j t d � Wd QXt | j d � � } | j d d � Wd QXt j | j d � �* } | j � } | j | t d d � Wd QXd S)Nr% r& rK s r= zIncorrect data in file)r r r r r( r} r r )r r" rA r r r �test_zero_padded_fileU s z TestGzip.test_zero_padded_filec C s� t d } t � } t j d | d d � � } | j | � Wd QX| j � } t | � } t j d | d d � � } | j | j � | � Wd QXd S)Nr& rv r! r% r= )r( r r r r �getvaluer r )r � uncompressed�bufr" �
compressedr r r �test_non_seekable_filea s
z TestGzip.test_non_seekable_filec C s t d } t j | j d � � } | j | � Wd QXd d � } t j | j d � �� } d | _ d } x_ | � D]T } | j | � } | d k r� Pn | j | j t | � � | � | t | � 7} qr W| j | j d � d � | j | t | � � Wd QXd S)
NrQ r% c s s- x& x t d d d � D] } | Vq Wq Wd S)N� r& rh )r5 )�nr r r �sizesp s z!TestGzip.test_peek.<locals>.sizesr= �! r rC �d )
r( r r r r Zmax_read_chunk�peekr r r )r r� r" r� rG r� �sr r r � test_peekk s
zTestGzip.test_peekc C s t d j d � j d d � } | j � t j | j d � �; } t j | d d �� } | j | j
� | � Wd QXWd QXd S)Nr& �asciir\ Tr �encoding) r( �decoder^ r. r r r r �
TextIOWrapperr rf )r r_ r" �tr r r �test_textio_readlines� s
z TestGzip.test_textio_readlinesc C s_ t j | j t j t j B� } t j | d � �( } t j d | d d � � } Wd QXWd QXd S)Nr% rv r! r )r, r} r �O_WRONLY�O_CREAT�fdopenr r )r �fdr" �gr r r �test_fileobj_from_fdopen� s z!TestGzip.test_fileobj_from_fdopenc C s� | j } y | j d � } Wn t k
r= | j d � Yn Xt j | d � � } | j t d � Wd QXt j | d � �! } | j | j � t d � Wd QXt j | d � �! } | j | j � t d � Wd QXd S)Nr� z%Temporary file name needs to be ASCIIr% r& r= )
r r �UnicodeEncodeErrorZskipTestr r r r( r r )r Z str_filenameZbytes_filenamer" r r r �test_bytes_filename� s
z TestGzip.test_bytes_filenamec C s� x� t t g D]� } x� f d d d g D]n } t j | | � } | j t | � t � t j d t j | � d d � � } | j | j
� | � Wd QXq& Wq
Wd S)
Nr< � rO rv r! r= )r< )r� )rO ) r( rM r �compressr �typer r r �BytesIOr )r r r �datacr" r r r �
test_compress� s $zTestGzip.test_compressc
C s� x� t t f D]� } t j � } t j d | d d � � } | j | � Wd QX| j t j | j � � | � t j
| � } | j t j | � | � q
Wd S)Nrv r! r% ) r( rM r r� r r r r �
decompressr� r� )r r r� r" r� r r r �test_decompress� s zTestGzip.test_decompressc C s t d } t j | � d d � } t j d t j | � � � } | j t | j � Wd QXt j d t j | � � �< } | j | j t
| � � | � | j t | j d � Wd QXx[ t d d � D]J } t j d t j | d | � � � � } | j t | j d � Wd QXq� Wd S)Nr& r2 rv r< r| rh i����) r( r r� r r r� r> �EOFErrorr r r r5 )r r Z truncatedr" rS r r r �test_read_truncated� s
(z TestGzip.test_read_truncatedc
C sD d } t j d t j | � � � } | j | j � d � Wd QXd S)Ns ��cQ� Extra I-. 2�Mx rv s Test)r r r r� r r )r Zgzdatar" r r r �test_read_with_extra� s z TestGzip.test_read_with_extrac C sZ t j | j d � � } | j t � Wd QXt j | j d � � } | j j � Wd QXd S)Nr% r= )r r} r r r( rv Zprepend)r r" r r r �test_prepend_error� s zTestGzip.test_prepend_errorN)%r r r r$ r. r8 r: r@ rB rH rJ rN rU rX r` rd rg ri rn rq rs ru rw r{ r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r - sD
-
r c @ sp e Z 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)�TestOpenc 4 C s� t d } t j | j d � � } | j | � Wd QXt | j d � �, } t j | j � � } | j | | � Wd QXt j | j d � � } | j | j � | � Wd QXt j | j d � � } | j | � Wd QXt | j d � �0 } t j | j � � } | j | | d � Wd QX| j t � � t j | j d � Wd QXt
j | j � t j | j d � � } | j | � Wd QXt | j d � �, } t j | j � � } | j | | � Wd QXd S)Nr& r% r= rK r| rV ) r( r r} r r r� r r r> rW r r )r r� r" � file_datar r r �test_binary_modes� s*
zTestOpen.test_binary_modesc 4 C s� t d } t j | j d � � } | j | � Wd QXt | j d � �, } t j | j � � } | j | | � Wd QXt j | j d � � } | j | j � | � Wd QXt j | j d � � } | j | � Wd QXt | j d � �0 } t j | j � � } | j | | d � Wd QX| j t � � t j | j d � Wd QXt
j | j � t j | j d � � } | j | � Wd QXt | j d � �, } t j | j � � } | j | | � Wd QXd S)Nr& r r= r r; r| rr ) r( r r} r r r� r r r> rW r r )r r� r" r� r r r �test_implicit_binary_modes� s*
z#TestOpen.test_implicit_binary_modesc # C sL t j d � d } | j d t j � } t j | j d � � } | j | � Wd QXt | j d � �5 } t j | j
� � j d � } | j | | � Wd QXt j | j d � � } | j | j
� | � Wd QXt j | j d � � } | j | � Wd QXt | j d � �9 } t j | j
� � j d � } | j | | d � Wd QXd S) Nr� r& �
�wtr= �rt�atr| ) r( r� �replacer, �linesepr r} r r r� r r )r r� �uncompressed_rawr" r� r r r �test_text_modes s zTestOpen.test_text_modesc C s� t d } | j d � } t j | � } t j t j | � d � � } | j | j � | � Wd QXt j t j | � d � � } | j | j � | � Wd QXt j t j | � d � � } | j | j � | � Wd QXd S)Nr& r� r r= r� ) r( r� r r� r} r r� r r )r Zuncompressed_bytesZuncompressed_strr� r" r r r � test_fileobj s
zTestOpen.test_fileobjc * C s | j t � � t j d � Wd QX| j t � � t j | j d � Wd QX| j t � � t j | j d � Wd QX| j t � � t j | j d d d �Wd QX| j t � � t j | j d d d �Wd QX| j t � � t j | j d d d
�Wd QXd S) Ngw��/�^@ZwbtZxbtr= r� zutf-8�errors�ignore�newliner� )r> r? r r} rI r )r r r r �test_bad_params# s zTestOpen.test_bad_paramsc C s� t j d � d } | j d t j � } t j | j d d d �� } | j | � Wd QXt | j d � �5 } t j | j
� � j d � } | j | | � Wd QXt j | j d d d �� } | j | j
� | � Wd QXd S) Nr� r& r� r� r� zutf-16r= r� ) r( r� r� r, r� r r} r r r� r r )r r� r� r" r� r r r �
test_encoding2 s zTestOpen.test_encodingc C so t j | j d � � } | j d � Wd QXt j | j d d d d d �� } | j | j � d � Wd QXd S) Nr% s foo�barr� r� r� r� r� Zfoobar)r r} r r r r )r r" r r r �test_encoding_error_handler> s
!z$TestOpen.test_encoding_error_handlerc C s� t j d � d } t j | j d d d �� } | j | � Wd QXt j | j d d d �� } | j | j � | g � Wd QXd S)Nr� r& r� r� r� r� �
)r( r� r r} r r r rf )r r� r" r r r � test_newlineF s
zTestOpen.test_newlineN) r r r r� r� r� r� r� r� r� r� r r r r r� � s r� c C s t j t t � d S)N)r Z run_unittestr r� )�verboser r r � test_mainN s r� �__main__r� T)�__doc__ZunittestZtestr r, r r~ �
import_moduler r( rM r� r ZTestCaser r r� r� r r r r r �<module> s
� �~