Improve type hints in sma tests (#123889)
This commit is contained in:
parent
57902fed22
commit
f8879a51fe
1 changed files with 5 additions and 2 deletions
|
@ -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)
|
||||
|
||||
|
|
Loading…
Add table
Reference in a new issue