diff options
Diffstat (limited to 'benchmarks/socket_read_size.py')
| -rw-r--r-- | benchmarks/socket_read_size.py | 27 |
1 files changed, 10 insertions, 17 deletions
diff --git a/benchmarks/socket_read_size.py b/benchmarks/socket_read_size.py index 72a1b0a..3427956 100644 --- a/benchmarks/socket_read_size.py +++ b/benchmarks/socket_read_size.py @@ -1,34 +1,27 @@ -from redis.connection import PythonParser, HiredisParser from base import Benchmark +from redis.connection import HiredisParser, PythonParser + class SocketReadBenchmark(Benchmark): ARGUMENTS = ( + {"name": "parser", "values": [PythonParser, HiredisParser]}, { - 'name': 'parser', - 'values': [PythonParser, HiredisParser] - }, - { - 'name': 'value_size', - 'values': [10, 100, 1000, 10000, 100000, 1000000, 10000000, - 100000000] + "name": "value_size", + "values": [10, 100, 1000, 10000, 100000, 1000000, 10000000, 100000000], }, - { - 'name': 'read_size', - 'values': [4096, 8192, 16384, 32768, 65536, 131072] - } + {"name": "read_size", "values": [4096, 8192, 16384, 32768, 65536, 131072]}, ) def setup(self, value_size, read_size, parser): - r = self.get_client(parser_class=parser, - socket_read_size=read_size) - r.set('benchmark', 'a' * value_size) + r = self.get_client(parser_class=parser, socket_read_size=read_size) + r.set("benchmark", "a" * value_size) def run(self, value_size, read_size, parser): r = self.get_client() - r.get('benchmark') + r.get("benchmark") -if __name__ == '__main__': +if __name__ == "__main__": SocketReadBenchmark().run_benchmark() |