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

test

Browse Source
This commit is contained in:
Tryndaron
2024-01-05 12:32:03 +01:00
parent f90d16859e
commit ffe4dc8c58
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scraibe/test/test_diarisation.py
+1
View File
@@ -16,6 +16,7 @@ def test_Diariser_init(diariser_instance):
assert diariser_instance.model == 'pyannote' assert diariser_instance.model == 'pyannote'
def test_diarisation_function(diariser_instance): def test_diarisation_function(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')