import struct
import numpy as np
from sttlib.audio import pcm_bytes_to_float32


def test_known_value():
    # 16384 in int16 -> 0.5 in float32
    pcm = struct.pack("<h", 16384)
    result = pcm_bytes_to_float32(pcm)
    assert abs(result[0] - 0.5) < 1e-5


def test_silence():
    pcm = b"\x00\x00" * 10
    result = pcm_bytes_to_float32(pcm)
    assert np.all(result == 0.0)


def test_full_scale():
    # max int16 = 32767 -> ~1.0
    pcm = struct.pack("<h", 32767)
    result = pcm_bytes_to_float32(pcm)
    assert abs(result[0] - (32767 / 32768.0)) < 1e-5


def test_negative():
    # min int16 = -32768 -> -1.0
    pcm = struct.pack("<h", -32768)
    result = pcm_bytes_to_float32(pcm)
    assert result[0] == -1.0


def test_round_trip_shape():
    # 100 samples worth of bytes
    pcm = b"\x00\x00" * 100
    result = pcm_bytes_to_float32(pcm)
    assert result.shape == (100,)
    assert result.dtype == np.float32