From f8879a51fede3229f71ba9f4583c464c4396205b Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Wed, 14 Aug 2024 12:49:53 +0200
Subject: [PATCH] Improve type hints in sma tests (#123889)

---
 tests/components/sma/conftest.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/tests/components/sma/conftest.py b/tests/components/sma/conftest.py
index a98eda673e4..a54f478a31d 100644
--- a/tests/components/sma/conftest.py
+++ b/tests/components/sma/conftest.py
@@ -9,6 +9,7 @@ import pytest
 
 from homeassistant import config_entries
 from homeassistant.components.sma.const import DOMAIN
+from homeassistant.core import HomeAssistant
 
 from . import MOCK_DEVICE, MOCK_USER_INPUT
 
@@ -16,7 +17,7 @@ from tests.common import MockConfigEntry
 
 
 @pytest.fixture
-def mock_config_entry():
+def mock_config_entry() -> MockConfigEntry:
     """Return the default mocked config entry."""
     return MockConfigEntry(
         domain=DOMAIN,
@@ -28,7 +29,9 @@ def mock_config_entry():
 
 
 @pytest.fixture
-async def init_integration(hass, mock_config_entry):
+async def init_integration(
+    hass: HomeAssistant, mock_config_entry: MockConfigEntry
+) -> MockConfigEntry:
     """Create a fake SMA Config Entry."""
     mock_config_entry.add_to_hass(hass)