Fixup

revdotcom · Jun 17, 2024 · 4af1623 · 4af1623
1 parent e9f3ba3
commit 4af1623
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/tests/test_job.py b/tests/test_job.py
@@ -120,7 +120,10 @@ def test_submit_job_url_with_success(self, mock_session, make_mock_response):
             'delete_after_seconds': 0,
             'language': LANGUAGE,
             'transcriber': TRANSCRIBER,
-            'skip_postprocessing': True
+            'skip_postprocessing': True,
+            'remove_atmospherics': True,
+            'speakers_count': 123,
+            'diarization_type': "premium"
         }
         response = make_mock_response(url=JOB_ID_URL, json_data=data)
         mock_session.request.return_value = response
@@ -295,7 +298,10 @@ def test_submit_job_local_file_with_success(self, mocker, mock_session, make_moc
             'delete_after_seconds': 0,
             'language': LANGUAGE,
             'transcriber': TRANSCRIBER,
-            'skip_postprocessing': True
+            'skip_postprocessing': True,
+            'remove_atmospherics': True,
+            'speakers_count': 123,
+            'diarization_type': "premium"
         }
         response = make_mock_response(url=JOB_ID_URL, json_data=data)
         mock_session.request.return_value = response