jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-23 16:57:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve type hints in benchmark script (#136259)

Browse Source
This commit is contained in:
epenet 2025-01-22 22:29:00 +01:00 committed by GitHub
parent 33f966a12e
commit 3a493bb6c0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
homeassistant/scripts/benchmark/__init__.py
View File

@ -58,7 +58,7 @@ def benchmark[_CallableT: Callable](func: _CallableT) -> _CallableT:
@benchmark
async def fire_events(hass):
async def fire_events(hass: core.HomeAssistant) -> float:
"""Fire a million events."""
count = 0
event_name = "benchmark_event"
@ -85,7 +85,7 @@ async def fire_events(hass):
@benchmark
async def fire_events_with_filter(hass):
async def fire_events_with_filter(hass: core.HomeAssistant) -> float:
"""Fire a million events with a filter that rejects them."""
count = 0
event_name = "benchmark_event"
@ -117,7 +117,7 @@ async def fire_events_with_filter(hass):
@benchmark
async def state_changed_helper(hass):
async def state_changed_helper(hass: core.HomeAssistant) -> float:
"""Run a million events through state changed helper with 1000 entities."""
count = 0
entity_id = "light.kitchen"
@ -141,7 +141,7 @@ async def state_changed_helper(hass):
}
for _ in range(10**6):
hass.bus.async_fire(EVENT_STATE_CHANGED, event_data)
hass.bus.async_fire(EVENT_STATE_CHANGED, event_data) # type: ignore[misc]
start = timer()
@ -151,7 +151,7 @@ async def state_changed_helper(hass):
@benchmark
async def state_changed_event_helper(hass):
async def state_changed_event_helper(hass: core.HomeAssistant) -> float:
"""Run a million events through state changed event helper with 1000 entities."""
count = 0
entity_id = "light.kitchen"
@ -174,7 +174,7 @@ async def state_changed_event_helper(hass):
}
for _ in range(events_to_fire):
hass.bus.async_fire(EVENT_STATE_CHANGED, event_data)
hass.bus.async_fire(EVENT_STATE_CHANGED, event_data) # type: ignore[misc]
start = timer()
@ -186,7 +186,7 @@ async def state_changed_event_helper(hass):
@benchmark
async def state_changed_event_filter_helper(hass):
async def state_changed_event_filter_helper(hass: core.HomeAssistant) -> float:
"""Run a million events through state changed event helper.
With 1000 entities that all get filtered.
@ -212,7 +212,7 @@ async def state_changed_event_filter_helper(hass):
}
for _ in range(events_to_fire):
hass.bus.async_fire(EVENT_STATE_CHANGED, event_data)
hass.bus.async_fire(EVENT_STATE_CHANGED, event_data) # type: ignore[misc]
start = timer()
@ -224,7 +224,7 @@ async def state_changed_event_filter_helper(hass):
@benchmark
async def filtering_entity_id(hass):
async def filtering_entity_id(hass: core.HomeAssistant) -> float:
"""Run a 100k state changes through entity filter."""
config = {
"include": {
@ -289,7 +289,7 @@ async def filtering_entity_id(hass):
@benchmark
async def valid_entity_id(hass):
async def valid_entity_id(hass: core.HomeAssistant) -> float:
"""Run valid entity ID a million times."""
start = timer()
for _ in range(10**6):
@ -298,7 +298,7 @@ async def valid_entity_id(hass):
@benchmark
async def json_serialize_states(hass):
async def json_serialize_states(hass: core.HomeAssistant) -> float:
"""Serialize million states with websocket default encoder."""
states = [
core.State("light.kitchen", "on", {"friendly_name": "Kitchen Lights"})