From 67f761c0e9538c28a63d7102f8150c7f597be64f Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Wed, 14 Aug 2024 14:58:16 +0200
Subject: [PATCH] Improve type hints in template tests (#123915)

---
 tests/components/template/test_light.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/tests/components/template/test_light.py b/tests/components/template/test_light.py
index ad97146d0fb..065a1488dc9 100644
--- a/tests/components/template/test_light.py
+++ b/tests/components/template/test_light.py
@@ -1,5 +1,7 @@
 """The tests for the  Template light platform."""
 
+from typing import Any
+
 import pytest
 
 from homeassistant.components import light
@@ -152,7 +154,9 @@ OPTIMISTIC_RGBWW_COLOR_LIGHT_CONFIG = {
 }
 
 
-async def async_setup_light(hass, count, light_config):
+async def async_setup_light(
+    hass: HomeAssistant, count: int, light_config: dict[str, Any]
+) -> None:
     """Do setup of light integration."""
     config = {"light": {"platform": "template", "lights": light_config}}
 
@@ -169,7 +173,9 @@ async def async_setup_light(hass, count, light_config):
 
 
 @pytest.fixture
-async def setup_light(hass, count, light_config):
+async def setup_light(
+    hass: HomeAssistant, count: int, light_config: dict[str, Any]
+) -> None:
     """Do setup of light integration."""
     await async_setup_light(hass, count, light_config)