changed folder position

This commit is contained in:
Tryndaron
2024-03-28 10:23:24 +01:00
parent beab5dff05
commit b01818212f
9 changed files with 0 additions and 0 deletions
+29
View File
@@ -0,0 +1,29 @@
import pytest
from unittest.mock import patch
from scraibe import Transcriber
@pytest.mark.parametrize("audio_file, expected_transcription",[("path_to_test_audiofile", "test_transcription")] )
@patch("scraibe.Transcriber.load_model")
def test_transcriber(mock_load_model, audio_file, expected_transcription):
"""_summary_
Args:
mock_load_model (_type_): _description_
audio_file (_type_): _description_
expected_transcription (_type_): _description_
"""
mock_model = mock_load_model.return_value
mock_model.transcribe.return_value ={"text": expected_transcription}
transcriber = Transcriber.load_model(model="medium")
transcription_result = transcriber.transcribe(audio=audio_file)
assert transcription_result == expected_transcription