Tryndaron
38 lines
828 B
Python
38 lines
828 B
Python
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}
|
|
|
|
|
|
|
|
|
|
|