use all tests

This commit is contained in:
Tryndaron
2024-04-09 10:22:27 +02:00
parent 328f7d4a0f
commit f03069b9d2
2 changed files with 1 additions and 38 deletions
-37
View File
@@ -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}