mirror of
https://github.com/chidiwilliams/buzz.git
synced 2026-03-14 14:45:46 +01:00
Fixed tests
This commit is contained in:
parent
c42f458fe7
commit
eae123db68
2 changed files with 3 additions and 3 deletions
|
|
@ -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)
|
||||
|
|
|
|||
|
|
@ -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):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue