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

Handle negative numbers in sensor long term statistics (#55708)

Browse Source 
* Handle negative numbers in sensor long term statistics

* Use negative states in tests
This commit is contained in:
Erik Montnemery 2021-09-04 10:47:42 +02:00 committed by GitHub
parent b3181a0ab2
commit 38d42de2c0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 52 additions and 56 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
homeassistant/components/sensor/recorder.py
View File

@ -129,13 +129,6 @@ def _get_entities(hass: HomeAssistant) -> list[tuple[str, str, str | None]]:
return entity_ids return entity_ids
# Faster than try/except
# From https://stackoverflow.com/a/23639915
def _is_number(s: str) -> bool: # pylint: disable=invalid-name
"""Return True if string is a number."""
return s.replace(".", "", 1).isdigit()
def _time_weighted_average( def _time_weighted_average(
fstates: list[tuple[float, State]], start: datetime.datetime, end: datetime.datetime fstates: list[tuple[float, State]], start: datetime.datetime, end: datetime.datetime
) -> float: ) -> float:
@ -190,9 +183,13 @@ def _normalize_states(
if device_class not in UNIT_CONVERSIONS: if device_class not in UNIT_CONVERSIONS:
# We're not normalizing this device class, return the state as they are # We're not normalizing this device class, return the state as they are
fstates = [ fstates = []
(float(el.state), el) for el in entity_history if _is_number(el.state) for state in entity_history:
] try:
fstates.append((float(state.state), state))
except ValueError:
continue
if fstates: if fstates:
all_units = _get_units(fstates) all_units = _get_units(fstates)
if len(all_units) > 1: if len(all_units) > 1:
@ -220,10 +217,7 @@ def _normalize_states(
fstates = [] fstates = []
for state in entity_history: for state in entity_history:
# Exclude non numerical states from statistics try:
if not _is_number(state.state):
continue
fstate = float(state.state) fstate = float(state.state)
unit = state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) unit = state.attributes.get(ATTR_UNIT_OF_MEASUREMENT)
# Exclude unsupported units from statistics # Exclude unsupported units from statistics
@ -236,6 +230,8 @@ def _normalize_states(
continue continue
fstates.append((UNIT_CONVERSIONS[device_class][unit](fstate), state)) fstates.append((UNIT_CONVERSIONS[device_class][unit](fstate), state))
except ValueError:
continue
return DEVICE_CLASS_UNITS[device_class], fstates return DEVICE_CLASS_UNITS[device_class], fstates

64
tests/components/sensor/test_recorder.py
View File

@ -50,18 +50,18 @@ GAS_SENSOR_ATTRIBUTES = {
@pytest.mark.parametrize( @pytest.mark.parametrize(
"device_class,unit,native_unit,mean,min,max", "device_class,unit,native_unit,mean,min,max",
[ [
(None, "%", "%", 16.440677, 10, 30), (None, "%", "%", 13.050847, -10, 30),
("battery", "%", "%", 16.440677, 10, 30), ("battery", "%", "%", 13.050847, -10, 30),
("battery", None, None, 16.440677, 10, 30), ("battery", None, None, 13.050847, -10, 30),
("humidity", "%", "%", 16.440677, 10, 30), ("humidity", "%", "%", 13.050847, -10, 30),
("humidity", None, None, 16.440677, 10, 30), ("humidity", None, None, 13.050847, -10, 30),
("pressure", "Pa", "Pa", 16.440677, 10, 30), ("pressure", "Pa", "Pa", 13.050847, -10, 30),
("pressure", "hPa", "Pa", 1644.0677, 1000, 3000), ("pressure", "hPa", "Pa", 1305.0847, -1000, 3000),
("pressure", "mbar", "Pa", 1644.0677, 1000, 3000), ("pressure", "mbar", "Pa", 1305.0847, -1000, 3000),
("pressure", "inHg", "Pa", 55674.53, 33863.89, 101591.67), ("pressure", "inHg", "Pa", 44195.25, -33863.89, 101591.67),
("pressure", "psi", "Pa", 113354.48, 68947.57, 206842.71), ("pressure", "psi", "Pa", 89982.42, -68947.57, 206842.71),
("temperature", "°C", "°C", 16.440677, 10, 30), ("temperature", "°C", "°C", 13.050847, -10, 30),
("temperature", "°F", "°C", -8.644068, -12.22222, -1.111111), ("temperature", "°F", "°C", -10.52731, -23.33333, -1.111111),
], ],
) )
def test_compile_hourly_statistics( def test_compile_hourly_statistics(
@ -155,8 +155,8 @@ def test_compile_hourly_statistics_unsupported(hass_recorder, caplog, attributes
{ {
"statistic_id": "sensor.test1", "statistic_id": "sensor.test1",
"start": process_timestamp_to_utc_isoformat(zero), "start": process_timestamp_to_utc_isoformat(zero),
"mean": approx(16.440677966101696), "mean": approx(13.050847),
"min": approx(10.0), "min": approx(-10.0),
"max": approx(30.0), "max": approx(30.0),
"last_reset": None, "last_reset": None,
"state": None, "state": None,
@ -167,8 +167,8 @@ def test_compile_hourly_statistics_unsupported(hass_recorder, caplog, attributes
{ {
"statistic_id": "sensor.test6", "statistic_id": "sensor.test6",
"start": process_timestamp_to_utc_isoformat(zero), "start": process_timestamp_to_utc_isoformat(zero),
"mean": approx(16.440677966101696), "mean": approx(13.050847),
"min": approx(10.0), "min": approx(-10.0),
"max": approx(30.0), "max": approx(30.0),
"last_reset": None, "last_reset": None,
"state": None, "state": None,
@ -179,8 +179,8 @@ def test_compile_hourly_statistics_unsupported(hass_recorder, caplog, attributes
{ {
"statistic_id": "sensor.test7", "statistic_id": "sensor.test7",
"start": process_timestamp_to_utc_isoformat(zero), "start": process_timestamp_to_utc_isoformat(zero),
"mean": approx(16.440677966101696), "mean": approx(13.050847),
"min": approx(10.0), "min": approx(-10.0),
"max": approx(30.0), "max": approx(30.0),
"last_reset": None, "last_reset": None,
"state": None, "state": None,
@ -988,10 +988,10 @@ def test_list_statistic_ids_unsupported(hass_recorder, caplog, _attributes):
@pytest.mark.parametrize( @pytest.mark.parametrize(
"device_class,unit,native_unit,mean,min,max", "device_class,unit,native_unit,mean,min,max",
[ [
(None, None, None, 16.440677, 10, 30), (None, None, None, 13.050847, -10, 30),
(None, "%", "%", 16.440677, 10, 30), (None, "%", "%", 13.050847, -10, 30),
("battery", "%", "%", 16.440677, 10, 30), ("battery", "%", "%", 13.050847, -10, 30),
("battery", None, None, 16.440677, 10, 30), ("battery", None, None, 13.050847, -10, 30),
], ],
) )
def test_compile_hourly_statistics_changing_units_1( def test_compile_hourly_statistics_changing_units_1(
@ -1074,10 +1074,10 @@ def test_compile_hourly_statistics_changing_units_1(
@pytest.mark.parametrize( @pytest.mark.parametrize(
"device_class,unit,native_unit,mean,min,max", "device_class,unit,native_unit,mean,min,max",
[ [
(None, None, None, 16.440677, 10, 30), (None, None, None, 13.050847, -10, 30),
(None, "%", "%", 16.440677, 10, 30), (None, "%", "%", 13.050847, -10, 30),
("battery", "%", "%", 16.440677, 10, 30), ("battery", "%", "%", 13.050847, -10, 30),
("battery", None, None, 16.440677, 10, 30), ("battery", None, None, 13.050847, -10, 30),
], ],
) )
def test_compile_hourly_statistics_changing_units_2( def test_compile_hourly_statistics_changing_units_2(
@ -1119,10 +1119,10 @@ def test_compile_hourly_statistics_changing_units_2(
@pytest.mark.parametrize( @pytest.mark.parametrize(
"device_class,unit,native_unit,mean,min,max", "device_class,unit,native_unit,mean,min,max",
[ [
(None, None, None, 16.440677, 10, 30), (None, None, None, 13.050847, -10, 30),
(None, "%", "%", 16.440677, 10, 30), (None, "%", "%", 13.050847, -10, 30),
("battery", "%", "%", 16.440677, 10, 30), ("battery", "%", "%", 13.050847, -10, 30),
("battery", None, None, 16.440677, 10, 30), ("battery", None, None, 13.050847, -10, 30),
], ],
) )
def test_compile_hourly_statistics_changing_units_3( def test_compile_hourly_statistics_changing_units_3(
@ -1203,7 +1203,7 @@ def test_compile_hourly_statistics_changing_units_3(
@pytest.mark.parametrize( @pytest.mark.parametrize(
"device_class,unit,native_unit,mean,min,max", "device_class,unit,native_unit,mean,min,max",
[ [
(None, None, None, 16.440677, 10, 30), (None, None, None, 13.050847, -10, 30),
], ],
) )
def test_compile_hourly_statistics_changing_statistics( def test_compile_hourly_statistics_changing_statistics(
@ -1309,7 +1309,7 @@ def record_states(hass, zero, entity_id, attributes):
states = {entity_id: []} states = {entity_id: []}
with patch("homeassistant.components.recorder.dt_util.utcnow", return_value=one): with patch("homeassistant.components.recorder.dt_util.utcnow", return_value=one):
states[entity_id].append(set_state(entity_id, "10", attributes=attributes)) states[entity_id].append(set_state(entity_id, "-10", attributes=attributes))
with patch("homeassistant.components.recorder.dt_util.utcnow", return_value=two): with patch("homeassistant.components.recorder.dt_util.utcnow", return_value=two):
states[entity_id].append(set_state(entity_id, "15", attributes=attributes)) states[entity_id].append(set_state(entity_id, "15", attributes=attributes))