test if file gets found

2024-04-11 09:58:16 +02:00
parent ac9d34136c
commit fb0c7a9fa6
2 changed files with 7 additions and 3 deletions
@@ -39,5 +39,5 @@ jobs:
        env: 
          HF_TOKEN : ${{ secrets.HF_TOKEN }}
        run: |
-            pytest
+            pytest 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)