26 lines
734 B
Python
26 lines
734 B
Python
from faster_whisper import WhisperModel
|
|
|
|
|
|
def test_transcribe(tiny_model_dir, jfk_path):
|
|
model = WhisperModel(tiny_model_dir)
|
|
segments, info = model.transcribe(jfk_path, word_timestamps=True)
|
|
|
|
assert info.language == "en"
|
|
assert info.language_probability > 0.9
|
|
assert info.duration == 11
|
|
|
|
segments = list(segments)
|
|
|
|
assert len(segments) == 1
|
|
|
|
segment = segments[0]
|
|
|
|
assert segment.text == (
|
|
" And so my fellow Americans ask not what your country can do for you, "
|
|
"ask what you can do for your country."
|
|
)
|
|
|
|
assert segment.text == "".join(word.word for word in segment.words)
|
|
assert segment.start == segment.words[0].start
|
|
assert segment.end == segment.words[-1].end
|