From eae123db68b1d6268952ecbeee4dcc4d166c2e83 Mon Sep 17 00:00:00 2001 From: Raivis Dejus Date: Sat, 28 Feb 2026 18:37:45 +0200 Subject: [PATCH] Fixed tests --- tests/recording_transcriber_test.py | 4 ++-- tests/transcriber/recording_transcriber_test.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/recording_transcriber_test.py b/tests/recording_transcriber_test.py index 9c2f66c4..8b394cd2 100644 --- a/tests/recording_transcriber_test.py +++ b/tests/recording_transcriber_test.py @@ -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) diff --git a/tests/transcriber/recording_transcriber_test.py b/tests/transcriber/recording_transcriber_test.py index dbd17eee..daa77789 100644 --- a/tests/transcriber/recording_transcriber_test.py +++ b/tests/transcriber/recording_transcriber_test.py @@ -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):