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
+1 -1
View File
@@ -1,5 +1,5 @@
import pytest import pytest
from .audio import AudioProcessor from ..scraibe.audio import AudioProcessor
import torch import torch
+7 -5
View File
@@ -1,14 +1,16 @@
import pytest import pytest
import torch import torch
from scraibe import Scraibe, Diariser, Transcriber, Transcript, AudioProcessor from ..scraibe import Scraibe, Diariser, Transcriber, Transcript, AudioProcessor
from unittest.mock import MagicMOck, patch from unittest.mock import MagicMock, patch
""" """
@pytest.fixture @pytest.fixture
def example_audio_file(tmp_path): def example_audio_file(audio_test_2.mp4):
audio_path = tmp_path audio_path = audio_test_2.mp4
""" """
@pytest.fixture @pytest.fixture
def create_scraibe_instance(): def create_scraibe_instance():
return Scraibe() return Scraibe()
@@ -23,7 +25,7 @@ def test_scraibe_init(create_scraibe_instance):
def test_scraibe_autotranscribe(create_scraibe_instance, example_audio_file): def test_scraibe_autotranscribe(create_scraibe_instance, example_audio_file):
model = create_scraibe_instance model = create_scraibe_instance
transcript = example_audio_file transcript = model.autotranscribe(example_audio_file)
assert isinstance(transcript, Transcript) assert isinstance(transcript, Transcript)
def test_scraibe_diarization(create_scraibe_instance, example_audio_file): def test_scraibe_diarization(create_scraibe_instance, example_audio_file):
+1 -1
View File
@@ -1,7 +1,7 @@
import pytest import pytest
import os import os
from unittest import mock from unittest import mock
from scraibe import Diariser from ..scraibe import Diariser
+5 -5
View File
@@ -1,21 +1,21 @@
import pytest import pytest
from unittest.mock import patch 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")] ) @pytest.mark.parametrize("audio_file, expected_transcription",[("path_to_test_audiofile", "test_transcription")] )
@patch("scraibe.Transcriber.load_model") @patch("scraibe.Transcriber.load_model")
def test_transcriber(mock_load_model, audio_file, expected_transcription): def test_transcriber(mock_load_model, audio_file, expected_transcription):
"""_summary_
Args: Args:
mock_load_model (_type_): _description_ mock_load_model (_type_): _description_
audio_file (_type_): _description_ audio_file (_type_): _description_
expected_transcription (_type_): _description_ expected_transcription (_type_): _description_
"""
mock_model = mock_load_model.return_value mock_model = mock_load_model.return_value
mock_model.transcribe.return_value ={"text": expected_transcription} 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) transcription_result = transcriber.transcribe(audio=audio_file)
assert transcription_result == expected_transcription assert transcription_result == expected_transcription """