From 5f694d9a8449e25aa3aff0ff252fd83c5afe54c3 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Tue, 13 Aug 2024 11:56:18 +0200
Subject: [PATCH] Improve type hints in mochad tests (#123794)

---
 tests/components/mochad/test_light.py  | 2 +-
 tests/components/mochad/test_switch.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/components/mochad/test_light.py b/tests/components/mochad/test_light.py
index 872bd3a9d61..49beebbaec6 100644
--- a/tests/components/mochad/test_light.py
+++ b/tests/components/mochad/test_light.py
@@ -18,7 +18,7 @@ def pymochad_mock():
 
 
 @pytest.fixture
-def light_mock(hass, brightness):
+def light_mock(hass: HomeAssistant, brightness: int) -> mochad.MochadLight:
     """Mock light."""
     controller_mock = mock.MagicMock()
     dev_dict = {"address": "a1", "name": "fake_light", "brightness_levels": brightness}
diff --git a/tests/components/mochad/test_switch.py b/tests/components/mochad/test_switch.py
index 750dd48296e..9fea3b5c14c 100644
--- a/tests/components/mochad/test_switch.py
+++ b/tests/components/mochad/test_switch.py
@@ -21,7 +21,7 @@ def pymochad_mock():
 
 
 @pytest.fixture
-def switch_mock(hass):
+def switch_mock(hass: HomeAssistant) -> mochad.MochadSwitch:
     """Mock switch."""
     controller_mock = mock.MagicMock()
     dev_dict = {"address": "a1", "name": "fake_switch"}