@@ -130,7 +130,12 @@ def test_submit_job_url_with_success(self, mock_session, make_mock_response):
130130 NOTIFICATION_URL , True ,
131131 True , 1 , CUSTOM_VOCAB , True ,
132132 True , 0 , LANGUAGE , CUSTOM_VOCAB_ID ,
133- TRANSCRIBER , skip_postprocessing = True )
133+ TRANSCRIBER ,
134+ skip_postprocessing = True ,
135+ remove_atmospherics = True ,
136+ speakers_count = 123 ,
137+ diarization_type = "premium"
138+ )
134139
135140 assert res == Job (JOB_ID ,
136141 CREATED_ON ,
@@ -143,7 +148,11 @@ def test_submit_job_url_with_success(self, mock_session, make_mock_response):
143148 remove_disfluencies = True ,
144149 delete_after_seconds = 0 ,
145150 language = LANGUAGE ,
146- transcriber = TRANSCRIBER )
151+ transcriber = TRANSCRIBER ,
152+ remove_atmospherics = True ,
153+ speakers_count = 123 ,
154+ diarization_type = "premium"
155+ )
147156 mock_session .request .assert_called_once_with (
148157 "POST" ,
149158 JOBS_URL ,
@@ -161,7 +170,10 @@ def test_submit_job_url_with_success(self, mock_session, make_mock_response):
161170 'language' : LANGUAGE ,
162171 'custom_vocabulary_id' : CUSTOM_VOCAB_ID ,
163172 'transcriber' : TRANSCRIBER ,
164- 'skip_postprocessing' : True
173+ 'skip_postprocessing' : True ,
174+ 'remove_atmospherics' : True ,
175+ 'speakers_count' : 123 ,
176+ 'diarization_type' : "premium"
165177 },
166178 headers = client .default_headers )
167179
0 commit comments