1 files changed, 10 insertions, 0 deletions
diff --git a/tools.py b/tools.py
index 51ef2e0..2a44104 100644
--- a/tools.py
+++ b/tools.py
@@ -1,4 +1,14 @@
 import numpy as np
+import time
+import datetime
+
+def timeit(func):
+    def f(*args, **kwargs):
+        t0 = time.time()
+        func(*args, **kwargs)
+        elapsed = time.time() - t0
+        print '%s elapsed in %s().' % (datetime.timedelta(seconds=elapsed), func.__name__)
+    return f
 
 def read_csv(filename):
     """Return an array of comma-separated values from `filename`."""