Your IP : 216.73.216.213
�
g f.� � @ s� d d l m Z d d l m Z m Z d d l Z d d l m Z 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
y d d l Z Wn e k
r� d Z Yn Xe j d � Z d d l m Z m Z m Z Gd d � d e j � Z Gd
d � d e � Z Gd d
� d
e � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z d d � Z e d k r�e � n d S)� )�support)�
bigmemtest�_4GN)�BytesIO)�unlink�bz2)�BZ2File�
BZ2Compressor�BZ2Decompressorc @ s� e Z d Z d Z d d d d d d d d d
d d d
d d d d d d d d d g Z d j e � Z d Z d Z d Z d d � Z
d d � Z e j
d k r� d d! � Z n d" d! � Z d# S)$�BaseTestzBase for other testcases.s root:x:0:0:root:/root:/bin/bash
s bin:x:1:1:bin:/bin:
s daemon:x:2:2:daemon:/sbin:
s adm:x:3:4:adm:/var/adm:
s lp:x:4:7:lp:/var/spool/lpd:
s sync:x:5:0:sync:/sbin:/bin/sync
s- shutdown:x:6:0:shutdown:/sbin:/sbin/shutdown
s! halt:x:7:0:halt:/sbin:/sbin/halt
s" mail:x:8:12:mail:/var/spool/mail:
s" news:x:9:13:news:/var/spool/news:
s# uucp:x:10:14:uucp:/var/spool/uucp:
s operator:x:11:0:operator:/root:
s! games:x:12:100:games:/usr/games:
s, gopher:x:13:30:gopher:/usr/lib/gopher-data:
s( ftp:x:14:50:FTP User:/var/ftp:/bin/bash
s# nobody:x:65534:65534:Nobody:/home:
s. postfix:x:100:101:postfix:/var/spool/postfix:
s- niemeyer:x:500:500::/home/niemeyer:/bin/bash
s> postgres:x:101:102:PostgreSQL Server:/var/lib/pgsql:/bin/bash
s6 mysql:x:102:103:MySQL server:/var/lib/mysql:/bin/bash
s# www:x:103:104::/var/www:/bin/false
� s� BZh91AY&SY.�N >_� @��n ?��0�� �F��#&�F��Ц�MQ���4hh$�i4�4S�<Q�H���Շ���
��I��/E pu�]ɻDL
Q 1ߠ���2O9�� 7d�I �J�L#����/��#C��W����M��Xa<e�t�L�����˄.�����`n}�lh E,�ZVH>���9gd6
�Պ��.K�v��xw�ߡ���|/";���ձ#A���o��y\�O煚��f8�2��"%�>V,��z⟣�"�E)I�^��r�U�2�6���U��0��^��DF����ݸ����ankR���3]���f��"6l$�j@���l1..�k�ls����E��
��k��n��Q�%y�v��z��̹z n7.G�]��B@�!8`s BZh9rE8P� s this is not a valid bzip2 filec C s t j | _ d S)N)r ZTESTFN�filename)�self� r �2/opt/alt/python34/lib64/python3.4/test/test_bz2.py�setUp6 s zBaseTest.setUpc C s, t j j | j � r( t j | j � n d S)N)�os�path�isfiler
r )r r r r �tearDown9 s zBaseTest.tearDown�win32c C s
t j | � S)N)r �
decompress)r �datar r r r ? s zBaseTest.decompressc
C s� t j d d d d t j d t j d t j �} | j j | � | j j � | j j � } | j j � | j � d k r� t
j | � } n | S)NZbunzip2�shellT�stdin�stdout�stderrr ) �
subprocess�Popen�PIPEZSTDOUTr �write�closer �read�waitr r )r r �popZretr r r r B s
N)�__name__�
__module__� __qualname__�__doc__�
TEXT_LINES�join�TEXT�DATA�
EMPTY_DATA�BAD_DATAr r �sys�platformr r r r r r s> r c @ s� e Z d Z d Z d 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 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 dG � Z% dH dI � Z& dJ dK � Z' dL dM � Z( dN dO � Z) dP dQ � Z* dR dS � Z+ dT dU � Z, dV dW � Z- dX dY � Z. dZ d[ � Z/ d\ d] � Z0 d^ d_ � Z1 d` da � Z2 e3 j4 e5 db � dc dd � � Z6 de df � Z7 dg dh � Z8 di dj � Z9 dk dl � Z: dm dn � Z; do dp � Z< dq dr � Z= ds dt � Z> du dv � Z? dw dx � Z@ dy S)z� BZ2FileTestzTest the BZ2File class.� r c
C s@ t | j d � �( } | j | j | � | j | � Wd QXd S)N�wb)�openr
r r, )r �streams�suffix�fr r r �createTempFileS s zBZ2FileTest.createTempFilec C s� | j t t d � | j t t t j d � | j t t t j d � | j t t t j d � | j t t t j d d �| j t t t j d d �d S)Ngw��/�^@�zZrxZrbt�
compresslevelr �
)� assertRaises� TypeErrorr �
ValueErrorr �devnull)r r r r � testBadArgsX s zBZ2FileTest.testBadArgsc C sU | j � t | j � �6 } | j t | j d � | j | j � | j � Wd QXd S)N)r8 r r
r< r= r" � assertEqualr+ )r �bz2fr r r �testRead` s
zBZ2FileTest.testReadc
C sH | j d d d | j � t | j � � } | j t | j � Wd QXd S)Nr5 r r6 )r8 r. r r
r< �OSErrorr" )r rB r r r �testReadBadFilef s zBZ2FileTest.testReadBadFilec C s_ | j d d � t | j � �: } | j t | j d � | j | j � | j d � Wd QXd S)Nr5 � )r8 r r
r< r= r" rA r+ )r rB r r r �testReadMultiStreamk s z BZ2FileTest.testReadMultiStreamc C s� t j } t | j � t _ z_ | j d d � t | j � �: } | j t | j d � | j
| j � | j d � Wd QXWd | t _ Xd S)Nr5 rF ) r Z _BUFFER_SIZE�lenr, r8 r r
r< r= r" rA r+ )r Z buffer_sizerB r r r �testReadMonkeyMultiStreamq s 'z%BZ2FileTest.testReadMonkeyMultiStreamc
C sH | j d | j � t | j � � } | j | j � | j � Wd QXd S)Nr6 )r8 r. r r
rA r" r+ )r rB r r r �testReadTrailingJunk~ s z BZ2FileTest.testReadTrailingJunkc C sR | j d d d | j � t | j � �$ } | j | j � | j d � Wd QXd S)Nr5 rF r6 )r8 r. r r
rA r" r+ )r rB r r r � testReadMultiStreamTrailingJunk� s z+BZ2FileTest.testReadMultiStreamTrailingJunkc C sU | j � t | j � �6 } | j t | j d � | j | j d � d � Wd QXd S)Nr r )r8 r r
r< r= r" rA )r rB r r r � testRead0� s
zBZ2FileTest.testRead0c
C si | j � t | j � �J } d } x' | j d � } | s> Pn | | 7} q% W| j | | j � Wd QXd S)Nr r; )r8 r r
r" rA r+ )r rB �text�strr r r �testReadChunk10� s
zBZ2FileTest.testReadChunk10c C ss | j d d � t | j � �N } d } x' | j d � } | sD Pn | | 7} q+ W| j | | j d � Wd QXd S)Nr5 rF r r; )r8 r r
r" rA r+ )r rB rM rN r r r �testReadChunk10MultiStream� s z&BZ2FileTest.testReadChunk10MultiStreamc C sL | j � t | j � �- } | j | j d � | j d d � � Wd QXd S)N�d )r8 r r
rA r" r+ )r rB r r r � testRead100� s
zBZ2FileTest.testRead100c
C sz | j � t | j � �[ } | j � } | j t | � d � | j | j j | � � | j | j
� | j � Wd QXd S)Nr ) r8 r r
�peek�assertNotEqualrH �
assertTruer+ �
startswithrA r" )r rB �pdatar r r �testPeek� s
zBZ2FileTest.testPeekc C s� | j � t | j � �� } d } t | � } | j | j | � | � | j | | j d | � � t | j � | } t t | j � � } | j | j | � | � | j | d | � | j | d � � Wd QXd S)N� )r8 r r
� bytearrayrA �readintor+ rH )r rB �n�br r r � testReadInto� s
zBZ2FileTest.testReadIntoc C sf | j � t | j � �G } | j t | j d � x' | j D] } | j | j � | � q< WWd QXd S)N)r8 r r
r< r= �readliner) rA )r rB �liner r r � testReadLine� s
zBZ2FileTest.testReadLinec C sp | j d d � t | j � �K } | j t | j d � x+ | j d D] } | j | j � | � qF WWd QXd S)Nr5 rF )r8 r r
r< r= r_ r) rA )r rB r` r r r �testReadLineMultiStream� s
z#BZ2FileTest.testReadLineMultiStreamc C sU | j � t | j � �6 } | j t | j d � | j | j � | j � Wd QXd S)N)r8 r r
r< r= � readlinesrA r) )r rB r r r �
testReadLines� s
zBZ2FileTest.testReadLinesc C s_ | j d d � t | j � �: } | j t | j d � | j | j � | j d � Wd QXd S)Nr5 rF )r8 r r
r< r= rc rA r) )r rB r r r �testReadLinesMultiStream� s z$BZ2FileTest.testReadLinesMultiStreamc C sE | j � t | j � �& } | j t t | � � | j � Wd QXd S)N)r8 r r
rA �list�iterr) )r rB r r r � testIterator� s
zBZ2FileTest.testIteratorc C sO | j d d � t | j � �* } | j t t | � � | j d � Wd QXd S)Nr5 rF )r8 r r
rA rf rg r) )r rB r r r �testIteratorMultiStream� s z#BZ2FileTest.testIteratorMultiStreamc C sM | j � t | j � } | j � | j t t | � | j t | j � d S)N)r8 r r
r! r< r> �nextrc )r rB r r r �testClosedIteratorDeadlock� s
z&BZ2FileTest.testClosedIteratorDeadlockc C s t | j d � �* } | j t | j � | j | j � Wd QXt | j d � �) } | j | j | j � � | j � Wd QXd S)N�w�rb)
r r
r< r= r r+ r4 rA r r" )r rB r7 r r r � testWrite� s
zBZ2FileTest.testWritec C s� t | j d � �T } d } xD | j | d | d d � } | sG Pn | j | � | d 7} q WWd QXt | j d � �) } | j | j | j � � | j � Wd QXd S)Nrl r r; r2 rm )r r
r+ r r4 rA r r" )r rB r\ rN r7 r r r �testWriteChunks10� s
z BZ2FileTest.testWriteChunks10c C s~ t j | j d d �} t | j d d d �� } | j | j � Wd QXt | j d � � } | j | j � | � Wd QXd S)Nr: rF rl rm ) r �compressr+ r r
r r4 rA r" )r ZexpectedrB r7 r r r � testWriteNonDefaultCompressLevel� s
z,BZ2FileTest.testWriteNonDefaultCompressLevelc C s� t | j d � �* } | j t | j � | j | j � Wd QX| j t | j d g � t | j d � �) } | j | j | j
� � | j � Wd QXd S)Nrl �arm ) r r
r< r= �
writelinesr) r> r4 rA r r" r+ )r rB r7 r r r �testWriteLines s zBZ2FileTest.testWriteLinesc C sv t | j d � � } | j d � Wd QXt | j d � �6 } | j t | j d � | j t | j d g � Wd QXd S)Nrl s abc�r� a)r r
r r< rD rs )r rB r r r � testWriteMethodsOnReadOnlyFile s
z*BZ2FileTest.testWriteMethodsOnReadOnlyFilec C s� t | j d � �* } | j t | j � | j | j � Wd QXt | j d � �* } | j t | j � | j | j � Wd QXt | j d � �- } | j | j | j � � | j d � Wd QXd S)Nrl rr rm � )
r r
r< r= r r+ r4 rA r r" )r rB r7 r r r �
testAppend s zBZ2FileTest.testAppendc C si | j � t | j � �J } | j t | j � | j d � | j | j � | j d d � � Wd QXd S)N� ) r8 r r
r< r= �seekrA r" r+ )r rB r r r �testSeekForward s
zBZ2FileTest.testSeekForwardc C s| | j d d � t | j � �W } | j t | j � | j t | j � d � | j | j � | j d d � � Wd QXd S)Nr5 rx rz )
r8 r r
r< r= r{ rH r+ rA r" )r rB r r r � testSeekForwardAcrossStreams' s
z(BZ2FileTest.testSeekForwardAcrossStreamsc C sf | j � t | j � �G } | j d � | j d d � | j | j � | j d d � � Wd QXd S)Ni� rz r2 ij���i^ )r8 r r
r" r{ rA r+ )r rB r r r �testSeekBackwards. s
z BZ2FileTest.testSeekBackwardsc C s� | j d d � t | j � �� } t | j � d } x) | d k r` | t | j | � � 8} q8 W| j d d � | j | j � | j d d � | j � Wd QXd S) Nr5 rx rQ r rz r2 ij���i���)r8 r r
rH r+ r" r{ rA )r rB Zreadtor r r � testSeekBackwardsAcrossStreams5 s z*BZ2FileTest.testSeekBackwardsAcrossStreamsc C sf | j � t | j � �G } | j d d � | j | j � | j t | j � d d � � Wd QXd S)Nrz rx ij���)r8 r r
r{ rA r" r+ rH )r rB r r r �testSeekBackwardsFromEnd> s
z$BZ2FileTest.testSeekBackwardsFromEndc C sc | j d d � t | j � �> } | j d d � | j | j � | j d d d � � Wd QXd S)Nr5 rx i� i���i���)r8 r r
r{ rA r" r+ )r rB r r r �%testSeekBackwardsFromEndAcrossStreamsD s z1BZ2FileTest.testSeekBackwardsFromEndAcrossStreamsc C sh | j � t | j � �I } | j d � | j | j � t | j � � | j | j � d � Wd QXd S)Ni�I r ) r8 r r
r{ rA �tellrH r+ r" )r rB r r r �testSeekPostEndJ s
zBZ2FileTest.testSeekPostEndc C sr | j d d � t | j � �M } | j d � | j | j � t | j � d � | j | j � d � Wd QXd S)Nr5 rF i�I r ) r8 r r
r{ rA r� rH r+ r" )r rB r r r �testSeekPostEndMultiStreamQ s
#z&BZ2FileTest.testSeekPostEndMultiStreamc C su | j � t | j � �V } | j d � | j d � | j | j � t | j � � | j | j � d � Wd QXd S)Ni�I r ) r8 r r
r{ rA r� rH r+ r" )r rB r r r �testSeekPostEndTwiceX s
z BZ2FileTest.testSeekPostEndTwicec C s | j d d � t | j � �Z } | j d � | j d � | j | j � t | j � d � | j | j � d � Wd QXd S)Nr5 rF i�I r ) r8 r r
r{ rA r� rH r+ r" )r rB r r r � testSeekPostEndTwiceMultiStream` s
#z+BZ2FileTest.testSeekPostEndTwiceMultiStreamc
C sb | j � t | j � �C } | j d � | j | j � d � | j | j � | j � Wd QXd S)Nrz r ij���)r8 r r
r{ rA r� r" r+ )r rB r r r �testSeekPreStarth s
z BZ2FileTest.testSeekPreStartc C sl | j d d � t | j � �G } | j d � | j | j � d � | j | j � | j d � Wd QXd S)Nr5 rx rz r ij���)r8 r r
r{ rA r� r" r+ )r rB r r r �testSeekPreStartMultiStreamo s
z'BZ2FileTest.testSeekPreStartMultiStreamc C sv | j � t | j d � �A } t | � } z | j | j � | j � � Wd | j � XWd QX| j t | j � d S)Nrm ) r8 r4 r
r rA �filenor! r< r> )r ZrawfrB r r r �
testFilenov s
zBZ2FileTest.testFilenoc C s t t | j � � } z4 | j | j � � | j � | j | j � � Wd | j � X| j t | j � t t � d � } z | j | j � � Wd | j � X| j t | j � t | j � } d d � | _ t | � } z | j | j � � Wd | j � X| j t | j � d S)Nrl c S s d S)NFr r r r r �<lambda>� s z*BZ2FileTest.testSeekable.<locals>.<lambda>)
r r r, rU �seekabler" r! r< r> � assertFalse)r rB �srcr r r � testSeekable� s&
zBZ2FileTest.testSeekablec C s� t t | j � � } z4 | j | j � � | j � | j | j � � Wd | j � X| j t | j � t t � d � } z | j | j � � Wd | j � X| j t | j � d S)Nrl )
r r r, rU �readabler" r! r< r> r� )r rB r r r � testReadable� s
zBZ2FileTest.testReadablec C s� t t | j � � } z4 | j | j � � | j � | j | j � � Wd | j � X| j t | j � t t � d � } z | j | j � � Wd | j � X| j t | j � d S)Nrl )
r r r, r� �writabler" r! r< r> rU )r rB r r r � testWritable� s
zBZ2FileTest.testWritablec C s7 | j � x& t d � D] } t | j � } ~ q Wd S)Ni' )r8 �ranger r
)r �i�or r r � testOpenDel� s
zBZ2FileTest.testOpenDelc C s | j t t d � d S)Nz
/non/existent)r< rD r )r r r r �testOpenNonexistent� s z BZ2FileTest.testOpenNonexistentc C s� d } t | j d � � } | j | � Wd QXt | j � � } | j � } Wd QX| j | d g � t | j � � } t | j � � } Wd QX| j | d g � d S)Ns, BZh91AY&SY�b�] � !�h3M<]��BCe�%tr3 s Test)r4 r
r r rc rA rf )r r r7 rB �linesZxlinesr r r �testReadlinesNoNewline� s z"BZ2FileTest.testReadlinesNoNewlinec ! C s� d } t | j d � � } | j d � Wd QXt | j d � } | j � y | � Wd QXWn t k
ro Yn X| j d � y' t | j d � � } d d Wd QXWn t k
r� Yn X| j d � d S)Nr3 s xxxrm z4__enter__ on a closed file didn't raise an exceptionr2 r z 1/0 didn't raise an exception)r r
r r! r> Zfail�ZeroDivisionError)r r7 r r r �testContextProtocol� s"
z BZ2FileTest.testContextProtocolz!Threading required for this test.c sy d d
� d } t | j d � �Q � � � f d d � � � f d d � t | � D� } t j | � � Wd QXWd QXd S) N� 1rx � r; r3 c s( x! t d � D] } � j � � q
Wd S)NrF )r� r )r� )r r7 r r �comp� s z'BZ2FileTest.testThreading.<locals>.compc s" g | ] } t j d � � � q S)�target)� threadingZThread)�.0r� )r� r r �
<listcomp>� s z-BZ2FileTest.testThreading.<locals>.<listcomp>i )r r
r� r Z
start_threads)r ZnthreadsZthreadsr )r� r r7 r �
testThreading� s
zBZ2FileTest.testThreadingc C sx t j d d d �} | j | j d � � } | j d � Wd QX| j | j d � � } | j | j � d � Wd QXd S)Nr Zblockedr� r3 s abcrm )z threading)r Zimport_fresh_moduler r
r rA r" )r �moduler7 r r r �testWithoutThreading� s
z BZ2FileTest.testWithoutThreadingc C sA | j � t | j d � } | d } t | j � �^ } | j | � | j t | � | j d | d � � | j | j � | j | d � � Wd QXt | j � �K } | j � | j t | � | j d � | j | j � | j d � Wd QXt | j � �: } | j
� | j t t | � | j | j
� g � Wd QXd S)Nr rx r2 )
r8 rH r) r r
r" rA rj r+ r_ rc r< �
StopIteration)r ZlinelenZhalflenrB r r r �testMixedIterationAndReads� s
')
#
z&BZ2FileTest.testMixedIterationAndReadsc C s� d d } d d } t | j d � � } | j | � Wd QXt | j d � � } | j | � Wd QXt | j � �! } | j | j � | | � Wd QXd S)Ns fooi� s barrl rr )r r
r rA r" )r Zdata1Zdata2rB r r r �testMultiStreamOrdering s
z#BZ2FileTest.testMultiStreamOrderingc C s� | j } y | j d � } Wn t k
r= | j d � Yn Xt | d � � } | j | j � Wd QXt | d � � } | j | j � | j � Wd QXt | d � � } | j | j � | j � Wd QXd S)N�asciiz%Temporary file name needs to be ASCIIr3 rm ) r
�encode�UnicodeEncodeErrorZskipTestr r r, rA r" )r Z str_filenameZbytes_filenamer7 r r r �testOpenBytesFilename s
z!BZ2FileTest.testOpenBytesFilenamec C sp t | j � �[ } t | � �6 } | j t | j d � | j | j � | j � Wd QX| j | j � Wd QXd S)N)
r r, r r< r= r" rA r+ r� �closed)r �biorB r r r �testReadBytesIO' s
zBZ2FileTest.testReadBytesIOc C s� t | j � �p } t | � �[ } | j � } | j t | � d � | j | j j | � � | j | j
� | j � Wd QXWd QXd S)Nr ) r r, r rS rT rH rU r+ rV rA r" )r r� rB rW r r r �testPeekBytesIO. s zBZ2FileTest.testPeekBytesIOc C s� t � �t } t | d � �* } | j t | j � | j | j � Wd QX| j | j | j � � | j � | j | j
� Wd QXd S)Nrl ) r r r< r= r r+ rA r �getvaluer� r� )r r� rB r r r �testWriteBytesIO6 s "z BZ2FileTest.testWriteBytesIOc C st t | j � �_ } t | � �J } | j t | j � | j d � | j | j � | j d d � � Wd QXWd QXd S)Nrz ) r r, r r< r= r{ rA r" r+ )r r� rB r r r �testSeekForwardBytesIO> s
z"BZ2FileTest.testSeekForwardBytesIOc C sq t | j � �\ } t | � �G } | j d � | j d d � | j | j � | j d d � � Wd QXWd QXd S)Ni� rz r2 ij���i^ )r r, r r" r{ rA r+ )r r� rB r r r �testSeekBackwardsBytesIOE s
z$BZ2FileTest.testSeekBackwardsBytesIOc C s� | j d d � } t t | � � � } | j t | j � Wd QXt t | � � �B } | j | j t | j � � | j � | j t | j d � Wd QXxO t d � D]A } t t | d | � � � � } | j t | j d � Wd QXq� Wd S)Nr; r2 � i����)
r, r r r< �EOFErrorr" rA rH r+ r� )r Z truncatedr7 r� r r r �test_read_truncatedL s % z BZ2FileTest.test_read_truncatedN)Ar% r& r' r( r8 r@ rC rE rG rI rJ rK rL rO rP rR rX r^ ra rb rd re rh ri rk rn ro rq rt rw ry r| r} r~ r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �unittestZ
skipUnlessr� r� r� r� r� r� r� r� r� r� r� r� r r r r r1 P sv
r1 c @ se e Z d Z d d � Z d d � Z d d � Z e d e d d d
� d d � � Z d
d � Z d S)�BZ2CompressorTestc C s^ t � } | j t | j � | j | j � } | | j � 7} | j | j | � | j � d S)N)r r< r= rp r+ �flushrA r )r �bz2cr r r r � testCompress[ s
z BZ2CompressorTest.testCompressc C s? t � } | j d � } | | j � 7} | j | | j � d S)Nr )r rp r� rA r- )r r� r r r r �testCompressEmptyStringb s z)BZ2CompressorTest.testCompressEmptyStringc C s� t � } d } d } xJ | j | d | d d � } | sA Pn | | j | � 7} | d 7} q W| | j � 7} | j | j | � | j � d S)Nr r r; r2 )r r+ rp r� rA r )r r� r\ r rN r r r �testCompressChunks10h s z&BZ2CompressorTest.testCompressChunks10�sizerQ �memuserx c C s� t � } d | } z# | j | � } | | j � 7} Wd d } Xt j | � } z9 | j t | � | � | j t | j d � � d � Wd d } Xd S)N� xr )r rp r� r r rA rH �strip)r r� r� r �
compressedr r r �testCompress4Gu s
#z BZ2CompressorTest.testCompress4Gc C sK xD t t j d � D]/ } | j t � � t j t � | � Wd QXq Wd S)Nr2 )r� �pickle�HIGHEST_PROTOCOLr< r= �dumpsr )r �protor r r �
testPickle� s z BZ2CompressorTest.testPickleN)
r% r& r' r� r� r� r r r� r� r r r r r� Z s
%r� c @ s} e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z e d e d d
d � d d � � Z
d d � Z d S)�BZ2DecompressorTestc C s | j t t d � d S)N�* )r< r= r
)r r r r �test_Constructor� s z$BZ2DecompressorTest.test_Constructorc C sE t � } | j t | j � | j | j � } | j | | j � d S)N)r
r< r= r r, rA r+ )r �bz2drM r r r �testDecompress� s z"BZ2DecompressorTest.testDecompressc C sy t � } d } d } xJ | j | d | d d � } | sA Pn | | j | � 7} | d 7} q W| j | | j � d S)Nr r r; r2 )r
r, r rA r+ )r r� rM r\ rN r r r �testDecompressChunks10� s z*BZ2DecompressorTest.testDecompressChunks10c C sO t � } d } | j | j | � } | j | | j � | j | j | � d S)Ns this is unused data)r
r r, rA r+ � unused_data)r r� r� rM r r r �testDecompressUnusedData� s
z,BZ2DecompressorTest.testDecompressUnusedDatac C sK t � } | j | j � } | j t | j d � | j t | j d � d S)Ns anythingr )r
r r, r< r� )r r� rM r r r � testEOFError� s z BZ2DecompressorTest.testEOFErrorr� rQ r� gffffff
@c
C s� d } t j | d � j | d � } zP | | | d } t j | � } t � } | j | � } | j | | k � Wd d } d } d } Xd S)Nr; i � �littler2 i ( i � )�randomZ getrandbits�to_bytesr rp r
r rU )r r� Z blocksize�blockr r� r� Z decompressedr r r �testDecompress4G� s z$BZ2DecompressorTest.testDecompress4Gc C sK xD t t j d � D]/ } | j t � � t j t � | � Wd QXq Wd S)Nr2 )r� r� r� r< r= r� r
)r r� r r r r� � s z BZ2DecompressorTest.testPickleN) r% r& r' r� r� r� r� r� r r r� r� r r r r r� � s %r� c @ s� 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 d � Z d d � Z d S)�CompressDecompressTestc C s2 t j | j � } | j | j | � | j � d S)N)r rp r+ rA r )r r r r r r� � s z#CompressDecompressTest.testCompressc C s& t j d � } | j | | j � d S)Nr )r rp rA r- )r rM r r r r� � s z.CompressDecompressTest.testCompressEmptyStringc C s) t j | j � } | j | | j � d S)N)r r r, rA r+ )r rM r r r r� � s z%CompressDecompressTest.testDecompressc C s# t j d � } | j | d � d S)Nr )r r rA )r rM r r r �testDecompressEmpty� s z*CompressDecompressTest.testDecompressEmptyc C s& t j | j � } | j | d � d S)Nr )r r r- rA )r rM r r r �testDecompressToEmptyString� s z2CompressDecompressTest.testDecompressToEmptyStringc C s' | j t t j | j d d � � d S)Nr; i����)r< r> r r r, )r r r r �testDecompressIncomplete� s z/CompressDecompressTest.testDecompressIncompletec C s | j t t j | j � d S)N)r< rD r r r. )r r r r �testDecompressBadData� s z,CompressDecompressTest.testDecompressBadDatac C s1 t j | j d � } | j | | j d � d S)NrF )r r r, rA r+ )r rM r r r �testDecompressMultiStream� s z0CompressDecompressTest.testDecompressMultiStreamc C s0 t j | j | j � } | j | | j � d S)N)r r r, r. rA r+ )r rM r r r �testDecompressTrailingJunk� s z1CompressDecompressTest.testDecompressTrailingJunkc C s8 t j | j d | j � } | j | | j d � d S)NrF )r r r, r. rA r+ )r rM r r r �%testDecompressMultiStreamTrailingJunk� s z<CompressDecompressTest.testDecompressMultiStreamTrailingJunkN)
r% r& r' r� r� r� r� r� r� r� r� r� r� r r r r r� � s r� c @ s� e Z d 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 d � Z d d � Z
d S)�OpenTestzTest the open function.c O s t j | | � S)N)r r4 )r �args�kwargsr r r r4 � s z
OpenTest.openc $ C sN xGd D]?} | d k r) t | j � n | j | j | � � } | j | j � Wd QXt | j d � �/ } | j | j � � } | j | | j � Wd QX| j | j d � � } | j | j � | j � Wd QX| j | j d � � } | j | j � Wd QXt | j d � �3 } | j | j � � } | j | | j d � Wd QXq Wd S)Nr3 �xbrm Zabrx )zwbr� )r r
r4 r r+ r r" rA )r �moder7 � file_datar r r �test_binary_modes� s
zOpenTest.test_binary_modesc $ C sN xGd D]?} | d k r) t | j � n | j | j | � � } | j | j � Wd QXt | j d � �/ } | j | j � � } | j | | j � Wd QX| j | j d � � } | j | j � | j � Wd QX| j | j d � � } | j | j � Wd QXt | j d � �3 } | j | j � � } | j | | j d � Wd QXq Wd S)Nrl �xrm ru rr rx )rl r� )r r
r4 r r+ r r" rA )r r� r7 r� r r r �test_implicit_binary_modes s
z#OpenTest.test_implicit_binary_modesc $ C sx | j j d � } | j d t j � } xJd D]B} | d k rP t | j � n | j | j | � � } | j | � Wd QXt | j d � �5 } | j | j
� � j d � } | j | | � Wd QX| j | j d � � } | j | j
� | � Wd QX| j | j d � � } | j | � Wd QXt | j d � �9 } | j | j
� � j d � } | j | | d � Wd QXq. Wd S)
Nr� �
�wt�xtrm �rt�atrx )r� r� ) r+ �decode�replacer �linesepr r
r4 r r r" rA )r rM �text_native_eolr� r7 r� r r r �test_text_modes s"
zOpenTest.test_text_modesc C st xm d D]e } t | j � | j | j | � � } Wd QX| j t � �# | j | j | � � } Wd QXWd QXq Wd S)Nr� r� r� )r� zxbzxt)r r
r4 r< �FileExistsError)r r� r7 r r r � test_x_mode) s
zOpenTest.test_x_modec C s� | j t | j � d � � } | j | j � | j � Wd QX| j t | j � d � � } | j | j � | j � Wd QX| j j d � } | j t | j � d � � } | j | j � | � Wd QXd S)Nru rm r� r� )r4 r r, rA r" r+ r� )r r7 rM r r r � test_fileobj2 s zOpenTest.test_fileobjc C s� | j t | j | j d � | j t | j | j d � | j t | j | j d d d �| j t | j | j d d d �| j t | j | j d d d �d S)
NZwbtZxbtrm �encodingzutf-8�errors�ignore�newliner� )r< r> r4 r
)r r r r �test_bad_params; s zOpenTest.test_bad_paramsc C s� | j j d � } | j d t j � } | j | j d d d �� } | j | � Wd QXt | j d � �5 } | j | j � � j d � } | j
| | � Wd QX| j | j d d d �� } | j
| j � | � Wd QXd S)Nr� r� r� r� z utf-16-lerm r� ) r+ r� r� r r� r4 r
r r r" rA )r rM r� r7 r� r r r �
test_encodingH s zOpenTest.test_encodingc C so | j | j d � � } | j d � Wd QX| j | j d d d d d �� } | j | j � d � Wd QXd S) Nr3 s foo�barr� r� r� r� r� Zfoobar)r4 r
r rA r" )r r7 r r r �test_encoding_error_handlerT s
!z$OpenTest.test_encoding_error_handlerc C s� | j j d � } | j | j d d d �� } | j | � Wd QX| j | j d d d �� } | j | j � | g � Wd QXd S)Nr� r� r� r� r� �
)r+ r� r4 r
r rA rc )r rM r7 r r r � test_newline\ s
zOpenTest.test_newlineN)r% r& r' r( r4 r� r� r� r� r� r� r� r� r r r r r r� � s
r� c C s' t j t t t t t � t j � d S)N)r Z run_unittestr1 r� r� r� r� Z
reap_childrenr r r r � test_maine s r �__main__) Ztestr Z test.supportr r r� �ior r r� r� r r/ r r� � ImportError�
import_moduler r r r
ZTestCaser r1 r� r� r� r� r r% r r r r �<module> s4
9� � 29(x