From fb0c7a9fa6d8b7601f8160e4bcbb8a5cefb1c4e3 Mon Sep 17 00:00:00 2001 From: Tryndaron Date: Thu, 11 Apr 2024 09:58:16 +0200 Subject: [PATCH] test if file gets found --- .github/workflows/pytest.yaml | 2 +- test/test_autotranscript.py | 8 ++++++-- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/.github/workflows/pytest.yaml b/.github/workflows/pytest.yaml index 4826cb1..7ebf890 100644 --- a/.github/workflows/pytest.yaml +++ b/.github/workflows/pytest.yaml @@ -39,5 +39,5 @@ jobs: env: HF_TOKEN : ${{ secrets.HF_TOKEN }} run: | - pytest + pytest test/test_autotranscript.py \ No newline at end of file diff --git a/test/test_autotranscript.py b/test/test_autotranscript.py index 8864c5f..01a883c 100644 --- a/test/test_autotranscript.py +++ b/test/test_autotranscript.py @@ -3,6 +3,10 @@ import torch from scraibe import Scraibe, Diariser, Transcriber, Transcript, AudioProcessor from unittest.mock import MagicMock, patch import os +import imp + + + """ @pytest.fixture @@ -31,13 +35,13 @@ def test_scraibe_init(create_scraibe_instance): def test_scraibe_autotranscribe(create_scraibe_instance): model = create_scraibe_instance - transcript = model.autotranscribe("audio_test_2.mp4") + transcript = model.autotranscribe(util) assert isinstance(transcript, Transcript) def test_scraibe_diarization(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)