diff --git a/homeassistant/components/arlo/sensor.py b/homeassistant/components/arlo/sensor.py index 883fd011c52..e78c8b7bf49 100644 --- a/homeassistant/components/arlo/sensor.py +++ b/homeassistant/components/arlo/sensor.py @@ -38,7 +38,6 @@ SENSOR_TYPES = ( key="battery_level", name="Battery Level", unit_of_measurement=PERCENTAGE, - icon="mdi:battery-50", device_class=DEVICE_CLASS_BATTERY, ), SensorEntityDescription( @@ -48,14 +47,12 @@ SENSOR_TYPES = ( key="temperature", name="Temperature", unit_of_measurement=TEMP_CELSIUS, - icon="mdi:thermometer", device_class=DEVICE_CLASS_TEMPERATURE, ), SensorEntityDescription( key="humidity", name="Humidity", unit_of_measurement=PERCENTAGE, - icon="mdi:water-percent", device_class=DEVICE_CLASS_HUMIDITY, ), SensorEntityDescription( diff --git a/tests/components/arlo/test_sensor.py b/tests/components/arlo/test_sensor.py index 34d5088397d..bf67ab21c97 100644 --- a/tests/components/arlo/test_sensor.py +++ b/tests/components/arlo/test_sensor.py @@ -8,6 +8,7 @@ from homeassistant.components.arlo import DATA_ARLO, sensor as arlo from homeassistant.components.arlo.sensor import SENSOR_TYPES from homeassistant.const import ( ATTR_ATTRIBUTION, + DEVICE_CLASS_BATTERY, DEVICE_CLASS_HUMIDITY, DEVICE_CLASS_TEMPERATURE, PERCENTAGE, @@ -161,14 +162,14 @@ def test_sensor_state_default(default_sensor): assert default_sensor.state is None -def test_sensor_icon_battery(battery_sensor): - """Test the battery icon.""" - assert battery_sensor.icon == "mdi:battery-50" +def test_sensor_device_class__battery(battery_sensor): + """Test the battery device_class.""" + assert battery_sensor.device_class == DEVICE_CLASS_BATTERY -def test_sensor_icon(temperature_sensor): - """Test the icon property.""" - assert temperature_sensor.icon == "mdi:thermometer" +def test_sensor_device_class(temperature_sensor): + """Test the device_class property.""" + assert temperature_sensor.device_class == DEVICE_CLASS_TEMPERATURE def test_unit_of_measure(default_sensor, battery_sensor):