This commit is contained in:
Tryndaron
2024-01-05 12:32:03 +01:00
parent f90d16859e
commit ffe4dc8c58
+1
View File
@@ -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')