diff --git a/test/test_autotranscript.py b/test/test_autotranscript.py index 4ff74a2..05c1aa3 100644 --- a/test/test_autotranscript.py +++ b/test/test_autotranscript.py @@ -1,21 +1,11 @@ import pytest -import torch -from scraibe import Scraibe, Diariser, Transcriber, Transcript, AudioProcessor +from scraibe import Scraibe, Diariser, Transcriber, Transcript from unittest.mock import MagicMock, patch import os -import imp -""" -@pytest.fixture -def example_audio_file(audio_test_2.mp4): - audio_path = audio_test_2.mp4 -""" - - - @pytest.fixture def create_scraibe_instance(): @@ -35,7 +25,7 @@ def test_scraibe_init(create_scraibe_instance): def test_scraibe_autotranscribe(create_scraibe_instance): model = create_scraibe_instance - transcript = model.autotranscribe('test/audio_test_2.mp4') + transcript = model.autotranscribe('test/audio_test.mp4') assert isinstance(transcript, Transcript)