Add tests to check speaker diarization config from file is used

Author: Claire Schaefer

tests/data/transcription_config.json

@@ -2,6 +2,12 @@
   "language": "xy",
   "domain": "fake",
   "enable_entities": true,
+  "diarization": "speaker",
+  "speaker_diarization_config": {
+    "prefer_current_speaker": true,
+    "max_speakers": 5,
+    "speaker_sensitivity": 0.3
+  },
   "translation_config": {
     "target_languages": ["es"],
     "enable_partials": false

tests/test_cli.py

@@ -6,8 +6,7 @@
 import pytest
 import toml
 
-from speechmatics import cli
-from speechmatics import cli_parser
+from speechmatics import cli, cli_parser
 from speechmatics.constants import (
     BATCH_SELF_SERVICE_URL,
     RT_SELF_SERVICE_URL,
@@ -771,6 +770,12 @@ def test_rt_main_with_config_file(mock_server):
     assert msg["transcription_config"]["domain"] == "fake"
     assert msg["transcription_config"]["enable_entities"] is True
     assert msg["transcription_config"].get("operating_point") is None
+    assert msg["transcription_config"]["diarization"] == "speaker"
+    assert msg["transcription_config"]["speaker_diarization_config"] == {
+        "prefer_current_speaker": True,
+        "max_speakers": 5,
+        "speaker_sensitivity": 0.3
+    }
     assert msg["translation_config"] is not None
     assert msg["translation_config"]["enable_partials"] is False
     assert msg["translation_config"]["target_languages"] == ["es"]
@@ -795,6 +800,8 @@ def test_rt_main_with_config_file_cmdline_override(mock_server):
         "--output-locale=en-US",
         "--domain=different",
         "--operating-point=enhanced",
+        "--speaker-diarization-max-speakers=3",
+        "--speaker-diarization-sensitivity=0.7",
         audio_path,
     ]
 
@@ -816,6 +823,12 @@ def test_rt_main_with_config_file_cmdline_override(mock_server):
     assert msg["transcription_config"]["enable_entities"] is True
     assert msg["transcription_config"]["output_locale"] == "en-US"
     assert msg["transcription_config"]["operating_point"] == "enhanced"
+    assert msg["transcription_config"]["diarization"] == "speaker"
+    assert msg["transcription_config"]["speaker_diarization_config"] == {
+        "prefer_current_speaker": True,
+        "max_speakers": 3,
+        "speaker_sensitivity": 0.7
+    }
     assert msg["translation_config"] is not None
     assert msg["translation_config"]["enable_partials"] is True
     assert msg["translation_config"]["target_languages"] == ["fr"]