tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-11 14:22:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

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
litellm/tests/test_completion.py
+3 -8
View File
@@ -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}")