File size: 3,035 Bytes
e3615a2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 | import sys
import os
import numpy as np
import pytest
sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "backend"))
from signal_processor import (
extract_green_signal,
extract_chrom_signal,
extract_pos_signal,
extract_gr_signal,
extract_grgb_signal,
)
@pytest.fixture
def clean_72bpm_rgb():
fps = 30
duration = 15
t = np.linspace(0, duration, fps * duration)
freq = 72.0 / 60.0
noise = np.random.normal(0, 0.3, 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 r, g, b
def test_extract_green_signal_returns_array_of_same_length(clean_72bpm_rgb):
r, g, b = clean_72bpm_rgb
result = extract_green_signal(g)
assert isinstance(result, np.ndarray)
assert len(result) == len(g)
def test_extract_green_signal_nonzero_with_varying_input(clean_72bpm_rgb):
_, g, _ = clean_72bpm_rgb
result = extract_green_signal(g)
assert np.std(result) > 0
def test_extract_chrom_signal_returns_array(clean_72bpm_rgb):
r, g, b = clean_72bpm_rgb
result = extract_chrom_signal(r, g, b)
assert isinstance(result, np.ndarray)
assert len(result) == len(r)
def test_extract_chrom_signal_nonzero_with_varying_input(clean_72bpm_rgb):
r, g, b = clean_72bpm_rgb
result = extract_chrom_signal(r, g, b)
assert np.std(result) > 0
def test_extract_pos_signal_returns_array(clean_72bpm_rgb):
r, g, b = clean_72bpm_rgb
result = extract_pos_signal(r, g, b)
assert isinstance(result, np.ndarray)
assert len(result) == len(r)
def test_extract_pos_signal_nonzero_with_varying_input(clean_72bpm_rgb):
r, g, b = clean_72bpm_rgb
result = extract_pos_signal(r, g, b)
assert np.std(result) > 0
def test_extract_gr_signal_returns_array(clean_72bpm_rgb):
r, g, _ = clean_72bpm_rgb
result = extract_gr_signal(r, g)
assert isinstance(result, np.ndarray)
assert len(result) == len(r)
def test_extract_gr_signal_nonzero_with_varying_input(clean_72bpm_rgb):
r, g, _ = clean_72bpm_rgb
result = extract_gr_signal(r, g)
assert np.std(result) > 0
def test_extract_grgb_signal_returns_array(clean_72bpm_rgb):
r, g, b = clean_72bpm_rgb
result = extract_grgb_signal(r, g, b)
assert isinstance(result, np.ndarray)
assert len(result) == len(r)
def test_extract_grgb_signal_nonzero_with_varying_input(clean_72bpm_rgb):
r, g, b = clean_72bpm_rgb
result = extract_grgb_signal(r, g, b)
assert np.std(result) > 0
def test_all_methods_zero_on_constant_input():
n = 300
r = np.full(n, 140.0)
g = np.full(n, 142.0)
b = np.full(n, 138.0)
assert np.std(extract_green_signal(g)) < 1e-6
assert np.std(extract_chrom_signal(r, g, b)) < 1e-6
assert np.std(extract_pos_signal(r, g, b)) < 1e-6
assert np.std(extract_gr_signal(r, g)) < 1e-6
assert np.std(extract_grgb_signal(r, g, b)) < 1e-6
|