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):