jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-05-01 12:47:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Migrate yeelight light tests to use Kelvin (#133033)

Browse Source
This commit is contained in:
epenet 2024-12-12 20:29:15 +01:00 committed by GitHub
parent f0391f4963
commit de35bfce77
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
tests/components/yeelight/test_light.py
View File

@ -24,7 +24,7 @@ from yeelight.main import _MODEL_SPECS
from homeassistant.components.light import ( from homeassistant.components.light import (
ATTR_BRIGHTNESS, ATTR_BRIGHTNESS,
ATTR_BRIGHTNESS_PCT, ATTR_BRIGHTNESS_PCT,
ATTR_COLOR_TEMP, ATTR_COLOR_TEMP_KELVIN,
ATTR_EFFECT, ATTR_EFFECT,
ATTR_FLASH, ATTR_FLASH,
ATTR_HS_COLOR, ATTR_HS_COLOR,
@ -107,7 +107,6 @@ from homeassistant.util.color import (
color_RGB_to_hs, color_RGB_to_hs,
color_RGB_to_xy, color_RGB_to_xy,
color_temperature_kelvin_to_mired, color_temperature_kelvin_to_mired,
color_temperature_mired_to_kelvin,
) )
from . import ( from . import (
@ -289,7 +288,7 @@ async def test_services(hass: HomeAssistant, caplog: pytest.LogCaptureFixture) -
# turn_on color_temp # turn_on color_temp
brightness = 100 brightness = 100
color_temp = 200 color_temp = 5000
transition = 1 transition = 1
mocked_bulb.last_properties["power"] = "off" mocked_bulb.last_properties["power"] = "off"
await hass.services.async_call( await hass.services.async_call(
@ -298,7 +297,7 @@ async def test_services(hass: HomeAssistant, caplog: pytest.LogCaptureFixture) -
{ {
ATTR_ENTITY_ID: ENTITY_LIGHT, ATTR_ENTITY_ID: ENTITY_LIGHT,
ATTR_BRIGHTNESS: brightness, ATTR_BRIGHTNESS: brightness,
ATTR_COLOR_TEMP: color_temp, ATTR_COLOR_TEMP_KELVIN: color_temp,
ATTR_FLASH: FLASH_LONG, ATTR_FLASH: FLASH_LONG,
ATTR_EFFECT: EFFECT_STOP, ATTR_EFFECT: EFFECT_STOP,
ATTR_TRANSITION: transition, ATTR_TRANSITION: transition,
@ -316,7 +315,7 @@ async def test_services(hass: HomeAssistant, caplog: pytest.LogCaptureFixture) -
brightness / 255 * 100, duration=transition * 1000, light_type=LightType.Main brightness / 255 * 100, duration=transition * 1000, light_type=LightType.Main
) )
mocked_bulb.async_set_color_temp.assert_called_once_with( mocked_bulb.async_set_color_temp.assert_called_once_with(
color_temperature_mired_to_kelvin(color_temp), color_temp,
duration=transition * 1000, duration=transition * 1000,
light_type=LightType.Main, light_type=LightType.Main,
) )
@ -327,7 +326,7 @@ async def test_services(hass: HomeAssistant, caplog: pytest.LogCaptureFixture) -
# turn_on color_temp - flash short # turn_on color_temp - flash short
brightness = 100 brightness = 100
color_temp = 200 color_temp = 5000
transition = 1 transition = 1
mocked_bulb.async_start_music.reset_mock() mocked_bulb.async_start_music.reset_mock()
mocked_bulb.async_set_brightness.reset_mock() mocked_bulb.async_set_brightness.reset_mock()
@ -342,7 +341,7 @@ async def test_services(hass: HomeAssistant, caplog: pytest.LogCaptureFixture) -
{ {
ATTR_ENTITY_ID: ENTITY_LIGHT, ATTR_ENTITY_ID: ENTITY_LIGHT,
ATTR_BRIGHTNESS: brightness, ATTR_BRIGHTNESS: brightness,
ATTR_COLOR_TEMP: color_temp, ATTR_COLOR_TEMP_KELVIN: color_temp,
ATTR_FLASH: FLASH_SHORT, ATTR_FLASH: FLASH_SHORT,
ATTR_EFFECT: EFFECT_STOP, ATTR_EFFECT: EFFECT_STOP,
ATTR_TRANSITION: transition, ATTR_TRANSITION: transition,
@ -360,7 +359,7 @@ async def test_services(hass: HomeAssistant, caplog: pytest.LogCaptureFixture) -
brightness / 255 * 100, duration=transition * 1000, light_type=LightType.Main brightness / 255 * 100, duration=transition * 1000, light_type=LightType.Main
) )
mocked_bulb.async_set_color_temp.assert_called_once_with( mocked_bulb.async_set_color_temp.assert_called_once_with(
color_temperature_mired_to_kelvin(color_temp), color_temp,
duration=transition * 1000, duration=transition * 1000,
light_type=LightType.Main, light_type=LightType.Main,
) )
@ -691,7 +690,7 @@ async def test_state_already_set_avoid_ratelimit(hass: HomeAssistant) -> None:
await hass.services.async_call( await hass.services.async_call(
"light", "light",
SERVICE_TURN_ON, SERVICE_TURN_ON,
{ATTR_ENTITY_ID: ENTITY_LIGHT, ATTR_COLOR_TEMP: 250}, {ATTR_ENTITY_ID: ENTITY_LIGHT, ATTR_COLOR_TEMP_KELVIN: 4000},
blocking=True, blocking=True,
) )
assert mocked_bulb.async_set_hsv.mock_calls == [] assert mocked_bulb.async_set_hsv.mock_calls == []
@ -707,7 +706,7 @@ async def test_state_already_set_avoid_ratelimit(hass: HomeAssistant) -> None:
await hass.services.async_call( await hass.services.async_call(
"light", "light",
SERVICE_TURN_ON, SERVICE_TURN_ON,
{ATTR_ENTITY_ID: ENTITY_LIGHT, ATTR_COLOR_TEMP: 250}, {ATTR_ENTITY_ID: ENTITY_LIGHT, ATTR_COLOR_TEMP_KELVIN: 4000},
blocking=True, blocking=True,
) )
assert mocked_bulb.async_set_hsv.mock_calls == [] assert mocked_bulb.async_set_hsv.mock_calls == []
@ -720,7 +719,7 @@ async def test_state_already_set_avoid_ratelimit(hass: HomeAssistant) -> None:
await hass.services.async_call( await hass.services.async_call(
"light", "light",
SERVICE_TURN_ON, SERVICE_TURN_ON,
{ATTR_ENTITY_ID: ENTITY_LIGHT, ATTR_COLOR_TEMP: 250}, {ATTR_ENTITY_ID: ENTITY_LIGHT, ATTR_COLOR_TEMP_KELVIN: 4000},
blocking=True, blocking=True,
) )
assert mocked_bulb.async_set_hsv.mock_calls == [] assert mocked_bulb.async_set_hsv.mock_calls == []