Diarisation
basics tests for the Diarisation object
This commit is contained in:
@@ -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')
|
||||||
|
|||||||
Reference in New Issue
Block a user