aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorDavid Robillard <d@drobilla.net>2020-11-15 12:32:07 +0100
committerDavid Robillard <d@drobilla.net>2020-11-15 12:32:07 +0100
commit1b8dafe596788d52c94eb69839d5689e78f51abf (patch)
tree7e781ccece5105b7da878081ccd9aa9ec3573a33
parent7d1b98ad1b2f5202126d78eaeb471263444a1384 (diff)
downloadserd-1b8dafe596788d52c94eb69839d5689e78f51abf.tar.gz
serd-1b8dafe596788d52c94eb69839d5689e78f51abf.tar.bz2
serd-1b8dafe596788d52c94eb69839d5689e78f51abf.zip
Format serd_bench.py with black
-rwxr-xr-xscripts/serd_bench.py159
1 files changed, 94 insertions, 65 deletions
diff --git a/scripts/serd_bench.py b/scripts/serd_bench.py
index b34545b0..0e309b15 100755
--- a/scripts/serd_bench.py
+++ b/scripts/serd_bench.py
@@ -12,6 +12,7 @@ import sys
class WorkingDirectory:
"Scoped context for changing working directory"
+
def __init__(self, working_dir):
self.original_dir = os.getcwd()
self.working_dir = working_dir
@@ -28,34 +29,34 @@ class WorkingDirectory:
def filename(n):
"Filename for a generated file with n statements"
- return 'gen%d.ttl' % n
+ return "gen%d.ttl" % n
def gen(sp2b_dir, n_min, n_max, step):
"Generate files with n_min ... n_max statements if they are not present"
with WorkingDirectory(sp2b_dir) as dir:
for n in range(n_min, n_max + step, step):
- out_path = os.path.join(dir.original_dir, 'build', filename(n))
+ out_path = os.path.join(dir.original_dir, "build", filename(n))
if not os.path.exists(out_path):
- subprocess.call(['./sp2b_gen', '-t', str(n), out_path])
+ subprocess.call(["./sp2b_gen", "-t", str(n), out_path])
def write_header(results, progs):
"Write the header line for TSV output"
- results.write('n')
+ results.write("n")
for prog in progs:
- results.write('\t' + os.path.basename(prog.split()[0]))
- results.write('\n')
+ results.write("\t" + os.path.basename(prog.split()[0]))
+ results.write("\n")
def parse_time(report):
"Return user time and max RSS from a /usr/bin/time -v report"
time = memory = None
- for line in report.split('\n'):
- if line.startswith('\tUser time'):
- time = float(line[line.find(':') + 1:])
- elif line.startswith('\tMaximum resident set'):
- memory = float(line[line.find(':') + 1:]) * 1024
+ for line in report.split("\n"):
+ if line.startswith("\tUser time"):
+ time = float(line[line.find(":") + 1 :])
+ elif line.startswith("\tMaximum resident set"):
+ memory = float(line[line.find(":") + 1 :]) * 1024
return (time, memory)
@@ -65,9 +66,9 @@ def get_dashes():
dash = 2.0
space = dot = 0.75
- yield [] # Solid
+ yield [] # Solid
yield [dash, space] # Dashed
- yield [dot, space] # Dotted
+ yield [dot, space] # Dotted
# Dash-dots, with increasing number of dots for each line
for i in itertools.count(2):
@@ -77,14 +78,14 @@ def get_dashes():
def plot(in_file, out_filename, x_label, y_label, y_max=None):
"Plot a TSV file as SVG"
- matplotlib.use('agg')
+ matplotlib.use("agg")
import matplotlib.pyplot as plt
fig_height = 4.0
dashes = get_dashes()
- markers = itertools.cycle(['o', 's', 'v', 'D', '*', 'p', 'P', 'h', 'X'])
+ markers = itertools.cycle(["o", "s", "v", "D", "*", "p", "P", "h", "X"])
- reader = csv.reader(in_file, delimiter='\t')
+ reader = csv.reader(in_file, delimiter="\t")
header = next(reader)
cols = [x for x in zip(*list(reader))]
@@ -98,32 +99,36 @@ def plot(in_file, out_filename, x_label, y_label, y_max=None):
if y_max is not None:
ax.set_ylim([0.0, y_max])
- ax.grid(linewidth=0.25, linestyle=':', color='0', dashes=[0.2, 1.6])
- ax.ticklabel_format(style='sci', scilimits=(4, 0), useMathText=True)
- ax.tick_params(axis='both', width=0.75)
+ ax.grid(linewidth=0.25, linestyle=":", color="0", dashes=[0.2, 1.6])
+ ax.ticklabel_format(style="sci", scilimits=(4, 0), useMathText=True)
+ ax.tick_params(axis="both", width=0.75)
x = list(map(float, cols[0]))
for i, y in enumerate(cols[1::]):
- ax.plot(x,
- list(map(float, y)),
- label=header[i + 1],
- marker=next(markers),
- dashes=next(dashes),
- markersize=3.0,
- linewidth=1.0)
+ ax.plot(
+ x,
+ list(map(float, y)),
+ label=header[i + 1],
+ marker=next(markers),
+ dashes=next(dashes),
+ markersize=3.0,
+ linewidth=1.0,
+ )
plt.legend()
- plt.savefig(out_filename, bbox_inches='tight', pad_inches=0.025)
+ plt.savefig(out_filename, bbox_inches="tight", pad_inches=0.025)
plt.close()
- sys.stderr.write('wrote {}\n'.format(out_filename))
+ sys.stderr.write("wrote {}\n".format(out_filename))
def run(progs, n_min, n_max, step):
"Benchmark each program with n_min ... n_max statements"
- with WorkingDirectory('build'):
- results = {'time': open('serdi-time.txt', 'w'),
- 'throughput': open('serdi-throughput.txt', 'w'),
- 'memory': open('serdi-memory.txt', 'w')}
+ with WorkingDirectory("build"):
+ results = {
+ "time": open("serdi-time.txt", "w"),
+ "throughput": open("serdi-throughput.txt", "w"),
+ "memory": open("serdi-memory.txt", "w"),
+ }
# Write TSV header for all output files
for name, f in results.items():
@@ -137,70 +142,94 @@ def run(progs, n_min, n_max, step):
# Run each program and fill rows with measurements
for prog in progs:
- cmd = '/usr/bin/time -v ' + prog + ' ' + filename(n)
- with open(filename(n) + '.out', 'w') as out:
- sys.stderr.write(cmd + '\n')
+ cmd = "/usr/bin/time -v " + prog + " " + filename(n)
+ with open(filename(n) + ".out", "w") as out:
+ sys.stderr.write(cmd + "\n")
proc = subprocess.Popen(
- cmd.split(), stdout=out, stderr=subprocess.PIPE)
+ cmd.split(), stdout=out, stderr=subprocess.PIPE
+ )
time, memory = parse_time(proc.communicate()[1].decode())
- rows['time'] += ['%.07f' % time]
- rows['throughput'] += ['%d' % (n / time)]
- rows['memory'] += [str(memory)]
+ rows["time"] += ["%.07f" % time]
+ rows["throughput"] += ["%d" % (n / time)]
+ rows["memory"] += [str(memory)]
# Write rows to output files
for name, f in results.items():
- f.write('\t'.join(rows[name]) + '\n')
+ f.write("\t".join(rows[name]) + "\n")
for name, f in results.items():
- tsv_filename = 'serdi-%s.txt' % name
- sys.stderr.write('wrote %s\n' % tsv_filename)
+ tsv_filename = "serdi-%s.txt" % name
+ sys.stderr.write("wrote %s\n" % tsv_filename)
def plot_results():
"Plot all benchmark results"
- with WorkingDirectory('build'):
- plot(open('serdi-time.txt', 'r'), 'serdi-time.svg',
- 'Statements', 'Time (s)')
- plot(open('serdi-throughput.txt', 'r'), 'serdi-throughput.svg',
- 'Statements', 'Statements / s')
- plot(open('serdi-memory.txt', 'r'), 'serdi-memory.svg',
- 'Statements', 'Bytes')
+ with WorkingDirectory("build"):
+ plot(
+ open("serdi-time.txt", "r"),
+ "serdi-time.svg",
+ "Statements",
+ "Time (s)",
+ )
+ plot(
+ open("serdi-throughput.txt", "r"),
+ "serdi-throughput.svg",
+ "Statements",
+ "Statements / s",
+ )
+ plot(
+ open("serdi-memory.txt", "r"),
+ "serdi-memory.svg",
+ "Statements",
+ "Bytes",
+ )
if __name__ == "__main__":
+
class OptParser(optparse.OptionParser):
def format_epilog(self, formatter):
return self.expand_prog_name(self.epilog)
opt = OptParser(
- usage='%prog [OPTION]... SP2B_DIR',
- description='Benchmark RDF reading and writing commands\n',
- epilog='''
+ usage="%prog [OPTION]... SP2B_DIR",
+ description="Benchmark RDF reading and writing commands\n",
+ epilog="""
Example:
%prog --max 100000 \\
--run 'rapper -i turtle -o turtle' \\
--run 'riot --output=ttl' \\
--run 'rdfpipe -i turtle -o turtle' /path/to/sp2b/src/
-''')
-
- opt.add_option('--max', type='int', default=1000000,
- help='maximum triple count')
- opt.add_option('--run', type='string', action='append', default=[],
- help='additional command to run (input file is appended)')
- opt.add_option('--no-generate', action='store_true',
- help='do not generate data')
- opt.add_option('--no-execute', action='store_true',
- help='do not run benchmarks')
- opt.add_option('--no-plot', action='store_true',
- help='do not plot benchmarks')
+""",
+ )
+
+ opt.add_option(
+ "--max", type="int", default=1000000, help="maximum triple count"
+ )
+ opt.add_option(
+ "--run",
+ type="string",
+ action="append",
+ default=[],
+ help="additional command to run (input file is appended)",
+ )
+ opt.add_option(
+ "--no-generate", action="store_true", help="do not generate data"
+ )
+ opt.add_option(
+ "--no-execute", action="store_true", help="do not run benchmarks"
+ )
+ opt.add_option(
+ "--no-plot", action="store_true", help="do not plot benchmarks"
+ )
(options, args) = opt.parse_args()
if len(args) != 1:
opt.print_usage()
sys.exit(1)
- progs = ['serdi -b -f -i turtle -o turtle'] + options.run
+ progs = ["serdi -b -f -i turtle -o turtle"] + options.run
min_n = int(options.max / 10)
max_n = options.max
step = min_n