parent
fe22aa0b4b
commit
c50c920589
|
@ -8,11 +8,7 @@ from typing import TYPE_CHECKING
|
|||
|
||||
import voluptuous as vol
|
||||
|
||||
from homeassistant.components.sensor import (
|
||||
PLATFORM_SCHEMA,
|
||||
SensorEntity,
|
||||
SensorStateClass,
|
||||
)
|
||||
from homeassistant.components.sensor import PLATFORM_SCHEMA, SensorEntity
|
||||
from homeassistant.config_entries import ConfigEntry
|
||||
from homeassistant.const import (
|
||||
ATTR_UNIT_OF_MEASUREMENT,
|
||||
|
@ -135,7 +131,6 @@ class DerivativeSensor(RestoreEntity, SensorEntity):
|
|||
|
||||
_attr_icon = ICON
|
||||
_attr_should_poll = False
|
||||
_attr_state_class = SensorStateClass.MEASUREMENT
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
|
|
|
@ -4,7 +4,6 @@ from math import sin
|
|||
import random
|
||||
from unittest.mock import patch
|
||||
|
||||
from homeassistant.components.sensor import ATTR_STATE_CLASS, SensorStateClass
|
||||
from homeassistant.const import UnitOfPower, UnitOfTime
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.setup import async_setup_component
|
||||
|
@ -79,7 +78,6 @@ async def setup_tests(hass, config, times, values, expected_state):
|
|||
assert state is not None
|
||||
|
||||
assert round(float(state.state), config["sensor"]["round"]) == expected_state
|
||||
assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
|
||||
|
||||
return state
|
||||
|
||||
|
|
Loading…
Reference in New Issue