test if file gets found
This commit is contained in:
@@ -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
|
||||||
|
|
||||||
@@ -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)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user