From 0b00881ce470282d74c1c77a54f45cf8e2348cbe Mon Sep 17 00:00:00 2001 From: Gokul Mohanarangan Date: Mon, 25 Sep 2023 10:22:41 +0530 Subject: [PATCH] update tests: LLM mock to return LLM TITLE for all cases --- server/tests/conftest.py | 2 +- server/tests/test_transcripts_rtc_ws.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/server/tests/conftest.py b/server/tests/conftest.py index d0b3a26f..27417bcb 100644 --- a/server/tests/conftest.py +++ b/server/tests/conftest.py @@ -30,7 +30,7 @@ def dummy_processors(): "reflector.processors.transcript_final_short_summary.TranscriptFinalShortSummaryProcessor.get_short_summary" ) as mock_short_summary: mock_topic.return_value = {"title": "LLM TITLE", "summary": "LLM SUMMARY"} - mock_title.return_value = {"title": "LLM FINAL TITLE"} + mock_title.return_value = {"title": "LLM TITLE"} mock_long_summary.return_value = "LLM LONG SUMMARY" mock_short_summary.return_value = {"short_summary": "LLM SHORT SUMMARY"} diff --git a/server/tests/test_transcripts_rtc_ws.py b/server/tests/test_transcripts_rtc_ws.py index d6816192..d691dacb 100644 --- a/server/tests/test_transcripts_rtc_ws.py +++ b/server/tests/test_transcripts_rtc_ws.py @@ -185,7 +185,7 @@ async def test_transcript_rtc_and_websocket( assert "FINAL_TITLE" in eventnames ev = events[eventnames.index("FINAL_TITLE")] - assert ev["data"]["title"] == "LLM FINAL TITLE" + assert ev["data"]["title"] == "LLM TITLE" # check status order statuses = [e["data"]["value"] for e in events if e["event"] == "STATUS"] @@ -330,7 +330,7 @@ async def test_transcript_rtc_and_websocket_and_fr( assert "FINAL_TITLE" in eventnames ev = events[eventnames.index("FINAL_TITLE")] - assert ev["data"]["title"] == "LLM FINAL TITLE" + assert ev["data"]["title"] == "LLM TITLE" # check status order statuses = [e["data"]["value"] for e in events if e["event"] == "STATUS"]