diff options
Diffstat (limited to 'benchmarks/benchmark.py')
-rw-r--r-- | benchmarks/benchmark.py | 42 |
1 files changed, 0 insertions, 42 deletions
diff --git a/benchmarks/benchmark.py b/benchmarks/benchmark.py deleted file mode 100644 index 047379451..000000000 --- a/benchmarks/benchmark.py +++ /dev/null @@ -1,42 +0,0 @@ -from timeit import Timer - -class Benchmark(dict): - """Benchmark a feature in different modules.""" - - def __init__(self,modules,title='',runs=3,reps=1000): - self.module_test = dict((m,'') for m in modules) - self.runs = runs - self.reps = reps - self.title = title - - def __setitem__(self,module,(test_str,setup_str)): - """Set the test code for modules.""" - if module == 'all': - modules = self.module_test.keys() - else: - modules = [module] - - for m in modules: - setup_str = 'import %s; import %s as np; ' % (m,m) \ - + setup_str - self.module_test[m] = Timer(test_str, setup_str) - - def run(self): - """Run the benchmark on the different modules.""" - module_column_len = max(len(mod) for mod in self.module_test) - - if self.title: - print self.title - print 'Doing %d runs, each with %d reps.' % (self.runs,self.reps) - print '-'*79 - - for mod in sorted(self.module_test): - modname = mod.ljust(module_column_len) - try: - print "%s: %s" % (modname, \ - self.module_test[mod].repeat(self.runs,self.reps)) - except Exception as e: - print "%s: Failed to benchmark (%s)." % (modname,e) - - print '-'*79 - print |