relativ imports

This commit is contained in:
Tryndaron
2024-04-02 10:10:52 +02:00
parent 6f9c0c8cc8
commit d114841425
4 changed files with 14 additions and 12 deletions
+5 -5
View File
@@ -1,21 +1,21 @@
import pytest
from unittest.mock import patch
from scraibe import Transcriber
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}
@@ -23,7 +23,7 @@ def test_transcriber(mock_load_model, audio_file, expected_transcription):
transcription_result = transcriber.transcribe(audio=audio_file)
assert transcription_result == expected_transcription
assert transcription_result == expected_transcription """