From 4d485c82130a8156b47efe844a71bfd1a9f6d6cd Mon Sep 17 00:00:00 2001 From: alex Date: Wed, 6 Sep 2023 15:34:50 +0200 Subject: [PATCH] add intermediary results ssave --- results_tools.py | 7 +++++-- run.py | 2 +- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/results_tools.py b/results_tools.py index f855308..cf534c4 100644 --- a/results_tools.py +++ b/results_tools.py @@ -7,17 +7,20 @@ from uuid import uuid4 class Results: def __init__(self, *args, **kwargs): self.result_list = [["Run num", "test_case", "files", "MB/s", "time", "full size"]] + self.name = f"results{uuid4()}.csv" - def add(self, row: list) -> None: + def add(self, row: list, save: bool = True) -> None: """Add a row to the results lists. Args: row (list): a row of results. """ self.result_list.append(row) + if save: + self.save() def save_to_csv(self) -> None: """Save results to csv file in the current directory.""" - with open(f"results{uuid4()}.csv", "w") as f: + with open(self.name, "w") as f: cw = csv.writer(f) cw.writerows(self.result_list) diff --git a/run.py b/run.py index 6e4e038..6fe6686 100644 --- a/run.py +++ b/run.py @@ -21,7 +21,7 @@ def local(): click.echo("Running locally") for runnum in range(50): for files_type in BENCH_DIR_NAMES: - test_case = "Distant-InsSto to InsSto" + test_case = "Local-InsSto to CEPH" source = f"{INST_STO_ROOT}/{files_type}" target = CEPH_ROOT + files_type run_single_bench(