import sys
import os
import numpy as np
import pytest

sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "backend"))


@pytest.fixture
def sine_wave_72bpm():
    fps = 30
    duration = 20
    t = np.linspace(0, duration, fps * duration)
    freq = 72.0 / 60.0
    noise = np.random.normal(0, 0.5, len(t))
    r = 140.0 + 2.0 * np.sin(2 * np.pi * freq * t) + noise
    g = 142.0 + 4.0 * np.sin(2 * np.pi * freq * t + 0.3) + noise * 0.8
    b = 138.0 + 1.0 * np.sin(2 * np.pi * freq * t + 0.6) + noise * 1.2
    return list(zip(r, g, b))


@pytest.fixture
def sine_wave_90bpm():
    fps = 30
    duration = 20
    t = np.linspace(0, duration, fps * duration)
    freq = 90.0 / 60.0
    noise = np.random.normal(0, 0.5, len(t))
    r = 140.0 + 2.0 * np.sin(2 * np.pi * freq * t) + noise
    g = 142.0 + 4.0 * np.sin(2 * np.pi * freq * t + 0.3) + noise * 0.8
    b = 138.0 + 1.0 * np.sin(2 * np.pi * freq * t + 0.6) + noise * 1.2
    return list(zip(r, g, b))