From 5b5a2d12a8de283da26b02019732d001c774b28a Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Thu, 21 Aug 2025 00:32:16 +0530 Subject: [PATCH] remove test --- .../test_deepinfra_rerank_transformation.py | 29 ------------------- 1 file changed, 29 deletions(-) diff --git a/tests/test_litellm/llms/deepinfra/test_deepinfra_rerank_transformation.py b/tests/test_litellm/llms/deepinfra/test_deepinfra_rerank_transformation.py index e40a23206eb..252eb40532c 100644 --- a/tests/test_litellm/llms/deepinfra/test_deepinfra_rerank_transformation.py +++ b/tests/test_litellm/llms/deepinfra/test_deepinfra_rerank_transformation.py @@ -42,35 +42,6 @@ class TestDeepinfraRerankTransform: with pytest.raises(ValueError, match="Deepinfra API Base is required"): self.config.get_complete_url(None, model) - def test_validate_environment(self): - """Test environment validation with API key.""" - # Test with API key provided - headers = self.config.validate_environment( - headers={}, model="test", api_key="test_key" - ) - assert "Authorization" in headers - assert headers["Authorization"] == "Bearer test_key" - assert headers["accept"] == "application/json" - assert headers["content-type"] == "application/json" - - # Test headers override - custom_headers = {"custom": "header", "Authorization": "Bearer custom_key"} - headers = self.config.validate_environment( - headers=custom_headers, model="test", api_key="test_key" - ) - assert headers["custom"] == "header" - assert ( - headers["Authorization"] == "Bearer custom_key" - ) # Custom auth should override - - # Test without API key should use environment variable (DEEPINFRA_API_KEY from .env) - headers = self.config.validate_environment( - headers={}, model="test", api_key=None - ) - assert "Authorization" in headers - assert headers["Authorization"].startswith("Bearer ") - assert headers["accept"] == "application/json" - assert headers["content-type"] == "application/json" def test_map_cohere_rerank_params_basic(self): """Test basic parameter mapping for DeepInfra rerank."""