CLI tests

Author: JamesG-Speechmatics

speechmatics/cli.py

@@ -286,7 +286,9 @@ def get_transcription_config(
 
     if args.get("end_of_utterance_silence_trigger") is not None:
         config["conversation_config"] = {
-            "end_of_utterance_silence_trigger": args.get("end_of_utterance_silence_trigger")
+            "end_of_utterance_silence_trigger": args.get(
+                "end_of_utterance_silence_trigger"
+            )
         }
 
     if args.get("volume_threshold") is not None:

speechmatics/cli_parser.py

@@ -223,8 +223,9 @@ def get_arg_parser():
         dest="end_of_utterance_silence_trigger",
         type=float,
         default=None,
-        help=("Generate an EndOfUtterance message from the server after this many seconds of silence (0.0-2.0)"),
-
+        help=(
+            "Generate an EndOfUtterance message from the server after this many seconds of silence (0.0-2.0)"
+        ),
     )
     config_parser.add_argument(
         "--remove-disfluencies",

tests/mock_rt_server.py

@@ -120,6 +120,15 @@ def dummy_add_partial_transcript():
     }
 
 
+def dummy_end_of_utterance():
+    """Returns a dummy EndOfUtterance message."""
+    return {
+        "message": "EndOfUtterance",
+        "format": "2.1",
+        "metadata": {"start_time": 3.0, "end_time": 3.0},
+    }
+
+
 def dummy_add_transcript():
     """Returns a dummy AddTranscript message."""
     return {
@@ -194,9 +203,10 @@ def get_responses(message, is_binary=False):
             )
             mock_server_handler.next_audio_seq_no += 1
 
-            # Answer immediately with a partial and a final.
+            # Answer immediately with a partial and a final and an end of utterance.
             responses.append(dummy_add_partial_transcript())
             responses.append(dummy_add_transcript())
+            responses.append(dummy_end_of_utterance())
         else:
             msg_name = message.get("message")
             if not msg_name:

tests/test_cli.py

@@ -183,6 +183,10 @@
             {"enable_translation_partials": True},
         ),
         (["rt", "transcribe", "--enable-entities"], {"enable_entities": True}),
+        (
+            ["rt", "transcribe", "--end-of-utterance-silence-trigger=1.8"],
+            {"end_of_utterance_silence_trigger": 1.8},
+        ),
         (["batch", "transcribe", "--enable-entities"], {"enable_entities": True}),
         (
             ["batch", "transcribe", "--speaker-diarization-sensitivity=0.7"],

tests/test_models.py

@@ -220,3 +220,17 @@ def test_notification_config(params, want):
 def test_audio_events_config_config(params, want):
     audio_events_config = models.AudioEventsConfig(**params)
     assert audio_events_config.asdict() == want
+
+
+@mark.parametrize(
+    "params, want",
+    [
+        param(
+            {"end_of_utterance_silence_trigger": 1.8},
+            {"end_of_utterance_silence_trigger": 1.8},
+        ),
+    ],
+)
+def test_conversation_config(params, want):
+    conversation_config = models.ConversationConfig(**params)
+    assert asdict(conversation_config) == want