Your IP : 216.73.216.52
a
XC?hʓ � @ sJ d dl mZ d dlmZmZ d dlZd dlZd dlmZm Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZd dlZd dlZd dlZd dlZd dlmZ d dlZd dlZe�d�Zd dlmZmZmZ dadd � Z G 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�rFe�&� dS )� )�support)�
bigmemtest�_4GN)�BytesIO�DEFAULT_BUFFER_SIZE)�unlink�bz2)�BZ2File�
BZ2Compressor�BZ2Decompressorc C s8 t d u rtt�d��a t r*tjdg| d�S t�| �S d S )NZbunzip2)�input)�has_cmdline_bunzip2�bool�shutilZwhich�
subprocessZ check_outputr �
decompress)�data� r �2/opt/alt/python39/lib64/python3.9/test/test_bz2.py�ext_decompress s
r c @ s� e Zd ZdZg d�Zd�e�ZdZdZdZ dZ
e d�Z e
�
ej�e
�ej�e��d ��D ]TZeed
��*Ze
e�ee �e
d � �7 Z
W d � n1 s�0 Y e
dkrX q�qXeje d d
�Zdd� Zdd� 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� �Z
compresslevelc C s t �� \}| _t�|� d S �N)�tempfileZmkstemp�filename�os�close)�self�fdr r r �setUpP s zBaseTest.setUpc C s t | j� d S r )r r �r r r r �tearDownT s zBaseTest.tearDown) �__name__�
__module__� __qualname__�__doc__�
TEXT_LINES�join�TEXT�DATA�
EMPTY_DATA�BAD_DATAZ test_size� bytearray�BIG_TEXT�globr �path�escape�dirname�__file__Zfname�openZfh�readinto�
memoryviewr �compress�BIG_DATAr" r$ r r r r r $ s
& 8r 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>dzd{� Z?d|S )~� BZ2FileTestzTest the BZ2File class.r r c C sJ t | jd��*}|�| j| � |�|� W d � n1 s<0 Y d S )N�wb)r6 r �writer, )r �streams�suffix�fr r r �createTempFile[ s zBZ2FileTest.createTempFilec C s� | � ttd� | � tttjd� | � tttjd� | � tttjd� | j tttjdd� | j tttjdd� | � tttjdd � d S )
Ngw��/�^@�zZrxZrbtr r �
�r� )� assertRaises� TypeErrorr �
ValueErrorr �devnullr# r r r � testBadArgs` s zBZ2FileTest.testBadArgsc C sZ | � � t| j��4}| �t|jt� � | �|�� | j� W d � n1 sL0 Y d S r � rA r r rF rG �read�float� assertEqualr+ �r �bz2fr r r �testReadk s zBZ2FileTest.testReadc C sL | j d| jd� t| j�� }| �t|j� W d � n1 s>0 Y d S )Nr �r> r? )rA r. r r rF �OSErrorrL rO r r r �testReadBadFileq s zBZ2FileTest.testReadBadFilec C sb | j dd� t| j��8}| �t|jt� � | �|�� | jd � W d � n1 sT0 Y d S �N� �r> rK rO r r r �testReadMultiStreamv s z BZ2FileTest.testReadMultiStreamc C s� t j}t| j�t _zh| jdd� t| j��8}| �t|j t
� � | � |� � | j d � W d � n1 sh0 Y W |t _n|t _0 d S rU )
� _compression� BUFFER_SIZE�lenr, rA r r rF rG rL rM rN r+ )r Z buffer_sizerP r r r �testReadMonkeyMultiStream| s 6z%BZ2FileTest.testReadMonkeyMultiStreamc C sN | j | jd� t| j��"}| �|�� | j� W d � n1 s@0 Y d S )N)r? �rA r. r r rN rL r+ rO r r r �testReadTrailingJunk� s z BZ2FileTest.testReadTrailingJunkc C sT | j d| jd� t| j��&}| �|�� | jd � W d � n1 sF0 Y d S )NrV rR r] rO r r r � testReadMultiStreamTrailingJunk� s z+BZ2FileTest.testReadMultiStreamTrailingJunkc C sZ | � � t| j��4}| �t|jt� � | �|�d�d� W d � n1 sL0 Y d S )Nr r )rA r r rF rG rL rM rN rO r r r � testRead0� s zBZ2FileTest.testRead0c C sb | � � t| j��<}d}|�d�}|s(q2||7 }q| �|| j� W d � n1 sT0 Y d S )Nr rC �rA r r rL rN r+ �r rP �text�strr r r �testReadChunk10� s
zBZ2FileTest.testReadChunk10c C sj | j dd� t| j��@}d}|�d�}|s,q6||7 }q | �|| jd � W d � n1 s\0 Y d S )NrV rW r rC ra rb r r r �testReadChunk10MultiStream� s
z&BZ2FileTest.testReadChunk10MultiStreamc C sR | � � t| j��,}| �|�d�| jd d� � W d � n1 sD0 Y d S )N�d )rA r r rN rL r+ rO r r r � testRead100� s zBZ2FileTest.testRead100c C sr | � � t| j��L}|�� }| �t|�d� | �| j�|�� | � |�
� | j� W d � n1 sd0 Y d S �Nr ) rA r r �peek�assertNotEqualr[ �
assertTruer+ �
startswithrN rL )r rP �pdatar r r �testPeek� s zBZ2FileTest.testPeekc C s� | � � t| j���}d}t|�}| �|�|�|� | �|| jd |� � t| j�| }tt| j��}| �|�|�|� | �|d |� | j| d � � W d � n1 s�0 Y d S )N� )rA r r r/ rN r7 r+ r[ )r rP �n�br r r � testReadInto� s zBZ2FileTest.testReadIntoc C sb | � � t| j��<}| �t|jd � | jD ]}| �|�� |� q*W d � n1 sT0 Y d S r �rA r r rF rG �readliner) rN �r rP �liner r r � testReadLine� s
zBZ2FileTest.testReadLinec C sj | j dd� t| j��@}| �t|jd � | jd D ]}| �|�� |� q2W d � n1 s\0 Y d S rU rt rv r r r �testReadLineMultiStream� s
z#BZ2FileTest.testReadLineMultiStreamc C sX | � � t| j��2}| �t|jd � | �|�� | j� W d � n1 sJ0 Y d S r �rA r r rF rG � readlinesrN r) rO r r r �
testReadLines� s zBZ2FileTest.testReadLinesc C s` | j dd� t| j��6}| �t|jd � | �|�� | jd � W d � n1 sR0 Y d S rU rz rO r r r �testReadLinesMultiStream� s z$BZ2FileTest.testReadLinesMultiStreamc C sL | � � t| j��&}| �tt|��| j� W d � n1 s>0 Y d S r �rA r r rN �list�iterr) rO r r r � testIterator� s zBZ2FileTest.testIteratorc C sT | j dd� t| j��*}| �tt|��| jd � W d � n1 sF0 Y d S rU r~ rO r r r �testIteratorMultiStream� s z#BZ2FileTest.testIteratorMultiStreamc C s: | � � t| j�}|�� | �tt|� | �t|j� d S r )rA r r r rF rH �nextr{ rO r r r �testClosedIteratorDeadlock� s
z&BZ2FileTest.testClosedIteratorDeadlockc C s� t | jd��*}| �t|j� |�| j� W d � n1 s<0 Y t| jd��&}| �t|� � �| j� W d � n1 s~0 Y d S )N�wr �
r r rF rG r= r+ r6 rN r rL �r rP r@ r r r � testWrite� s
*zBZ2FileTest.testWritec C s� t | jd��H}d}| j|d |d d � }|s2qF|�|� |d7 }qW d � n1 sZ0 Y t| jd��&}| �t|�� �| j� W d � n1 s�0 Y d S )Nr� r rC r r )r r r+ r= r6 rN r rL )r rP rq rd r@ r r r �testWriteChunks10� s
(z BZ2FileTest.testWriteChunks10c C s� t j| jdd�}t| jddd�� }|�| j� W d � n1 sB0 Y t| jd�� }| �|�� |� W d � n1 s~0 Y d S )NrV r r� r ) r r9 r+ r r r= r6 rN rL )r ZexpectedrP r@ r r r � testWriteNonDefaultCompressLevel s
*z,BZ2FileTest.testWriteNonDefaultCompressLevelc C s� t | jd��*}| �t|j� |�| j� W d � n1 s<0 Y | �t|jdg� t| jd��&}| �t |�
� �| j � W d � n1 s�0 Y d S )Nr� �ar ) r r rF rG �
writelinesr) rH r6 rN r rL r+ r� r r r �testWriteLines s *zBZ2FileTest.testWriteLinesc C s� t | jd��}|�d� W d � n1 s,0 Y t | jd��2}| �t|jd� | �t|jdg� W d � n1 sz0 Y d S )Nr� s abcrD � a)r r r= rF rS r� rO r r r � testWriteMethodsOnReadOnlyFile s
(z*BZ2FileTest.testWriteMethodsOnReadOnlyFilec C s� t | jd��*}| �t|j� |�| j� W d � n1 s<0 Y t | jd��*}| �t|j� |�| j� W d � n1 s�0 Y t| jd��*}| �t|� � �| jd � W d � n1 s�0 Y d S )Nr� r� r � r� r� r r r �
testAppend! s **zBZ2FileTest.testAppendc C sh | � � t| j��B}| �t|j� |�d� | �|�� | jdd � � W d � n1 sZ0 Y d S �N� ) rA r r rF rG �seekrN rL r+ rO r r r �testSeekForward+ s
zBZ2FileTest.testSeekForwardc C sv | j dd� t| j��L}| �t|j� |�t| j�d � | �|� � | jdd � � W d � n1 sh0 Y d S )Nr� rW r� )
rA r r rF rG r� r[ r+ rN rL rO r r r � testSeekForwardAcrossStreams2 s
z(BZ2FileTest.testSeekForwardAcrossStreamsc C sf | � � t| j��@}|�d� |�dd� | �|�� | jdd � � W d � n1 sX0 Y d S �Ni� �j���r i^ )rA r r rL r� rN r+ rO r r r �testSeekBackwards9 s
z BZ2FileTest.testSeekBackwardsc C s� | j dd� t| j��f}t| j�d }|dkrB|t|�|��8 }q&|�dd� | �|�� | jdd � | j � W d � n1 s�0 Y d S )Nr� rW rg r r� r i���)rA r r r[ r+ rL r� rN )r rP Zreadtor r r � testSeekBackwardsAcrossStreams@ s z*BZ2FileTest.testSeekBackwardsAcrossStreamsc C sf | � � t| j��@}|�dd� | �|�� | jt| j�d d � � W d � n1 sX0 Y d S )Nr� r� r� )rA r r r� rN rL r+ r[ rO r r r �testSeekBackwardsFromEndI s z$BZ2FileTest.testSeekBackwardsFromEndc C sd | j dd� t| j��:}|�dd� | �|�� | jd dd � � W d � n1 sV0 Y d S )Nr� rW i���)rA r r r� rN rL r+ rO r r r �%testSeekBackwardsFromEndAcrossStreamsO s z1BZ2FileTest.testSeekBackwardsFromEndAcrossStreamsc C sf | � � t| j��@}|�d� | �|�� t| j�� | �|�� d� W d � n1 sX0 Y d S �N��I r � rA r r r� rN �tellr[ r+ rL rO r r r �testSeekPostEndU s
zBZ2FileTest.testSeekPostEndc C sn | j dd� t| j��D}|�d� | �|�� t| j�d � | �|�� d� W d � n1 s`0 Y d S �NrV rW r� r r� rO r r r �testSeekPostEndMultiStream\ s
z&BZ2FileTest.testSeekPostEndMultiStreamc C sp | � � t| j��J}|�d� |�d� | �|�� t| j�� | �|�� d� W d � n1 sb0 Y d S r� r� rO r r r �testSeekPostEndTwicec s
z BZ2FileTest.testSeekPostEndTwicec C sx | j dd� t| j��N}|�d� |�d� | �|�� t| j�d � | �|�� d� W d � n1 sj0 Y d S r� r� rO r r r � testSeekPostEndTwiceMultiStreamk s
z+BZ2FileTest.testSeekPostEndTwiceMultiStreamc C sb | � � t| j��<}|�d� | �|�� d� | �|�� | j� W d � n1 sT0 Y d S )Nr� r �rA r r r� rN r� rL r+ rO r r r �testSeekPreStarts s
z BZ2FileTest.testSeekPreStartc C sj | j dd� t| j��@}|�d� | �|�� d� | �|�� | jd � W d � n1 s\0 Y d S )Nr� rW r� r r� rO r r r �testSeekPreStartMultiStreamz s
z'BZ2FileTest.testSeekPreStartMultiStreamc C sz | � � t| jd��D}t|�}z | �|�� |�� � W |�� n
|�� 0 W d � n1 s^0 Y | �t|j� d S )Nr ) rA r6 r r rN �filenor rF rH )r ZrawfrP r r r �
testFileno� s 2zBZ2FileTest.testFilenoc C s� t t| j��}z0| �|�� � |�� | �|�� � W |�� n
|�� 0 | �t|j� t t� d�}z| � |�� � W |�� n
|�� 0 | �t|j� t| j�}dd� |_t |�}z| � |�� � W |�� n
|�� 0 | �t|j� d S )Nr� c S s dS )NFr r r r r �<lambda>� r z*BZ2FileTest.testSeekable.<locals>.<lambda>)
r r r, rl �seekablerL r rF rH � assertFalse)r rP �srcr r r � testSeekable� s&
zBZ2FileTest.testSeekablec C s� t t| j��}z0| �|�� � |�� | �|�� � W |�� n
|�� 0 | �t|j� t t� d�}z| � |�� � W |�� n
|�� 0 | �t|j� d S �Nr� )
r r r, rl �readablerL r rF rH r� rO r r r � testReadable� s zBZ2FileTest.testReadablec C s� t t| j��}z0| �|�� � |�� | �|�� � W |�� n
|�� 0 | �t|j� t t� d�}z| � |�� � W |�� n
|�� 0 | �t|j� d S r� )
r r r, r� �writablerL r rF rH rl rO r r r � testWritable� s zBZ2FileTest.testWritablec C s& | � � td�D ]}t| j�}~qd S )Ni' )rA �ranger r )r �i�or r r � testOpenDel� s
zBZ2FileTest.testOpenDelc C s | � ttd� d S )Nz
/non/existent)rF rS r r# r r r �testOpenNonexistent� s z BZ2FileTest.testOpenNonexistentc C s� d}t | jd��}|�|� W d � n1 s00 Y t| j��}|�� }W d � n1 sb0 Y | �|dg� t| j�� }t|�� �}W d � n1 s�0 Y | �|dg� d S )Ns, BZh91AY&SY�b�] � !�h3M<]��BCe�%tr< s Test)r6 r r= r r{ rN r )r r r@ rP �linesZxlinesr r r �testReadlinesNoNewline� s ( & *z"BZ2FileTest.testReadlinesNoNewlinec C s� d }t | jd��}|�d� W d � n1 s00 Y t | jd�}|�� z(|� W d � n1 sj0 Y W n ty� Y n 0 | �d� z8t | jd��}dd W d � n1 s�0 Y W n ty� Y n 0 | �d� d S )Nr< 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= r rH Zfail�ZeroDivisionError�r r@ r r r �testContextProtocol� s" ( "
* z BZ2FileTest.testContextProtocolc s� dd �d}t | jd��^���fdd�� � fdd�t|�D �}t�|�� W d � n1 s^0 Y W d � n1 s|0 Y d S ) N� 1i rC r< c s t d�D ]} ��� � qd S �NrV )r� r= )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>� r 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��R}|�|� | �t|�| jd |d � � | �|�� | j|d � � W d � n1 s�0 Y t| j��D}|� � | �t|�| jd � | �|� � | jd � W d � n1 s�0 Y t| j��6}|�
� | � t t|� | �|�
� g � W d � n1 �s00 Y d S )Nr r� r )
rA r[ r) r r rL rN r� r+ ru r{ rF �
StopIteration)r ZlinelenZhalflenrP r r r �testMixedIterationAndReads� s
8 4 z&BZ2FileTest.testMixedIterationAndReadsc C s� d}d}t | jd��}|�|� W d � n1 s40 Y t | jd��}|�|� W d � n1 sj0 Y t | j��$}| �|�� || � W d � n1 s�0 Y d S )Ns� foofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoos� barbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarr� r� )r r r= rN rL )r Zdata1Zdata2rP r r r �testMultiStreamOrdering s (( z#BZ2FileTest.testMultiStreamOrderingc C s� | j }z|�d�}W n ty0 | �d� Y n0 t|d�� }|�| j� W d � n1 s^0 Y t|d��"}| �|�� | j� W d � n1 s�0 Y t|d��"}| �|�� | j� W d � n1 s�0 Y d S )N�asciiz%Temporary file name needs to be ASCIIr< r ) r �encode�UnicodeEncodeErrorZskipTestr r= r, rN rL )r Z str_filenameZbytes_filenamer@ r r r �testOpenBytesFilename s * 0 z!BZ2FileTest.testOpenBytesFilenamec C s� t �| j�}t|d�� }|�| j� W d � n1 s80 Y t|d��"}| �|�� | j� W d � n1 st0 Y d S )Nr< r )�pathlib�Pathr r r= r, rN rL )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 r9 �intZ
assertLessr[ rY rZ r r rN rL r �assertLessEqualZ_buffer�rawr� )r ZbombZdecompZ
max_decompr r r �testDecompressLimited. s �z!BZ2FileTest.testDecompressLimitedc C s� t | j��h}t|��4}| �t|jt� � | �|�� | j� W d � n1 sN0 Y | � |j
� W d � n1 sx0 Y d S r ) r r, r rF rG rL rM rN r+ r� �closed�r �biorP r r r �testReadBytesIO< s
0zBZ2FileTest.testReadBytesIOc C s� t | j��t}t|��L}|�� }| �t|�d� | �| j�|�� | � |�
� | j� W d � n1 sf0 Y W d � n1 s�0 Y d S ri ) r r, r rj rk r[ rl r+ rm rN rL )r r� rP rn r r r �testPeekBytesIOC s
zBZ2FileTest.testPeekBytesIOc C s� t � �v}t|d��*}| �t|j� |�| j� W d � n1 sB0 Y | �t|�� �| j� | � |j
� W d � n1 s�0 Y d S r� ) r r rF rG r= r+ rN r �getvaluer� r� r� r r r �testWriteBytesIOK s *z BZ2FileTest.testWriteBytesIOc C s� t | j��j}t|��B}| �t|j� |�d� | �|�� | jdd � � W d � n1 s\0 Y W d � n1 sz0 Y d S r� ) r r, r rF rG r� rN rL r+ r� r r r �testSeekForwardBytesIOS s
z"BZ2FileTest.testSeekForwardBytesIOc C s� t | j��h}t|��@}|�d� |�dd� | �|�� | jdd � � W d � n1 sZ0 Y W d � n1 sx0 Y d S r� )r r, r rL r� rN r+ r� r r r �testSeekBackwardsBytesIOZ s
z$BZ2FileTest.testSeekBackwardsBytesIOc C s� | j d d� }tt|��� }| �t|j� W d � n1 s>0 Y tt|���:}| �|�t| j��| j� | �t|jd� W d � n1 s�0 Y t d�D ]H}tt|d |� ��� }| �t|jd� W d � q�1 s�0 Y q�d S )N���r � )
r, r r rF �EOFErrorrL rN r[ r+ r� )r Z truncatedr@ r� r r r �test_read_truncateda s ,. z BZ2FileTest.test_read_truncatedc C sp t � dg d��}t|�|j }tt� d��2}| �|�|�|� | �|�� |� W d � n1 sb0 Y d S )N�Q)r r� rE � rV r� )�arrayr[ �itemsizer r rN r= r� )r �qZLENGTHr@ r r r �test_issue44439n s
zBZ2FileTest.test_issue44439N)r r )@r% r& r' r( rA rJ rQ rT rX r\ r^ r_ r` re rf rh ro rs rx ry 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� r� r� r� r� r� r� r r r r r; X sz
r; c @ sJ e Zd Zdd� Zdd� Zdd� Zejee d dd �d
d � ��Z
d d
� Z dS )�BZ2CompressorTestc C sB t � }| �t|j� |�| j�}||�� 7 }| �t|�| j� d S r )r
rF rG r9 r+ �flushrN r �r �bz2cr r r r � testCompressx s
z BZ2CompressorTest.testCompressc C s. t � }|�d�}||�� 7 }| �|| j� d S �Nr )r
r9 r� rN r- r� r r r �testCompressEmptyString s
z)BZ2CompressorTest.testCompressEmptyStringc C sh t � }d}d}| j|d |d d � }|s.qF||�|�7 }|d7 }q||�� 7 }| �t|�| j� d S )Nr r rC r )r
r+ r9 r� rN r )r r� rq r rd r r r �testCompressChunks10� s
z&BZ2CompressorTest.testCompressChunks10rg r� ��sizeZmemusec C sx t � }d| }z |�|�}||�� 7 }W d }nd }0 t�|�}z.| �t|�|� | �t|�d��d� W d }nd }0 d S )N� xr )r
r9 r� r r rN r[ �strip)r r� r� r �
compressedr r r �testCompress4G� s
z BZ2CompressorTest.testCompress4Gc C sP t tjd �D ]<}| �t�� t�t� |� W d � q1 s@0 Y qd S �Nr )r� �pickle�HIGHEST_PROTOCOLrF rG �dumpsr
�r �protor r r �
testPickle� s z BZ2CompressorTest.testPickleN) r% r& r' r� r� r� r �skip_if_pgo_taskr r r� r� r r r r r� w s
r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zej e
e d d d
�dd� ��Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zejd d � �Zd S ) �BZ2DecompressorTestc C s | � ttd� d S )N�* )rF rG r r# r r r �test_Constructor� s z$BZ2DecompressorTest.test_Constructorc C s2 t � }| �t|j� |�| j�}| �|| j� d S r )r rF rG r r, rN r+ �r �bz2drc r r r �testDecompress� s z"BZ2DecompressorTest.testDecompressc C sX t � }d}d}| j|d |d d � }|s.qF||�|�7 }|d7 }q| �|| j� d S )Nr r rC r )r r, r rN r+ )r r rc rq rd 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, rN r+ � unused_data)r r r rc r r r �testDecompressUnusedData� s
z,BZ2DecompressorTest.testDecompressUnusedDatac C s6 t � }|�| j�}| �t|jd� | �t|jd� d S )Ns anythingr )r r r, rF r� r r r r � testEOFError� s z BZ2DecompressorTest.testEOFErrorrg gffffff
@r� c C sj d}t �|�}zH||| d }t�|�}t� }|�|�}| �||k� W d }d }d }nd }d }d }0 d S )Ni � r )�randomZ randbytesr r9 r r rl )r r� Z blocksize�blockr r� r Z decompressedr r r �testDecompress4G� s
�z$BZ2DecompressorTest.testDecompress4Gc C sP t tjd �D ]<}| �t�� t�t� |� W d � q1 s@0 Y qd S r� )r� r� r� rF rG 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 �|� |j s�|�|jd|d�� | �t|d �|� q�d�
|�}| �|| j � | �|j d� d S )Nrg �@ ��
max_length���r )
r r[ r: �appendr r� Z needs_inputrN r� �eofr* r0 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 )Nrg r r r r� �i � �r rN 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� �, 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 rV r r )r r r r, rN 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 rF � Exceptionr r. )r r r r r � test_failure= s z BZ2DecompressorTest.test_failurec C sH t �td�}t� }|� }td�D ] }|�� q | j|� | ddd� d S )N�gettotalrefcountrg r rC )Zdelta)r Z
get_attribute�sysr r� �__init__ZassertAlmostEqual)r r" r Z refs_beforer� r r r �test_refleaks_in___init__C 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 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 r )r r9 r+ rN r )r r r r r r� N s z#CompressDecompressTest.testCompressc C s t �d�}| �|| j� d S r� )r r9 rN r- �r rc r r r r� R s
z.CompressDecompressTest.testCompressEmptyStringc C s t �| j�}| �|| j� d S r �r r r, rN r+ r' r r r r V s z%CompressDecompressTest.testDecompressc C s t �d�}| �|d� d S r� )r r rN r' r r r �testDecompressEmptyZ s
z*CompressDecompressTest.testDecompressEmptyc C s t �| j�}| �|d� d S r� )r r r- rN r' r r r �testDecompressToEmptyString^ s z2CompressDecompressTest.testDecompressToEmptyStringc C s | � ttj| jd d� � d S )Nr� )rF rH r r r, r# r r r �testDecompressIncompleteb s z/CompressDecompressTest.testDecompressIncompletec C s | � ttj| j� d S r )rF rS r r r. r# r r r �testDecompressBadDatae s z,CompressDecompressTest.testDecompressBadDatac C s& t �| jd �}| �|| jd � d S r� r( r' r r r �testDecompressMultiStreamh s z0CompressDecompressTest.testDecompressMultiStreamc C s$ t �| j| j �}| �|| j� d S r �r r r, r. rN r+ r' r r r �testDecompressTrailingJunkl s z1CompressDecompressTest.testDecompressTrailingJunkc C s, t �| jd | j �}| �|| jd � d S r� r. r' r r r �%testDecompressMultiStreamTrailingJunkp s z<CompressDecompressTest.testDecompressMultiStreamTrailingJunkN)
r% r& r' r� r� r r) r* r+ r, r- r/ r0 r r r r r& M 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|i |��S r )r r6 )r �args�kwargsr r r r6 x s z
OpenTest.openc C sj dD �]^}|dkr t | j� | �| j|�� }|�| j� W d � n1 sL0 Y t| jd��*}t|�� �}| �|| j� W d � n1 s�0 Y | �| jd��"}| �|�� | j� W d � n1 s�0 Y | �| jd�� }|�| j� W d � n1 �s0 Y t| jd��.}t|�� �}| �|| jd � W d � q1 �sZ0 Y qd S )N)r< �xbr4 r Zabr� �r r r6 r= r+ r rL rN �r �moder@ � file_datar r r �test_binary_modes{ s
* ,0, zOpenTest.test_binary_modesc C sj dD �]^}|dkr t | j� | �| j|�� }|�| j� W d � n1 sL0 Y t| jd��*}t|�� �}| �|| j� W d � n1 s�0 Y | �| jd��"}| �|�� | j� W d � n1 s�0 Y | �| jd�� }|�| j� W d � n1 �s0 Y t| jd��.}t|�� �}| �|| jd � W d � q1 �sZ0 Y qd S )N)r� �xr: r rD r� r� r5 r6 r r r �test_implicit_binary_modes� s
* ,0, z#OpenTest.test_implicit_binary_modesc C s� | j �d�}|�dtj�}dD �]`}|dkr6t| j� | �| j|��}|�|� W d � n1 sd0 Y t| jd��.}t |�
� ��d�}| � ||� W d � n1 s�0 Y | �| jd�� }| � |�
� |� W d � n1 s�0 Y | �| jd��}|�|� W d � n1 �s&0 Y t| jd��2}t |�
� ��d�}| � ||d � W d � q 1 �sv0 Y q d S ) Nr� �
)�wt�xtr> r �rtZatr� ) r+ �decode�replacer �linesepr r r6 r= r rL rN )r rc �text_native_eolr7 r@ r8 r r r �test_text_modes� s"
(*.*zOpenTest.test_text_modesc
C s� dD ]�}t | j� | �| j|��}W d � n1 s60 Y | �t��> | �| j|��}W d � n1 sp0 Y W d � q1 s�0 Y qd S )N)r: r4 r> )r r r6 rF �FileExistsError)r r7 r@ r r r � test_x_mode� s
zOpenTest.test_x_modec C s� | � t| j�d��"}| �|�� | j� W d � n1 s:0 Y | � t| j�d��"}| �|�� | j� W d � n1 s~0 Y | j�d�}| � t| j�d�� }| �|�� |� W d � n1 s�0 Y d S )NrD r r� r? )r6 r r, rN rL r+ r@ )r r@ rc r r r � test_fileobj� s 00 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)rF rH r6 r r# r r r �test_bad_params� s
�
� � � �zOpenTest.test_bad_paramsc C s� | j �d�}|�dtj�}| j| jddd��}|�|� W d � n1 sL0 Y t| jd��.}t|� � ��d�}| �
||� W d � n1 s�0 Y | j| jddd�� }| �
|� � |� W d � n1 s�0 Y d S )Nr� r< r= z utf-16-lerH r r? ) r+ r@ rA r rB r6 r r= r rL rN )r rc rC r@ r8 r r r �
test_encoding� s (*zOpenTest.test_encodingc C s� | � | jd��}|�d� W d � n1 s.0 Y | j | jdddd�� }| �|�� d� W d � n1 sr0 Y d S )Nr< s foo�barr? r� rJ )rI rK Zfoobar)r6 r r= rN rL r� r r r �test_encoding_error_handler� s
(z$OpenTest.test_encoding_error_handlerc C s� | j �d�}| j| jddd��}|�|� W d � n1 s>0 Y | j| jddd��"}| �|�� |g� W d � n1 s�0 Y d S )Nr� r= r< rL r? �
)r+ r@ r6 r r= rN r{ )r rc r@ r r r � test_newline� s
(zOpenTest.test_newlineN)r% r& r' r( r6 r9 r; rD rF rG rN rO rP rR r r r r r1 u s
r1 c C s t �� d S r )r Z
reap_childrenr r r r �tearDownModule� s rS �__main__)'�testr Z test.supportr r r� Zunittest�ior r r r� r1 r r� r
r r r� r rY r# �
import_moduler r r
r r
r ZTestCaser r; r� r r& r1 rS r% �mainr r r r �<module> sD
4 #3 $(x