diff --git a/scraibe/test/test_diarisation.py b/scraibe/test/test_diarisation.py index 55b7276..7825552 100644 --- a/scraibe/test/test_diarisation.py +++ b/scraibe/test/test_diarisation.py @@ -16,6 +16,7 @@ def test_Diariser_init(diariser_instance): assert diariser_instance.model == 'pyannote' + def test_diarisation_function(diariser_instance): with mock.patch.object(diariser_instance.model, 'apply', return_value='diarization_result'): diarization_output = diariser_instance.diarization('example_audio_file.wav')