|
|
@@ -8,7 +8,7 @@ from homeassistant.components.vacuum import (
|
|
|
STATE_RETURNING,
|
|
|
VacuumEntityFeature,
|
|
|
)
|
|
|
-from homeassistant.const import AREA_SQUARE_METERS, PERCENTAGE, UnitOfTime
|
|
|
+from homeassistant.const import PERCENTAGE, UnitOfArea, UnitOfTime
|
|
|
|
|
|
from ..const import KYVOL_E30_VACUUM_PAYLOAD
|
|
|
from ..helpers import assert_device_properties_set
|
|
|
@@ -68,7 +68,7 @@ class TestKyvolE30Vacuum(MultiButtonTests, MultiSensorTests, TuyaDeviceTestCase)
|
|
|
{
|
|
|
"dps": AREA_DPS,
|
|
|
"name": "sensor_clean_area",
|
|
|
- "unit": AREA_SQUARE_METERS,
|
|
|
+ "unit": UnitOfArea.SQUARE_METERS,
|
|
|
"testdata": (30, 3.0),
|
|
|
},
|
|
|
{
|