mirror of
https://github.com/esphome/esphome.git
synced 2026-09-13 16:18:41 +00:00
J. Nick Koston
41c413d5f7
Add tests/component_tests/helpers.py with typed extract_packed_value() and INTERNAL_BIT constant, replacing 5 duplicate copies.
16 lines
480 B
Python
16 lines
480 B
Python
"""Shared helpers for component tests."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import re
|
|
|
|
INTERNAL_BIT = 1 << 24
|
|
|
|
|
|
def extract_packed_value(main_cpp: str, var_name: str) -> int:
|
|
"""Extract the third (packed) argument from a configure_entity_ call."""
|
|
pattern = rf"{re.escape(var_name)}->configure_entity_\([^,]+,\s*\w+,\s*(\d+)\)"
|
|
match = re.search(pattern, main_cpp)
|
|
assert match, f"configure_entity_ call not found for {var_name}"
|
|
return int(match.group(1))
|