From 5dfd7f7a8cdc5ce16edb0d39b92bf9a483c6a9c2 Mon Sep 17 00:00:00 2001 From: Christopher Date: Tue, 28 May 2019 14:24:03 +0200 Subject: [PATCH] Added a delimiter option, replaced int(i) with eval(i) --- benchpress/benchmarks/util.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/benchpress/benchmarks/util.py b/benchpress/benchmarks/util.py index a92b84de..a16c94ce 100644 --- a/benchpress/benchmarks/util.py +++ b/benchpress/benchmarks/util.py @@ -47,9 +47,10 @@ class Benchmark: the statements in-between start() and stop() are measured. """ - def __init__(self, description, size_pattern): + def __init__(self, description, size_pattern, delimiter="*"): self._elapsed = 0.0 # The quantity measured self._script = sys.argv[0] # The script being run + self.delimiter = delimiter # Construct argument parser p = argparse.ArgumentParser(description=description) @@ -124,7 +125,7 @@ def __init__(self, description, size_pattern): ) self.args = p.parse_args() # Parse the arguments - self.args.size = [int(i) for i in self.args.size.split("*")] if self.args.size else [] + self.args.size = [eval(i) for i in self.args.size.split(self.delimiter)] if self.args.size else [] self.dtype = eval("numpy.%s" % self.args.dtype) if self.args.visualize: self._visual_args = VisualArgs(self.args)