@@ -306,7 +306,11 @@ def test_submit_job_local_file_with_success(self, mocker, mock_session, make_moc
306306 NOTIFICATION_URL , True ,
307307 True , 1 , CUSTOM_VOCAB , True ,
308308 True , 0 , LANGUAGE , CUSTOM_VOCAB_ID ,
309- TRANSCRIBER , skip_postprocessing = True )
309+ TRANSCRIBER , skip_postprocessing = True ,
310+ remove_atmospherics = True ,
311+ speakers_count = 123 ,
312+ diarization_type = "premium"
313+ )
310314
311315 assert res == Job (JOB_ID ,
312316 CREATED_ON ,
@@ -320,7 +324,11 @@ def test_submit_job_local_file_with_success(self, mocker, mock_session, make_moc
320324 remove_disfluencies = True ,
321325 delete_after_seconds = 0 ,
322326 language = LANGUAGE ,
323- transcriber = TRANSCRIBER )
327+ transcriber = TRANSCRIBER ,
328+ remove_atmospherics = True ,
329+ speakers_count = 123 ,
330+ diarization_type = "premium"
331+ )
324332 mock_session .request .assert_called_once_with (
325333 "POST" ,
326334 JOBS_URL ,
@@ -341,7 +349,10 @@ def test_submit_job_local_file_with_success(self, mocker, mock_session, make_moc
341349 'language' : LANGUAGE ,
342350 'custom_vocabulary_id' : CUSTOM_VOCAB_ID ,
343351 'transcriber' : TRANSCRIBER ,
344- 'skip_postprocessing' : True
352+ 'skip_postprocessing' : True ,
353+ 'remove_atmospherics' : True ,
354+ 'speakers_count' : 123 ,
355+ 'diarization_type' : "premium"
345356 }, sort_keys = True )
346357 )
347358 },
0 commit comments