From 701ff22b5696e6dfb9cee24424e2ece08e3507b0 Mon Sep 17 00:00:00 2001 From: yujong-lee Date: Fri, 18 Aug 2023 17:41:31 +0900 Subject: [PATCH] better testing --- litellm/tests/test_completion.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index dfe9d5e8bb..00054d6a8f 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -94,14 +94,9 @@ def test_completion_openai(): response_str = response['choices'][0]['message']['content'] response_str_2 = response.choices[0].message.content - print(response_str) - print(response_str_2) - if type(response_str) != str: - pytest.fail(f"Error occurred: {e}") - if type(response_str_2) != str: - pytest.fail(f"Error occurred: {e}") - # Add any assertions here to check the response - print(response) + assert response_str == response_str_2 + assert type(response_str) == str + assert len(response_str) > 1 except Exception as e: pytest.fail(f"Error occurred: {e}")