Skip to content

Commit ae5e46a

Browse files
committed
Extra tests
1 parent fe59fa9 commit ae5e46a

File tree

1 file changed

+14
-3
lines changed

1 file changed

+14
-3
lines changed

tests/test_job.py

Lines changed: 14 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)