Changes in benchmark/rmit.py [7a2a3af:836c9925]

File:

• benchmark/rmit.py (modified) (4 diffs)

benchmark/rmit.py

@@ -10 +10 @@
 
 import argparse
+import datetime
 import itertools
 import os
@@ -101 +102 @@
         return nopts
 
+def actions_eta(actions):
+        time = 0
+        for a in actions:
+                i = 0
+                while i < len(a):
+                        if a[i] == '-d':
+                                i += 1
+                                if i != len(a):
+                                        time += int(a[i])
+                        i += 1
+        return time
+
 if __name__ == "__main__":
         # ================================================================================
@@ -160 +173 @@
         # ================================================================================
         # Prepare to run
+
+        # find expected time
+        time = actions_eta(actions)
+        print("Running {} trials{}".format(len(actions), "" if time == 0 else " (expecting to take {}".format(str(datetime.timedelta(seconds=int(time)))) ))
+
         random.shuffle(actions)
-
-        print("Running {} trials".format(len(actions)))
         result = []
 
@@ -209 +225 @@
                         d = [r[0], r[1]]
                         for k in headers[2:]:
-                                d.append(r[2][k])
+                                try:
+                                        d.append(r[2][k])
+                                except:
+                                        d.append(0.0)
 
                         data.append(d)