Format Benchmarks as Markdown Table

jemalloc.py

import numpy

benchmarks = {}

def flush_benchmark(fname, benchmark, times):
	if benchmark == None:
		return
	benchmarks[fname][benchmark] = numpy.mean(times)


benchmark = None
times = []

fnames = ['nojemalloc', 'jemalloc']

benchmark_names = []

for fname in fnames:
	benchmarks[fname] = {}
	with open(fname, 'r') as f:
		benchmark = None
		times = []
		for line in f:
			if '||' in line:
				flush_benchmark(fname, benchmark, times)
				benchmark = line.split('||')[1].strip()
				if benchmark not in benchmark_names:
					benchmark_names.append(benchmark)
				times = []
			if '...' in line:
				try:
					val = float(line.split('...')[1].strip())
					times.append(val)
				except:
					pass
		flush_benchmark(fname, benchmark, times)

benchmark_names.sort()

print('| Query | System | Jemalloc |Speedup|')
print('|:------|-------:|---------:|------:|')
for bname in benchmark_names:
	nojeval = benchmarks['nojemalloc'][bname]
	jeval = benchmarks['jemalloc'][bname]
	speedup = ((nojeval - jeval) / jeval) * 100

	print('|%s|%.3f|%.3f|%d%%' % (bname, nojeval, jeval, speedup))