from openrgb_hue.color import Color
from openrgb_hue.gradient import Gradient, Stop


def test_from_colors_even_spacing_single():
    gradient = Gradient.from_colors([Color(10, 20, 30)])
    assert [s.position for s in gradient.stops] == [0.0]


def test_from_colors_even_spacing_multiple():
    colors = [Color(0, 0, 0), Color(50, 50, 50), Color(255, 255, 255)]
    gradient = Gradient.from_colors(colors)
    assert [s.position for s in gradient.stops] == [0.0, 0.5, 1.0]


def test_sample_at_exact_stop_positions():
    colors = [Color(255, 0, 0), Color(0, 255, 0), Color(0, 0, 255)]
    gradient = Gradient.from_colors(colors, interpolation="rgb")
    assert gradient.sample(0.0) == colors[0]
    assert gradient.sample(0.5) == colors[1]
    assert gradient.sample(1.0) == colors[0]  # cyclic: wraps back to the first stop


def test_sample_wraps_cyclically():
    colors = [Color(255, 0, 0), Color(0, 0, 255)]
    gradient = Gradient.from_colors(colors, interpolation="rgb")
    # t=0.75 is halfway between the last stop (1.0 -> wraps to 0.0) and back
    # to the first stop; verify negative/over-1 t wrap to the same result.
    assert gradient.sample(-0.25) == gradient.sample(0.75)
    assert gradient.sample(1.25) == gradient.sample(0.25)


def test_single_stop_gradient_is_constant():
    gradient = Gradient([Stop(0.3, Color(1, 2, 3))])
    assert gradient.sample(0.0) == Color(1, 2, 3)
    assert gradient.sample(0.9) == Color(1, 2, 3)


def test_from_scene_dedupe_collapses_consecutive_duplicates(sample_scene):
    with_dedupe = Gradient.from_scene(sample_scene, dedupe=True, use_brightness=False)
    without_dedupe = Gradient.from_scene(sample_scene, dedupe=False, use_brightness=False)
    # sample_scene has 4 lights but lights 2 and 3 share a color.
    assert len(with_dedupe.stops) == 3
    assert len(without_dedupe.stops) == 4


def test_from_scene_use_brightness_scales_colors(sample_scene):
    gradient = Gradient.from_scene(sample_scene, dedupe=False, use_brightness=True)
    # Fourth light: Color(0, 0, 255) at brightness 128/255.
    last_stop_color = gradient.stops[-1].color
    expected = Color(0, 0, 255).scale_brightness(128 / 255)
    assert last_stop_color == expected


def test_preview_stops_length():
    gradient = Gradient.from_colors([Color(255, 0, 0), Color(0, 255, 0)])
    assert len(gradient.preview_stops(10)) == 10
    assert gradient.preview_stops(0) == []