[Ezbench-dev] [PATCH 6/7] smartezbench: allow ensuring runs on a testset

[Martin Peres]

---
 python-modules/ezbench/smartezbench.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/python-modules/ezbench/smartezbench.py b/python-modules/ezbench/smartezbench.py
index 66fb413..b6ccdae 100644
--- a/python-modules/ezbench/smartezbench.py
+++ b/python-modules/ezbench/smartezbench.py
@@ -519,15 +519,17 @@ class SmartEzbench:
         self.__release_lock()
         return total_rounds
 
-    def add_testset(self, commit, testset, rounds = 1):
+    def add_testset(self, commit, testset, rounds = 1, ensure=False):
         self.__reload_state(keep_lock=True)
 
         self.__log(Criticality.II, "Add the testset {} ({} tests)".format(testset.name,
                                                                          len(testset)))
 
         for test in sorted(testset.keys()):
-            self.__add_test_unlocked__(commit, test,
-                                            testset[test] * rounds)
+            if not ensure:
+                self.__add_test_unlocked__(commit, test, testset[test] * rounds)
+            else:
+                self.__force_test_rounds_unlocked__(commit, test, testset[test] * rounds)
 
         self.__save_state()
         self.__release_lock()
-- 
2.11.1