Include elapsed time in write benchmark

[bluesky.git] / microbench / writebench.py

diff --git a/microbench/writebench.py b/microbench/writebench.py
index b480886..e2dbf34 100755 (executable)
--- a/microbench/writebench.py
+++ b/microbench/writebench.py
@@ -26,15 +26,17 @@ def run_writebench(path, rate):
     start_time = time.time()
     last_time = start_time
     latencies = []
-    while last_time < start_time + 30:
+    while last_time < start_time + 120:
         now = time.time()
         next_time = start_time + (1.0/rate) * count
         time.sleep(max(0.0, next_time - now))
         last_time = time.time()
         latencies.append(write_file(path, count))
-        #print "create(%d) = %s" % (count, latencies[-1])
-        print latencies[-1]
+        print "%s\t%s" % (latencies[-1], time.time() - start_time)
         count += 1
+    end_time = time.time()
+    duration = end_time - start_time
+    print "# %s MB/s (%d MB/%s seconds)" % (count / duration, count, duration)
     return latencies
 
 if __name__ == '__main__':