commit d49112515ce338334b75a795ab6aaf08daa5b489
Author: Isis Lovecruft <isis@xxxxxxxxxxxxxx>
Date: Fri Apr 17 23:49:13 2015 +0000
Add a test utility for benchmarking code.
---
lib/bridgedb/test/util.py | 31 +++++++++++++++++++++++++++++++
1 file changed, 31 insertions(+)
diff --git a/lib/bridgedb/test/util.py b/lib/bridgedb/test/util.py
index 234a9b6..9e401a5 100644
--- a/lib/bridgedb/test/util.py
+++ b/lib/bridgedb/test/util.py
@@ -18,6 +18,7 @@ import errno
import ipaddr
import os
import random
+import time
from functools import wraps
@@ -146,3 +147,33 @@ def randomIPString():
#: being run as a ``TestCase`` by ``twisted.trial``.
TestCaseMixin = bdbutil.mixin
TestCaseMixin.register(unittest.TestCase)
+
+
+class Benchmarker(object):
+ """Wrap a context with a timer to benchmark execution time.
+
+ .. hint:: Use like so::
+
+ with Benchmarker():
+ foo(bar, baz)
+
+ Once the ``with`` context exits, something like::
+
+ Benchmark: 180.269957ms (0s)
+
+ will be printed to stdout (if **verbose** is set to ``True``).
+ """
+
+ def __init__(self, verbose=True):
+ self.verbose = verbose
+
+ def __enter__(self):
+ self.start = time.time()
+ return self
+
+ def __exit__(self, *args):
+ self.end = time.time()
+ self.seconds = self.end - self.start
+ self.milliseconds = self.seconds * 1000
+ if self.verbose:
+ print("Benchmark: %12fms %12fs" % (self.milliseconds, self.seconds))
_______________________________________________
tor-commits mailing list
tor-commits@xxxxxxxxxxxxxxxxxxxx
https://lists.torproject.org/cgi-bin/mailman/listinfo/tor-commits