phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(test) proxy: use fixtures

Browse source
This commit is contained in:
ishaan-jaff 2023-12-11 20:03:01 -08:00
parent c2f0a5c76a
commit da0ec40a9f
1 changed files with 10 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

22
litellm/tests/test_proxy_server.py
View file

@ -32,16 +32,17 @@ headers = {
"Authorization": f"Bearer {token}" "Authorization": f"Bearer {token}"
} }
def get_client(config_fp): @pytest.fixture
def client(config_fp):
filepath = os.path.dirname(os.path.abspath(__file__)) filepath = os.path.dirname(os.path.abspath(__file__))
config_fp = f"{filepath}/test_configs/{config_fp}" config_fp = f"{filepath}/test_configs/test_config_no_auth"
# initialize can get run in parallel, it sets specific variables for the fast api app, sinc eit gets run in parallel different tests use the wrong variables
initialize(config=config_fp) initialize(config=config_fp)
app = FastAPI() app = FastAPI()
app.include_router(router) # Include your router in the test app app.include_router(router) # Include your router in the test app
return TestClient(app) return TestClient(app)
def test_chat_completion(): def test_chat_completion(client):
client = get_client(config_fp="test_config_no_auth.yaml")
global headers global headers
try: try:
# Your test data # Your test data
@ -67,8 +68,8 @@ def test_chat_completion():
# Run the test # Run the test
def test_chat_completion_azure(): def test_chat_completion_azure(client):
client = get_client(config_fp="test_config_no_auth.yaml")
global headers global headers
try: try:
# Your test data # Your test data
@ -97,8 +98,7 @@ def test_chat_completion_azure():
# test_chat_completion_azure() # test_chat_completion_azure()
def test_embedding(): def test_embedding(client):
client = get_client(config_fp="test_config_no_auth.yaml")
global headers global headers
try: try:
test_data = { test_data = {
@ -119,8 +119,7 @@ def test_embedding():
# test_embedding() # test_embedding()
# @pytest.mark.skip(reason="hitting yaml load issues on circle-ci") # @pytest.mark.skip(reason="hitting yaml load issues on circle-ci")
def test_add_new_model(): def test_add_new_model(client):
client = get_client(config_fp="test_config_no_auth.yaml")
global headers global headers
try: try:
test_data = { test_data = {
@ -161,10 +160,9 @@ class MyCustomHandler(CustomLogger):
customHandler = MyCustomHandler() customHandler = MyCustomHandler()
def test_chat_completion_optional_params(): def test_chat_completion_optional_params(client):
# [PROXY: PROD TEST] - DO NOT DELETE # [PROXY: PROD TEST] - DO NOT DELETE
# This tests if all the /chat/completion params are passed to litellm # This tests if all the /chat/completion params are passed to litellm
client = get_client(config_fp="test_config_no_auth.yaml")
try: try:
# Your test data # Your test data
litellm.set_verbose=True litellm.set_verbose=True