diff --git a/.github/workflows/pytest.yaml b/.github/workflows/pytest.yaml index 7ebf890..da3e1a4 100644 --- a/.github/workflows/pytest.yaml +++ b/.github/workflows/pytest.yaml @@ -39,5 +39,5 @@ jobs: env: HF_TOKEN : ${{ secrets.HF_TOKEN }} run: | - pytest test/test_autotranscript.py + pytest test/test_diarisation2.py \ No newline at end of file diff --git a/test/test_diarisation2.py b/test/test_diarisation2.py new file mode 100644 index 0000000..55518ef --- /dev/null +++ b/test/test_diarisation2.py @@ -0,0 +1,37 @@ +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} + + + + +