From 1ddc7232745c8c66c95fc0483c679de78b056ef2 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Wed, 14 Aug 2024 13:12:24 +0200
Subject: [PATCH] Improve type hints in powerwall tests (#123872)

---
 tests/components/powerwall/test_switch.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/components/powerwall/test_switch.py b/tests/components/powerwall/test_switch.py
index b01f60210a6..b4ff0ca724e 100644
--- a/tests/components/powerwall/test_switch.py
+++ b/tests/components/powerwall/test_switch.py
@@ -1,6 +1,6 @@
 """Test for Powerwall off-grid switch."""
 
-from unittest.mock import patch
+from unittest.mock import MagicMock, patch
 
 import pytest
 from tesla_powerwall import GridStatus, PowerwallError
@@ -24,7 +24,7 @@ ENTITY_ID = "switch.mysite_off_grid_operation"
 
 
 @pytest.fixture(name="mock_powerwall")
-async def mock_powerwall_fixture(hass):
+async def mock_powerwall_fixture(hass: HomeAssistant) -> MagicMock:
     """Set up base powerwall fixture."""
 
     mock_powerwall = await _mock_powerwall_with_fixtures(hass)