diff --git a/.github/workflows/pytest.yaml b/.github/workflows/pytest.yaml index 7591ae5..4826cb1 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_transcriber.py + pytest \ No newline at end of file diff --git a/test/test_diarisation2.py b/test/test_diarisation2.py deleted file mode 100644 index 55518ef..0000000 --- a/test/test_diarisation2.py +++ /dev/null @@ -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} - - - - -