evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Migrate energy units to an enum (#80998)

Browse source
This commit is contained in:
epenet 2022-10-26 13:56:51 +02:00 • committed by GitHub
parent 9d3442055b
commit 842cb18d39
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
8 changed files with 85 additions and 93 deletions
Download patch file Download diff file Expand all files Collapse all files

25
homeassistant/components/energy/sensor.py
View file

@ -17,12 +17,9 @@ from homeassistant.components.sensor import (
from homeassistant.components.sensor.recorder import reset_detected from homeassistant.components.sensor.recorder import reset_detected
from homeassistant.const import ( from homeassistant.const import (
ATTR_UNIT_OF_MEASUREMENT, ATTR_UNIT_OF_MEASUREMENT,
ENERGY_GIGA_JOULE,
ENERGY_KILO_WATT_HOUR,
ENERGY_MEGA_WATT_HOUR,
ENERGY_WATT_HOUR,
VOLUME_CUBIC_FEET, VOLUME_CUBIC_FEET,
VOLUME_CUBIC_METERS, VOLUME_CUBIC_METERS,
UnitOfEnergy,
) )
from homeassistant.core import ( from homeassistant.core import (
HomeAssistant, HomeAssistant,
@ -46,10 +43,10 @@ SUPPORTED_STATE_CLASSES = [
SensorStateClass.TOTAL_INCREASING, SensorStateClass.TOTAL_INCREASING,
] ]
VALID_ENERGY_UNITS = [ VALID_ENERGY_UNITS = [
ENERGY_WATT_HOUR, UnitOfEnergy.WATT_HOUR,
ENERGY_KILO_WATT_HOUR, UnitOfEnergy.KILO_WATT_HOUR,
ENERGY_MEGA_WATT_HOUR, UnitOfEnergy.MEGA_WATT_HOUR,
ENERGY_GIGA_JOULE, UnitOfEnergy.GIGA_JOULE,
] ]
VALID_ENERGY_UNITS_GAS = [VOLUME_CUBIC_FEET, VOLUME_CUBIC_METERS] + VALID_ENERGY_UNITS VALID_ENERGY_UNITS_GAS = [VOLUME_CUBIC_FEET, VOLUME_CUBIC_METERS] + VALID_ENERGY_UNITS
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@ -286,17 +283,17 @@ class EnergyCostSensor(SensorEntity):
return return
if energy_price_state.attributes.get(ATTR_UNIT_OF_MEASUREMENT, "").endswith( if energy_price_state.attributes.get(ATTR_UNIT_OF_MEASUREMENT, "").endswith(
f"/{ENERGY_WATT_HOUR}" f"/{UnitOfEnergy.WATT_HOUR}"
): ):
energy_price *= 1000.0 energy_price *= 1000.0
if energy_price_state.attributes.get(ATTR_UNIT_OF_MEASUREMENT, "").endswith( if energy_price_state.attributes.get(ATTR_UNIT_OF_MEASUREMENT, "").endswith(
f"/{ENERGY_MEGA_WATT_HOUR}" f"/{UnitOfEnergy.MEGA_WATT_HOUR}"
): ):
energy_price /= 1000.0 energy_price /= 1000.0
if energy_price_state.attributes.get(ATTR_UNIT_OF_MEASUREMENT, "").endswith( if energy_price_state.attributes.get(ATTR_UNIT_OF_MEASUREMENT, "").endswith(
f"/{ENERGY_GIGA_JOULE}" f"/{UnitOfEnergy.GIGA_JOULE}"
): ):
energy_price /= 1000 / 3.6 energy_price /= 1000 / 3.6
@ -319,11 +316,11 @@ class EnergyCostSensor(SensorEntity):
if energy_unit not in VALID_ENERGY_UNITS_GAS: if energy_unit not in VALID_ENERGY_UNITS_GAS:
energy_unit = None energy_unit = None
if energy_unit == ENERGY_WATT_HOUR: if energy_unit == UnitOfEnergy.WATT_HOUR:
energy_price /= 1000 energy_price /= 1000
elif energy_unit == ENERGY_MEGA_WATT_HOUR: elif energy_unit == UnitOfEnergy.MEGA_WATT_HOUR:
energy_price *= 1000 energy_price *= 1000
elif energy_unit == ENERGY_GIGA_JOULE: elif energy_unit == UnitOfEnergy.GIGA_JOULE:
energy_price *= 1000 / 3.6 energy_price *= 1000 / 3.6
if energy_unit is None: if energy_unit is None:

21
homeassistant/components/energy/validate.py
View file

@ -9,14 +9,11 @@ from typing import Any
from homeassistant.components import recorder, sensor from homeassistant.components import recorder, sensor
from homeassistant.const import ( from homeassistant.const import (
ATTR_DEVICE_CLASS, ATTR_DEVICE_CLASS,
ENERGY_GIGA_JOULE,
ENERGY_KILO_WATT_HOUR,
ENERGY_MEGA_WATT_HOUR,
ENERGY_WATT_HOUR,
STATE_UNAVAILABLE, STATE_UNAVAILABLE,
STATE_UNKNOWN, STATE_UNKNOWN,
VOLUME_CUBIC_FEET, VOLUME_CUBIC_FEET,
VOLUME_CUBIC_METERS, VOLUME_CUBIC_METERS,
UnitOfEnergy,
) )
from homeassistant.core import HomeAssistant, callback, valid_entity_id from homeassistant.core import HomeAssistant, callback, valid_entity_id
@ -26,10 +23,10 @@ from .const import DOMAIN
ENERGY_USAGE_DEVICE_CLASSES = (sensor.SensorDeviceClass.ENERGY,) ENERGY_USAGE_DEVICE_CLASSES = (sensor.SensorDeviceClass.ENERGY,)
ENERGY_USAGE_UNITS = { ENERGY_USAGE_UNITS = {
sensor.SensorDeviceClass.ENERGY: ( sensor.SensorDeviceClass.ENERGY: (
ENERGY_KILO_WATT_HOUR, UnitOfEnergy.KILO_WATT_HOUR,
ENERGY_MEGA_WATT_HOUR, UnitOfEnergy.MEGA_WATT_HOUR,
ENERGY_WATT_HOUR, UnitOfEnergy.WATT_HOUR,
ENERGY_GIGA_JOULE, UnitOfEnergy.GIGA_JOULE,
) )
} }
ENERGY_PRICE_UNITS = tuple( ENERGY_PRICE_UNITS = tuple(
@ -43,10 +40,10 @@ GAS_USAGE_DEVICE_CLASSES = (
) )
GAS_USAGE_UNITS = { GAS_USAGE_UNITS = {
sensor.SensorDeviceClass.ENERGY: ( sensor.SensorDeviceClass.ENERGY: (
ENERGY_WATT_HOUR, UnitOfEnergy.WATT_HOUR,
ENERGY_KILO_WATT_HOUR, UnitOfEnergy.KILO_WATT_HOUR,
ENERGY_MEGA_WATT_HOUR, UnitOfEnergy.MEGA_WATT_HOUR,
ENERGY_GIGA_JOULE, UnitOfEnergy.GIGA_JOULE,
), ),
sensor.SensorDeviceClass.GAS: (VOLUME_CUBIC_METERS, VOLUME_CUBIC_FEET), sensor.SensorDeviceClass.GAS: (VOLUME_CUBIC_METERS, VOLUME_CUBIC_FEET),
} }

4
homeassistant/components/energy/websocket_api.py
View file

@ -13,7 +13,7 @@ from typing import Any, cast
import voluptuous as vol import voluptuous as vol
from homeassistant.components import recorder, websocket_api from homeassistant.components import recorder, websocket_api
from homeassistant.const import ENERGY_KILO_WATT_HOUR from homeassistant.const import UnitOfEnergy
from homeassistant.core import HomeAssistant, callback from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.integration_platform import ( from homeassistant.helpers.integration_platform import (
async_process_integration_platforms, async_process_integration_platforms,
@ -273,7 +273,7 @@ async def ws_get_fossil_energy_consumption(
statistic_ids, statistic_ids,
"hour", "hour",
True, True,
{"energy": ENERGY_KILO_WATT_HOUR}, {"energy": UnitOfEnergy.KILO_WATT_HOUR},
) )
def _combine_sum_statistics( def _combine_sum_statistics(

14
homeassistant/const.py
View file

@ -486,11 +486,23 @@ POWER_BTU_PER_HOUR: Final = "BTU/h"
# Reactive power units # Reactive power units
POWER_VOLT_AMPERE_REACTIVE: Final = "var" POWER_VOLT_AMPERE_REACTIVE: Final = "var"
# Energy units # Energy units
ENERGY_GIGA_JOULE: Final = "GJ" class UnitOfEnergy(StrEnum):
"""Energy units."""
GIGA_JOULE = "GJ"
KILO_WATT_HOUR = "kWh"
MEGA_WATT_HOUR = "MWh"
WATT_HOUR = "Wh"
ENERGY_KILO_WATT_HOUR: Final = "kWh" ENERGY_KILO_WATT_HOUR: Final = "kWh"
"""Deprecated: please use UnitOfEnergy.KILO_WATT_HOUR."""
ENERGY_MEGA_WATT_HOUR: Final = "MWh" ENERGY_MEGA_WATT_HOUR: Final = "MWh"
"""Deprecated: please use UnitOfEnergy.MEGA_WATT_HOUR."""
ENERGY_WATT_HOUR: Final = "Wh" ENERGY_WATT_HOUR: Final = "Wh"
"""Deprecated: please use UnitOfEnergy.WATT_HOUR."""
# Electric_current units # Electric_current units
ELECTRIC_CURRENT_MILLIAMPERE: Final = "mA" ELECTRIC_CURRENT_MILLIAMPERE: Final = "mA"

23
homeassistant/util/unit_conversion.py
View file

@ -2,10 +2,6 @@
from __future__ import annotations from __future__ import annotations
from homeassistant.const import ( from homeassistant.const import (
ENERGY_GIGA_JOULE,
ENERGY_KILO_WATT_HOUR,
ENERGY_MEGA_WATT_HOUR,
ENERGY_WATT_HOUR,
LENGTH_CENTIMETERS, LENGTH_CENTIMETERS,
LENGTH_FEET, LENGTH_FEET,
LENGTH_INCHES, LENGTH_INCHES,
@ -46,6 +42,7 @@ from homeassistant.const import (
VOLUME_GALLONS, VOLUME_GALLONS,
VOLUME_LITERS, VOLUME_LITERS,
VOLUME_MILLILITERS, VOLUME_MILLILITERS,
UnitOfEnergy,
UnitOfVolumetricFlux, UnitOfVolumetricFlux,
) )
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
@ -151,18 +148,18 @@ class EnergyConverter(BaseUnitConverter):
"""Utility to convert energy values.""" """Utility to convert energy values."""
UNIT_CLASS = "energy" UNIT_CLASS = "energy"
NORMALIZED_UNIT = ENERGY_KILO_WATT_HOUR NORMALIZED_UNIT = UnitOfEnergy.KILO_WATT_HOUR
_UNIT_CONVERSION: dict[str, float] = { _UNIT_CONVERSION: dict[str, float] = {
ENERGY_WATT_HOUR: 1 * 1000, UnitOfEnergy.WATT_HOUR: 1 * 1000,
ENERGY_KILO_WATT_HOUR: 1, UnitOfEnergy.KILO_WATT_HOUR: 1,
ENERGY_MEGA_WATT_HOUR: 1 / 1000, UnitOfEnergy.MEGA_WATT_HOUR: 1 / 1000,
ENERGY_GIGA_JOULE: 3.6 / 1000, UnitOfEnergy.GIGA_JOULE: 3.6 / 1000,
} }
VALID_UNITS = { VALID_UNITS = {
ENERGY_WATT_HOUR, UnitOfEnergy.WATT_HOUR,
ENERGY_KILO_WATT_HOUR, UnitOfEnergy.KILO_WATT_HOUR,
ENERGY_MEGA_WATT_HOUR, UnitOfEnergy.MEGA_WATT_HOUR,
ENERGY_GIGA_JOULE, UnitOfEnergy.GIGA_JOULE,
} }

35
tests/components/energy/test_sensor.py
View file

@ -16,13 +16,10 @@ from homeassistant.components.sensor.recorder import compile_statistics
from homeassistant.const import ( from homeassistant.const import (
ATTR_DEVICE_CLASS, ATTR_DEVICE_CLASS,
ATTR_UNIT_OF_MEASUREMENT, ATTR_UNIT_OF_MEASUREMENT,
ENERGY_GIGA_JOULE,
ENERGY_KILO_WATT_HOUR,
ENERGY_MEGA_WATT_HOUR,
ENERGY_WATT_HOUR,
STATE_UNKNOWN, STATE_UNKNOWN,
VOLUME_CUBIC_FEET, VOLUME_CUBIC_FEET,
VOLUME_CUBIC_METERS, VOLUME_CUBIC_METERS,
UnitOfEnergy,
) )
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -143,7 +140,7 @@ async def test_cost_sensor_price_entity_total_increasing(
return compile_statistics(hass, now, now + timedelta(seconds=1)).platform_stats return compile_statistics(hass, now, now + timedelta(seconds=1)).platform_stats
energy_attributes = { energy_attributes = {
ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, ATTR_UNIT_OF_MEASUREMENT: UnitOfEnergy.KILO_WATT_HOUR,
ATTR_STATE_CLASS: SensorStateClass.TOTAL_INCREASING, ATTR_STATE_CLASS: SensorStateClass.TOTAL_INCREASING,
} }
@ -347,7 +344,7 @@ async def test_cost_sensor_price_entity_total(
return compile_statistics(hass, now, now + timedelta(seconds=1)).platform_stats return compile_statistics(hass, now, now + timedelta(seconds=1)).platform_stats
energy_attributes = { energy_attributes = {
ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, ATTR_UNIT_OF_MEASUREMENT: UnitOfEnergy.KILO_WATT_HOUR,
ATTR_STATE_CLASS: energy_state_class, ATTR_STATE_CLASS: energy_state_class,
} }
@ -553,7 +550,7 @@ async def test_cost_sensor_price_entity_total_no_reset(
return compile_statistics(hass, now, now + timedelta(seconds=1)).platform_stats return compile_statistics(hass, now, now + timedelta(seconds=1)).platform_stats
energy_attributes = { energy_attributes = {
ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, ATTR_UNIT_OF_MEASUREMENT: UnitOfEnergy.KILO_WATT_HOUR,
ATTR_STATE_CLASS: energy_state_class, ATTR_STATE_CLASS: energy_state_class,
} }
@ -701,10 +698,10 @@ async def test_cost_sensor_price_entity_total_no_reset(
@pytest.mark.parametrize( @pytest.mark.parametrize(
"energy_unit,factor", "energy_unit,factor",
[ [
(ENERGY_WATT_HOUR, 1000), (UnitOfEnergy.WATT_HOUR, 1000),
(ENERGY_KILO_WATT_HOUR, 1), (UnitOfEnergy.KILO_WATT_HOUR, 1),
(ENERGY_MEGA_WATT_HOUR, 0.001), (UnitOfEnergy.MEGA_WATT_HOUR, 0.001),
(ENERGY_GIGA_JOULE, 0.001 * 3.6), (UnitOfEnergy.GIGA_JOULE, 0.001 * 3.6),
], ],
) )
async def test_cost_sensor_handle_energy_units( async def test_cost_sensor_handle_energy_units(
@ -767,10 +764,10 @@ async def test_cost_sensor_handle_energy_units(
@pytest.mark.parametrize( @pytest.mark.parametrize(
"price_unit,factor", "price_unit,factor",
[ [
(f"EUR/{ENERGY_WATT_HOUR}", 0.001), (f"EUR/{UnitOfEnergy.WATT_HOUR}", 0.001),
(f"EUR/{ENERGY_KILO_WATT_HOUR}", 1), (f"EUR/{UnitOfEnergy.KILO_WATT_HOUR}", 1),
(f"EUR/{ENERGY_MEGA_WATT_HOUR}", 1000), (f"EUR/{UnitOfEnergy.MEGA_WATT_HOUR}", 1000),
(f"EUR/{ENERGY_GIGA_JOULE}", 1000 / 3.6), (f"EUR/{UnitOfEnergy.GIGA_JOULE}", 1000 / 3.6),
], ],
) )
async def test_cost_sensor_handle_price_units( async def test_cost_sensor_handle_price_units(
@ -778,7 +775,7 @@ async def test_cost_sensor_handle_price_units(
) -> None: ) -> None:
"""Test energy cost price from sensor entity.""" """Test energy cost price from sensor entity."""
energy_attributes = { energy_attributes = {
ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, ATTR_UNIT_OF_MEASUREMENT: UnitOfEnergy.KILO_WATT_HOUR,
ATTR_STATE_CLASS: SensorStateClass.TOTAL_INCREASING, ATTR_STATE_CLASS: SensorStateClass.TOTAL_INCREASING,
} }
price_attributes = { price_attributes = {
@ -891,7 +888,7 @@ async def test_cost_sensor_handle_gas_kwh(
) -> None: ) -> None:
"""Test gas cost price from sensor entity.""" """Test gas cost price from sensor entity."""
energy_attributes = { energy_attributes = {
ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, ATTR_UNIT_OF_MEASUREMENT: UnitOfEnergy.KILO_WATT_HOUR,
ATTR_STATE_CLASS: SensorStateClass.TOTAL_INCREASING, ATTR_STATE_CLASS: SensorStateClass.TOTAL_INCREASING,
} }
energy_data = data.EnergyManager.default_preferences() energy_data = data.EnergyManager.default_preferences()
@ -942,7 +939,7 @@ async def test_cost_sensor_wrong_state_class(
) -> None: ) -> None:
"""Test energy sensor rejects sensor with wrong state_class.""" """Test energy sensor rejects sensor with wrong state_class."""
energy_attributes = { energy_attributes = {
ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, ATTR_UNIT_OF_MEASUREMENT: UnitOfEnergy.KILO_WATT_HOUR,
ATTR_STATE_CLASS: state_class, ATTR_STATE_CLASS: state_class,
} }
energy_data = data.EnergyManager.default_preferences() energy_data = data.EnergyManager.default_preferences()
@ -1003,7 +1000,7 @@ async def test_cost_sensor_state_class_measurement_no_reset(
) -> None: ) -> None:
"""Test energy sensor rejects state_class measurement with no last_reset.""" """Test energy sensor rejects state_class measurement with no last_reset."""
energy_attributes = { energy_attributes = {
ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, ATTR_UNIT_OF_MEASUREMENT: UnitOfEnergy.KILO_WATT_HOUR,
ATTR_STATE_CLASS: state_class, ATTR_STATE_CLASS: state_class,
} }
energy_data = data.EnergyManager.default_preferences() energy_data = data.EnergyManager.default_preferences()

21
tests/components/energy/test_validate.py
View file

@ -4,12 +4,7 @@ from unittest.mock import patch
import pytest import pytest
from homeassistant.components.energy import async_get_manager, validate from homeassistant.components.energy import async_get_manager, validate
from homeassistant.const import ( from homeassistant.const import UnitOfEnergy
ENERGY_GIGA_JOULE,
ENERGY_KILO_WATT_HOUR,
ENERGY_MEGA_WATT_HOUR,
ENERGY_WATT_HOUR,
)
from homeassistant.helpers.json import JSON_DUMP from homeassistant.helpers.json import JSON_DUMP
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -68,13 +63,13 @@ async def test_validation_empty_config(hass):
@pytest.mark.parametrize( @pytest.mark.parametrize(
"state_class, energy_unit, extra", "state_class, energy_unit, extra",
[ [
("total_increasing", ENERGY_KILO_WATT_HOUR, {}), ("total_increasing", UnitOfEnergy.KILO_WATT_HOUR, {}),
("total_increasing", ENERGY_MEGA_WATT_HOUR, {}), ("total_increasing", UnitOfEnergy.MEGA_WATT_HOUR, {}),
("total_increasing", ENERGY_WATT_HOUR, {}), ("total_increasing", UnitOfEnergy.WATT_HOUR, {}),
("total", ENERGY_KILO_WATT_HOUR, {}), ("total", UnitOfEnergy.KILO_WATT_HOUR, {}),
("total", ENERGY_KILO_WATT_HOUR, {"last_reset": "abc"}), ("total", UnitOfEnergy.KILO_WATT_HOUR, {"last_reset": "abc"}),
("measurement", ENERGY_KILO_WATT_HOUR, {"last_reset": "abc"}), ("measurement", UnitOfEnergy.KILO_WATT_HOUR, {"last_reset": "abc"}),
("total_increasing", ENERGY_GIGA_JOULE, {}), ("total_increasing", UnitOfEnergy.GIGA_JOULE, {}),
], ],
) )
async def test_validation( async def test_validation(

35
tests/util/test_unit_conversion.py
View file

@ -2,10 +2,6 @@
import pytest import pytest
from homeassistant.const import ( from homeassistant.const import (
ENERGY_GIGA_JOULE,
ENERGY_KILO_WATT_HOUR,
ENERGY_MEGA_WATT_HOUR,
ENERGY_WATT_HOUR,
LENGTH_CENTIMETERS, LENGTH_CENTIMETERS,
LENGTH_FEET, LENGTH_FEET,
LENGTH_INCHES, LENGTH_INCHES,
@ -44,6 +40,7 @@ from homeassistant.const import (
VOLUME_GALLONS, VOLUME_GALLONS,
VOLUME_LITERS, VOLUME_LITERS,
VOLUME_MILLILITERS, VOLUME_MILLILITERS,
UnitOfEnergy,
UnitOfVolumetricFlux, UnitOfVolumetricFlux,
) )
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
@ -73,10 +70,10 @@ INVALID_SYMBOL = "bob"
(DistanceConverter, LENGTH_YARD), (DistanceConverter, LENGTH_YARD),
(DistanceConverter, LENGTH_FEET), (DistanceConverter, LENGTH_FEET),
(DistanceConverter, LENGTH_INCHES), (DistanceConverter, LENGTH_INCHES),
(EnergyConverter, ENERGY_WATT_HOUR), (EnergyConverter, UnitOfEnergy.WATT_HOUR),
(EnergyConverter, ENERGY_KILO_WATT_HOUR), (EnergyConverter, UnitOfEnergy.KILO_WATT_HOUR),
(EnergyConverter, ENERGY_MEGA_WATT_HOUR), (EnergyConverter, UnitOfEnergy.MEGA_WATT_HOUR),
(EnergyConverter, ENERGY_GIGA_JOULE), (EnergyConverter, UnitOfEnergy.GIGA_JOULE),
(MassConverter, MASS_GRAMS), (MassConverter, MASS_GRAMS),
(MassConverter, MASS_KILOGRAMS), (MassConverter, MASS_KILOGRAMS),
(MassConverter, MASS_MICROGRAMS), (MassConverter, MASS_MICROGRAMS),
@ -120,7 +117,7 @@ def test_convert_same_unit(converter: type[BaseUnitConverter], valid_unit: str)
"converter,valid_unit", "converter,valid_unit",
[ [
(DistanceConverter, LENGTH_KILOMETERS), (DistanceConverter, LENGTH_KILOMETERS),
(EnergyConverter, ENERGY_KILO_WATT_HOUR), (EnergyConverter, UnitOfEnergy.KILO_WATT_HOUR),
(MassConverter, MASS_GRAMS), (MassConverter, MASS_GRAMS),
(PowerConverter, POWER_WATT), (PowerConverter, POWER_WATT),
(PressureConverter, PRESSURE_PA), (PressureConverter, PRESSURE_PA),
@ -146,7 +143,7 @@ def test_convert_invalid_unit(
"converter,from_unit,to_unit", "converter,from_unit,to_unit",
[ [
(DistanceConverter, LENGTH_KILOMETERS, LENGTH_METERS), (DistanceConverter, LENGTH_KILOMETERS, LENGTH_METERS),
(EnergyConverter, ENERGY_WATT_HOUR, ENERGY_KILO_WATT_HOUR), (EnergyConverter, UnitOfEnergy.WATT_HOUR, UnitOfEnergy.KILO_WATT_HOUR),
(MassConverter, MASS_GRAMS, MASS_KILOGRAMS), (MassConverter, MASS_GRAMS, MASS_KILOGRAMS),
(PowerConverter, POWER_WATT, POWER_KILO_WATT), (PowerConverter, POWER_WATT, POWER_KILO_WATT),
(PressureConverter, PRESSURE_HPA, PRESSURE_INHG), (PressureConverter, PRESSURE_HPA, PRESSURE_INHG),
@ -167,7 +164,7 @@ def test_convert_nonnumeric_value(
"converter,from_unit,to_unit,expected", "converter,from_unit,to_unit,expected",
[ [
(DistanceConverter, LENGTH_KILOMETERS, LENGTH_METERS, 1 / 1000), (DistanceConverter, LENGTH_KILOMETERS, LENGTH_METERS, 1 / 1000),
(EnergyConverter, ENERGY_WATT_HOUR, ENERGY_KILO_WATT_HOUR, 1000), (EnergyConverter, UnitOfEnergy.WATT_HOUR, UnitOfEnergy.KILO_WATT_HOUR, 1000),
(PowerConverter, POWER_WATT, POWER_KILO_WATT, 1000), (PowerConverter, POWER_WATT, POWER_KILO_WATT, 1000),
(PressureConverter, PRESSURE_HPA, PRESSURE_INHG, pytest.approx(33.86389)), (PressureConverter, PRESSURE_HPA, PRESSURE_INHG, pytest.approx(33.86389)),
( (
@ -261,14 +258,14 @@ def test_distance_convert(
@pytest.mark.parametrize( @pytest.mark.parametrize(
"value,from_unit,expected,to_unit", "value,from_unit,expected,to_unit",
[ [
(10, ENERGY_WATT_HOUR, 0.01, ENERGY_KILO_WATT_HOUR), (10, UnitOfEnergy.WATT_HOUR, 0.01, UnitOfEnergy.KILO_WATT_HOUR),
(10, ENERGY_WATT_HOUR, 0.00001, ENERGY_MEGA_WATT_HOUR), (10, UnitOfEnergy.WATT_HOUR, 0.00001, UnitOfEnergy.MEGA_WATT_HOUR),
(10, ENERGY_KILO_WATT_HOUR, 10000, ENERGY_WATT_HOUR), (10, UnitOfEnergy.KILO_WATT_HOUR, 10000, UnitOfEnergy.WATT_HOUR),
(10, ENERGY_KILO_WATT_HOUR, 0.01, ENERGY_MEGA_WATT_HOUR), (10, UnitOfEnergy.KILO_WATT_HOUR, 0.01, UnitOfEnergy.MEGA_WATT_HOUR),
(10, ENERGY_MEGA_WATT_HOUR, 10000000, ENERGY_WATT_HOUR), (10, UnitOfEnergy.MEGA_WATT_HOUR, 10000000, UnitOfEnergy.WATT_HOUR),
(10, ENERGY_MEGA_WATT_HOUR, 10000, ENERGY_KILO_WATT_HOUR), (10, UnitOfEnergy.MEGA_WATT_HOUR, 10000, UnitOfEnergy.KILO_WATT_HOUR),
(10, ENERGY_GIGA_JOULE, 10000 / 3.6, ENERGY_KILO_WATT_HOUR), (10, UnitOfEnergy.GIGA_JOULE, 10000 / 3.6, UnitOfEnergy.KILO_WATT_HOUR),
(10, ENERGY_GIGA_JOULE, 10 / 3.6, ENERGY_MEGA_WATT_HOUR), (10, UnitOfEnergy.GIGA_JOULE, 10 / 3.6, UnitOfEnergy.MEGA_WATT_HOUR),
], ],
) )
def test_energy_convert( def test_energy_convert(