Diarisation

basics tests for the Diarisation object
This commit is contained in:
Tryndaron
2024-01-23 13:21:42 +01:00
parent abec1e1726
commit 033ce8c92c
+9 -4
View File
@@ -7,26 +7,31 @@ from scraibe import Diariser
@pytest.fixture @pytest.fixture
def diariser_instance(): def diariser_instance():
"""Creates a instance of the Diariser Object for further testing
Returns:
_type_: _description_
"""
with mock.patch.object(Diariser, '_get_token', return_value = 'personal Hugging-Face token') with mock.patch.object(Diariser, '_get_token', return_value = 'personal Hugging-Face token')
return Diariser('pyannote') return Diariser('pyannote')
def test_Diariser_init(diariser_instance): def test_Diariser_init(diariser_instance):
"""_summary_ """Tests if the Diariser gets initiated correctly
Args: Args:
diariser_instance (_type_): _description_ diariser_instance
""" """
assert diariser_instance.model == 'pyannote' assert diariser_instance.model == 'pyannote'
def test_diarisation_function(diariser_instance): def test_diarisation_function(diariser_instance):
"""_summary_ """tests if the Diariser object with an example audio File
Args: Args:
diariser_instance (_type_): _description_ diariser_instance
""" """
with mock.patch.object(diariser_instance.model, 'apply', return_value='diarization_result'): with mock.patch.object(diariser_instance.model, 'apply', return_value='diarization_result'):
diarization_output = diariser_instance.diarization('example_audio_file.wav') diarization_output = diariser_instance.diarization('example_audio_file.wav')