From 4194bafae044b431934a2d4027cdb40a06eb88b1 Mon Sep 17 00:00:00 2001 From: Marc Abramowitz Date: Thu, 16 May 2024 11:34:22 -0700 Subject: [PATCH] Add nicer test ids when using pytest -v Replace: ``` test_key_generate_prisma.py::test_generate_and_call_with_valid_key[api_route0] PASSED test_key_generate_prisma.py::test_generate_and_call_with_valid_key[api_route10] PASSED test_key_generate_prisma.py::test_generate_and_call_with_valid_key[api_route11] PASSED test_key_generate_prisma.py::test_generate_and_call_with_valid_key[api_route12] PASSED test_key_generate_prisma.py::test_generate_and_call_with_valid_key[api_route13] PASSED test_key_generate_prisma.py::test_generate_and_call_with_valid_key[api_route14] PASSED ```` with: ``` litellm/tests/test_key_generate_prisma.py::test_generate_and_call_with_valid_key[{'route': 'audio_transcriptions', 'path': '/audio/transcriptions'}] PASSED litellm/tests/test_key_generate_prisma.py::test_generate_and_call_with_valid_key[{'route': 'audio_transcriptions', 'path': '/v1/audio/transcriptions'}] PASSED litellm/tests/test_key_generate_prisma.py::test_generate_and_call_with_valid_key[{'route': 'chat_completion', 'path': '/chat/completions'}] PASSED litellm/tests/test_key_generate_prisma.py::test_generate_and_call_with_valid_key[{'route': 'chat_completion', 'path': '/engines/{model}/chat/completions'}] PASSED litellm/tests/test_key_generate_prisma.py::test_generate_and_call_with_valid_key[{'route': 'chat_completion', 'path': '/openai/deployments/{model}/chat/completions'}] PASSED litellm/tests/test_key_generate_prisma.py::test_generate_and_call_with_valid_key[{'route': 'chat_completion', 'path': '/v1/chat/completions'}] PASSED ``` --- litellm/tests/test_key_generate_prisma.py | 1 + 1 file changed, 1 insertion(+) diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py index 051136256..1d3a487da 100644 --- a/litellm/tests/test_key_generate_prisma.py +++ b/litellm/tests/test_key_generate_prisma.py @@ -183,6 +183,7 @@ async def test_new_user_response(prisma_client): APIRoute(path= "/v1/models", endpoint=model_list), APIRoute(path= "/models", endpoint=model_list), ], + ids=lambda route: str(dict(route=route.endpoint.__name__, path=route.path)), ) def test_generate_and_call_with_valid_key(prisma_client, api_route): # 1. Generate a Key, and use it to make a call