Your IP : 216.73.216.85
���]�� � @ s� d d l m Z d d l m Z m Z d d l Z d d l m Z 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 Z d d l Z d d l m Z d d l 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 d a d d � 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" � d S)� )�support)�
bigmemtest�_4GN)�BytesIO�DEFAULT_BUFFER_SIZE)�unlink�bz2)�BZ2File�
BZ2Compressor�BZ2Decompressorc C sN t d k r! t t j d � � a t r= t j d g d | �St j | � Sd S)NZbunzip2�input)�has_cmdline_bunzip2�bool�shutil�which�
subprocess� check_outputr �
decompress)�data� r �-/opt/alt/python35/lib64/python3.5/test_bz2.py�ext_decompress s
r c @ s3 e 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 Z e
d& � Z x{ e j e
j j e
j j e � d � � D]O Z e e d � �* Z e e j e e � e d � � 7Z Wd QRXe d' k r� Pq� We j e d d! �Z d" d# � Z d$ d% � Z d S)(�BaseTests 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 filer � i z*.py�rbN�
compresslevel� c C s t j | _ d S)N)r ZTESTFN�filename)�selfr r r �setUpP s zBaseTest.setUpc C s) t j j | j � r% t j | j � d S)N)�os�path�isfiler r )r r r r �tearDownS s zBaseTest.tearDowni i )�__name__�
__module__� __qualname__�
TEXT_LINES�join�TEXT�DATA�
EMPTY_DATA�BAD_DATAZ test_size� bytearray�BIG_TEXT�globr! r" �dirname�__file__Zfname�openZfh�readinto�
memoryviewr �compress�BIG_DATAr r$ r r r r r $ sF 1* r c @ s� e 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 d/ d0 � Z d1 d2 � Z d3 d4 � Z d5 d6 � Z d7 d8 � Z d9 d: � Z d; d< � Z d= d> � Z d? d@ � Z! dA dB � Z" dC dD � Z# dE dF � Z$ dG dH � Z% dI dJ � Z& dK dL � Z' dM dN � Z( dO dP � Z) dQ dR � Z* dS dT � Z+ dU dV � Z, dW dX � Z- dY dZ � Z. d[ d\ � Z/ d] d^ � Z0 d_ d` � Z1 e2 j3 e4 da � db dc � � Z5 dd de � Z6 df dg � Z7 dh di � Z8 dj dk � Z9 dl dm � Z: dn do � Z; dp dq � Z< dr ds � Z= dt du � Z> dv dw � Z? dx dy � Z@ dz S){� BZ2FileTestr r c
C sA t | j d � �( } | j | j | � | j | � Wd QRXd S)N�wb)r3 r �writer+ )r �streams�suffix�fr r r �createTempFile[ 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��/�^@�z�rxZrbtr r �
)� assertRaises� TypeErrorr �
ValueErrorr! �devnull)r r r r � testBadArgs` s zBZ2FileTest.testBadArgsc C sY | j � t | j � �9 } | j t | j t � � | j | j � | j � Wd QRXd S)N) r> r r rB rC �read�float� assertEqualr* )r �bz2fr r r �testReadh s
zBZ2FileTest.testReadc
C sI | j d d d | j � t | j � � } | j t | j � Wd QRXd S)Nr; r r< )r> r- r r rB �OSErrorrG )r rJ r r r �testReadBadFilen s zBZ2FileTest.testReadBadFilec C sc | j d d � t | j � �= } | j t | j t � � | j | j � | j d � Wd QRXd S)Nr; � ) r> r r rB rC rG rH rI r* )r rJ r r r �testReadMultiStreams s z BZ2FileTest.testReadMultiStreamc C s� t j } t | j � t _ zc | j d d � t | j � �= } | j t | j t
� � | j | j � | j d � Wd QRXWd | t _ Xd S)Nr; rN )
� _compression� BUFFER_SIZE�lenr+ r> r r rB rC rG rH rI r* )r Z buffer_sizerJ r r r �testReadMonkeyMultiStreamy s (z%BZ2FileTest.testReadMonkeyMultiStreamc
C sI | j d | j � t | j � � } | j | j � | j � Wd QRXd S)Nr< )r> r- r r rI rG r* )r rJ r r r �testReadTrailingJunk� s z BZ2FileTest.testReadTrailingJunkc C sS | j d d d | j � t | j � �$ } | j | j � | j d � Wd QRXd S)Nr; rN r< )r> r- r r rI rG r* )r rJ r r r � testReadMultiStreamTrailingJunk� s z+BZ2FileTest.testReadMultiStreamTrailingJunkc C sY | j � t | j � �9 } | j t | j t � � | j | j d � d � Wd QRXd S)Nr r )r> r r rB rC rG rH rI )r rJ r r r � testRead0� s
zBZ2FileTest.testRead0c
C sg | j � t | j � �G } d } x$ | j d � } | s; P| | 7} q% W| j | | j � Wd QRXd S)Nr rA )r> r r rG rI r* )r rJ �text�strr r r �testReadChunk10� s
zBZ2FileTest.testReadChunk10c C sq | j d d � t | j � �K } d } x$ | j d � } | sA P| | 7} q+ W| j | | j d � Wd QRXd S)Nr; rN r rA )r> r r rG rI r* )r rJ rW rX r r r �testReadChunk10MultiStream� s z&BZ2FileTest.testReadChunk10MultiStreamc C sM | j � t | j � �- } | j | j d � | j d d � � Wd QRXd S)N�d )r> r r rI rG r* )r rJ r r r � testRead100� s
zBZ2FileTest.testRead100c
C s{ | j � t | j � �[ } | j � } | j t | � d � | j | j j | � � | j | j
� | j � Wd QRXd S)Nr ) r> r r �peek�assertNotEqualrR �
assertTruer* �
startswithrI rG )r rJ �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 QRXd S)Nr )r> r r r. rI r4 r* rR )r rJ �n�br r r � testReadInto� s
zBZ2FileTest.testReadIntoc C sg | j � t | j � �G } | j t | j d � x' | j D] } | j | j � | � q< WWd QRXd S)N)r> r r rB rC �readliner( rI )r rJ �liner r r � testReadLine� s
zBZ2FileTest.testReadLinec C sq | j d d � t | j � �K } | j t | j d � x+ | j d D] } | j | j � | � qF WWd QRXd S)Nr; rN )r> r r rB rC rf r( rI )r rJ rg r r r �testReadLineMultiStream� s
z#BZ2FileTest.testReadLineMultiStreamc C sV | j � t | j � �6 } | j t | j d � | j | j � | j � Wd QRXd S)N)r> r r rB rC � readlinesrI r( )r rJ r r r �
testReadLines� s
zBZ2FileTest.testReadLinesc C s` | j d d � t | j � �: } | j t | j d � | j | j � | j d � Wd QRXd S)Nr; rN )r> r r rB rC rj rI r( )r rJ r r r �testReadLinesMultiStream� s z$BZ2FileTest.testReadLinesMultiStreamc C sF | j � t | j � �& } | j t t | � � | j � Wd QRXd S)N)r> r r rI �list�iterr( )r rJ r r r � testIterator� s
zBZ2FileTest.testIteratorc C sP | j d d � t | j � �* } | j t t | � � | j d � Wd QRXd S)Nr; rN )r> r r rI rm rn r( )r rJ r r r �testIteratorMultiStream� s z#BZ2FileTest.testIteratorMultiStreamc C sM | j � t | j � } | j � | j t t | � | j t | j � d S)N)r> r r �closerB rD �nextrj )r rJ r r r �testClosedIteratorDeadlock� s
z&BZ2FileTest.testClosedIteratorDeadlockc C s~ t | j d � �* } | j t | j � | j | j � Wd QRXt | j d � �& } | j t | j � � | j � Wd QRXd S)N�wr )
r r rB rC r: r* r3 rI r rG )r rJ r= r r r � testWrite� s
zBZ2FileTest.testWritec C s� t | j d � �Q } d } xA | j | d | d d � } | sD P| j | � | d 7} q WWd QRXt | j d � �& } | j t | j � � | j � Wd QRXd S)Nrt r rA r r )r r r* r: r3 rI r rG )r rJ rc rX r= r r r �testWriteChunks10� s
z BZ2FileTest.testWriteChunks10c C s� t j | j d d �} t | j d d d �� } | j | j � Wd QRXt | j d � � } | j | j � | � Wd QRXd S)Nr rN rt r ) r r6 r* r r r: r3 rI rG )r ZexpectedrJ r= r r r � testWriteNonDefaultCompressLevel s
z,BZ2FileTest.testWriteNonDefaultCompressLevelc C s� t | j d � �* } | j t | j � | j | j � Wd QRX| j t | j d g � t | j d � �& } | j t | j
� � | j � Wd QRXd S)Nrt �ar ) r r rB rC �
writelinesr( rD r3 rI r rG r* )r rJ r= r r r �testWriteLines s zBZ2FileTest.testWriteLinesc C sx t | j d � � } | j d � Wd QRXt | j d � �6 } | j t | j d � | j t | j d g � Wd QRXd S)Nrt s abc�r� a)r r r: rB rL ry )r rJ r r r � testWriteMethodsOnReadOnlyFile s
z*BZ2FileTest.testWriteMethodsOnReadOnlyFilec C s� t | j d � �* } | j t | j � | j | j � Wd QRXt | j d � �* } | j t | j � | j | j � Wd QRXt | j d � �* } | j t | j � � | j d � Wd QRXd S)Nrt rx r � )
r r rB rC r: r* r3 rI r rG )r rJ r= r r r �
testAppend s zBZ2FileTest.testAppendc C sj | j � t | j � �J } | j t | j � | j d � | j | j � | j d d � � Wd QRXd S)N� ) r> r r rB rC �seekrI rG r* )r rJ 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 QRXd S)Nr; r~ r� )
r> r r rB rC r� rR r* rI rG )r rJ r r r � testSeekForwardAcrossStreams/ s
z(BZ2FileTest.testSeekForwardAcrossStreamsc C sg | j � t | j � �G } | j d � | j d d � | j | j � | j d d � � Wd QRXd S)Ni� r� r ij���i^ )r> r r rG r� rI r* )r rJ r r r �testSeekBackwards6 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 QRXd S) Nr; r~ r[ r r� r ij���i���)r> r r rR r* rG r� rI )r rJ Zreadtor r r � testSeekBackwardsAcrossStreams= s z*BZ2FileTest.testSeekBackwardsAcrossStreamsc C sg | j � t | j � �G } | j d d � | j | j � | j t | j � d d � � Wd QRXd S)Nr� r~ ij���)r> r r r� rI rG r* rR )r rJ r r r �testSeekBackwardsFromEndF s
z$BZ2FileTest.testSeekBackwardsFromEndc C sd | j d d � t | j � �> } | j d d � | j | j � | j d d d � � Wd QRXd S)Nr; r~ i� i���i���)r> r r r� rI rG r* )r rJ r r r �%testSeekBackwardsFromEndAcrossStreamsL s z1BZ2FileTest.testSeekBackwardsFromEndAcrossStreamsc C si | j � t | j � �I } | j d � | j | j � t | j � � | j | j � d � Wd QRXd S)Ni�I r ) r> r r r� rI �tellrR r* rG )r rJ r r r �testSeekPostEndR s
zBZ2FileTest.testSeekPostEndc C ss | j d d � t | j � �M } | j d � | j | j � t | j � d � | j | j � d � Wd QRXd S)Nr; rN i�I r ) r> r r r� rI r� rR r* rG )r rJ r r r �testSeekPostEndMultiStreamY s
#z&BZ2FileTest.testSeekPostEndMultiStreamc C sv | j � t | j � �V } | j d � | j d � | j | j � t | j � � | j | j � d � Wd QRXd S)Ni�I r ) r> r r r� rI r� rR r* rG )r rJ r r r �testSeekPostEndTwice` 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 QRXd S)Nr; rN i�I r ) r> r r r� rI r� rR r* rG )r rJ r r r � testSeekPostEndTwiceMultiStreamh s
#z+BZ2FileTest.testSeekPostEndTwiceMultiStreamc
C sc | j � t | j � �C } | j d � | j | j � d � | j | j � | j � Wd QRXd S)Nr� r ij���)r> r r r� rI r� rG r* )r rJ r r r �testSeekPreStartp s
z BZ2FileTest.testSeekPreStartc C sm | j d d � t | j � �G } | j d � | j | j � d � | j | j � | j d � Wd QRXd S)Nr; r~ r� r ij���)r> r r r� rI r� rG r* )r rJ r r r �testSeekPreStartMultiStreamw s
z'BZ2FileTest.testSeekPreStartMultiStreamc C sw | j � t | j d � �A } t | � } z | j | j � | j � � Wd | j � XWd QRX| j t | j � d S)Nr ) r> r3 r r rI �filenorq rB rD )r ZrawfrJ r r r �
testFileno~ 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)Nrt c S s d S)NFr r r r r �<lambda>� s z*BZ2FileTest.testSeekable.<locals>.<lambda>)
r r r+ r_ �seekablerG rq rB rD � assertFalse)r rJ �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)Nrt )
r r r+ r_ �readablerG rq rB rD r� )r rJ 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)Nrt )
r r r+ r� �writablerG rq rB rD r_ )r rJ r r r � testWritable� s
zBZ2FileTest.testWritablec C s7 | j � x& t d � D] } t | j � } ~ q Wd S)Ni' )r> �ranger r )r �i�or r r � testOpenDel� s
zBZ2FileTest.testOpenDelc C s | j t t d � d S)Nz
/non/existent)rB rL r )r r r r �testOpenNonexistent� s z BZ2FileTest.testOpenNonexistentc C s� d } t | j d � � } | j | � Wd QRXt | j � � } | j � } Wd QRX| j | d g � t | j � � } t | j � � } Wd QRX| j | d g � d S)Ns, BZh91AY&SY�b�] � !�h3M<]��BCe�%tr9 s Test)r3 r r: r rj rI rm )r r r= rJ �linesZxlinesr r r �testReadlinesNoNewline� s z"BZ2FileTest.testReadlinesNoNewlinec # C s� d } t | j d � � } | j d � Wd QRXt | j d � } | j � y | � Wd QRXWn t k
rq Yn X| j d � y( t | j d � � } d d Wd QRXWn t k
r� Yn X| j d � d S)Nr9 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: rq rD Zfail�ZeroDivisionError)r r= r r r �testContextProtocol� s"
z BZ2FileTest.testContextProtocolz!Threading required for this test.c s{ d d
� d } t | j d � �R � � � f d d � � � f d d � t | � D� } t j | � � Wd QRXWd QRXd S) N� 1r~ � rA r9 c s( x! t d � D] } � j � � q
Wd S)NrN )r� r: )r� )r r= r r �comp� s z'BZ2FileTest.testThreading.<locals>.compc s" g | ] } t j d � � � q S)�target)� threading�Thread)�.0r� )r� r r �
<listcomp>� s z-BZ2FileTest.testThreading.<locals>.<listcomp>i )r r r� r Z
start_threads)r Znthreads�threadsr )r� r r= r �
testThreading� s
zBZ2FileTest.testThreadingc C sz t j d d d �} | j | j d � � } | j d � Wd QRX| j | j d � � } | j | j � d � Wd QRXd S)Nr Zblockedr� r9 s abcr )r� )r Zimport_fresh_moduler r r: rI rG )r �moduler= r r r �testWithoutThreading� s
z BZ2FileTest.testWithoutThreadingc C sD | j � t | j d � } | d } t | j � �^ } | j | � | j t | � | j d | d � � | j | j � | j | d � � Wd QRXt | j � �K } | j � | j t | � | j d � | j | j � | j d � Wd QRXt | j � �: } | j
� | j t t | � | j | j
� g � Wd QRXd S)Nr r~ r )
r> rR r( r r rG rI rr r* rf rj rB �
StopIteration)r ZlinelenZhalflenrJ r r r �testMixedIterationAndReads s
'*
$
z&BZ2FileTest.testMixedIterationAndReadsc C s� d d } d d } t | j d � � } | j | � Wd QRXt | j d � � } | j | � Wd QRXt | j � �! } | j | j � | | � Wd QRXd S)Ns fooi� s barrt rx )r r r: rI rG )r Zdata1Zdata2rJ 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 QRXt | d � � } | j | j � | j � Wd QRXt | d � � } | j | j � | j � Wd QRXd S)N�asciiz%Temporary file name needs to be ASCIIr9 r ) r �encode�UnicodeEncodeErrorZskipTestr r: r+ rI rG )r Z str_filenameZbytes_filenamer= r r r �testOpenBytesFilename s
z!BZ2FileTest.testOpenBytesFilenamec C s� t j t t d � � d d �} | j t | � t j � t t | � � } | j
t d � | j d � � d t } | j
| j j j � | d � d S)Ng ��>Ar � r z)Excessive amount of data was decompressed)r r6 �bytes�intZ
assertLessrR rP rQ r r rI rG r �assertLessEqual�_buffer�rawr� )r Zbomb�decompZ
max_decompr r r �testDecompressLimited, s !
z!BZ2FileTest.testDecompressLimitedc C su t | j � �_ } t | � �9 } | j t | j t � � | j | j � | j � Wd QRX| j | j
� Wd QRXd S)N) r r+ r rB rC rG rH rI r* r� �closed)r �biorJ r r r �testReadBytesIO: s
zBZ2FileTest.testReadBytesIOc C s� t | j � �q } t | � �[ } | j � } | j t | � d � | j | j j | � � | j | j
� | j � Wd QRXWd QRXd S)Nr ) r r+ r r] r^ rR r_ r* r` rI rG )r r� rJ ra r r r �testPeekBytesIOA s zBZ2FileTest.testPeekBytesIOc C s� t � �r } t | d � �* } | j t | j � | j | j � Wd QRX| j t | j � � | j � | j | j
� Wd QRXd S)Nrt ) r r rB rC r: r* rI r �getvaluer� r� )r r� rJ r r r �testWriteBytesIOI s z BZ2FileTest.testWriteBytesIOc C sv t | j � �` } t | � �J } | j t | j � | j d � | j | j � | j d d � � Wd QRXWd QRXd S)Nr� ) r r+ r rB rC r� rI rG r* )r r� rJ r r r �testSeekForwardBytesIOQ s
z"BZ2FileTest.testSeekForwardBytesIOc C ss t | j � �] } t | � �G } | j d � | j d d � | j | j � | j d d � � Wd QRXWd QRXd S)Ni� r� r ij���i^ )r r+ r rG r� rI r* )r r� rJ r r r �testSeekBackwardsBytesIOX s
z$BZ2FileTest.testSeekBackwardsBytesIOc C s� | j d d � } t t | � � � } | j t | j � Wd QRXt t | � � �B } | j | j t | j � � | j � | j t | j d � Wd QRXxP t d � D]B } t t | d | � � � � } | j t | j d � Wd QRXq� Wd S)NrA r � i����)
r+ r r rB �EOFErrorrG rI rR r* r� )r Z truncatedr= r� r r r �test_read_truncated_ s % z BZ2FileTest.test_read_truncatedN)Ar% r& r' r> rF rK rM rO rS rT rU rV rY rZ r\ rb re rh ri rk rl ro rp rs ru rv rw rz r} r 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 r r8 X sv
r8 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 t | � | j � d S)N)r
rB rC r6 r* �flushrI r )r �bz2cr r r r � testCompressn s
z BZ2CompressorTest.testCompressc C s? t � } | j d � } | | j � 7} | j | | j � d S)Nr )r
r6 r� rI r, )r r� r r r r �testCompressEmptyStringu s z)BZ2CompressorTest.testCompressEmptyStringc C s� t � } d } d } xG | j | d | d d � } | s> P| | j | � 7} | d 7} q W| | j � 7} | j t | � | j � d S)Nr r rA r )r
r* r6 r� rI r )r r� rc r rX r r r �testCompressChunks10{ s z&BZ2CompressorTest.testCompressChunks10�sizer[ �memuser~ 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
r6 r� r r rI rR �strip)r r� r� r �
compressedr r r �testCompress4G� s
#z BZ2CompressorTest.testCompress4Gc C sL xE t t j d � D]0 } | j t � � t j t � | � Wd QRXq Wd S)Nr )r� �pickle�HIGHEST_PROTOCOLrB rC �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� m 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 d � Z d d � Z
d d � Z d d � Z d d � Z d S) �BZ2DecompressorTestc C s | j t t d � d S)N�* )rB rC 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 rB rC r r+ rI r* )r �bz2drW r r r �testDecompress� s z"BZ2DecompressorTest.testDecompressc C sv t � } d } d } xG | j | d | d d � } | s> P| | j | � 7} | d 7} q W| j | | j � d S)Nr r rA r )r r+ r rI r* )r r� rW rc rX 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+ rI r* � unused_data)r r� r� rW 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+ rB r� )r r� rW r r r � testEOFError� s z BZ2DecompressorTest.testEOFErrorr� r[ 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)NrA i � �littler i ( i � )�random� getrandbits�to_bytesr r6 r r r_ )r r� � blocksize�blockr r� r� Z decompressedr r r �testDecompress4G� s z$BZ2DecompressorTest.testDecompress4Gc C sL xE t t j d � D]0 } | j t � � t j t � | � Wd QRXq Wd S)Nr )r� r� r� rB rC r� r )r r� r r r r� � s z BZ2DecompressorTest.testPicklec C s� t � } d } g } t | j � d } | j | j | j d | � d | �� | j | j � | j t | d � | � | j | j d d | �� | j | j � | j t | d � | � | j | j | j | d � d | �� | j t | d � | � xC | j sI| j | j d d | �� | j t | d � | � qWd j
| � } | j | | j � | j | j d � d S)
Nr[ �@ �
max_lengthr r ���r� r� r� )
r rR r7 �appendr r� � needs_inputrI r� �eofr) r/ r� )r �bzdr� �outZlen_r r r � testDecompressorChunksMaxsize� s(
z1BZ2DecompressorTest.testDecompressorChunksMaxsizec C s� t � } g } | j | j | j d d � d d �d � | j | j d d � � | j | j | j d d � d � � | j | j | j d d � � � | j d j | � | j � d S)Nr[ r� r r r~ �i � )r rI r r+ r� r) r* )r r� r� r r r � test_decompressor_inputbuf_1� s
&#z0BZ2DecompressorTest.test_decompressor_inputbuf_1c C s� t � } g } | j | j | j d d � d d �d � | j | j d � � | j | j | j d d � d � � | j | j | j d d � d � � | j | j | j d d � � � | j d j | � | j � d S)N�� r� r r i r~ i, )r rI r r+ r� r) r* )r r� r� r r r � test_decompressor_inputbuf_2 s
&&#z0BZ2DecompressorTest.test_decompressor_inputbuf_2c C s� t � } g } | j | j | j d d � d � � | j | j | j d d � d � � | j | j | j d d � � � | j d j | � | j � d S)Nr� rN i, r )r r� r r+ rI r) r* )r r� r� r r r � test_decompressor_inputbuf_3! s &&#z0BZ2DecompressorTest.test_decompressor_inputbuf_3c C sG t � } | j t | j | j d � | j t | j | j d � d S)N� )r rB � Exceptionr r- )r r� r r r � test_failure1 s z BZ2DecompressorTest.test_failureN)r% r& r' r� 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 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 s/ t j | j � } | j t | � | j � d S)N)r r6 r* rI r )r r r r r r� 8 s z#CompressDecompressTest.testCompressc C s& t j d � } | j | | j � d S)Nr )r r6 rI r, )r rW r r r r� < s z.CompressDecompressTest.testCompressEmptyStringc C s) t j | j � } | j | | j � d S)N)r r r+ rI r* )r rW r r r r� @ s z%CompressDecompressTest.testDecompressc C s# t j d � } | j | d � d S)Nr )r r rI )r rW r r r �testDecompressEmptyD s z*CompressDecompressTest.testDecompressEmptyc C s& t j | j � } | j | d � d S)Nr )r r r, rI )r rW r r r �testDecompressToEmptyStringH s z2CompressDecompressTest.testDecompressToEmptyStringc C s' | j t t j | j d d � � d S)NrA i����)rB rD r r r+ )r r r r �testDecompressIncompleteL s z/CompressDecompressTest.testDecompressIncompletec C s | j t t j | j � d S)N)rB rL r r r- )r r r r �testDecompressBadDataO s z,CompressDecompressTest.testDecompressBadDatac C s1 t j | j d � } | j | | j d � d S)NrN )r r r+ rI r* )r rW r r r �testDecompressMultiStreamR s z0CompressDecompressTest.testDecompressMultiStreamc C s0 t j | j | j � } | j | | j � d S)N)r r r+ r- rI r* )r rW r r r �testDecompressTrailingJunkV s z1CompressDecompressTest.testDecompressTrailingJunkc C s8 t j | j d | j � } | j | | j d � d S)NrN )r r r+ r- rI r* )r rW r r r �%testDecompressMultiStreamTrailingJunkZ s z<CompressDecompressTest.testDecompressMultiStreamTrailingJunkN)
r% r& r' r� r� r� r� r r r r r r r r r r r� 7 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)�OpenTestc O s t j | | � S)N)r r3 )r �args�kwargsr r r r3 b s z
OpenTest.openc ( C sJ xCd D];} | d k r& t | j � | j | j | � � } | j | j � Wd QRXt | j d � �, } t | j � � } | j | | j � Wd QRX| j | j d � � } | j | j � | j � Wd QRX| j | j d � � } | j | j � Wd QRXt | j d � �0 } t | j � � } | j | | j d � Wd QRXq Wd S)Nr9 �xbr �abr~ )r9 r )r r r3 r: r* r rG rI )r �moder= � file_datar r r �test_binary_modese s
zOpenTest.test_binary_modesc ( C sJ xCd D];} | d k r& t | j � | j | j | � � } | j | j � Wd QRXt | j d � �, } t | j � � } | j | | j � Wd QRX| j | j d � � } | j | j � | j � Wd QRX| j | j d � � } | j | j � Wd QRXt | j d � �0 } t | j � � } | j | | j d � Wd QRXq Wd S)Nrt �xr r{ rx r~ )rt r )r r r3 r: r* r rG rI )r r r= r r r r �test_implicit_binary_modesv s
z#OpenTest.test_implicit_binary_modesc ( C st | j j d � } | j d t j � } xFd D]>} | d k rM t | j � | j | j | � � } | j | � Wd QRXt | j d � �2 } t | j
� � j d � } | j | | � Wd QRX| j | j d � � } | j | j
� | � Wd QRX| j | j d � � } | j | � Wd QRXt | j d � �6 } t | j
� � j d � } | j | | d � Wd QRXq. Wd S)
Nr� �
�wt�xtr �rtZatr~ )r r ) r* �decode�replacer! �linesepr r r3 r: r rG rI )r rW �text_native_eolr r= r r r r �test_text_modes� s"
zOpenTest.test_text_modesc C sw xp d D]h } t | j � | j | j | � � } Wd QRX| j t � �$ | j | j | � � } Wd QRXWd QRXq Wd S)Nr r r )r r r )r r r3 rB �FileExistsError)r r r= r r r � test_x_mode� s
zOpenTest.test_x_modec C s� | j t | j � d � � } | j | j � | j � Wd QRX| j t | j � d � � } | j | j � | j � Wd QRX| j j d � } | j t | j � d � � } | j | j � | � Wd QRXd S)Nr{ r r� r )r3 r r+ rI rG r* r )r r= rW r r r � test_fileobj� 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)
NZwbtZxbtr �encodingzutf-8�errors�ignore�newliner )rB rD r3 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 QRXt | j d � �2 } t | j � � j d � } | j
| | � Wd QRX| j | j d d d �� } | j
| j � | � Wd QRXd S)Nr� r r r z utf-16-ler r ) r* r r r! r r3 r r: r rG rI )r rW r r= r r r r �
test_encoding� s zOpenTest.test_encodingc C sq | j | j d � � } | j d � Wd QRX| j | j d d d d d �� } | j | j � d � Wd QRXd S) Nr9 s foo�barr r r� r r Zfoobar)r3 r r: rI rG )r r= r r r �test_encoding_error_handler� s
!z$OpenTest.test_encoding_error_handlerc C s� | j j d � } | j | j d d d �� } | j | � Wd QRX| j | j d d d �� } | j | j � | g � Wd QRXd S)Nr� r r r r �
)r* r r3 r r: rI rj )r rW r= r r r � test_newline� s
zOpenTest.test_newlineN)
r% r& r' r3 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_unittestr8 r� r� r� r Z
reap_childrenr r r r � test_main� s r% �__main__)$Ztestr Z test.supportr r r� �ior r r! r� r0 r� r r �sysr rP r� � ImportError�
import_moduler r r
r r
r ZTestCaser r8 r� r� r� r r% r% r r r r �<module> s>
4� � 2�(x