From 1b0c7ac7c7f34d8d8661ade688a3c71ca08f4945 Mon Sep 17 00:00:00 2001 From: ByronHsu Date: Mon, 27 Jan 2025 21:26:40 -0800 Subject: [PATCH] print output --- test/srt/test_session_control.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/test/srt/test_session_control.py b/test/srt/test_session_control.py index e835a056ecb..e29b8a7b851 100644 --- a/test/srt/test_session_control.py +++ b/test/srt/test_session_control.py @@ -216,7 +216,7 @@ def test_session_control(self, gen_len=12): print(outputs_from_session) print("outputs from normal queries:") print(outputs_normal) - assert outputs_from_session == outputs_normal + assert outputs_from_session == outputs_normal, f"outputs_from_session: {outputs_from_session}, outputs_normal: {outputs_normal}" async def async_generate(self, payload): url = self.base_url + "/generate" @@ -345,7 +345,7 @@ async def run_session_control_backtrack_with_abort(self, replace): output_no_session = response["text"] print("second request output without session:") print(output_no_session) - assert second_output == output_no_session + assert second_output == output_no_session, f"second_output: {second_output}, output_no_session: {output_no_session}" def test_session_control_backtrack_with_abort(self): asyncio.run(self.run_session_control_backtrack_with_abort(replace=True)) @@ -463,7 +463,7 @@ def run_session_control_with_branching( print(outputs_from_session) print("====== outputs from normal queries: =======") print(outputs_normal) - assert outputs_from_session == outputs_normal + assert outputs_from_session == outputs_normal, f"outputs_from_session: {outputs_from_session}, outputs_normal: {outputs_normal}" def test_session_control_with_branching(self): root_prompt = "First, let me explain in one sentence about AI" @@ -696,7 +696,7 @@ def test_session_control(self): print(outputs_from_session) print("outputs from normal queries:") print(outputs_normal) - assert outputs_from_session == outputs_normal + assert outputs_from_session == outputs_normal, f"outputs_from_session: {outputs_from_session}, outputs_normal: {outputs_normal}" if __name__ == "__main__":