jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 21:27:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Migrate Ambee to new entity naming style (#74877)

Browse Source
This commit is contained in:
Franck Nijhof 2022-07-11 09:51:01 +02:00 committed by GitHub
parent eb92f0e16c
commit 34b8f2b283
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 73 additions and 56 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
homeassistant/components/ambee/const.py
View File

@ -27,7 +27,7 @@ SERVICE_AIR_QUALITY: Final = "air_quality"
SERVICE_POLLEN: Final = "pollen" SERVICE_POLLEN: Final = "pollen"
SERVICES: dict[str, str] = { SERVICES: dict[str, str] = {
SERVICE_AIR_QUALITY: "Air Quality", SERVICE_AIR_QUALITY: "Air quality",
SERVICE_POLLEN: "Pollen", SERVICE_POLLEN: "Pollen",
} }
@ -35,25 +35,25 @@ SENSORS: dict[str, list[SensorEntityDescription]] = {
SERVICE_AIR_QUALITY: [ SERVICE_AIR_QUALITY: [
SensorEntityDescription( SensorEntityDescription(
key="particulate_matter_2_5", key="particulate_matter_2_5",
name="Particulate Matter < 2.5 μm", name="Particulate matter < 2.5 μm",
native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
), ),
SensorEntityDescription( SensorEntityDescription(
key="particulate_matter_10", key="particulate_matter_10",
name="Particulate Matter < 10 μm", name="Particulate matter < 10 μm",
native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
), ),
SensorEntityDescription( SensorEntityDescription(
key="sulphur_dioxide", key="sulphur_dioxide",
name="Sulphur Dioxide (SO2)", name="Sulphur dioxide (SO2)",
native_unit_of_measurement=CONCENTRATION_PARTS_PER_BILLION, native_unit_of_measurement=CONCENTRATION_PARTS_PER_BILLION,
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
), ),
SensorEntityDescription( SensorEntityDescription(
key="nitrogen_dioxide", key="nitrogen_dioxide",
name="Nitrogen Dioxide (NO2)", name="Nitrogen dioxide (NO2)",
native_unit_of_measurement=CONCENTRATION_PARTS_PER_BILLION, native_unit_of_measurement=CONCENTRATION_PARTS_PER_BILLION,
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
), ),
@ -65,60 +65,60 @@ SENSORS: dict[str, list[SensorEntityDescription]] = {
), ),
SensorEntityDescription( SensorEntityDescription(
key="carbon_monoxide", key="carbon_monoxide",
name="Carbon Monoxide (CO)", name="Carbon monoxide (CO)",
device_class=SensorDeviceClass.CO, device_class=SensorDeviceClass.CO,
native_unit_of_measurement=CONCENTRATION_PARTS_PER_MILLION, native_unit_of_measurement=CONCENTRATION_PARTS_PER_MILLION,
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
), ),
SensorEntityDescription( SensorEntityDescription(
key="air_quality_index", key="air_quality_index",
name="Air Quality Index (AQI)", name="Air quality index (AQI)",
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
), ),
], ],
SERVICE_POLLEN: [ SERVICE_POLLEN: [
SensorEntityDescription( SensorEntityDescription(
key="grass", key="grass",
name="Grass Pollen", name="Grass",
icon="mdi:grass", icon="mdi:grass",
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER,
), ),
SensorEntityDescription( SensorEntityDescription(
key="tree", key="tree",
name="Tree Pollen", name="Tree",
icon="mdi:tree", icon="mdi:tree",
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER,
), ),
SensorEntityDescription( SensorEntityDescription(
key="weed", key="weed",
name="Weed Pollen", name="Weed",
icon="mdi:sprout", icon="mdi:sprout",
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER,
), ),
SensorEntityDescription( SensorEntityDescription(
key="grass_risk", key="grass_risk",
name="Grass Pollen Risk", name="Grass risk",
icon="mdi:grass", icon="mdi:grass",
device_class=DEVICE_CLASS_AMBEE_RISK, device_class=DEVICE_CLASS_AMBEE_RISK,
), ),
SensorEntityDescription( SensorEntityDescription(
key="tree_risk", key="tree_risk",
name="Tree Pollen Risk", name="Tree risk",
icon="mdi:tree", icon="mdi:tree",
device_class=DEVICE_CLASS_AMBEE_RISK, device_class=DEVICE_CLASS_AMBEE_RISK,
), ),
SensorEntityDescription( SensorEntityDescription(
key="weed_risk", key="weed_risk",
name="Weed Pollen Risk", name="Weed risk",
icon="mdi:sprout", icon="mdi:sprout",
device_class=DEVICE_CLASS_AMBEE_RISK, device_class=DEVICE_CLASS_AMBEE_RISK,
), ),
SensorEntityDescription( SensorEntityDescription(
key="grass_poaceae", key="grass_poaceae",
name="Poaceae Grass Pollen", name="Poaceae grass",
icon="mdi:grass", icon="mdi:grass",
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER,
@ -126,7 +126,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = {
), ),
SensorEntityDescription( SensorEntityDescription(
key="tree_alder", key="tree_alder",
name="Alder Tree Pollen", name="Alder tree",
icon="mdi:tree", icon="mdi:tree",
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER,
@ -134,7 +134,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = {
), ),
SensorEntityDescription( SensorEntityDescription(
key="tree_birch", key="tree_birch",
name="Birch Tree Pollen", name="Birch tree",
icon="mdi:tree", icon="mdi:tree",
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER,
@ -142,7 +142,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = {
), ),
SensorEntityDescription( SensorEntityDescription(
key="tree_cypress", key="tree_cypress",
name="Cypress Tree Pollen", name="Cypress tree",
icon="mdi:tree", icon="mdi:tree",
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER,
@ -150,7 +150,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = {
), ),
SensorEntityDescription( SensorEntityDescription(
key="tree_elm", key="tree_elm",
name="Elm Tree Pollen", name="Elm tree",
icon="mdi:tree", icon="mdi:tree",
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER,
@ -158,7 +158,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = {
), ),
SensorEntityDescription( SensorEntityDescription(
key="tree_hazel", key="tree_hazel",
name="Hazel Tree Pollen", name="Hazel tree",
icon="mdi:tree", icon="mdi:tree",
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER,
@ -166,7 +166,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = {
), ),
SensorEntityDescription( SensorEntityDescription(
key="tree_oak", key="tree_oak",
name="Oak Tree Pollen", name="Oak tree",
icon="mdi:tree", icon="mdi:tree",
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER,
@ -174,7 +174,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = {
), ),
SensorEntityDescription( SensorEntityDescription(
key="tree_pine", key="tree_pine",
name="Pine Tree Pollen", name="Pine tree",
icon="mdi:tree", icon="mdi:tree",
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER,
@ -182,7 +182,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = {
), ),
SensorEntityDescription( SensorEntityDescription(
key="tree_plane", key="tree_plane",
name="Plane Tree Pollen", name="Plane tree",
icon="mdi:tree", icon="mdi:tree",
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER,
@ -190,7 +190,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = {
), ),
SensorEntityDescription( SensorEntityDescription(
key="tree_poplar", key="tree_poplar",
name="Poplar Tree Pollen", name="Poplar tree",
icon="mdi:tree", icon="mdi:tree",
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER,
@ -198,7 +198,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = {
), ),
SensorEntityDescription( SensorEntityDescription(
key="weed_chenopod", key="weed_chenopod",
name="Chenopod Weed Pollen", name="Chenopod weed",
icon="mdi:sprout", icon="mdi:sprout",
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER,
@ -206,7 +206,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = {
), ),
SensorEntityDescription( SensorEntityDescription(
key="weed_mugwort", key="weed_mugwort",
name="Mugwort Weed Pollen", name="Mugwort weed",
icon="mdi:sprout", icon="mdi:sprout",
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER,
@ -214,7 +214,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = {
), ),
SensorEntityDescription( SensorEntityDescription(
key="weed_nettle", key="weed_nettle",
name="Nettle Weed Pollen", name="Nettle weed",
icon="mdi:sprout", icon="mdi:sprout",
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER,
@ -222,7 +222,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = {
), ),
SensorEntityDescription( SensorEntityDescription(
key="weed_ragweed", key="weed_ragweed",
name="Ragweed Weed Pollen", name="Ragweed weed",
icon="mdi:sprout", icon="mdi:sprout",
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER,

2
homeassistant/components/ambee/sensor.py
View File

@ -42,6 +42,8 @@ async def async_setup_entry(
class AmbeeSensorEntity(CoordinatorEntity, SensorEntity): class AmbeeSensorEntity(CoordinatorEntity, SensorEntity):
"""Defines an Ambee sensor.""" """Defines an Ambee sensor."""
_attr_has_entity_name = True
def __init__( def __init__(
self, self,
*, *,

73
tests/components/ambee/test_sensor.py
View File

@ -41,7 +41,10 @@ async def test_air_quality(
assert state assert state
assert entry.unique_id == f"{entry_id}_air_quality_particulate_matter_2_5" assert entry.unique_id == f"{entry_id}_air_quality_particulate_matter_2_5"
assert state.state == "3.14" assert state.state == "3.14"
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Particulate Matter < 2.5 μm" assert (
state.attributes.get(ATTR_FRIENDLY_NAME)
== "Air quality Particulate matter < 2.5 μm"
)
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
assert ( assert (
state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) state.attributes.get(ATTR_UNIT_OF_MEASUREMENT)
@ -56,7 +59,10 @@ async def test_air_quality(
assert state assert state
assert entry.unique_id == f"{entry_id}_air_quality_particulate_matter_10" assert entry.unique_id == f"{entry_id}_air_quality_particulate_matter_10"
assert state.state == "5.24" assert state.state == "5.24"
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Particulate Matter < 10 μm" assert (
state.attributes.get(ATTR_FRIENDLY_NAME)
== "Air quality Particulate matter < 10 μm"
)
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
assert ( assert (
state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) state.attributes.get(ATTR_UNIT_OF_MEASUREMENT)
@ -71,7 +77,9 @@ async def test_air_quality(
assert state assert state
assert entry.unique_id == f"{entry_id}_air_quality_sulphur_dioxide" assert entry.unique_id == f"{entry_id}_air_quality_sulphur_dioxide"
assert state.state == "0.031" assert state.state == "0.031"
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Sulphur Dioxide (SO2)" assert (
state.attributes.get(ATTR_FRIENDLY_NAME) == "Air quality Sulphur dioxide (SO2)"
)
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
assert ( assert (
state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) state.attributes.get(ATTR_UNIT_OF_MEASUREMENT)
@ -86,7 +94,9 @@ async def test_air_quality(
assert state assert state
assert entry.unique_id == f"{entry_id}_air_quality_nitrogen_dioxide" assert entry.unique_id == f"{entry_id}_air_quality_nitrogen_dioxide"
assert state.state == "0.66" assert state.state == "0.66"
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Nitrogen Dioxide (NO2)" assert (
state.attributes.get(ATTR_FRIENDLY_NAME) == "Air quality Nitrogen dioxide (NO2)"
)
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
assert ( assert (
state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) state.attributes.get(ATTR_UNIT_OF_MEASUREMENT)
@ -101,7 +111,7 @@ async def test_air_quality(
assert state assert state
assert entry.unique_id == f"{entry_id}_air_quality_ozone" assert entry.unique_id == f"{entry_id}_air_quality_ozone"
assert state.state == "17.067" assert state.state == "17.067"
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Ozone" assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Air quality Ozone"
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
assert ( assert (
state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) state.attributes.get(ATTR_UNIT_OF_MEASUREMENT)
@ -117,7 +127,9 @@ async def test_air_quality(
assert entry.unique_id == f"{entry_id}_air_quality_carbon_monoxide" assert entry.unique_id == f"{entry_id}_air_quality_carbon_monoxide"
assert state.state == "0.105" assert state.state == "0.105"
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.CO assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.CO
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Carbon Monoxide (CO)" assert (
state.attributes.get(ATTR_FRIENDLY_NAME) == "Air quality Carbon monoxide (CO)"
)
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
assert ( assert (
state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) state.attributes.get(ATTR_UNIT_OF_MEASUREMENT)
@ -131,7 +143,10 @@ async def test_air_quality(
assert state assert state
assert entry.unique_id == f"{entry_id}_air_quality_air_quality_index" assert entry.unique_id == f"{entry_id}_air_quality_air_quality_index"
assert state.state == "13" assert state.state == "13"
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Air Quality Index (AQI)" assert (
state.attributes.get(ATTR_FRIENDLY_NAME)
== "Air quality Air quality index (AQI)"
)
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
assert ATTR_DEVICE_CLASS not in state.attributes assert ATTR_DEVICE_CLASS not in state.attributes
assert ATTR_UNIT_OF_MEASUREMENT not in state.attributes assert ATTR_UNIT_OF_MEASUREMENT not in state.attributes
@ -142,7 +157,7 @@ async def test_air_quality(
assert device_entry assert device_entry
assert device_entry.identifiers == {(DOMAIN, f"{entry_id}_air_quality")} assert device_entry.identifiers == {(DOMAIN, f"{entry_id}_air_quality")}
assert device_entry.manufacturer == "Ambee" assert device_entry.manufacturer == "Ambee"
assert device_entry.name == "Air Quality" assert device_entry.name == "Air quality"
assert device_entry.entry_type is dr.DeviceEntryType.SERVICE assert device_entry.entry_type is dr.DeviceEntryType.SERVICE
assert not device_entry.model assert not device_entry.model
assert not device_entry.sw_version assert not device_entry.sw_version
@ -163,7 +178,7 @@ async def test_pollen(
assert state assert state
assert entry.unique_id == f"{entry_id}_pollen_grass" assert entry.unique_id == f"{entry_id}_pollen_grass"
assert state.state == "190" assert state.state == "190"
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Grass Pollen" assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Pollen Grass"
assert state.attributes.get(ATTR_ICON) == "mdi:grass" assert state.attributes.get(ATTR_ICON) == "mdi:grass"
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
assert ( assert (
@ -178,7 +193,7 @@ async def test_pollen(
assert state assert state
assert entry.unique_id == f"{entry_id}_pollen_tree" assert entry.unique_id == f"{entry_id}_pollen_tree"
assert state.state == "127" assert state.state == "127"
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Tree Pollen" assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Pollen Tree"
assert state.attributes.get(ATTR_ICON) == "mdi:tree" assert state.attributes.get(ATTR_ICON) == "mdi:tree"
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
assert ( assert (
@ -193,7 +208,7 @@ async def test_pollen(
assert state assert state
assert entry.unique_id == f"{entry_id}_pollen_weed" assert entry.unique_id == f"{entry_id}_pollen_weed"
assert state.state == "95" assert state.state == "95"
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Weed Pollen" assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Pollen Weed"
assert state.attributes.get(ATTR_ICON) == "mdi:sprout" assert state.attributes.get(ATTR_ICON) == "mdi:sprout"
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
assert ( assert (
@ -209,7 +224,7 @@ async def test_pollen(
assert entry.unique_id == f"{entry_id}_pollen_grass_risk" assert entry.unique_id == f"{entry_id}_pollen_grass_risk"
assert state.state == "high" assert state.state == "high"
assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_AMBEE_RISK assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_AMBEE_RISK
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Grass Pollen Risk" assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Pollen Grass risk"
assert state.attributes.get(ATTR_ICON) == "mdi:grass" assert state.attributes.get(ATTR_ICON) == "mdi:grass"
assert ATTR_STATE_CLASS not in state.attributes assert ATTR_STATE_CLASS not in state.attributes
assert ATTR_UNIT_OF_MEASUREMENT not in state.attributes assert ATTR_UNIT_OF_MEASUREMENT not in state.attributes
@ -221,7 +236,7 @@ async def test_pollen(
assert entry.unique_id == f"{entry_id}_pollen_tree_risk" assert entry.unique_id == f"{entry_id}_pollen_tree_risk"
assert state.state == "moderate" assert state.state == "moderate"
assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_AMBEE_RISK assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_AMBEE_RISK
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Tree Pollen Risk" assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Pollen Tree risk"
assert state.attributes.get(ATTR_ICON) == "mdi:tree" assert state.attributes.get(ATTR_ICON) == "mdi:tree"
assert ATTR_STATE_CLASS not in state.attributes assert ATTR_STATE_CLASS not in state.attributes
assert ATTR_UNIT_OF_MEASUREMENT not in state.attributes assert ATTR_UNIT_OF_MEASUREMENT not in state.attributes
@ -233,7 +248,7 @@ async def test_pollen(
assert entry.unique_id == f"{entry_id}_pollen_weed_risk" assert entry.unique_id == f"{entry_id}_pollen_weed_risk"
assert state.state == "high" assert state.state == "high"
assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_AMBEE_RISK assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_AMBEE_RISK
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Weed Pollen Risk" assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Pollen Weed risk"
assert state.attributes.get(ATTR_ICON) == "mdi:sprout" assert state.attributes.get(ATTR_ICON) == "mdi:sprout"
assert ATTR_STATE_CLASS not in state.attributes assert ATTR_STATE_CLASS not in state.attributes
assert ATTR_UNIT_OF_MEASUREMENT not in state.attributes assert ATTR_UNIT_OF_MEASUREMENT not in state.attributes
@ -286,20 +301,20 @@ async def test_pollen_disabled_by_default(
@pytest.mark.parametrize( @pytest.mark.parametrize(
"key,icon,name,value", "key,icon,name,value",
[ [
("grass_poaceae", "mdi:grass", "Poaceae Grass Pollen", "190"), ("grass_poaceae", "mdi:grass", "Poaceae grass", "190"),
("tree_alder", "mdi:tree", "Alder Tree Pollen", "0"), ("tree_alder", "mdi:tree", "Alder tree", "0"),
("tree_birch", "mdi:tree", "Birch Tree Pollen", "35"), ("tree_birch", "mdi:tree", "Birch tree", "35"),
("tree_cypress", "mdi:tree", "Cypress Tree Pollen", "0"), ("tree_cypress", "mdi:tree", "Cypress tree", "0"),
("tree_elm", "mdi:tree", "Elm Tree Pollen", "0"), ("tree_elm", "mdi:tree", "Elm tree", "0"),
("tree_hazel", "mdi:tree", "Hazel Tree Pollen", "0"), ("tree_hazel", "mdi:tree", "Hazel tree", "0"),
("tree_oak", "mdi:tree", "Oak Tree Pollen", "55"), ("tree_oak", "mdi:tree", "Oak tree", "55"),
("tree_pine", "mdi:tree", "Pine Tree Pollen", "30"), ("tree_pine", "mdi:tree", "Pine tree", "30"),
("tree_plane", "mdi:tree", "Plane Tree Pollen", "5"), ("tree_plane", "mdi:tree", "Plane tree", "5"),
("tree_poplar", "mdi:tree", "Poplar Tree Pollen", "0"), ("tree_poplar", "mdi:tree", "Poplar tree", "0"),
("weed_chenopod", "mdi:sprout", "Chenopod Weed Pollen", "0"), ("weed_chenopod", "mdi:sprout", "Chenopod weed", "0"),
("weed_mugwort", "mdi:sprout", "Mugwort Weed Pollen", "1"), ("weed_mugwort", "mdi:sprout", "Mugwort weed", "1"),
("weed_nettle", "mdi:sprout", "Nettle Weed Pollen", "88"), ("weed_nettle", "mdi:sprout", "Nettle weed", "88"),
("weed_ragweed", "mdi:sprout", "Ragweed Weed Pollen", "3"), ("weed_ragweed", "mdi:sprout", "Ragweed weed", "3"),
], ],
) )
async def test_pollen_enable_disable_by_defaults( async def test_pollen_enable_disable_by_defaults(
@ -335,7 +350,7 @@ async def test_pollen_enable_disable_by_defaults(
assert state assert state
assert entry.unique_id == f"{entry_id}_pollen_{key}" assert entry.unique_id == f"{entry_id}_pollen_{key}"
assert state.state == value assert state.state == value
assert state.attributes.get(ATTR_FRIENDLY_NAME) == name assert state.attributes.get(ATTR_FRIENDLY_NAME) == f"Pollen {name}"
assert state.attributes.get(ATTR_ICON) == icon assert state.attributes.get(ATTR_ICON) == icon
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
assert ( assert (