From 90e7d820497ec20e6614afa14e45ba83cf41445b Mon Sep 17 00:00:00 2001
From: Marc Mueller <30130371+cdce8p@users.noreply.github.com>
Date: Sat, 20 Jul 2024 17:33:48 +0200
Subject: [PATCH] Use correct enum in UnitSystem tests (#122256)

---
 tests/util/test_unit_system.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/tests/util/test_unit_system.py b/tests/util/test_unit_system.py
index 033631563f4..15500777212 100644
--- a/tests/util/test_unit_system.py
+++ b/tests/util/test_unit_system.py
@@ -15,6 +15,7 @@ from homeassistant.const import (
     WIND_SPEED,
     UnitOfLength,
     UnitOfMass,
+    UnitOfPrecipitationDepth,
     UnitOfPressure,
     UnitOfSpeed,
     UnitOfTemperature,
@@ -42,7 +43,7 @@ def test_invalid_units() -> None:
     with pytest.raises(ValueError):
         UnitSystem(
             SYSTEM_NAME,
-            accumulated_precipitation=UnitOfLength.MILLIMETERS,
+            accumulated_precipitation=UnitOfPrecipitationDepth.MILLIMETERS,
             conversions={},
             length=UnitOfLength.METERS,
             mass=UnitOfMass.GRAMS,
@@ -55,7 +56,7 @@ def test_invalid_units() -> None:
     with pytest.raises(ValueError):
         UnitSystem(
             SYSTEM_NAME,
-            accumulated_precipitation=UnitOfLength.MILLIMETERS,
+            accumulated_precipitation=UnitOfPrecipitationDepth.MILLIMETERS,
             conversions={},
             length=INVALID_UNIT,
             mass=UnitOfMass.GRAMS,
@@ -68,7 +69,7 @@ def test_invalid_units() -> None:
     with pytest.raises(ValueError):
         UnitSystem(
             SYSTEM_NAME,
-            accumulated_precipitation=UnitOfLength.MILLIMETERS,
+            accumulated_precipitation=UnitOfPrecipitationDepth.MILLIMETERS,
             conversions={},
             length=UnitOfLength.METERS,
             mass=UnitOfMass.GRAMS,
@@ -81,7 +82,7 @@ def test_invalid_units() -> None:
     with pytest.raises(ValueError):
         UnitSystem(
             SYSTEM_NAME,
-            accumulated_precipitation=UnitOfLength.MILLIMETERS,
+            accumulated_precipitation=UnitOfPrecipitationDepth.MILLIMETERS,
             conversions={},
             length=UnitOfLength.METERS,
             mass=UnitOfMass.GRAMS,
@@ -94,7 +95,7 @@ def test_invalid_units() -> None:
     with pytest.raises(ValueError):
         UnitSystem(
             SYSTEM_NAME,
-            accumulated_precipitation=UnitOfLength.MILLIMETERS,
+            accumulated_precipitation=UnitOfPrecipitationDepth.MILLIMETERS,
             conversions={},
             length=UnitOfLength.METERS,
             mass=INVALID_UNIT,
@@ -107,7 +108,7 @@ def test_invalid_units() -> None:
     with pytest.raises(ValueError):
         UnitSystem(
             SYSTEM_NAME,
-            accumulated_precipitation=UnitOfLength.MILLIMETERS,
+            accumulated_precipitation=UnitOfPrecipitationDepth.MILLIMETERS,
             conversions={},
             length=UnitOfLength.METERS,
             mass=UnitOfMass.GRAMS,