diff --git a/homeassistant/components/number/const.py b/homeassistant/components/number/const.py index e9c4c741e54..c10d677b773 100644 --- a/homeassistant/components/number/const.py +++ b/homeassistant/components/number/const.py @@ -289,7 +289,7 @@ class NumberDeviceClass(StrEnum): TEMPERATURE = "temperature" """Temperature. - Unit of measurement: `°C`, `°F` + Unit of measurement: `°C`, `°F`, `K` """ VOLATILE_ORGANIC_COMPOUNDS = "volatile_organic_compounds" @@ -384,10 +384,7 @@ DEVICE_CLASS_UNITS: dict[NumberDeviceClass, set[type[StrEnum] | str | None]] = { NumberDeviceClass.SOUND_PRESSURE: set(UnitOfSoundPressure), NumberDeviceClass.SPEED: set(UnitOfSpeed).union(set(UnitOfVolumetricFlux)), NumberDeviceClass.SULPHUR_DIOXIDE: {CONCENTRATION_MICROGRAMS_PER_CUBIC_METER}, - NumberDeviceClass.TEMPERATURE: { - UnitOfTemperature.CELSIUS, - UnitOfTemperature.FAHRENHEIT, - }, + NumberDeviceClass.TEMPERATURE: set(UnitOfTemperature), NumberDeviceClass.VOLATILE_ORGANIC_COMPOUNDS: { CONCENTRATION_MICROGRAMS_PER_CUBIC_METER }, diff --git a/homeassistant/components/sensor/const.py b/homeassistant/components/sensor/const.py index 3c3a073a308..66d231bea24 100644 --- a/homeassistant/components/sensor/const.py +++ b/homeassistant/components/sensor/const.py @@ -326,7 +326,7 @@ class SensorDeviceClass(StrEnum): TEMPERATURE = "temperature" """Temperature. - Unit of measurement: `°C`, `°F` + Unit of measurement: `°C`, `°F`, `K` """ VOLATILE_ORGANIC_COMPOUNDS = "volatile_organic_compounds" @@ -484,10 +484,7 @@ DEVICE_CLASS_UNITS: dict[SensorDeviceClass, set[type[StrEnum] | str | None]] = { SensorDeviceClass.SOUND_PRESSURE: set(UnitOfSoundPressure), SensorDeviceClass.SPEED: set(UnitOfSpeed).union(set(UnitOfVolumetricFlux)), SensorDeviceClass.SULPHUR_DIOXIDE: {CONCENTRATION_MICROGRAMS_PER_CUBIC_METER}, - SensorDeviceClass.TEMPERATURE: { - UnitOfTemperature.CELSIUS, - UnitOfTemperature.FAHRENHEIT, - }, + SensorDeviceClass.TEMPERATURE: set(UnitOfTemperature), SensorDeviceClass.VOLATILE_ORGANIC_COMPOUNDS: { CONCENTRATION_MICROGRAMS_PER_CUBIC_METER }, diff --git a/tests/components/number/test_websocket_api.py b/tests/components/number/test_websocket_api.py index f920fddd420..1cba89c42a1 100644 --- a/tests/components/number/test_websocket_api.py +++ b/tests/components/number/test_websocket_api.py @@ -22,7 +22,7 @@ async def test_device_class_units(hass: HomeAssistant, hass_ws_client) -> None: ) msg = await client.receive_json() assert msg["success"] - assert msg["result"] == {"units": unordered(["°F", "°C"])} + assert msg["result"] == {"units": unordered(["°F", "°C", "K"])} # Device class with units which number doesn't allow customizing & converting await client.send_json(