mirror of
https://github.com/python-kasa/python-kasa.git
synced 2024-12-23 03:33:35 +00:00
fde156c859
* Add benchmarks for speedups * Update README.md * Update README.md Co-authored-by: Teemu R. <tpr@iki.fi> * relo * Update README.md * document benchmark * Update README.md --------- Co-authored-by: Teemu R. <tpr@iki.fi>
31 lines
899 B
Python
31 lines
899 B
Python
"""Benchmark the new parser against the old parser."""
|
|
|
|
import json
|
|
import timeit
|
|
|
|
import orjson
|
|
from kasa_crypt import decrypt, encrypt
|
|
from utils.data import REQUEST, WIRE_RESPONSE
|
|
from utils.original import OriginalTPLinkSmartHomeProtocol
|
|
|
|
|
|
def original_request_response() -> None:
|
|
"""Benchmark the original parser."""
|
|
OriginalTPLinkSmartHomeProtocol.encrypt(json.dumps(REQUEST))
|
|
json.loads(OriginalTPLinkSmartHomeProtocol.decrypt(WIRE_RESPONSE[4:]))
|
|
|
|
|
|
def new_request_response() -> None:
|
|
"""Benchmark the new parser."""
|
|
encrypt(orjson.dumps(REQUEST).decode())
|
|
orjson.loads(decrypt(WIRE_RESPONSE[4:]))
|
|
|
|
|
|
count = 100000
|
|
|
|
time = timeit.Timer(new_request_response).timeit(count)
|
|
print(f"New parser, parsing {count} messages took {time} seconds")
|
|
|
|
time = timeit.Timer(original_request_response).timeit(count)
|
|
print(f"Old parser, parsing {count} messages took {time} seconds")
|