Add constants

This commit is contained in:
G Johansson
2026-05-03 19:46:58 +00:00
parent 6308a2e691
commit e4afd71d16
4 changed files with 15 additions and 7 deletions
@@ -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]
@@ -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(
@@ -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")
@@ -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,
},