Your IP : 216.73.216.213
3
��2ap� � @ s^ d dl mZ d dlmZmZ d dlZd dlmZmZ d dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZd dlZd dlZd dlmZ d dlZd dlZy d dlZW n ek
r� dZY nX ejd�Zd dlmZmZmZ dadd � ZG d
d � d ej �Z G d d
� d
e �Z G dd� de �Z G dd� de �Z G dd� de �Z!G dd� de �Z"dd� Z#e$dk�rZe#� dS )� )�support)�
bigmemtest�_4GN)�BytesIO�DEFAULT_BUFFER_SIZE)�unlink�bz2)�BZ2File�
BZ2Compressor�BZ2Decompressorc C s8 t d krttjd��a t r*tjdg| d�S tj| �S d S )NZbunzip2)�input)�has_cmdline_bunzip2�bool�shutilZwhich�
subprocessZ check_outputr �
decompress)�data� r �-/opt/alt/python36/lib64/python3.6/test_bz2.py�ext_decompress s
r c @ s� e Zd Zddddddddd d
d d d
ddddddddgZdje�ZdZdZdZdZ e
d&�Z x^e j e
jje
jje�d ��D ]>Zeed �� Ze ejee �e d � �7 Z W d Q R X e d'krvP qvW eje d d!�Zd"d#� Zd$d%� Zd 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� )�
compresslevelc C s t j| _d S )N)r ZTESTFN�filename)�selfr r r �setUpR s zBaseTest.setUpc C s t jj| j�rt j| j� d S )N)�os�path�isfiler r )r r r r �tearDownU 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
" $r c @ s� e Zd Zd}dd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Zd5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z d?d@� Z!dAdB� Z"dCdD� Z#dEdF� Z$dGdH� Z%dIdJ� Z&dKdL� Z'dMdN� Z(dOdP� Z)dQdR� Z*dSdT� Z+dUdV� Z,dWdX� Z-dYdZ� Z.d[d\� Z/d]d^� Z0d_d`� Z1e2j3e4da�dbdc� �Z5ddde� Z6dfdg� Z7dhdi� Z8djdk� Z9dldm� Z:dndo� Z;dpdq� Z<drds� Z=dtdu� Z>dvdw� Z?dxdy� Z@dzd{� ZAd|S )~� BZ2FileTestr r c
C s6 t | jd�� }|j| j| � |j|� W d Q R X d S )N�wb)r1 r �writer) )r �streams�suffix�fr r r �createTempFile] s zBZ2FileTest.createTempFilec C sp | j ttd� | j tttjd� | j tttjd� | j tttjd� | j tttjdd� | j tttjdd� d S )Ngw��/�^@�z�rxZrbtr )r �
)� assertRaises� TypeErrorr �
ValueErrorr �devnull)r r r r � testBadArgsb s zBZ2FileTest.testBadArgsc C sF | j � t| j��*}| jt|jt� � | j|j� | j� W d Q R X d S )N) r< r r r@ rA �read�float� assertEqualr( )r �bz2fr r r �testReadj s zBZ2FileTest.testReadc
C s8 | j d| jd� t| j��}| jt|j� W d Q R X d S )Nr )r9 r: )r< r+ r r r@ �OSErrorrE )r rH r r r �testReadBadFilep s zBZ2FileTest.testReadBadFilec C sN | j dd� t| j��.}| jt|jt� � | j|j� | jd � W d Q R X d S )N� )r9 ) r< r r r@ rA rE rF rG r( )r rH r r r �testReadMultiStreamu s z BZ2FileTest.testReadMultiStreamc C sn t j}t| j�t _zN| jdd� t| j��.}| jt|j t
� � | j |j � | j d � W d Q R X W d |t _X d S )NrL )r9 )
� _compression� BUFFER_SIZE�lenr) r< r r r@ rA rE rF rG r( )r � buffer_sizerH r r r �testReadMonkeyMultiStream{ s $z%BZ2FileTest.testReadMonkeyMultiStreamc
C s: | j | jd� t| j��}| j|j� | j� W d Q R X d S )N)r: )r< r+ r r rG rE r( )r rH r r r �testReadTrailingJunk� s z BZ2FileTest.testReadTrailingJunkc C s@ | j d| jd� t| j�� }| j|j� | jd � W d Q R X d S )NrL )r9 r: )r< r+ r r rG rE r( )r rH r r r � testReadMultiStreamTrailingJunk� s z+BZ2FileTest.testReadMultiStreamTrailingJunkc C sF | j � t| j��*}| jt|jt� � | j|jd�d� W d Q R X d S )Nr r )r< r r r@ rA rE rF rG )r rH r r r � testRead0� s zBZ2FileTest.testRead0c
C sR | j � t| j��6}d}x |jd�}|s*P ||7 }qW | j|| j� W d Q R X d S )Nr r? )r< r r rE rG r( )r rH �text�strr r r �testReadChunk10� s
zBZ2FileTest.testReadChunk10c C sZ | j dd� t| j��:}d}x |jd�}|s.P ||7 }q W | j|| jd � W d Q R X d S )NrL )r9 r r? )r< r r rE rG r( )r rH rV rW r r r �testReadChunk10MultiStream� s
z&BZ2FileTest.testReadChunk10MultiStreamc C s> | j � t| j��"}| j|jd�| jd d� � W d Q R X d S )N�d )r< r r rG rE r( )r rH r r r � testRead100� s zBZ2FileTest.testRead100c
C s^ | j � t| j��B}|j� }| jt|�d� | j| jj|�� | j |j
� | j� W d Q R X d 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|| jd |� � t| j�| }tt| j��}| j|j|�|� | j|d |� | j| d � � W d Q R X d S )Nr )r< r r r, rG r2 r( rP )r rH �n�br r r � testReadInto� s zBZ2FileTest.testReadIntoc C sR | j � t| j��6}| jt|jd � x | jD ]}| j|j� |� q,W W d Q R X d S )N)r< r r r@ rA �readliner& rG )r rH �liner r r � testReadLine� s
zBZ2FileTest.testReadLinec C sZ | j dd� t| j��:}| jt|jd � x"| jd D ]}| j|j� |� q4W W d Q R X d S )NrL )r9 )r< r r r@ rA re r& rG )r rH rf r r r �testReadLineMultiStream� s
z#BZ2FileTest.testReadLineMultiStreamc C sD | j � t| j��(}| jt|jd � | j|j� | j� W d Q R X d S )N)r< r r r@ rA � readlinesrG r& )r rH r r r �
testReadLines� s zBZ2FileTest.testReadLinesc C sL | j dd� t| j��,}| jt|jd � | j|j� | jd � W d Q R X d S )NrL )r9 )r< r r r@ rA ri rG r& )r rH r r r �testReadLinesMultiStream� s z$BZ2FileTest.testReadLinesMultiStreamc C s8 | j � t| j�� }| jtt|��| j� W d Q R X d S )N)r< r r rG �list�iterr& )r rH r r r � testIterator� s zBZ2FileTest.testIteratorc C s@ | j dd� t| j�� }| jtt|��| jd � W d Q R X d S )NrL )r9 )r< r r rG rl rm r& )r rH r r r �testIteratorMultiStream� s z#BZ2FileTest.testIteratorMultiStreamc C s: | j � t| j�}|j� | jtt|� | jt|j� d S )N)r< r r �closer@ rB �nextri )r rH r r r �testClosedIteratorDeadlock� s
z&BZ2FileTest.testClosedIteratorDeadlockc C sd t | jd�� }| jt|j� |j| j� W d Q R X t| jd�� }| jt|j � �| j� W d Q R X d S )N�wr )
r r r@ rA r8 r( r1 rG r rE )r rH r; r r r � testWrite� s
zBZ2FileTest.testWritec C s� t | jd��B}d}x6| j|d |d d � }|s4P |j|� |d7 }qW W d Q R X t| jd�� }| jt|j� �| j� W d Q R X d S )Nrs r r? r r )r r r( r8 r1 rG r rE )r rH rb rW r; r r r �testWriteChunks10� s
z BZ2FileTest.testWriteChunks10c C sd t j| jdd�}t| jddd��}|j| j� W d Q R X t| jd��}| j|j� |� W d Q R X d S )NrL )r rs r ) r r4 r( r r r8 r1 rG rE )r ZexpectedrH r; r r r � testWriteNonDefaultCompressLevel s
z,BZ2FileTest.testWriteNonDefaultCompressLevelc C sv t | jd�� }| jt|j� |j| j� W d Q R X | jt|jdg� t| jd�� }| jt |j
� �| j � W d Q R X d S )Nrs �ar ) r r r@ rA �
writelinesr& rB r1 rG r rE r( )r rH r; r r r �testWriteLines s zBZ2FileTest.testWriteLinesc C s` t | jd��}|jd� W d Q R X t | jd��(}| jt|jd� | jt|jdg� W d Q R X d S )Nrs s abc�r� a)r r r8 r@ rJ rx )r rH r r r � testWriteMethodsOnReadOnlyFile s
z*BZ2FileTest.testWriteMethodsOnReadOnlyFilec C s� t | jd�� }| jt|j� |j| j� W d Q R X t | jd�� }| jt|j� |j| j� W d Q R X t| jd�� }| jt|j � �| jd � W d Q R X d S )Nrs rw r � )
r r r@ rA r8 r( r1 rG r rE )r rH r; r r r �
testAppend s zBZ2FileTest.testAppendc C sT | j � t| j��8}| jt|j� |jd� | j|j� | jdd � � W d Q R X d S )N� ) r< r r r@ rA �seekrG rE r( )r rH r r r �testSeekForward* s
zBZ2FileTest.testSeekForwardc C sb | j dd� t| j��B}| jt|j� |jt| j�d � | j|j � | jdd � � W d Q R X d S )Nr} )r9 r )
r< r r r@ rA r� rP r( rG rE )r rH r r r � testSeekForwardAcrossStreams1 s
z(BZ2FileTest.testSeekForwardAcrossStreamsc C sR | j � t| j��6}|jd� |jdd� | j|j� | jdd � � W d Q R X d S )Ni� r r ij���i^ )r< r r rE r� rG r( )r rH r r r �testSeekBackwards8 s
z BZ2FileTest.testSeekBackwardsc C s� | j dd� t| j��`}t| j�d }x |dkrD|t|j|��8 }q(W |jdd� | j|j� | jdd � | j � W d Q R X d S ) Nr} )r9 rZ r r r ij���i���)r< r r rP r( rE r� rG )r rH Zreadtor r r � testSeekBackwardsAcrossStreams? s
z*BZ2FileTest.testSeekBackwardsAcrossStreamsc C sR | j � t| j��6}|jdd� | j|j� | jt| j�d d � � W d Q R X d S )Nr r} ij���)r< r r r� rG rE r( rP )r rH r r r �testSeekBackwardsFromEndH s z$BZ2FileTest.testSeekBackwardsFromEndc C sP | j dd� t| j��0}|jdd� | j|j� | jd dd � � W d Q R X d S )Nr} )r9 i� i���i���)r< r r r� rG rE r( )r rH r r r �%testSeekBackwardsFromEndAcrossStreamsN s z1BZ2FileTest.testSeekBackwardsFromEndAcrossStreamsc C sR | j � t| j��6}|jd� | j|j� t| j�� | j|j� d� W d Q R X d S )Ni�I r ) r< r r r� rG �tellrP r( rE )r rH r r r �testSeekPostEndT s
zBZ2FileTest.testSeekPostEndc C sZ | j dd� t| j��:}|jd� | j|j� t| j�d � | j|j� d� W d Q R X d S )NrL )r9 i�I r ) r< r r r� rG r� rP r( rE )r rH r r r �testSeekPostEndMultiStream[ s
z&BZ2FileTest.testSeekPostEndMultiStreamc C s\ | j � t| j��@}|jd� |jd� | j|j� t| j�� | j|j� d� W d Q R X d S )Ni�I r ) r< r r r� rG r� rP r( rE )r rH r r r �testSeekPostEndTwiceb s
z BZ2FileTest.testSeekPostEndTwicec C sd | j dd� t| j��D}|jd� |jd� | j|j� t| j�d � | j|j� d� W d Q R X d S )NrL )r9 i�I r ) r< r r r� rG r� rP r( rE )r rH r r r � testSeekPostEndTwiceMultiStreamj s
z+BZ2FileTest.testSeekPostEndTwiceMultiStreamc
C sN | j � t| j��2}|jd� | j|j� d� | j|j� | j� W d Q R X d S )Nr r ij���)r< r r r� rG r� rE r( )r rH r r r �testSeekPreStartr s
z BZ2FileTest.testSeekPreStartc C sV | j dd� t| j��6}|jd� | j|j� d� | j|j� | jd � W d Q R X d S )Nr} )r9 r r ij���)r< r r r� rG r� rE r( )r rH r r r �testSeekPreStartMultiStreamy s
z'BZ2FileTest.testSeekPreStartMultiStreamc C s^ | j � t| jd��2}t|�}z| j|j� |j� � W d |j� X W d Q R X | jt|j� d S )Nr ) r< r1 r r rG �filenorp r@ rB )r ZrawfrH r r r �
testFileno� s zBZ2FileTest.testFilenoc C s� t t| j��}z(| j|j� � |j� | j|j� � W d |j� X | jt|j� t t� d�}z| j |j� � W d |j� X | jt|j� t| j�}dd� |_t |�}z| j |j� � W d |j� X | jt|j� d S )Nrs c S s dS )NFr r r r r �<lambda>� s z*BZ2FileTest.testSeekable.<locals>.<lambda>)
r r r) r^ �seekablerE rp r@ rB � assertFalse)r rH �srcr r r � testSeekable� s&
zBZ2FileTest.testSeekablec C s� t t| j��}z(| j|j� � |j� | j|j� � W d |j� X | jt|j� t t� d�}z| j |j� � W d |j� X | jt|j� d S )Nrs )
r r r) r^ �readablerE rp r@ rB r� )r rH r r r � testReadable� s
zBZ2FileTest.testReadablec C s� t t| j��}z(| j|j� � |j� | j|j� � W d |j� X | jt|j� t t� d�}z| j |j� � W d |j� X | jt|j� d S )Nrs )
r r r) r� �writablerE rp r@ rB r^ )r rH r r r � testWritable� s
zBZ2FileTest.testWritablec C s* | j � x td�D ]}t| j�}~qW d S )Ni' )r< �ranger r )r �i�or r r � testOpenDel� s
zBZ2FileTest.testOpenDelc C s | j ttd� d S )Nz
/non/existent)r@ rJ r )r r r r �testOpenNonexistent� s z BZ2FileTest.testOpenNonexistentc C s� d}t | jd��}|j|� W d Q R X t| j��}|j� }W d Q R X | j|dg� t| j��}t|j� �}W d Q R X | j|dg� d S )Ns, BZh91AY&SY�b�] � !�h3M<]��BCe�%tr7 s Test)r1 r r8 r ri rG rl )r r r; rH �linesZxlinesr r r �testReadlinesNoNewline� s z"BZ2FileTest.testReadlinesNoNewlinec # C s� d }t | jd��}|jd� W d Q R X t | jd�}|j� y|� W d Q R X W n tk
rb Y n X | jd� y$t | jd��}dd W d Q R X W n tk
r� Y n X | jd� d S )Nr7 s xxxr z4__enter__ on a closed file didn't raise an exceptionr r z 1/0 didn't raise an exception)r r r8 rp rB Zfail�ZeroDivisionError)r r; r r r �testContextProtocol� s"
z BZ2FileTest.testContextProtocolz!Threading required for this test.c sb dd
�d}t | jd��@���fdd�� � fdd �t|�D �}tj|�� W d Q R X W d Q R X d S ) N� 1r} � r? r7 c s xt d�D ]} �j� � q
W d S )NrL )r� r8 )r� )r r; r r �comp� s z'BZ2FileTest.testThreading.<locals>.compc s g | ]}t j� d ��qS ))�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 s` t jddd�}|j| jd��}|jd� W d Q R X |j| jd��}| j|j� d� W d Q R X d S )Nr r� )Zblockedr7 s abcr )r� )r Zimport_fresh_moduler r r8 rG rE )r �moduler; r r r �testWithoutThreading� s
z BZ2FileTest.testWithoutThreadingc C s | j � t| jd �}|d }t| j��H}|j|� | jt|�| jd |d � � | j|j� | j|d � � W d Q R X t| j��:}|j � | jt|�| jd � | j|j � | jd � W d Q R X t| j��,}|j
� | j t t|� | j|j
� g � W d Q R X d S )Nr r} r )
r< rP r& r r rE rG rq r( re ri r@ �
StopIteration)r ZlinelenZhalflenrH r r r �testMixedIterationAndReads s
$ z&BZ2FileTest.testMixedIterationAndReadsc C s� dd }dd }t | jd��}|j|� W d Q R X t | jd��}|j|� W d Q R X t | j��}| j|j� || � W d Q R X d S )Ns fooi� s barrs rw )r r r8 rG rE )r Zdata1Zdata2rH r r r �testMultiStreamOrdering s z#BZ2FileTest.testMultiStreamOrderingc C s� | j }y|jd�}W n tk
r2 | jd� Y nX t|d��}|j| j� W d Q R X t|d��}| j|j� | j� W d Q R X t|d��}| j|j� | j� W d Q R X d S )N�asciiz%Temporary file name needs to be ASCIIr7 r ) r �encode�UnicodeEncodeErrorZskipTestr r8 r) rG rE )r Z str_filenameZbytes_filenamer; r r r �testOpenBytesFilename s z!BZ2FileTest.testOpenBytesFilenamec C sZ t j| j�}t|d��}|j| j� W d Q R X t|d��}| j|j� | j� W d Q R X d S )Nr7 r )�pathlibZPathr r r8 r) rG rE )r r r; r r r �testOpenPathLikeFilename. s
z$BZ2FileTest.testOpenPathLikeFilenamec C sh t jdtd� dd�}| jt|�tj� tt|��}| j |j
d�d� dt }| j |j
jj� |d� d S )N� g ��>A� )r r z)Excessive amount of data was decompressed)r r4 �intZ
assertLessrP rN rO r r rG rE r �assertLessEqualZ_buffer�rawr� )r ZbombZdecompZ
max_decompr r r �testDecompressLimited5 s z!BZ2FileTest.testDecompressLimitedc C s^ t | j��J}t|��*}| jt|jt� � | j|j� | j� W d Q R X | j |j
� W d Q R X d S )N) r r) r r@ rA rE rF rG r( r� �closed)r �biorH r r r �testReadBytesIOC s
zBZ2FileTest.testReadBytesIOc C sj t | j��V}t|��B}|j� }| jt|�d� | j| jj|�� | j |j
� | j� W d Q R X W d Q R X d S )Nr ) r r) r r\ r] rP r^ r( r_ rG rE )r r� rH r` r r r �testPeekBytesIOJ s
zBZ2FileTest.testPeekBytesIOc C sh t � �X}t|d�� }| jt|j� |j| j� W d Q R X | jt|j� �| j� | j |j
� W d Q R X d S )Nrs ) r r r@ rA r8 r( rG r �getvaluer� r� )r r� rH r r r �testWriteBytesIOR s z BZ2FileTest.testWriteBytesIOc C s` t | j��L}t|��8}| jt|j� |jd� | j|j� | jdd � � W d Q R X W d Q R X d S )Nr ) r r) r r@ rA r� rG rE r( )r r� rH r r r �testSeekForwardBytesIOZ s
z"BZ2FileTest.testSeekForwardBytesIOc C s^ t | j��J}t|��6}|jd� |jdd� | j|j� | jdd � � W d Q R X W d Q R X d S )Ni� r r ij���i^ )r r) r rE r� rG r( )r r� rH r r r �testSeekBackwardsBytesIOa s
z$BZ2FileTest.testSeekBackwardsBytesIOc C s� | j d d� }tt|���}| jt|j� W d Q R X tt|���0}| j|jt| j��| j� | jt|jd� W d Q R X x@t d�D ]4}tt|d |� ���}| jt|jd� W d Q R X q�W d S )Nr? r � i����)
r) r r r@ �EOFErrorrE rG rP r( r� )r Z truncatedr; r� r r r �test_read_truncatedh s z BZ2FileTest.test_read_truncatedN)r r )Br# r$ r% r< rD rI rK rM rR rS rT rU rX rY r[ ra rd rg rh rj rk rn ro rr rt ru rv ry 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 r r6 Z sx
r6 c @ sD e Zd Zdd� Zdd� Zdd� Zeed dd �d
d � �Zd d
� Z dS )�BZ2CompressorTestc C sB t � }| jt|j� |j| j�}||j� 7 }| jt|�| j� d S )N)r
r@ rA r4 r( �flushrG r )r �bz2cr r r r � testCompressw s
z BZ2CompressorTest.testCompressc C s. t � }|jd�}||j� 7 }| j|| j� d S )Nr )r
r4 r� rG r* )r r� r r r r �testCompressEmptyString~ s
z)BZ2CompressorTest.testCompressEmptyStringc C sl t � }d}d}x:| j|d |d d � }|s0P ||j|�7 }|d7 }qW ||j� 7 }| jt|�| j� d S )Nr r r? r )r
r( r4 r� rG r )r r� rb r rW r r r �testCompressChunks10� s z&BZ2CompressorTest.testCompressChunks10rZ r} )�size�memusec C sp t � }d| }z|j|�}||j� 7 }W d d }X tj|�}z*| jt|�|� | jt|jd��d� W d d }X d S )N� xr )r
r4 r� r r rG rP �strip)r r� r� r �
compressedr r r �testCompress4G� s
z BZ2CompressorTest.testCompress4Gc C s@ x:t tjd �D ](}| jt�� tjt� |� W d Q R X qW d 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� v s
r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zee d d d
�dd� �Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zejd d � �Zd S ) �BZ2DecompressorTestc C s | j ttd� d S )N�* )r@ rA r )r r r r �test_Constructor� s z$BZ2DecompressorTest.test_Constructorc C s2 t � }| jt|j� |j| j�}| j|| j� d S )N)r r@ rA r r) rG r( )r �bz2drV r r r �testDecompress� s z"BZ2DecompressorTest.testDecompressc C s\ t � }d}d}x:| j|d |d d � }|s0P ||j|�7 }|d7 }qW | j|| j� d S )Nr r r? r )r r) r rG r( )r r� rV rb rW r r r �testDecompressChunks10� s z*BZ2DecompressorTest.testDecompressChunks10c C s: 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� rV r r r �testDecompressUnusedData� s
z,BZ2DecompressorTest.testDecompressUnusedDatac C s6 t � }|j| j�}| jt|jd� | jt|jd� d S )Ns anythingr )r r r) r@ r� )r r� rV r r r � testEOFError� s z BZ2DecompressorTest.testEOFErrorrZ gffffff
@)r� r� c
C sj d}t j|d �j|d�}z<||| d }tj|�}t� }|j|�}| j||k� W d d }d }d }X d S )Nr? i � �littler i ( i � )�randomZ getrandbits�to_bytesr r4 r r r^ )r r� Z blocksize�blockr r� r� Z decompressedr r r �testDecompress4G� s
z$BZ2DecompressorTest.testDecompress4Gc C s@ x:t tjd �D ](}| jt�� tjt� |� W d Q R X qW d 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| jd |� |d�� | j|j� | jt|d �|� |j|jd|d�� | j|j� | jt|d �|� |j|j| j|d � |d�� | jt|d �|� x2|j s�|j|jd|d�� | jt|d �|� q�W dj
|�}| j|| j � | j|j d� d S )
NrZ �@ )�
max_lengthr r ���r� r� r� )
r rP r5 �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| jd d� dd�d� |j|jdd�� |j|j| jdd� d�� |j|j| jdd � �� | jdj|�| j� d S )NrZ 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| jd d� dd�d� |j|jd�� |j|j| jdd� d�� |j|j| jdd� d�� |j|j| jdd � �� | jdj|�| 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 st t � }g }|j|j| jd d� d�� |j|j| jdd� d�� |j|j| jdd � �� | jdj|�| 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 s6 t � }| jt|j| jd � | jt|j| jd � d S )N� )r r@ � Exceptionr r+ )r r� r r r � test_failure: s z BZ2DecompressorTest.test_failurec C sL t jtd�}t� }|� }xtd�D ] }|j� q"W | j|� | ddd� d S )N�gettotalrefcountrZ r r? )Zdelta)r Z
get_attribute�sysr r� �__init__ZassertAlmostEqual)r r� r� Z refs_beforer� r r r �test_refleaks_in___init__@ s z-BZ2DecompressorTest.test_refleaks_in___init__N)r# r$ r% r� r� r� r� r� r r r� r� r� r� r� r� r� r Z
refcount_testr� r r r r r� � s r� c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�CompressDecompressTestc C s" t j| j�}| jt|�| j� d S )N)r r4 r( rG r )r r r r r r� K s z#CompressDecompressTest.testCompressc C s t jd�}| j|| j� d S )Nr )r r4 rG r* )r rV r r r r� O s
z.CompressDecompressTest.testCompressEmptyStringc C s t j| j�}| j|| j� d S )N)r r r) rG r( )r rV r r r r� S s z%CompressDecompressTest.testDecompressc C s t jd�}| j|d� d S )Nr )r r rG )r rV r r r �testDecompressEmptyW s
z*CompressDecompressTest.testDecompressEmptyc C s t j| j�}| j|d� d S )Nr )r r r* rG )r rV r r r �testDecompressToEmptyString[ s z2CompressDecompressTest.testDecompressToEmptyStringc C s | j ttj| jd d� � d S )Nr? i����)r@ rB r r r) )r r r r �testDecompressIncomplete_ s z/CompressDecompressTest.testDecompressIncompletec C s | j ttj| j� d S )N)r@ rJ r r r+ )r r r r �testDecompressBadDatab s z,CompressDecompressTest.testDecompressBadDatac C s& t j| jd �}| j|| jd � d S )NrL )r r r) rG r( )r rV r r r �testDecompressMultiStreame s z0CompressDecompressTest.testDecompressMultiStreamc C s$ t j| j| j �}| j|| j� d S )N)r r r) r+ rG r( )r rV r r r �testDecompressTrailingJunki s z1CompressDecompressTest.testDecompressTrailingJunkc C s, t j| jd | j �}| j|| jd � d S )NrL )r r r) r+ rG r( )r rV r r r �%testDecompressMultiStreamTrailingJunkm s z<CompressDecompressTest.testDecompressMultiStreamTrailingJunkN)
r# r$ r% r� r� r� r� r� r� r r r r r r r r r� J s r� c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�OpenTestc O s t j||�S )N)r r1 )r �args�kwargsr r r r1 u s z
OpenTest.openc ( C s � x�dD ]�}|dkr t | j� | j| j|��}|j| j� W d Q R X t| jd�� }t|j� �}| j|| j� W d Q R X | j| jd��}| j|j� | j� W d Q R X | j| jd��}|j| j� W d Q R X t| jd��$}t|j� �}| j|| jd � W d Q R X qW d S )Nr7 �xbr Zabr} )r7 r )r r r1 r8 r( r rE rG )r �moder; � file_datar r r �test_binary_modesx s
zOpenTest.test_binary_modesc ( C s � x�dD ]�}|dkr t | j� | j| j|��}|j| j� W d Q R X t| jd�� }t|j� �}| j|| j� W d Q R X | j| jd��}| j|j� | j� W d Q R X | j| jd��}|j| j� W d Q R X t| jd��$}t|j� �}| j|| jd � W d Q R X qW d S )Nrs �xr rz rw r} )rs r )r r r1 r8 r( r rE rG )r r r; r r r r �test_implicit_binary_modes� s
z#OpenTest.test_implicit_binary_modesc ( C s" | j jd�}|jdtj�}�x d D ]�}|dkr8t| j� | j| j|��}|j|� W d Q R X t| jd��$}t |j
� �jd�}| j ||� W d Q R X | j| jd��}| j |j
� |� W d Q R X | j| jd��}|j|� W d Q R X t| jd��(}t |j
� �jd�}| j ||d � W d Q R X q"W d S )
Nr� �
�wt�xtr �rtZatr} )r r ) r( �decode�replacer �linesepr r r1 r8 r rE rG )r rV �text_native_eolr r; r r r r �test_text_modes� s"
zOpenTest.test_text_modesc C sf x`dD ]X}t | j� | j| j|��}W d Q R X | jt�� | j| j|��}W d Q R X W d Q R X qW d S )Nr r r )r r r )r r r1 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� W d Q R X | j t| j�d��}| j|j� | j� W d Q R X | jjd�}| j t| j�d��}| j|j� |� W d Q R X d S )Nrz r r� r )r1 r r) rG rE r( r )r r; rV r r r � test_fileobj� s zOpenTest.test_fileobjc C st | j t| j| jd� | j t| j| jd� | j t| j| jddd� | j t| j| jddd� | j t| j| jddd � d S )
NZwbtZxbtr zutf-8)�encoding�ignore)�errorsr
)�newline)r@ rB r1 r )r r r r �test_bad_params� s zOpenTest.test_bad_paramsc C s� | j jd�}|jdtj�}| j| jddd��}|j|� W d Q R X t| jd��$}t|j � �jd�}| j
||� W d Q R X | j| jddd��}| j
|j � |� W d Q R X d S )Nr� r
r z utf-16-le)r r r ) r( r r r r r1 r r8 r rE rG )r rV r r; r r r r �
test_encoding� s zOpenTest.test_encodingc C sX | j | jd��}|jd� W d Q R X | j | jdddd��}| j|j� d� W d Q R X d S )Nr7 s foo�barr r� r )r r Zfoobar)r1 r r8 rG rE )r r; r r r �test_encoding_error_handler� s
z$OpenTest.test_encoding_error_handlerc C sh | j jd�}| j| jddd��}|j|� W d Q R X | j| jddd��}| j|j� |g� W d Q R X d S )Nr� r r
)r r �
)r( r r1 r r8 rG ri )r rV r; r r r � test_newline� s
zOpenTest.test_newlineN)
r# r$ r% r1 r
r r r r r r r r! r r r r r r s
r c C s t jttttt� t j� d S )N)r Z run_unittestr6 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 r r� r rN r� � ImportError�
import_moduler r r
r r
r ZTestCaser r6 r� r� r� r r" r# r r r r �<module> sH
4 2 #(x