File: //usr/lib64/python3.9/site-packages/borg/testsuite/__pycache__/benchmark.cpython-39.opt-1.pyc
a
HZ�hx � @ s� d Z ddlZddlZddlmZmZ ddlmZ ejdd� �Z ejd d
gd�dd
� �Z
ejdddgd�dd� �Zejd dgd�dd� �Zdd� Z
dd� Zdd� Zdd� Zdd � Zd!d"� Zd#d$� Zd%d&� ZdS )'zM
Do benchmarks using pytest-benchmark.
Usage:
py.test --benchmark-only
� N� )� changedir�cmd� )�zerosc c s| |� dd� |� dd� |� dd� |� dd� |� dt|�d ��� |� d
t|�d��� t|�d��V |jd
d� d S )NZBORG_PASSPHRASEZ123456Z!BORG_CHECK_I_KNOW_WHAT_I_AM_DOINGZYESZ"BORG_DELETE_I_KNOW_WHAT_I_AM_DOINGZ*BORG_UNKNOWN_UNENCRYPTED_REPO_ACCESS_IS_OKZyesZ
BORG_KEYS_DIR�keysZBORG_CACHE_DIR�cacheZ
repositoryr �Zrec)Zsetenv�str�join�remove)�request�tmpdirZmonkeypatch� r �>/usr/lib64/python3.9/site-packages/borg/testsuite/benchmark.py�repo_url s r �noneZrepokey)�paramsc C s |dd| j |� |S )N�initz--encryption��param)r
r r r r r �repo s r Zsessionr �random)Zscoper c c s� d\}}|� d�}| j}|dkr*dd� }n|dkr<dd� }ntd��t|�D ]J}tt|�t|���d ��}|�||�� W d � qL1 s�0 Y qLt|�V |jd
d� d S )N)�
i@B �datar c S s t t�d | � S �N)�
memoryviewr ��sizer r r r + s ztestdata.<locals>.datar c S s
t �| �S r )�os�urandomr r r r r . s z&data_type must be 'random' or 'zeros'.�wbr r ) Zmktempr �
ValueError�range�openr
r �writer ) r
Ztmpdir_factory�countr �pZ data_typer �i�fr r r �testdata# s
.
r* �lz4c C s |d }|dd| j ||� |S )N�::test�create�
--compressionr )r
r r r* Zarchive_urlr r r �archive9 s r/ c C s"