admin/scribe
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

docstring added and variable type

Browse Source
This commit is contained in:
Tryndaron
2024-03-21 09:29:37 +01:00
parent 033ce8c92c
commit e40860c92f
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scraibe/test/test_diarisation.py
+4 -4
View File
@@ -10,7 +10,7 @@ def diariser_instance():
"""Creates a instance of the Diariser Object for further testing """Creates a instance of the Diariser Object for further testing
Returns: Returns:
_type_: _description_ Object: Diariser Object for handling the diarization process of an audio file using a pretrained model from pyannote.audio. Diarization is the task of determining "who spoke when."
""" """
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')
@@ -21,17 +21,17 @@ def test_Diariser_init(diariser_instance):
"""Tests if the Diariser gets initiated correctly """Tests if the Diariser gets initiated correctly
Args: Args:
diariser_instance diariser_instance (obj): instance of the Diariser object
""" """
assert diariser_instance.model == 'pyannote' assert diariser_instance.model == 'pyannote'
def test_diarisation_function(diariser_instance): def test_diarisation_function(diariser_instance):
"""tests if the Diariser object with an example audio File """tests if the Diariser works object with an example audio File
Args: Args:
diariser_instance diariser_instance (obj): instance of the Diariser object
""" """
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')