1
0
mirror of https://github.com/esphome/esphome.git synced 2025-10-21 19:23:45 +01:00

fix flakey

This commit is contained in:
J. Nick Koston
2025-10-16 06:09:44 -10:00
parent 7be04916ac
commit 0cff6acdf4

View File

@@ -1,27 +0,0 @@
"""Shared utilities for sensor integration tests."""
from __future__ import annotations
from aioesphomeapi import EntityInfo
def build_key_to_sensor_mapping(
entities: list[EntityInfo], sensor_names: list[str]
) -> dict[int, str]:
"""Build a mapping from entity keys to sensor names.
Args:
entities: List of entity info objects from the API
sensor_names: List of sensor names to search for in object_ids
Returns:
Dictionary mapping entity keys to sensor names
"""
key_to_sensor: dict[int, str] = {}
for entity in entities:
obj_id = entity.object_id.lower()
for sensor_name in sensor_names:
if sensor_name in obj_id:
key_to_sensor[entity.key] = sensor_name
break
return key_to_sensor