File: //usr/local/lib/python3.7/test/__pycache__/test_bz2.cpython-37.opt-1.pyc
B
��gǒ � @ s8 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 dlZe�d�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�r4e"� dS )� )�support)�
bigmemtest�_4GN)�BytesIO�DEFAULT_BUFFER_SIZE)�unlink�bz2)�BZ2File�
BZ2Compressor�BZ2Decompressorc C s8 t d krtt�d��a t r*tjdg| d�S t�| �S d S )NZbunzip2)�input)�has_cmdline_bunzip2�bool�shutilZwhich�
subprocessZ check_outputr �
decompress)�data� r �)/usr/local/lib/python3.7/test/test_bz2.py�ext_decompress s
r c @ s� e Zd ZdZdddddddd d
d d d
dddddddddgZd�e�ZdZdZdZ dZ
e d �Z x^e
�
ej�ej�e�d ��D ]>Zeed �� Ze
e�ee �e
d � �7 Z
W d Q R X e
d krzP qzW eje d d!�Zd"d#� Zd$d%� Zd 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� )�
compresslevelc C s t j| _d S )N)r ZTESTFN�filename)�selfr r r �setUpN s zBaseTest.setUpc C s t j�| j�rt �| j� d S )N)�os�path�isfiler r )r r r r �tearDownQ s zBaseTest.tearDown) �__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
" $r c @ s� e Zd ZdZd{dd�Zdd� Zdd � 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.d/� Zd0d1� Zd2d3� Zd4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z d>d?� Z!d@dA� Z"dBdC� Z#dDdE� Z$dFdG� Z%dHdI� Z&dJdK� Z'dLdM� Z(dNdO� Z)dPdQ� Z*dRdS� Z+dTdU� Z,dVdW� Z-dXdY� Z.dZd[� Z/d\d]� Z0d^d_� Z1d`da� Z2dbdc� Z3ddde� Z4dfdg� Z5dhdi� Z6djdk� Z7dldm� Z8dndo� Z9dpdq� Z:drds� Z;dtdu� Z<dvdw� Z=dxdy� Z>dzS )|� BZ2FileTestzTest the BZ2File class.r r c C s6 t | jd�� }|�| j| � |�|� W d Q R X d S )N�wb)r1 r �writer) )r �streams�suffix�fr r r �createTempFileY s zBZ2FileTest.createTempFilec C sp | � ttd� | � tttjd� | � tttjd� | � tttjd� | j tttjdd� | j tttjdd� d S )Ngw��/�^@�z�rxZrbtr )r �
)� assertRaises� TypeErrorr �
ValueErrorr �devnull)r r r r � testBadArgs^ s zBZ2FileTest.testBadArgsc C sF | � � t| j��*}| �t|jt� � | �|�� | j� W d Q R X d S )N) r< r r r@ rA �read�float� assertEqualr( )r �bz2fr r r �testReadf s zBZ2FileTest.testReadc C s8 | j d| jd� t| j��}| �t|j� W d Q R X d S )Nr )r9 r: )r< r+ r r r@ �OSErrorrE )r rH r r r �testReadBadFilel s zBZ2FileTest.testReadBadFilec C sN | j dd� t| j��.}| �t|jt� � | �|�� | jd � W d Q R X d S )N� )r9 ) r< r r r@ rA rE rF rG r( )r rH r r r �testReadMultiStreamq s z BZ2FileTest.testReadMultiStreamc C sn t j}t| j�t _zN| jdd� t| j��.}| �t|j t
� � | � |� � | 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 �testReadMonkeyMultiStreamw s $z%BZ2FileTest.testReadMonkeyMultiStreamc C s: | j | jd� t| 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�� }| �|�� | 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 | � � t| j��*}| �t|jt� � | �|�d�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 | � � t| j��6}d}x |�d�}|s*P ||7 }qW | �|| 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 |�d�}|s.P ||7 }q W | �|| 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> | � � t| j��"}| �|�d�| 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^ | � � t| j��B}|�� }| �t|�d� | �| 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� | � � t| j���}d}t|�}| �|�|�|� | �|| jd |� � t| j�| }tt| j��}| �|�|�|� | �|d |� | j| d � � W d Q R X d S )N� )r< r r r, rG r2 r( rP )r rH �n�br r r � testReadInto� s zBZ2FileTest.testReadIntoc C sR | � � t| j��6}| �t|jd � x | jD ]}| �|�� |� 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��:}| �t|jd � x"| jd D ]}| �|�� |� q4W W d Q R X d S )NrL )r9 )r< r r r@ rA rf r& rG )r rH rg r r r �testReadLineMultiStream� s
z#BZ2FileTest.testReadLineMultiStreamc C sD | � � t| j��(}| �t|jd � | �|�� | 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��,}| �t|jd � | �|�� | jd � W d Q R X d S )NrL )r9 )r< r r r@ rA rj rG r& )r rH r r r �testReadLinesMultiStream� s z$BZ2FileTest.testReadLinesMultiStreamc C s8 | � � t| j�� }| �tt|��| 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�� }| �tt|��| jd � W d Q R X d S )NrL )r9 )r< r r rG rm rn r& )r rH r r r �testIteratorMultiStream� s z#BZ2FileTest.testIteratorMultiStreamc C s: | � � t| j�}|�� | �tt|� | �t|j� d S )N)r< r r �closer@ rB �nextrj )r rH r r r �testClosedIteratorDeadlock� s
z&BZ2FileTest.testClosedIteratorDeadlockc C sd t | jd�� }| �t|j� |�| j� W d Q R X t| jd�� }| �t|� � �| 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 |�|� |d7 }qW W d Q R X t| jd�� }| �t|�� �| j� W d Q R X d S )Nrt r r? r r )r r r( r8 r1 rG r rE )r rH rc rW r; r r r �testWriteChunks10� s
z BZ2FileTest.testWriteChunks10c C sd t j| jdd�}t| jddd��}|�| j� W d Q R X t| jd��}| �|�� |� W d Q R X d S )NrL )r rt 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�� }| �t|j� |�| j� W d Q R X | �t|jdg� t| jd�� }| �t |�
� �| j � W d Q R X d S )Nrt �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��}|�d� W d Q R X t | jd��(}| �t|jd� | �t|jdg� W d Q R X d S )Nrt s abc�r� a)r r r8 r@ rJ ry )r rH r r r � testWriteMethodsOnReadOnlyFile s
z*BZ2FileTest.testWriteMethodsOnReadOnlyFilec C s� t | jd�� }| �t|j� |�| j� W d Q R X t | jd�� }| �t|j� |�| j� W d Q R X t| jd�� }| �t|� � �| jd � W d Q R X d S )Nrt rx r � )
r r r@ rA r8 r( r1 rG r rE )r rH r; r r r �
testAppend s zBZ2FileTest.testAppendc C sT | � � t| j��8}| �t|j� |�d� | �|�� | 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}| �t|j� |�t| j�d � | �|� � | 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 � testSeekForwardAcrossStreams- s
z(BZ2FileTest.testSeekForwardAcrossStreamsc C sR | � � t| j��6}|�d� |�dd� | �|�� | jdd � � W d Q R X d S )Ni� ij���r i^ )r< r r rE r� rG r( )r rH r r r �testSeekBackwards4 s
z BZ2FileTest.testSeekBackwardsc C s� | j dd� t| j��`}t| j�d }x |dkrD|t|�|��8 }q(W |�dd� | �|�� | jdd � | j � W d Q R X d S )Nr~ )r9 rZ r ij���r i���)r< r r rP r( rE r� rG )r rH Zreadtor r r � testSeekBackwardsAcrossStreams; s
z*BZ2FileTest.testSeekBackwardsAcrossStreamsc C sR | � � t| j��6}|�dd� | �|�� | jt| j�d d � � W d Q R X d S )Nij���r~ r� )r< r r r� rG rE r( rP )r rH r r r �testSeekBackwardsFromEndD s z$BZ2FileTest.testSeekBackwardsFromEndc C sP | j dd� t| j��0}|�dd� | �|�� | jd dd � � W d Q R X d S )Nr~ )r9 i���)r< r r r� rG rE r( )r rH r r r �%testSeekBackwardsFromEndAcrossStreamsJ s z1BZ2FileTest.testSeekBackwardsFromEndAcrossStreamsc C sR | � � t| j��6}|�d� | �|�� t| 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 �testSeekPostEndP s
zBZ2FileTest.testSeekPostEndc C sZ | j dd� t| j��:}|�d� | �|�� t| j�d � | �|�� 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 �testSeekPostEndMultiStreamW s
z&BZ2FileTest.testSeekPostEndMultiStreamc C s\ | � � t| j��@}|�d� |�d� | �|�� t| 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 �testSeekPostEndTwice^ s
z BZ2FileTest.testSeekPostEndTwicec C sd | j dd� t| j��D}|�d� |�d� | �|�� t| j�d � | �|�� 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 � testSeekPostEndTwiceMultiStreamf s
z+BZ2FileTest.testSeekPostEndTwiceMultiStreamc C sN | � � t| j��2}|�d� | �|�� d� | �|�� | j� W d Q R X d S )Nij���r )r< r r r� rG r� rE r( )r rH r r r �testSeekPreStartn s
z BZ2FileTest.testSeekPreStartc C sV | j dd� t| j��6}|�d� | �|�� d� | �|�� | jd � W d Q R X d S )Nr~ )r9 ij���r )r< r r r� rG r� rE r( )r rH r r r �testSeekPreStartMultiStreamu s
z'BZ2FileTest.testSeekPreStartMultiStreamc C s^ | � � t| jd��2}t|�}z| �|�� |�� � W d |�� X W d Q R X | �t|j� d S )Nr ) r< r1 r r rG �filenorq r@ rB )r ZrawfrH r r r �
testFileno| s zBZ2FileTest.testFilenoc C s� t t| j��}z(| �|�� � |�� | �|�� � W d |�� X | �t|j� t t� d�}z| � |�� � W d |�� X | �t|j� t| j�}dd� |_t |�}z| � |�� � W d |�� X | �t|j� d S )Nrt c S s dS )NFr r r r r �<lambda>� r z*BZ2FileTest.testSeekable.<locals>.<lambda>)
r r r) r^ �seekablerE rq r@ rB � assertFalse)r rH �srcr r r � testSeekable� s&
zBZ2FileTest.testSeekablec C s� t t| j��}z(| �|�� � |�� | �|�� � W d |�� X | �t|j� t t� d�}z| � |�� � W d |�� X | �t|j� d S )Nrt )
r r r) r^ �readablerE rq r@ rB r� )r rH r r r � testReadable� s
zBZ2FileTest.testReadablec C s� t t| j��}z(| �|�� � |�� | �|�� � W d |�� X | �t|j� t t� d�}z| � |�� � W d |�� X | �t|j� d S )Nrt )
r r r) r� �writablerE rq r@ rB r^ )r rH r r r � testWritable� s
zBZ2FileTest.testWritablec C s* | � � x td�D ]}t| j�}~qW d S )Ni' )r< �ranger r )r �i�or r r � testOpenDel� s
zBZ2FileTest.testOpenDelc C s | � ttd� d S )Nz
/non/existent)r@ rJ r )r r r r �testOpenNonexistent� s z BZ2FileTest.testOpenNonexistentc C s� d}t | jd��}|�|� W d Q R X t| j��}|�� }W d Q R X | �|dg� t| j��}t|�� �}W d Q R X | �|dg� d S )Ns, BZh91AY&SY�b�] � !�h3M<]��BCe�%tr7 s Test)r1 r r8 r rj rG rm )r r r; rH �linesZxlinesr r r �testReadlinesNoNewline� s z"BZ2FileTest.testReadlinesNoNewlinec C s� d }t | jd��}|�d� W d Q R X t | jd�}|�� y|� W d Q R X W n tk
rb Y n X | �d� y$t | jd��}dd W d Q R X W n tk
r� Y n X | �d� 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 rq rB Zfail�ZeroDivisionError)r r; r r r �testContextProtocol� s"
z BZ2FileTest.testContextProtocolc
sb dd �d}t | jd��@���fdd�� � fdd�t|�D �}t�|�� W d Q R X W d Q R X d S ) N� 1i r? r7 c s xt d�D ]} ��� � 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>)r r r� r Z
start_threads)r ZnthreadsZthreadsr )r� r r; r �
testThreading� s zBZ2FileTest.testThreadingc C s | � � t| jd �}|d }t| j��H}|�|� | �t|�| jd |d � � | �|�� | j|d � � W d Q R X t| j��:}|� � | �t|�| jd � | �|� � | jd � W d Q R X t| j��,}|�
� | � t t|� | �|�
� g � W d Q R X d S )Nr r~ r )
r< rP r& r r rE rG rr r( rf rj r@ �
StopIteration)r ZlinelenZhalflenrH r r r �testMixedIterationAndReads� s
$ z&BZ2FileTest.testMixedIterationAndReadsc C sz d}d}t | jd��}|�|� W d Q R X t | jd��}|�|� W d Q R X t | j��}| �|�� || � W d Q R X d S )Ns� foofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoos� barbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarrt rx )r r r8 rG rE )r Zdata1Zdata2rH r r r �testMultiStreamOrdering s z#BZ2FileTest.testMultiStreamOrderingc C s� | j }y|�d�}W n tk
r2 | �d� Y nX t|d��}|�| j� W d Q R X t|d��}| �|�� | j� W d Q R X t|d��}| �|�� | 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�}t|d��}|�| j� W d Q R X t|d��}| �|�� | 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�}| �t|�tj� tt|��}| � |�
d�d� dt }| � |j
j�� |d� dS )z-Decompressed data buffering should be limited� g ��>A� )r r z)Excessive amount of data was decompressedN)r r4 �intZ
assertLessrP rN rO r r rG rE r �assertLessEqualZ_buffer�rawr� )r ZbombZdecompZ
max_decompr r r �testDecompressLimited) s z!BZ2FileTest.testDecompressLimitedc
C s^ t | j��J}t|��*}| �t|jt� � | �|�� | j� W d Q R X | � |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 �testReadBytesIO7 s
zBZ2FileTest.testReadBytesIOc
C sj t | j��V}t|��B}|�� }| �t|�d� | �| 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 �testPeekBytesIO> s
zBZ2FileTest.testPeekBytesIOc
C sh t � �X}t|d�� }| �t|j� |�| j� W d Q R X | �t|�� �| j� | � |j
� W d Q R X d S )Nrt ) r r r@ rA r8 r( rG r �getvaluer� r� )r r� rH r r r �testWriteBytesIOF s z BZ2FileTest.testWriteBytesIOc
C s` t | j��L}t|��8}| �t|j� |�d� | �|�� | 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 �testSeekForwardBytesION s
z"BZ2FileTest.testSeekForwardBytesIOc
C s^ t | j��J}t|��6}|�d� |�dd� | �|�� | jdd � � W d Q R X W d Q R X d S )Ni� ij���r i^ )r r) r rE r� rG r( )r r� rH r r r �testSeekBackwardsBytesIOU s
z$BZ2FileTest.testSeekBackwardsBytesIOc
C s� | j d d� }tt|���}| �t|j� W d Q R X tt|���0}| �|�t| j��| j� | �t|jd� W d Q R X x@t d�D ]4}tt|d |� ���}| �t|jd� W d Q R X q�W d S )Ni����r � )
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)r r )?r" r# r$ r% r< rD rI rK rM rR rS rT rU rX rY r[ ra 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� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r6 V sx
r6 c @ sD e Zd Zdd� Zdd� Zdd� Zeed dd �d
d � �Zd d
� Z dS )�BZ2CompressorTestc C sB t � }| �t|j� |�| j�}||�� 7 }| �t|�| j� d S )N)r
r@ rA r4 r( �flushrG r )r �bz2cr r r r � testCompressk s
z BZ2CompressorTest.testCompressc C s. t � }|�d�}||�� 7 }| �|| j� d S )Nr )r
r4 r� rG r* )r r� r r r r �testCompressEmptyStringr s
z)BZ2CompressorTest.testCompressEmptyStringc C sl t � }d}d}x:| j|d |d d � }|s0P ||�|�7 }|d7 }qW ||�� 7 }| �t|�| j� d S )Nr r r? r )r
r( r4 r� rG r )r r� rc r rW r r r �testCompressChunks10x s z&BZ2CompressorTest.testCompressChunks10rZ r~ )�size�memusec C sp t � }d| }z|�|�}||�� 7 }W d d }X t�|�}z*| �t|�|� | �t|�d��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 ](}| �t�� t�t� |� 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� j 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 | � ttd� d S )N�* )r@ rA r )r r r r �test_Constructor� s z$BZ2DecompressorTest.test_Constructorc C s2 t � }| �t|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 ||�|�7 }|d7 }qW | �|| j� d S )Nr r r? r )r r) r rG r( )r r� rV rc rW r r r �testDecompressChunks10� s z*BZ2DecompressorTest.testDecompressChunks10c C s: t � }d}|�| 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�}| �t|jd� | �t|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 �|d ��|d�}z<||| d }t�|�}t� }|�|�}| �||k� W d d }d }d }X d S )Ni � � �littler )�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 ](}| �t�� t�t� |� 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| jd |� |d�� | �|j� | �t|d �|� |�|jd|d�� | �|j� | �t|d �|� |�|j| j|d � |d�� | �t|d �|� x2|j s�|�|jd|d�� | �t|d �|� q�W d�
|�}| �|| j � | �|j d� d S )NrZ �@ )�
max_length���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| jd d� dd�d� |�|�dd�� |�|�| jdd� d�� |�|�| jdd � �� | �d�|�| 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| jd d� dd�d� |�|�d�� |�|�| jdd� d�� |�|�| jdd� d�� |�|�| jdd � �� | �d�|�| 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 }|�|�| jd d� d�� |�|�| jdd� d�� |�|�| jdd � �� | �d�|�| 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 � }| �t|j| jd � | �t|j| jd � d S )N� )r r@ � Exceptionr r+ )r r� r r r � test_failure. s z BZ2DecompressorTest.test_failurec C sL t �td�}t� }|� }xtd�D ] }|�� 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__4 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�}| �t|�| j� d S )N)r r4 r( rG r )r r r r r r� ? s z#CompressDecompressTest.testCompressc C s t �d�}| �|| j� d S )Nr )r r4 rG r* )r rV r r r r� C s
z.CompressDecompressTest.testCompressEmptyStringc C s t �| j�}| �|| j� d S )N)r r r) rG r( )r rV r r r r� G s z%CompressDecompressTest.testDecompressc C s t �d�}| �|d� d S )Nr )r r rG )r rV r r r �testDecompressEmptyK s
z*CompressDecompressTest.testDecompressEmptyc C s t �| j�}| �|d� d S )Nr )r r r* rG )r rV r r r �testDecompressToEmptyStringO s z2CompressDecompressTest.testDecompressToEmptyStringc C s | � ttj| jd d� � d S )Ni����)r@ rB r r r) )r r r r �testDecompressIncompleteS s z/CompressDecompressTest.testDecompressIncompletec C s | � ttj| j� d S )N)r@ rJ r r r+ )r r r r �testDecompressBadDataV s z,CompressDecompressTest.testDecompressBadDatac C s& t �| jd �}| �|| jd � d S )NrL )r r r) rG r( )r rV r r r �testDecompressMultiStreamY s z0CompressDecompressTest.testDecompressMultiStreamc C s$ t �| j| j �}| �|| j� d S )N)r r r) r+ rG r( )r rV r r r �testDecompressTrailingJunk] s z1CompressDecompressTest.testDecompressTrailingJunkc C s, t �| jd | j �}| �|| jd � d S )NrL )r r r) r+ rG r( )r rV r r r �%testDecompressMultiStreamTrailingJunka 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 Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )�OpenTestzTest the open function.c O s t j||�S )N)r r1 )r �args�kwargsr r r r1 i s z
OpenTest.openc
C s x�dD ]�}|dkr t | j� | �| j|��}|�| j� W d Q R X t| jd�� }t|�� �}| �|| j� W d Q R X | �| jd��}| �|�� | j� W d Q R X | �| jd��}|�| j� W d Q R X t| jd��$}t|�� �}| �|| jd � W d Q R X qW d S )N)r7 �xbr r Zabr~ )r r r1 r8 r( r rE rG )r �moder; � file_datar r r �test_binary_modesl s
zOpenTest.test_binary_modesc
C s x�dD ]�}|dkr t | j� | �| j|��}|�| j� W d Q R X t| jd�� }t|�� �}| �|| j� W d Q R X | �| jd��}| �|�� | j� W d Q R X | �| jd��}|�| j� W d Q R X t| jd��$}t|�� �}| �|| jd � W d Q R X qW d S )N)rt �xr r r{ rx 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 �d�}|�dtj�}�x dD ]�}|dkr8t| j� | �| j|��}|�|� W d Q R X t| jd��$}t |�
� ��d�}| � ||� W d Q R X | �| jd��}| � |�
� |� W d Q R X | �| jd��}|�|� W d Q R X t| jd��(}t |�
� ��d�}| � ||d � W d Q R X q"W d S ) Nr� �
)�wt�xtr r �rtZatr~ ) 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|��}W d Q R X | �t�� | �| j|��}W d Q R X W d Q R X qW d S )N)r r r )r r r1 r@ �FileExistsError)r r r; r r r � test_x_mode� s
zOpenTest.test_x_modec C s� | � t| j�d��}| �|�� | j� W d Q R X | � t| j�d��}| �|�� | j� W d Q R X | j�d�}| � t| j�d��}| �|�� |� W d Q R X d S )Nr{ r r� r
)r1 r r) rG rE r( r )r r; rV r r r � test_fileobj� s zOpenTest.test_fileobjc C st | � t| j| jd� | � 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 �d�}|�dtj�}| j| jddd��}|�|� W d Q R X t| jd��$}t|� � ��d�}| �
||� W d Q R X | j| jddd��}| �
|� � |� 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 | � | jd��}|�d� W d Q R X | j | jdddd��}| �|�� 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 �d�}| j| jddd��}|�|� W d Q R X | j| jddd��}| �|�� |g� W d Q R X d S )Nr� r r
)r r
�
)r( r r1 r r8 rG rj )r rV r; r r r � test_newline� s
zOpenTest.test_newlineN)r" r# r$ r% r1 r r r r r r r r r r r r r r f s
r c C s t �ttttt� t �� 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 Zunittest�ior r r r� r. r� r� r r r� r rN r� �
import_moduler r r
r r
r ZTestCaser r6 r� r� r� r r r" r r r r �<module> s@
4 2 #(x