mirror of
https://github.com/smittix/intercept.git
synced 2026-04-25 07:10:00 -07:00
26 lines
716 B
Python
26 lines
716 B
Python
"""Tests for SSTV scope waveform encoding."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import numpy as np
|
|
|
|
from utils.sstv.sstv_decoder import _encode_scope_waveform
|
|
|
|
|
|
def test_encode_scope_waveform_respects_window_and_bounds():
|
|
samples = np.array([-32768, -16384, 0, 16384, 32767], dtype=np.int16)
|
|
waveform = _encode_scope_waveform(samples, window_size=4)
|
|
|
|
assert len(waveform) == 4
|
|
assert waveform[0] == -64
|
|
assert waveform[1] == 0
|
|
assert waveform[2] == 64
|
|
assert waveform[3] == 127
|
|
assert max(waveform) <= 127
|
|
assert min(waveform) >= -127
|
|
|
|
|
|
def test_encode_scope_waveform_empty_input():
|
|
waveform = _encode_scope_waveform(np.array([], dtype=np.int16))
|
|
assert waveform == []
|