From b84f25ab35f6858fc5575bd929a5d7bd595c53f6 Mon Sep 17 00:00:00 2001 From: Boxuan Li Date: Mon, 13 May 2024 20:03:09 -0700 Subject: [PATCH] Integration test: exit if no prompt match (#1772) --- tests/integration/conftest.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/integration/conftest.py b/tests/integration/conftest.py index f6f74d9272..2abf127f53 100644 --- a/tests/integration/conftest.py +++ b/tests/integration/conftest.py @@ -1,6 +1,7 @@ import io import os import re +import sys from functools import partial import pytest @@ -79,9 +80,10 @@ def mock_completion(*args, test_name, **kwargs): for message in messages: message_str += message['content'] mock_response = get_mock_response(test_name, message_str) - assert mock_response is not None, ( - 'Mock response for prompt is not found:\n\n' + message_str - ) + if mock_response is None: + print('Mock response for prompt is not found:\n\n' + message_str) + print('Exiting...') + sys.exit(1) response = completion(**kwargs, mock_response=mock_response) return response