Improve type hints in mfi tests (#123792)

This commit is contained in:
epenet 2024-08-13 11:56:55 +02:00 committed by GitHub
parent 5f694d9a84
commit 2b968dfd9a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -116,13 +116,13 @@ async def test_setup_adds_proper_devices(hass: HomeAssistant) -> None:
@pytest.fixture(name="port")
def port_fixture():
def port_fixture() -> mock.MagicMock:
"""Port fixture."""
return mock.MagicMock()
@pytest.fixture(name="sensor")
def sensor_fixture(hass, port):
def sensor_fixture(hass: HomeAssistant, port: mock.MagicMock) -> mfi.MfiSensor:
"""Sensor fixture."""
sensor = mfi.MfiSensor(port, hass)
sensor.hass = hass