mirror of
https://github.com/chidiwilliams/buzz.git
synced 2026-03-16 23:55:51 +01:00
19 lines
541 B
Python
19 lines
541 B
Python
import platform
|
|
import pytest
|
|
|
|
from buzz.transformers_whisper import TransformersWhisper
|
|
from tests.audio import test_audio_path
|
|
|
|
|
|
class TestTransformersWhisper:
|
|
@pytest.mark.skipif(
|
|
platform.system() == "Darwin",
|
|
reason="Not supported on Darwin",
|
|
)
|
|
def test_should_transcribe(self):
|
|
model = TransformersWhisper("openai/whisper-tiny")
|
|
result = model.transcribe(
|
|
audio=test_audio_path, language="fr", task="transcribe"
|
|
)
|
|
|
|
assert "Bienvenue dans Passe" in result["text"]
|