diff options
Diffstat (limited to 'benchmarks/base.py')
| -rw-r--r-- | benchmarks/base.py | 17 |
1 files changed, 8 insertions, 9 deletions
diff --git a/benchmarks/base.py b/benchmarks/base.py index 519c9cc..f52657f 100644 --- a/benchmarks/base.py +++ b/benchmarks/base.py @@ -1,9 +1,10 @@ import functools import itertools -import redis import sys import timeit +import redis + class Benchmark: ARGUMENTS = () @@ -15,9 +16,7 @@ class Benchmark: # eventually make this more robust and take optional args from # argparse if self._client is None or kwargs: - defaults = { - 'db': 9 - } + defaults = {"db": 9} defaults.update(kwargs) pool = redis.ConnectionPool(**kwargs) self._client = redis.Redis(connection_pool=pool) @@ -30,16 +29,16 @@ class Benchmark: pass def run_benchmark(self): - group_names = [group['name'] for group in self.ARGUMENTS] - group_values = [group['values'] for group in self.ARGUMENTS] + group_names = [group["name"] for group in self.ARGUMENTS] + group_values = [group["values"] for group in self.ARGUMENTS] for value_set in itertools.product(*group_values): pairs = list(zip(group_names, value_set)) - arg_string = ', '.join(f'{p[0]}={p[1]}' for p in pairs) - sys.stdout.write(f'Benchmark: {arg_string}... ') + arg_string = ", ".join(f"{p[0]}={p[1]}" for p in pairs) + sys.stdout.write(f"Benchmark: {arg_string}... ") sys.stdout.flush() kwargs = dict(pairs) setup = functools.partial(self.setup, **kwargs) run = functools.partial(self.run, **kwargs) t = timeit.timeit(stmt=run, setup=setup, number=1000) - sys.stdout.write(f'{t:f}\n') + sys.stdout.write(f"{t:f}\n") sys.stdout.flush() |