diff options
author | Micah Anderson <micah@riseup.net> | 2014-11-11 11:53:55 -0500 |
---|---|---|
committer | Micah Anderson <micah@riseup.net> | 2014-11-11 11:53:55 -0500 |
commit | 7d5c3dcd969161322deed6c43f8a6a3cb92c3369 (patch) | |
tree | 109b05c88c7252d7609ef324d62ef9dd7f06123f /examples/bench/benchmark.py | |
parent | 44be832c5708baadd146cb954befbc3dcad8d463 (diff) |
upgrade to 14.4.1upstream/14.4.1
Diffstat (limited to 'examples/bench/benchmark.py')
-rw-r--r-- | examples/bench/benchmark.py | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/examples/bench/benchmark.py b/examples/bench/benchmark.py new file mode 100644 index 0000000..c379af9 --- /dev/null +++ b/examples/bench/benchmark.py @@ -0,0 +1,25 @@ +from timeit import default_timer as timer + +def benchmark(f, size, reps): + msg = size*'0' + t1 = timer() + for i in range(reps): + msg2 = f(msg) + assert msg == msg2 + t2 = timer() + diff = (t2-t1) + latency = diff/reps + return latency*1000000 + +kB = [1000*2**n for n in range(10)] +MB = [1000000*2**n for n in range(8)] +sizes = [1] + kB + MB + +def benchmark_set(f, sizes, reps): + latencies = [] + for size, rep in zip(sizes, reps): + print "Running benchmark with %r reps of %r bytes" % (rep, size) + lat = benchmark(f, size, rep) + latencies.append(lat) + return sizes, latencies + |