From fea3966d8eb2fd0418822bfbb4998ede79efccc3 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 12 May 2025 22:04:55 -0700 Subject: [PATCH] refactor: rename tests --- ...on.py => test_novita_chat_transformation.py} | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) rename tests/litellm/llms/novita/chat/{test_transformation.py => test_novita_chat_transformation.py} (90%) diff --git a/tests/litellm/llms/novita/chat/test_transformation.py b/tests/litellm/llms/novita/chat/test_novita_chat_transformation.py similarity index 90% rename from tests/litellm/llms/novita/chat/test_transformation.py rename to tests/litellm/llms/novita/chat/test_novita_chat_transformation.py index 5216ea9488..7a00b36125 100644 --- a/tests/litellm/llms/novita/chat/test_transformation.py +++ b/tests/litellm/llms/novita/chat/test_novita_chat_transformation.py @@ -27,7 +27,7 @@ class TestNovitaConfig: config = NovitaConfig() headers = {} api_key = "fake-novita-key" - + result = config.validate_environment( headers=headers, model="novita/meta-llama/llama-3.3-70b-instruct", @@ -35,9 +35,9 @@ class TestNovitaConfig: optional_params={}, litellm_params={}, api_key=api_key, - api_base="https://api.novita.ai/v3/openai" + api_base="https://api.novita.ai/v3/openai", ) - + # Verify headers assert result["Authorization"] == f"Bearer {api_key}" assert result["Content-Type"] == "application/json" @@ -46,7 +46,7 @@ class TestNovitaConfig: def test_missing_api_key(self): """Test error handling when API key is missing""" config = NovitaConfig() - + with pytest.raises(ValueError) as excinfo: config.validate_environment( headers={}, @@ -55,15 +55,16 @@ class TestNovitaConfig: optional_params={}, litellm_params={}, api_key=None, - api_base="https://api.novita.ai/v3/openai" + api_base="https://api.novita.ai/v3/openai", ) - + assert "Missing Novita AI API Key" in str(excinfo.value) def test_inheritance(self): """Test proper inheritance from OpenAIGPTConfig""" config = NovitaConfig() - + from litellm.llms.openai.chat.gpt_transformation import OpenAIGPTConfig + assert isinstance(config, OpenAIGPTConfig) - assert hasattr(config, "get_supported_openai_params") \ No newline at end of file + assert hasattr(config, "get_supported_openai_params")