| Current File : //usr/lib64/python3.6/site-packages/borg/testsuite/__pycache__/compress.cpython-36.pyc |
3
y��b� � @ s� d dl Z d dlZy d dlZW n ek
r4 dZY nX d dlZddlmZmZmZm Z m
Z
m Z m Z m
Z
mZ ed"�Zdd Zeddd �Zd
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Z d d � Z d d � Z d d!� Z dS )#� N� ) �get_compressor�
Compressor�CompressionSpec�CNONE�ZLIB�LZ4�LZMA�ZSTD�Auto� s fooooooooobaaaaaaaar�
�zlib� )�name�levelc
C sl t dd�} t| t�st�t dd�} t| t�s0t�t dd�} t| t�sHt�tjt�� t dd� W d Q R X d S )N�none)r �lz4r Zfoobar) r �
isinstancer �AssertionErrorr r �pytest�raises�KeyError)�c� r � /usr/lib64/python3.6/compress.py�test_get_compressor s
r c C sb t dd�} | jt�}t|�tt�ks(t�t|ks4t�t| j|�ksFt�ttf t�j|�ks^t�d S )Nr )r )r �compress�data�lenr �
decompressr �params)r �cdatar r r �
test_cnull s
r# c C sV t dd�} | jt�}t|�tt�k s(t�t| j|�ks:t�ttf t�j|�ksRt�d S )Nr )r )r r r r r r r r! )r r" r r r �test_lz4& s
r$ c C s\ t jd �d } t| �d ks t�tdd�}|j| �}t|�t| �ksFt�| |j|�ksXt�d S ) N� r � r
�2 r )r i i P i i )�os�urandomr r r r r )r r r" r r r �test_lz4_buffer_allocation. s
r* c C sV t dd�} | jt�}t|�tt�k s(t�t| j|�ks:t�ttf t�j|�ksRt�d S )Nr )r )r r r r r r r r! )r r" r r r � test_zlib8 s
r+ c C sh t d krtjd� tdd�} | jt�}t|�tt�k s:t�t| j|�ksLt�tt f t
�j|�ksdt�d S )NzNo lzma support found.�lzma)r ) r, r �skipr r r r r r r r! )r r" r r r � test_lzma@ s
r. c C sV t dd�} | jt�}t|�tt�k s(t�t| j|�ks:t�ttf t�j|�ksRt�d S )N�zstd)r )r r r r r r r r! )r r" r r r � test_zstdJ s
r0 c C sP t jt�� tf t�jd� W d Q R X t jt�� tf t�jd� W d Q R X d S )Ns ��totalcraps notreallyzlib)r r �
ValueErrorr r! r r r r r �test_autodetect_invalidR s r2 c C sv xpt d�D ]d} td| d�}|jt�}tjt| �}||ks<t�|j|�}t|ksRt�tf t�j|�}t|ks
t�q
W d S )Nr
r )r r ) �ranger r r r r r r r! )r r Zcdata1Zcdata2Zdata2r r r �test_zlib_compatY s
r4 c
C s� t dd�t dd�t ddd�t ddd�t dd d�t dd
d�t dd d�g} trf| t d d d�t d d
d�g7 } x.| D ]&}tf |�}t|j|jt��kslt�qlW d S )
Nr )r r r/ � )r r � r r r � r, )�dictr, r r r r r )Z params_listr! r r r r �test_compressorg s
r9 c C s� t d�j} t d�j}t d�j}td�}| j|�}|j|�}|j|�}t|�t|� }|dk rjtj|�tkntsrt �d}| j|�}tj|�t
ks�t �d S )Nz auto,zlib,9r zzlib,9i� g�G�z��?s ���-O�i��1�8\�b��E���hm�~) r �
compressor�bytesr r r Zdetectr r r r ) Zcompressor_auto_zlibZcompressor_lz4Zcompressor_zlibr Zcompressed_auto_zlibZcompressed_lz4Zcompressed_zlibZratioZ
compressedr r r � test_auto} s
r< c C s t jt�� td� W d Q R X ttd�jt�s2t�ttd�jt�sFt�td�j} t| t �s^t�| j
dkslt�td�j} t| t �s�t�| j
dks�t�td�j} t| t �s�t�| j
d ks�t�t jt�� td
� W d Q R X td �j}t|t �s�t�|j
dks�t�td �j}t|t ��st�|j
dk�s&t�td
�j}t|t ��s@t�|j
d k�sPt�td�j}t|t ��sjt�|j
dk�szt�td�j}t|t ��s�t�|j
dk�s�t�td�j}t|t ��s�t�|j
dk�s�t�t jt�� td� W d Q R X t jt�� td� W d Q R X d S )N� r r r r zzlib,0r zzlib,9r7 zzlib,9,invalidr, zlzma,0zlzma,9r/ r6 zzstd,1r5 zzstd,22� zlzma,9,invalidZinvalid)
r r r1 r r r: r r r r r r r
)r r, r/ r r r �test_compression_specs� sJ
r? i ) r( r r, � ImportErrorr r r r r r r r r r
r r; �bufferr r8 r! r r# r$ r* r+ r. r0 r2 r4 r9 r< r? r r r r �<module> s,
,