use all tests
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 test/test_transcriber.py
|
pytest
|
||||||
|
|
||||||
@@ -1,37 +0,0 @@
|
|||||||
import pytest
|
|
||||||
import os
|
|
||||||
import unittest
|
|
||||||
from unittest import mock
|
|
||||||
from scraibe import Diariser
|
|
||||||
import torch
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
|
||||||
def diariser_instance():
|
|
||||||
return Diariser('pyannote')
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def test_diariser_init(diariser_instance):
|
|
||||||
assert diariser_instance.model == 'pyannote'
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
""" def test_format_diarization_output():
|
|
||||||
dialogue = [("speaker1", "segment1"),("speaker2", "segment2"), ("speaker1","segment3")]
|
|
||||||
formatted_output = Diariser.format_diarization_output(dialogue)
|
|
||||||
assert formatted_output == {"speakers": ["speaker1", "speaker2", "speaker1"], "segments": ["segment1", "segment2", "segment3"]} """
|
|
||||||
|
|
||||||
|
|
||||||
def test_get_diarisation_kwargs():
|
|
||||||
kwargs = {"arg1": 1, "arg3": 3}
|
|
||||||
valid_kwargs = Diariser._get_diarisation_kwargs(**kwargs)
|
|
||||||
assert not valid_kwargs == {"arg1": 1, "arg3": 3}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
Reference in New Issue
Block a user