From 1413e8e2d25b4cbb7831745180cf32de0c91b1cd Mon Sep 17 00:00:00 2001 From: Lars Buitinck Date: Thu, 16 Jan 2014 21:15:26 +0100 Subject: [PATCH] simple benchmark script --- benchmark.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 benchmark.py diff --git a/benchmark.py b/benchmark.py new file mode 100644 index 0000000..02c9762 --- /dev/null +++ b/benchmark.py @@ -0,0 +1,16 @@ +import sys +from timeit import timeit + +from leven import levenshtein + +n_runs = 30 + +with open("/usr/share/dict/words") as f: + strs = [ln.strip() for ln in f] +print("Running %d times on %d reference strings" % (n_runs, len(strs))) + +arg = sys.argv[1] +t = timeit(stmt='for s in strs: levenshtein(arg, s)', + setup='from __main__ import arg, levenshtein, strs', + number=n_runs) +print("%.3f seconds" % t)