45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
import numpy as np
|
|
|
|
from ha_deconz_bridge.color import Color
|
|
|
|
|
|
def test_rgb_roundtrip_is_stable() -> None:
|
|
color = Color.from_rgb((32, 128, 255), brightness=0.5)
|
|
roundtrip = Color.from_xy(color.xy, brightness=color.brightness)
|
|
assert np.linalg.norm(color.uv - roundtrip.uv) < 0.05
|
|
assert roundtrip.brightness == color.brightness
|
|
|
|
|
|
def test_weighted_sum_of_ct_colors_uses_xyz_mixing() -> None:
|
|
warm = Color.from_color_temp(450, brightness=0.8)
|
|
cool = Color.from_color_temp(200, brightness=0.2)
|
|
mixed = Color.weighted_sum((warm, cool), (1.0, 1.0))
|
|
assert mixed.mode == "rgb"
|
|
assert 200 < mixed.color_temp < 450
|
|
assert mixed.brightness == 1.0
|
|
|
|
|
|
def test_weighted_sum_promotes_to_rgb_when_any_rgb_present() -> None:
|
|
white = Color.from_color_temp(300, brightness=0.5)
|
|
red = Color.from_rgb((1.0, 0.0, 0.0), brightness=0.25)
|
|
mixed = Color.weighted_sum((white, red), (1.0, 1.0))
|
|
assert mixed.mode == "rgb"
|
|
assert mixed.brightness > 0.0
|
|
|
|
|
|
def test_ct_mix_inverts_back_to_midpoint() -> None:
|
|
color = Color.from_color_temp(326, brightness=1.0, temp_range=(153, 500))
|
|
cool, warm = color.ct_mix((153, 500))
|
|
assert 0.0 < cool <= 1.0
|
|
assert 0.0 < warm <= 1.0
|
|
|
|
|
|
def test_color_temp_preserves_requested_mireds_for_serialization() -> None:
|
|
color = Color.from_color_temp(455, brightness=1.0, temp_range=(153, 500))
|
|
dimmed = color.with_brightness(0.2)
|
|
|
|
assert color.color_temp == 455
|
|
assert dimmed.color_temp == 455
|