1 files changed, 66 insertions, 0 deletions
diff --git a/tests/helper/BenchmarkTest.py b/tests/helper/BenchmarkTest.py
new file mode 100644
index 000000000..d28c52959
--- /dev/null
+++ b/tests/helper/BenchmarkTest.py
@@ -0,0 +1,66 @@
+# -*- coding: utf-8 -*-
+
+from time import time
+
+
+class BenchmarkTest:
+
+    bench = []
+    results = {}
+
+    @classmethod
+    def timestamp(cls, name, a):
+        t = time()
+        r = cls.results.get(name, [])
+        r.append((t-a) * 1000)
+        cls.results[name] = r
+
+    @classmethod
+    def benchmark(cls, n=1):
+
+        print "Benchmarking %s" % cls.__name__
+        print
+
+        for i in range(n):
+            cls.collect_results()
+
+        if "setUpClass" in cls.results:
+            cls.bench.insert(0, "setUpClass")
+
+        if "tearDownClass" in cls.results:
+            cls.bench.append("tearDownClass")
+
+        length = str(max([len(k) for k in cls.bench]) + 1)
+        total = 0
+
+        for k in cls.bench:
+            v = cls.results[k]
+
+            if len(v) > 1:
+                print ("%" + length +"s: %s | average: %.2f ms") % (k, ", ".join(["%.2f" % x for x in v]), sum(v)/len(v))
+                total += sum(v)/len(v)
+            else:
+                print ("%" + length +"s: %.2f ms") % (k, v[0])
+                total += v[0]
+
+        print "\ntotal: %.2f ms" % total
+
+
+    @classmethod
+    def collect_results(cls):
+        if hasattr(cls, "setUpClass"):
+            a = time()
+            cls.setUpClass()
+            cls.timestamp("setUpClass", a)
+
+        obj = cls()
+
+        for f in cls.bench:
+            a = time()
+            getattr(obj, "test_" + f)()
+            cls.timestamp(f, a)
+
+        if hasattr(cls, "tearDownClass"):
+            a = time()
+            cls.tearDownClass()
+            cls.timestamp("tearDownClass", a)
+\ No newline at end of file