diff options
author | Jordan Cook <jordan.cook@pioneer.com> | 2021-09-14 17:06:33 -0500 |
---|---|---|
committer | Jordan Cook <jordan.cook@pioneer.com> | 2021-09-14 19:03:18 -0500 |
commit | 5a907bc0d82c17901afddb603267cd798b87fb6b (patch) | |
tree | adfa439977181ed81939df299399a077c270936d | |
parent | e3c54551101a8dcf431aea474af86fe99a5d66fb (diff) | |
download | requests-cache-5a907bc0d82c17901afddb603267cd798b87fb6b.tar.gz |
Update benchmark script to optionally run for a specific serializer
-rwxr-xr-x | examples/benchmark.py | 14 | ||||
-rw-r--r-- | pyproject.toml | 1 |
2 files changed, 11 insertions, 4 deletions
diff --git a/examples/benchmark.py b/examples/benchmark.py index 6992b9a..c0b5d9d 100755 --- a/examples/benchmark.py +++ b/examples/benchmark.py @@ -2,9 +2,9 @@ """ An example of benchmarking cache write speeds with semi-randomized response content -Usage: +Usage (optionally for a specific backend and/or serializer): ``` -python benchmark.py -b <backend name> +python benchmark.py -b <backend> -s <serializer> ``` """ from argparse import ArgumentParser @@ -84,8 +84,9 @@ def get_randomized_response(i=0): if __name__ == '__main__': parser = ArgumentParser() parser.add_argument('-b', '--backend', default='sqlite') + parser.add_argument('-s', '--serializer', default='pickle') args = parser.parse_args() - print(f'[cyan]Benchmarking {args.backend} backend') + print(f'[cyan]Benchmarking {args.backend} backend with {args.serializer} serializer') kwargs = {} if args.backend == 'dynamodb': @@ -94,6 +95,11 @@ if __name__ == '__main__': args.backend = 'sqlite' kwargs = {'use_memory': True} - session = CachedSession(CACHE_NAME, backend=args.backend, **kwargs) + session = CachedSession( + CACHE_NAME, + backend=args.backend, + serializer=args.serializer, + **kwargs, + ) test_write_speed(session) test_read_speed(session) diff --git a/pyproject.toml b/pyproject.toml index ef510d9..496d0d8 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -13,6 +13,7 @@ keywords = ["requests", "cache", "http", "persistence", "sqlite", "redis", "mong classifiers = [ "Development Status :: 4 - Beta", "Intended Audience :: Developers", + "Topic :: Software Development :: Libraries :: Python Modules", "Typing :: Typed", ] include = [ |