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

Adjust unit_system type hints (#80946)

Browse Source
This commit is contained in:
epenet 2022-10-25 15:56:18 +02:00 committed by GitHub
parent 3aa64aaaf1
commit 0018939142
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 67 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
homeassistant/util/unit_system.py
View File

@ -100,7 +100,7 @@ class UnitSystem:
name: str, name: str,
*, *,
accumulated_precipitation: str, accumulated_precipitation: str,
conversions: dict[tuple[str | None, str | None], str], conversions: dict[tuple[SensorDeviceClass | str | None, str | None], str],
length: str, length: str,
mass: str, mass: str,
pressure: str, pressure: str,

93
tests/util/test_unit_system.py
View File

@ -1,6 +1,9 @@
"""Test the unit system helper.""" """Test the unit system helper."""
from __future__ import annotations
import pytest import pytest
from homeassistant.components.sensor import SensorDeviceClass
from homeassistant.const import ( from homeassistant.const import (
ACCUMULATED_PRECIPITATION, ACCUMULATED_PRECIPITATION,
LENGTH, LENGTH,
@ -388,38 +391,74 @@ def test_get_unit_system_invalid(key: str) -> None:
@pytest.mark.parametrize( @pytest.mark.parametrize(
"unit_system, device_class, original_unit, state_unit", "unit_system, device_class, original_unit, state_unit",
( (
# Test distance conversion (METRIC_SYSTEM, SensorDeviceClass.DISTANCE, LENGTH_FEET, LENGTH_METERS),
(METRIC_SYSTEM, "distance", LENGTH_FEET, LENGTH_METERS), (METRIC_SYSTEM, SensorDeviceClass.DISTANCE, LENGTH_INCHES, LENGTH_MILLIMETERS),
(METRIC_SYSTEM, "distance", LENGTH_INCHES, LENGTH_MILLIMETERS), (METRIC_SYSTEM, SensorDeviceClass.DISTANCE, LENGTH_MILES, LENGTH_KILOMETERS),
(METRIC_SYSTEM, "distance", LENGTH_MILES, LENGTH_KILOMETERS), (METRIC_SYSTEM, SensorDeviceClass.DISTANCE, LENGTH_YARD, LENGTH_METERS),
(METRIC_SYSTEM, "distance", LENGTH_YARD, LENGTH_METERS), (METRIC_SYSTEM, SensorDeviceClass.DISTANCE, LENGTH_KILOMETERS, None),
(METRIC_SYSTEM, "distance", LENGTH_KILOMETERS, None), (METRIC_SYSTEM, SensorDeviceClass.DISTANCE, "very_long", None),
(METRIC_SYSTEM, "distance", "very_long", None),
# Test speed conversion # Test speed conversion
(METRIC_SYSTEM, "speed", SPEED_FEET_PER_SECOND, SPEED_KILOMETERS_PER_HOUR), (
(METRIC_SYSTEM, "speed", SPEED_MILES_PER_HOUR, SPEED_KILOMETERS_PER_HOUR), METRIC_SYSTEM,
(METRIC_SYSTEM, "speed", SPEED_KILOMETERS_PER_HOUR, None), SensorDeviceClass.SPEED,
(METRIC_SYSTEM, "speed", SPEED_KNOTS, None), SPEED_FEET_PER_SECOND,
(METRIC_SYSTEM, "speed", SPEED_METERS_PER_SECOND, None), SPEED_KILOMETERS_PER_HOUR,
(METRIC_SYSTEM, "speed", "very_fast", None), ),
# Test distance conversion (
(US_CUSTOMARY_SYSTEM, "distance", LENGTH_CENTIMETERS, LENGTH_INCHES), METRIC_SYSTEM,
(US_CUSTOMARY_SYSTEM, "distance", LENGTH_KILOMETERS, LENGTH_MILES), SensorDeviceClass.SPEED,
(US_CUSTOMARY_SYSTEM, "distance", LENGTH_METERS, LENGTH_FEET), SPEED_MILES_PER_HOUR,
(US_CUSTOMARY_SYSTEM, "distance", LENGTH_MILLIMETERS, LENGTH_INCHES), SPEED_KILOMETERS_PER_HOUR,
(US_CUSTOMARY_SYSTEM, "distance", LENGTH_MILES, None), ),
(US_CUSTOMARY_SYSTEM, "distance", "very_long", None), (METRIC_SYSTEM, SensorDeviceClass.SPEED, SPEED_KILOMETERS_PER_HOUR, None),
(METRIC_SYSTEM, SensorDeviceClass.SPEED, SPEED_KNOTS, None),
(METRIC_SYSTEM, SensorDeviceClass.SPEED, SPEED_METERS_PER_SECOND, None),
(METRIC_SYSTEM, SensorDeviceClass.SPEED, "very_fast", None),
(
US_CUSTOMARY_SYSTEM,
SensorDeviceClass.DISTANCE,
LENGTH_CENTIMETERS,
LENGTH_INCHES,
),
(
US_CUSTOMARY_SYSTEM,
SensorDeviceClass.DISTANCE,
LENGTH_KILOMETERS,
LENGTH_MILES,
),
(US_CUSTOMARY_SYSTEM, SensorDeviceClass.DISTANCE, LENGTH_METERS, LENGTH_FEET),
(
US_CUSTOMARY_SYSTEM,
SensorDeviceClass.DISTANCE,
LENGTH_MILLIMETERS,
LENGTH_INCHES,
),
(US_CUSTOMARY_SYSTEM, SensorDeviceClass.DISTANCE, LENGTH_MILES, None),
(US_CUSTOMARY_SYSTEM, SensorDeviceClass.DISTANCE, "very_long", None),
# Test speed conversion # Test speed conversion
(US_CUSTOMARY_SYSTEM, "speed", SPEED_METERS_PER_SECOND, SPEED_MILES_PER_HOUR), (
(US_CUSTOMARY_SYSTEM, "speed", SPEED_KILOMETERS_PER_HOUR, SPEED_MILES_PER_HOUR), US_CUSTOMARY_SYSTEM,
(US_CUSTOMARY_SYSTEM, "speed", SPEED_FEET_PER_SECOND, None), SensorDeviceClass.SPEED,
(US_CUSTOMARY_SYSTEM, "speed", SPEED_KNOTS, None), SPEED_METERS_PER_SECOND,
(US_CUSTOMARY_SYSTEM, "speed", SPEED_MILES_PER_HOUR, None), SPEED_MILES_PER_HOUR,
(US_CUSTOMARY_SYSTEM, "speed", "very_fast", None), ),
(
US_CUSTOMARY_SYSTEM,
SensorDeviceClass.SPEED,
SPEED_KILOMETERS_PER_HOUR,
SPEED_MILES_PER_HOUR,
),
(US_CUSTOMARY_SYSTEM, SensorDeviceClass.SPEED, SPEED_FEET_PER_SECOND, None),
(US_CUSTOMARY_SYSTEM, SensorDeviceClass.SPEED, SPEED_KNOTS, None),
(US_CUSTOMARY_SYSTEM, SensorDeviceClass.SPEED, SPEED_MILES_PER_HOUR, None),
(US_CUSTOMARY_SYSTEM, SensorDeviceClass.SPEED, "very_fast", None),
), ),
) )
def test_get_converted_unit( def test_get_converted_unit(
unit_system, device_class, original_unit, state_unit unit_system: UnitSystem,
device_class: SensorDeviceClass,
original_unit: str,
state_unit: str | None,
) -> None: ) -> None:
"""Test unit conversion rules.""" """Test unit conversion rules."""
assert unit_system.get_converted_unit(device_class, original_unit) == state_unit assert unit_system.get_converted_unit(device_class, original_unit) == state_unit