summaryrefslogtreecommitdiff
path: root/benchmarks/base.py
diff options
context:
space:
mode:
authorAnas <anas.el.amraoui@live.com>2021-11-30 18:05:51 +0200
committerGitHub <noreply@github.com>2021-11-30 18:05:51 +0200
commitb94e230b17d08e6c89d134e933c706256b79bc4a (patch)
tree993bd7565169229326b810b66939587431ab9dc6 /benchmarks/base.py
parent368a25f9d163d784a8896f1c087582405e98e006 (diff)
downloadredis-py-b94e230b17d08e6c89d134e933c706256b79bc4a.tar.gz
Added black and isort (#1734)
Diffstat (limited to 'benchmarks/base.py')
-rw-r--r--benchmarks/base.py17
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()