Your IP : 216.73.216.52
��Yf�� � @ 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�shutilZwhich�
subprocessZ check_outputr �
decompress)�data� r �2/opt/alt/python35/lib64/python3.5/test/test_bz2.py�ext_decompress s
r c @ s9 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 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))�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 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__�__doc__�
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 $ sH 1* 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 dz � ZA d{ S)|� BZ2FileTestzTest the BZ2File class.r r c
C sA t | j d � �( } | j | j | � | j | � Wd QRXd S)N�wb)r2 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��/�^@�zZrxZrbtr 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 r@ rA �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 r@ �OSErrorrE )r rH 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 r@ rA rE rF rG r) )r rH 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: rL )
� _compression� BUFFER_SIZE�lenr* r= r r r@ rA rE rF rG r) )r Z buffer_sizerH 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 rG rE r) )r rH 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: rL r; )r= r, r r rG rE r) )r rH 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 r@ rA rE rF rG )r rH 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 r? )r= r r rE rG r) )r rH �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: rL r r? )r= r r rE rG r) )r rH rU rV 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 rG rE r) )r rH 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�assertNotEqualrP �
assertTruer) �
startswithrG rE )r rH �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- rG r3 r) rP )r rH �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 r@ rA �readliner' rG )r rH �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: rL )r= r r r@ rA rd r' rG )r rH re 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 r@ rA � readlinesrG r' )r rH 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: rL )r= r r r@ rA rh rG r' )r rH r r r �testReadLinesMultiStream� s z$BZ2FileTest.testReadLinesMultiStreamc C sF | j � t | j � �& } | j t t | � � | j � Wd QRXd S)N)r= r r rG �list�iterr' )r rH r r r � testIterator� s
zBZ2FileTest.testIteratorc C sP | j d d � t | j � �* } | j t t | � � | j d � Wd QRXd S)Nr: rL )r= r r rG rk rl r' )r rH 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 �closer@ rB �nextrh )r rH 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 r@ rA r9 r) r2 rG r rE )r rH 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)Nrr r r? r r )r r r) r9 r2 rG r rE )r rH ra rV 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 rL rr r ) r r5 r) r r r9 r2 rG rE )r ZexpectedrH 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)Nrr �ar ) r r r@ rA �
writelinesr' rB r2 rG r rE r) )r rH 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)Nrr s abc�r� a)r r r9 r@ rJ rw )r rH 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)Nrr rv r � )
r r r@ rA r9 r) r2 rG r rE )r rH 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 r@ rA �seekrG rE r) )r rH 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 r@ rA r rP r) rG rE )r rH 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 rE r rG r) )r rH 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| rY r r~ r ij���i���)r= r r rP r) rE r rG )r rH 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 rG rE r) rP )r rH 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 rG rE r) )r rH 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 rG �tellrP r) rE )r rH 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: rL i�I r ) r= r r r rG r� rP r) rE )r rH 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 rG r� rP r) rE )r rH 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: rL i�I r ) r= r r r rG r� rP r) rE )r rH 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 rG r� rE r) )r rH 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 rG r� rE r) )r rH 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= r2 r r rG �filenoro r@ rB )r ZrawfrH 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)Nrr c S s d S)NFr r r r r �<lambda>� s z*BZ2FileTest.testSeekable.<locals>.<lambda>)
r r r* r] �seekablerE ro r@ rB � assertFalse)r rH �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)Nrr )
r r r* r] �readablerE ro r@ rB r� )r rH 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)Nrr )
r r r* r� �writablerE ro r@ rB r] )r rH 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)r@ rJ 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�%tr8 s Test)r2 r r9 r rh rG rk )r r r< rH �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)Nr8 s xxxr z4__enter__ on a closed file didn't raise an exceptionr r z 1/0 didn't raise an exception)r r r9 ro rB 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| � r? r8 c s( x! t d � D] } � j � � q
Wd S)NrL )r� r9 )r� )r r< 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 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� r8 s abcr )r� )r Zimport_fresh_moduler r r9 rG rE )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= rP r' r r rE rG rp r) rd rh r@ �
StopIteration)r ZlinelenZhalflenrH 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 barrr rv )r r r9 rG rE )r Zdata1Zdata2rH 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 ASCIIr8 r ) r �encode�UnicodeEncodeErrorZskipTestr r9 r* rG rE )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)z-Decompressed data buffering should be limitedg ��>Ar � r z)Excessive amount of data was decompressedN)r r5 �bytes�intZ
assertLessrP rN rO r r rG rE r �assertLessEqualZ_buffer�rawr� )r ZbombZdecompZ
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 r@ rA rE rF rG r) r� �closed)r �biorH 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\ rP r] r) r^ rG rE )r r� rH r_ 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)Nrr ) r r r@ rA r9 r) rG r �getvaluer� r� )r r� rH 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 r@ rA r rG rE r) )r r� rH 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 rE r rG r) )r r� rH 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)Nr? r � i����)
r* r r r@ �EOFErrorrE rG rP r) r� )r Z truncatedr< r� r r r �test_read_truncated_ s % z BZ2FileTest.test_read_truncatedN)Br# r$ r% r& r= rD rI rK rM rQ rR rS rT rW rX rZ r` rc rf rg ri rj rm rn rq rs rt ru rx 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 r7 X sx
r7 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
r@ rA r5 r) �flushrG r )r �bz2cr r r r � testCompressn s
z BZ2CompressorTest.testCompressc C s? t � } | j d � } | | j � 7} | j | | j � d S)Nr )r
r5 r� rG 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 r? r )r
r) r5 r� rG r )r r� ra r rV r r r �testCompressChunks10{ s z&BZ2CompressorTest.testCompressChunks10�sizerY �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
r5 r� r r rG rP �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_PROTOCOLr@ rA �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�* )r@ rA 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@ rA r r* rG r) )r �bz2drU 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 r? r )r r* r rG r) )r r� rU ra rV 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* rG r) � unused_data)r r� r� rU 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� rU r r r � testEOFError� s z BZ2DecompressorTest.testEOFErrorr� rY 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 � �littler i ( i � )�randomZ getrandbits�to_bytesr r5 r r r] )r r� Z 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� r@ rA 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)
NrY �@ �
max_lengthr r ���r� r� r� )
r rP r6 �appendr r� Z needs_inputrG 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)NrY r� r r r| �i � )r rG 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 rG 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� rL i, r )r r� r r* rG 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 r@ � 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 r5 r) rG r )r r r r r r� 8 s z#CompressDecompressTest.testCompressc C s& t j d � } | j | | j � d S)Nr )r r5 rG r+ )r rU r r r r� < s z.CompressDecompressTest.testCompressEmptyStringc C s) t j | j � } | j | | j � d S)N)r r r* rG r) )r rU r r r r� @ s z%CompressDecompressTest.testDecompressc C s# t j d � } | j | d � d S)Nr )r r rG )r rU r r r �testDecompressEmptyD s z*CompressDecompressTest.testDecompressEmptyc C s& t j | j � } | j | d � d S)Nr )r r r+ rG )r rU r r r �testDecompressToEmptyStringH s z2CompressDecompressTest.testDecompressToEmptyStringc C s' | j t t j | j d d � � d S)Nr? i����)r@ rB r r r* )r r r r �testDecompressIncompleteL s z/CompressDecompressTest.testDecompressIncompletec C s | j t t j | j � d S)N)r@ rJ r r r, )r r r r �testDecompressBadDataO s z,CompressDecompressTest.testDecompressBadDatac C s1 t j | j d � } | j | | j d � d S)NrL )r r r* rG r) )r rU r r r �testDecompressMultiStreamR s z0CompressDecompressTest.testDecompressMultiStreamc C s0 t j | j | j � } | j | | j � d S)N)r r r* r, rG r) )r rU r r r �testDecompressTrailingJunkV s z1CompressDecompressTest.testDecompressTrailingJunkc C s8 t j | j d | j � } | j | | j d � d S)NrL )r r r* r, rG r) )r rU 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 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 r2 )r �args�kwargsr r r r2 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)Nr8 �xbr Zabr| )r8 r )r r r2 r9 r) r rE rG )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)Nrr �xr ry rv r| )rr r )r r r2 r9 r) r rE rG )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 r2 r9 r rE rG )r rU �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 r2 r@ �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)Nry r r� r )r2 r r* rG rE r) r
)r r< rU 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 )r@ rB r2 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 r2 r r9 r rE rG )r rU 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) Nr8 s foo�barr r r� r r Zfoobar)r2 r r9 rG rE )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
r2 r r9 rG rh )r rU r< r r r � test_newline� s
zOpenTest.test_newlineN)r# r$ r% r& r2 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_unittestr7 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� r/ r� r r �sysr rN r� � ImportError�
import_moduler r r
r r
r ZTestCaser r7 r� r� r� r� r r# r r r r �<module> s>
4� � 2�(x