relativ imports
This commit is contained in:
+1
-1
@@ -1,5 +1,5 @@
|
|||||||
import pytest
|
import pytest
|
||||||
from .audio import AudioProcessor
|
from ..scraibe.audio import AudioProcessor
|
||||||
import torch
|
import torch
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -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,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
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -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 """
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user