Skip to content

Commit fe59fa9

Browse files
committed
Extra tests
1 parent 7fbd6d9 commit fe59fa9

File tree

1 file changed

+15
-3
lines changed

1 file changed

+15
-3
lines changed

tests/test_job.py

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

Comments
 (0)