Improve type hints in mochad tests (#123794)

This commit is contained in:
epenet 2024-08-13 11:56:18 +02:00 committed by GitHub
parent 992de497f2
commit 5f694d9a84
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions

View File

@ -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}

View File

@ -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"}