test if file gets found

This commit is contained in:
Tryndaron
2024-04-11 09:58:16 +02:00
parent ac9d34136c
commit fb0c7a9fa6
2 changed files with 7 additions and 3 deletions
+1 -1
View File
@@ -39,5 +39,5 @@ jobs:
env: env:
HF_TOKEN : ${{ secrets.HF_TOKEN }} HF_TOKEN : ${{ secrets.HF_TOKEN }}
run: | run: |
pytest pytest test/test_autotranscript.py
+6 -2
View File
@@ -3,6 +3,10 @@ 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
import os import os
import imp
""" """
@pytest.fixture @pytest.fixture
@@ -31,13 +35,13 @@ def test_scraibe_init(create_scraibe_instance):
def test_scraibe_autotranscribe(create_scraibe_instance): def test_scraibe_autotranscribe(create_scraibe_instance):
model = create_scraibe_instance model = create_scraibe_instance
transcript = model.autotranscribe("audio_test_2.mp4") transcript = model.autotranscribe(util)
assert isinstance(transcript, Transcript) assert isinstance(transcript, Transcript)
def test_scraibe_diarization(create_scraibe_instance): def test_scraibe_diarization(create_scraibe_instance):
model = create_scraibe_instance model = create_scraibe_instance
diarisation_result = model.diarization("audio_test_2.mp4") diarisation_result = model.diarization('test/audio_test_2.mp4')
assert isinstance(diarisation_result, dict) assert isinstance(diarisation_result, dict)