michaelosthege · July 6, 2017 12:05 · Jul 6, 2017
diff --git a/benchmark.py b/benchmark.py
@@ -0,0 +1,15 @@
+import time
+
+class Measurement(object):
+    def __init__(self, name):
+        self.name = name
+        return
+
+    def __enter__(self):
+        self.t_start = time.time()
+        return
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        t_end = time.time()
+        print("{} took {:.3f} seconds".format(self.name, t_end - self.t_start))
+        return