From e4afd71d16dad4f78cae00229a1546c85f815ceb Mon Sep 17 00:00:00 2001 From: G Johansson Date: Sun, 3 May 2026 19:46:58 +0000 Subject: [PATCH] Add constants --- homeassistant/components/compensation/__init__.py | 4 +++- homeassistant/components/compensation/config_flow.py | 6 ++++-- homeassistant/components/compensation/const.py | 2 ++ tests/components/compensation/test_config_flow.py | 10 ++++++---- 4 files changed, 15 insertions(+), 7 deletions(-) diff --git a/homeassistant/components/compensation/__init__.py b/homeassistant/components/compensation/__init__.py index e092143c17d9..901c3af5c30a 100644 --- a/homeassistant/components/compensation/__init__.py +++ b/homeassistant/components/compensation/__init__.py @@ -30,6 +30,7 @@ from homeassistant.helpers.discovery import async_load_platform from homeassistant.helpers.typing import ConfigType from .const import ( + CONF_COMPENSATED_VALUE, CONF_COMPENSATION, CONF_DATAPOINTS, CONF_DEGREE, @@ -37,6 +38,7 @@ from .const import ( CONF_POLYNOMIAL, CONF_POLYNOMIAL_CONFIG, CONF_PRECISION, + CONF_UNCOMPENSATED_VALUE, CONF_UPPER_LIMIT, DATA_COMPENSATION, DEFAULT_DEGREE, @@ -173,7 +175,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: CONF_DATAPOINTS ] new_data_points = [ - [data_point["uncompensated_value"], data_point["compensated_value"]] + [data_point[CONF_UNCOMPENSATED_VALUE], data_point[CONF_COMPENSATED_VALUE]] for data_point in data_points ] config[CONF_DEGREE] = config[CONF_POLYNOMIAL_CONFIG][CONF_DEGREE] diff --git a/homeassistant/components/compensation/config_flow.py b/homeassistant/components/compensation/config_flow.py index a37d810dda74..b25f28c1867f 100644 --- a/homeassistant/components/compensation/config_flow.py +++ b/homeassistant/components/compensation/config_flow.py @@ -35,11 +35,13 @@ from homeassistant.helpers.selector import ( ) from .const import ( + CONF_COMPENSATED_VALUE, CONF_DATAPOINTS, CONF_DEGREE, CONF_LOWER_LIMIT, CONF_POLYNOMIAL_CONFIG, CONF_PRECISION, + CONF_UNCOMPENSATED_VALUE, CONF_UPPER_LIMIT, DEFAULT_DEGREE, DEFAULT_NAME, @@ -64,7 +66,7 @@ async def get_options_schema(handler: SchemaCommonFlowHandler) -> vol.Schema: multiple=True, translation_key=CONF_DATAPOINTS, fields={ - "uncompensated_value": ObjectSelectorField( + CONF_UNCOMPENSATED_VALUE: ObjectSelectorField( required=True, selector=NumberSelector( NumberSelectorConfig( @@ -72,7 +74,7 @@ async def get_options_schema(handler: SchemaCommonFlowHandler) -> vol.Schema: ) ), ), - "compensated_value": ObjectSelectorField( + CONF_COMPENSATED_VALUE: ObjectSelectorField( required=True, selector=NumberSelector( NumberSelectorConfig( diff --git a/homeassistant/components/compensation/const.py b/homeassistant/components/compensation/const.py index 7e08202ba437..dbb40a3ce12e 100644 --- a/homeassistant/components/compensation/const.py +++ b/homeassistant/components/compensation/const.py @@ -18,6 +18,8 @@ CONF_DEGREE = "degree" CONF_PRECISION = "precision" CONF_POLYNOMIAL = "polynomial" CONF_POLYNOMIAL_CONFIG = "polynomial_config" +CONF_COMPENSATED_VALUE = "compensated_value" +CONF_UNCOMPENSATED_VALUE = "uncompensated_value" DATA_COMPENSATION: HassKey[dict[str, Any]] = HassKey("compensation_data") diff --git a/tests/components/compensation/test_config_flow.py b/tests/components/compensation/test_config_flow.py index 01ce20b01ae2..58ef53e0c1d7 100644 --- a/tests/components/compensation/test_config_flow.py +++ b/tests/components/compensation/test_config_flow.py @@ -3,11 +3,13 @@ from unittest.mock import AsyncMock from homeassistant.components.compensation.const import ( + CONF_COMPENSATED_VALUE, CONF_DATAPOINTS, CONF_DEGREE, CONF_LOWER_LIMIT, CONF_POLYNOMIAL_CONFIG, CONF_PRECISION, + CONF_UNCOMPENSATED_VALUE, CONF_UPPER_LIMIT, DEFAULT_NAME, DOMAIN, @@ -43,8 +45,8 @@ async def test_user_flow( CONF_LOWER_LIMIT: False, CONF_POLYNOMIAL_CONFIG: { CONF_DATAPOINTS: [ - {"compensated_value": 6, "uncompensated_value": 5}, - {"compensated_value": 8, "uncompensated_value": 7}, + {CONF_COMPENSATED_VALUE: 6, CONF_UNCOMPENSATED_VALUE: 5}, + {CONF_COMPENSATED_VALUE: 8, CONF_UNCOMPENSATED_VALUE: 7}, ], CONF_DEGREE: 1, }, @@ -62,8 +64,8 @@ async def test_user_flow( CONF_NAME: "Compensation", CONF_POLYNOMIAL_CONFIG: { CONF_DATAPOINTS: [ - {"compensated_value": 6, "uncompensated_value": 5}, - {"compensated_value": 8, "uncompensated_value": 7}, + {CONF_COMPENSATED_VALUE: 6, CONF_UNCOMPENSATED_VALUE: 5}, + {CONF_COMPENSATED_VALUE: 8, CONF_UNCOMPENSATED_VALUE: 7}, ], CONF_DEGREE: 1, },