phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

better testing

Browse source
This commit is contained in:
yujong-lee 2023-08-18 17:41:31 +09:00
parent 1b78202fc0
commit 701ff22b56
1 changed files with 3 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/tests/test_completion.py
View file

@ -94,14 +94,9 @@ def test_completion_openai():
response_str = response['choices'][0]['message']['content'] response_str = response['choices'][0]['message']['content']
response_str_2 = response.choices[0].message.content response_str_2 = response.choices[0].message.content
print(response_str) assert response_str == response_str_2
print(response_str_2) assert type(response_str) == str
if type(response_str) != str: assert len(response_str) > 1
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)
except Exception as e: except Exception as e:
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")