Add a convenience use_on() on TestPersistence

Signed-off-by: Stefan Marr <git@stefan-marr.de>
smarr · Mar 20, 2023 · 3495e99 · 3495e99
1 parent 5ace1e6
commit 3495e99
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 6 deletions.
diff --git a/rebench/tests/executor_test.py b/rebench/tests/executor_test.py
@@ -74,8 +74,7 @@ def _basic_execution(self, cnf):
 
         runs = cnf.get_runs()
         persistence = TestPersistence()
-        for run in runs:
-            run.add_persistence(persistence)
+        persistence.use_on(runs)
 
         ex = Executor(runs, cnf.do_builds, self.ui)
         ex.execute()

diff --git a/rebench/tests/features/issue_16_multiple_data_points_test.py b/rebench/tests/features/issue_16_multiple_data_points_test.py
@@ -41,8 +41,8 @@ def _records_data_points(self, exp_name, num_data_points):
                            data_file=self._tmp_file)
         runs = cnf.get_runs()
         persistence = TestPersistence()
-        for run in runs:
-            run.add_persistence(persistence)
+        persistence.use_on(runs)
+
         ex = Executor(cnf.get_runs(), False, self.ui)
         ex.execute()
         self.assertEqual(1, len(cnf.get_runs()))

diff --git a/rebench/tests/features/issue_31_multivariate_data_points_test.py b/rebench/tests/features/issue_31_multivariate_data_points_test.py
@@ -41,8 +41,7 @@ def _records_data_points(self, exp_name, num_data_points):
                            data_file=self._tmp_file)
         runs = cnf.get_runs()
         persistence = TestPersistence()
-        for run in runs:
-            run.add_persistence(persistence)
+        persistence.use_on(runs)
 
         ex = Executor(runs, False, self.ui)
         ex.execute()

diff --git a/rebench/tests/persistence.py b/rebench/tests/persistence.py
@@ -4,6 +4,10 @@ class TestPersistence(object):
     def __init__(self):
         self._data_points = []
 
+    def use_on(self, runs):
+        for run in runs:
+            run.add_persistence(self)
+
     def get_data_points(self, run_id=None):
         if run_id:
             return [dp for dp in self._data_points if dp.run_id is run_id]