Add constants

Author: gjohansson-ST

homeassistant/components/compensation/__init__.py

@@ -30,13 +30,15 @@
 from homeassistant.helpers.typing import ConfigType
 
 from .const import (
+    CONF_COMPENSATED_VALUE,
     CONF_COMPENSATION,
     CONF_DATAPOINTS,
     CONF_DEGREE,
     CONF_LOWER_LIMIT,
     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]

homeassistant/components/compensation/config_flow.py

@@ -35,11 +35,13 @@
 )
 
 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,15 +66,15 @@ 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(
                                                 mode=NumberSelectorMode.BOX
                                             )
                                         ),
                                     ),
-                                    "compensated_value": ObjectSelectorField(
+                                    CONF_COMPENSATED_VALUE: ObjectSelectorField(
                                         required=True,
                                         selector=NumberSelector(
                                             NumberSelectorConfig(

homeassistant/components/compensation/const.py

@@ -18,6 +18,8 @@
 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")

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,
         },