Fixed tests

This commit is contained in:
Raivis Dejus 2026-02-28 18:37:45 +02:00
commit eae123db68
2 changed files with 3 additions and 3 deletions

View file

@ -43,10 +43,10 @@ class TestRecordingTranscriberInit:
t = make_transcriber(mode_index=0)
assert t.n_batch_samples == 5 * t.sample_rate
def test_append_and_correct_mode_batch_size_is_3_seconds(self):
def test_append_and_correct_mode_batch_size_uses_transcription_step(self):
mode_index = list(RecordingTranscriberMode).index(RecordingTranscriberMode.APPEND_AND_CORRECT)
t = make_transcriber(mode_index=mode_index)
assert t.n_batch_samples == 3 * t.sample_rate
assert t.n_batch_samples == int(t.transcription_options.transcription_step * t.sample_rate)
def test_append_and_correct_mode_keep_sample_seconds(self):
mode_index = list(RecordingTranscriberMode).index(RecordingTranscriberMode.APPEND_AND_CORRECT)

View file

@ -197,7 +197,7 @@ class TestRecordingTranscriberInit:
)
# APPEND_AND_CORRECT mode should use smaller batch size and longer keep duration
assert transcriber.n_batch_samples == 3 * 16000
assert transcriber.n_batch_samples == int(transcription_options.transcription_step * 16000)
assert transcriber.keep_sample_seconds == 1.5
def test_init_stores_silence_threshold(self):