diff --git a/test/test_autotranscript.py b/test/test_autotranscript.py index 78442b3..9d04c9b 100644 --- a/test/test_autotranscript.py +++ b/test/test_autotranscript.py @@ -6,7 +6,7 @@ import os @pytest.fixture def create_scraibe_instance(): if "HF_TOKEN" in os.environ: - return Scraibe(use_auth_token=os.environ["HF_TOKEN"]) + return Scraibe(use_auth_token=os.environ["HF_TOKEN"], whisper_model= "tiny") else: return Scraibe() @@ -19,19 +19,19 @@ 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_2.mp4') assert isinstance(transcript, Transcript) def test_scraibe_diarization(create_scraibe_instance): model = create_scraibe_instance - diarisation_result = model.diarization('test/audio_test_2.mp4') + diarisation_result = model.diarization('./test/audio_test_2.mp4') assert isinstance(diarisation_result, dict) def test_scraibe_transcribe(create_scraibe_instance): model = create_scraibe_instance - transcription_result = model.transcribe('test/audio_test_2.mp4') + transcription_result = model.transcribe('./test/audio_test_2.mp4') assert isinstance(transcription_result, str) diff --git a/test/test_transcriber.py b/test/test_transcriber.py index 5bfe3cf..80f79d2 100644 --- a/test/test_transcriber.py +++ b/test/test_transcriber.py @@ -69,12 +69,12 @@ def test_get_whisper_kwargs(): def test_whisper_transcribe(whisper_instance): model = whisper_instance # mocker.patch.object(transcriber_instance.model, 'transcribe', return_value={'Hello, World !'} ) - transcript = model.transcribe('test/audio_test_2.mp4') + transcript = model.transcribe('./test/audio_test_2.mp4') assert isinstance(transcript, str) def test_faster_whisper_transcribe(faster_whisper_instance): model = faster_whisper_instance # mocker.patch.object(transcriber_instance.model, 'transcribe', return_value={'Hello, World !'} ) - transcript = model.transcribe('test/audio_test_2.mp4') + transcript = model.transcribe('./test/audio_test_2.mp4') assert isinstance(transcript, str)