summaryrefslogtreecommitdiff
path: root/tests/test_runtime.py
blob: d0d45fb072278f24860aa188364ce1c1fbb96bec (plain)
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
from datetime import datetime, timezone

import numpy as np

from mic_clipper.audio_input import PulseAudioInput
from mic_clipper.runner import record_frames
from mic_clipper.segmenter import Segmenter


class FakeVad:
    def __init__(self, probabilities: list[float]) -> None:
        self.probabilities = iter(probabilities)

    def probability(self, _samples: np.ndarray) -> float:
        return next(self.probabilities)


class FakeClip:
    def __init__(self) -> None:
        self.writes: list[np.ndarray] = []
        self.closed = False

    def write(self, samples: np.ndarray) -> None:
        self.writes.append(samples)

    def close(self) -> None:
        self.closed = True

    def abort(self) -> None:
        self.closed = True


class FakeWriter:
    def __init__(self) -> None:
        self.opened: list[FakeClip] = []

    def open(self, _started_at: datetime) -> FakeClip:
        clip = FakeClip()
        self.opened.append(clip)
        return clip


def samples() -> np.ndarray:
    return np.zeros(512, dtype=np.float32)


def test_silence_never_opens_an_output_clip():
    writer = FakeWriter()

    record_frames(
        [samples(), samples()],
        vad=FakeVad([0.1, 0.1]),
        writer=writer,
        segmenter=Segmenter(pre_roll_seconds=0.5, silence_seconds=1.5),
        now=lambda: datetime(2026, 9, 29, tzinfo=timezone.utc),
    )

    assert writer.opened == []


def test_end_event_closes_the_active_encoder():
    writer = FakeWriter()

    record_frames(
        [samples()] + [samples()] * 47,
        vad=FakeVad([0.9] + [0.1] * 47),
        writer=writer,
        segmenter=Segmenter(pre_roll_seconds=0.5, silence_seconds=1.5),
        now=lambda: datetime(2026, 9, 29, tzinfo=timezone.utc),
    )

    assert len(writer.opened) == 1
    assert writer.opened[0].closed


def test_pulse_input_uses_dynamic_default_device_at_16khz_mono():
    command = PulseAudioInput.command()

    assert command[0] == "ffmpeg"
    assert command[command.index("-f") + 1] == "pulse"
    assert command[command.index("-i") + 1] == "default"
    assert command[-2:] == ["f32le", "pipe:1"]