Skip to content

Commit

Permalink
Merge pull request #20 from Shadesfear/master
Browse files Browse the repository at this point in the history
Added a delimiter option, replaced int(i) with eval(i)
  • Loading branch information
madsbk committed May 31, 2019
2 parents 805be70 + 5dfd7f7 commit e1dcda4
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions benchpress/benchmarks/util.py
Expand Up @@ -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)
Expand Down Expand Up @@ -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)
Expand Down

0 comments on commit e1dcda4

Please sign in to comment.