implementing results csv and local to local
This commit is contained in:
36
benchmark.py
Normal file
36
benchmark.py
Normal file
@@ -0,0 +1,36 @@
|
||||
# copyright: Canonical
|
||||
"Run the bench cases (remote to local)."
|
||||
import datetime
|
||||
import subprocess
|
||||
from pathlib import Path
|
||||
|
||||
from loguru import logger as log
|
||||
|
||||
|
||||
def run_single_bench(
|
||||
runnum,
|
||||
test_case,
|
||||
files_type,
|
||||
source,
|
||||
target,
|
||||
results,
|
||||
):
|
||||
log.info(f"Syncing {runnum} {test_case} {files_type}")
|
||||
start = datetime.datetime.now()
|
||||
subprocess.run(
|
||||
[
|
||||
"rsync",
|
||||
"-a",
|
||||
f"{source}",
|
||||
f"{target}",
|
||||
],
|
||||
capture_output=True,
|
||||
# shell=True,
|
||||
)
|
||||
time = datetime.datetime.now() - start
|
||||
size = sum(f.stat().st_size for f in Path(target).glob("**/*") if f.is_file())
|
||||
size_MB = size / 1000 / 1000 / 8
|
||||
speed = size_MB / time.total_seconds()
|
||||
log.info(f"{files_type} - {speed} MB/s")
|
||||
subprocess.run(["rm", "-rf", target])
|
||||
results.add([runnum, test_case, files_type, speed, time.total_seconds(), size_MB])
|
||||
Reference in New Issue
Block a user