diff --git a/tests/unit/conftest.py b/tests/unit/conftest.py
index 017e63ed1b8..fa91437964f 100644
--- a/tests/unit/conftest.py
+++ b/tests/unit/conftest.py
@@ -1,6 +1,7 @@
 import os
 from typing import Final
 
+import litellm
 import pytest
 from pytest_socket import enable_socket, socket_allow_hosts
 
@@ -9,6 +10,19 @@ os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
 LOOPBACK_HOSTS: Final = ["127.0.0.1", "::1"]
 
 
+@pytest.fixture
+def local_model_cost_map(monkeypatch):
+    original_model_cost = litellm.model_cost
+    monkeypatch.setenv("LITELLM_LOCAL_MODEL_COST_MAP", "True")
+    litellm.model_cost = litellm.get_model_cost_map(url="")
+    litellm.get_model_info.cache_clear()
+    try:
+        yield
+    finally:
+        litellm.model_cost = original_model_cost
+        litellm.get_model_info.cache_clear()
+
+
 def _allow_loopback_only() -> None:
     socket_allow_hosts(LOOPBACK_HOSTS, allow_unix_socket=True)
 
diff --git a/tests/test_litellm/llms/openrouter/responses/test_openrouter_responses_transformation.py b/tests/unit/llms/openrouter/responses/test_openrouter_responses_transformation.py
similarity index 100%
rename from tests/test_litellm/llms/openrouter/responses/test_openrouter_responses_transformation.py
rename to tests/unit/llms/openrouter/responses/test_openrouter_responses_transformation.py
diff --git a/tests/test_litellm/llms/parallel_ai/test_parallel_ai_search.py b/tests/unit/llms/parallel_ai/test_parallel_ai_search.py
similarity index 100%
rename from tests/test_litellm/llms/parallel_ai/test_parallel_ai_search.py
rename to tests/unit/llms/parallel_ai/test_parallel_ai_search.py
diff --git a/tests/test_litellm/llms/parallel_ai/test_parallel_ai_search_gateway.py b/tests/unit/llms/parallel_ai/test_parallel_ai_search_gateway.py
similarity index 100%
rename from tests/test_litellm/llms/parallel_ai/test_parallel_ai_search_gateway.py
rename to tests/unit/llms/parallel_ai/test_parallel_ai_search_gateway.py
diff --git a/tests/test_litellm/llms/parasail/test_parasail.py b/tests/unit/llms/parasail/test_parasail.py
similarity index 100%
rename from tests/test_litellm/llms/parasail/test_parasail.py
rename to tests/unit/llms/parasail/test_parasail.py
diff --git a/tests/test_litellm/llms/perplexity/chat/test_perplexity_chat_transformation.py b/tests/unit/llms/perplexity/chat/test_perplexity_chat_transformation.py
similarity index 100%
rename from tests/test_litellm/llms/perplexity/chat/test_perplexity_chat_transformation.py
rename to tests/unit/llms/perplexity/chat/test_perplexity_chat_transformation.py
diff --git a/tests/test_litellm/llms/perplexity/embedding/test_perplexity_embedding_transformation.py b/tests/unit/llms/perplexity/embedding/test_perplexity_embedding_transformation.py
similarity index 100%
rename from tests/test_litellm/llms/perplexity/embedding/test_perplexity_embedding_transformation.py
rename to tests/unit/llms/perplexity/embedding/test_perplexity_embedding_transformation.py
diff --git a/tests/test_litellm/llms/perplexity/responses/test_perplexity_responses_transformation.py b/tests/unit/llms/perplexity/responses/test_perplexity_responses_transformation.py
similarity index 100%
rename from tests/test_litellm/llms/perplexity/responses/test_perplexity_responses_transformation.py
rename to tests/unit/llms/perplexity/responses/test_perplexity_responses_transformation.py
diff --git a/tests/test_litellm/llms/publicai/test_publicai_chat_transformation.py b/tests/unit/llms/publicai/test_publicai_chat_transformation.py
similarity index 100%
rename from tests/test_litellm/llms/publicai/test_publicai_chat_transformation.py
rename to tests/unit/llms/publicai/test_publicai_chat_transformation.py
diff --git a/tests/test_litellm/llms/ragflow/chat/test_ragflow_chat_transformation.py b/tests/unit/llms/ragflow/chat/test_ragflow_chat_transformation.py
similarity index 100%
rename from tests/test_litellm/llms/ragflow/chat/test_ragflow_chat_transformation.py
rename to tests/unit/llms/ragflow/chat/test_ragflow_chat_transformation.py
diff --git a/tests/test_litellm/llms/recraft/image_edit/test_recraft_image_edit_transformation.py b/tests/unit/llms/recraft/image_edit/test_recraft_image_edit_transformation.py
similarity index 100%
rename from tests/test_litellm/llms/recraft/image_edit/test_recraft_image_edit_transformation.py
rename to tests/unit/llms/recraft/image_edit/test_recraft_image_edit_transformation.py
diff --git a/tests/test_litellm/llms/recraft/image_generation/test_recraft_image_gen_transformation.py b/tests/unit/llms/recraft/image_generation/test_recraft_image_gen_transformation.py
similarity index 100%
rename from tests/test_litellm/llms/recraft/image_generation/test_recraft_image_gen_transformation.py
rename to tests/unit/llms/recraft/image_generation/test_recraft_image_gen_transformation.py
diff --git a/tests/test_litellm/llms/runwayml/test_text_to_speech_transformation.py b/tests/unit/llms/runwayml/test_text_to_speech_transformation.py
similarity index 100%
rename from tests/test_litellm/llms/runwayml/test_text_to_speech_transformation.py
rename to tests/unit/llms/runwayml/test_text_to_speech_transformation.py
diff --git a/tests/test_litellm/llms/runwayml/videos/test_runway_video_transformation.py b/tests/unit/llms/runwayml/videos/test_runway_video_transformation.py
similarity index 100%
rename from tests/test_litellm/llms/runwayml/videos/test_runway_video_transformation.py
rename to tests/unit/llms/runwayml/videos/test_runway_video_transformation.py
diff --git a/tests/test_litellm/llms/s3_vectors/vector_stores/test_s3_vectors_transformation.py b/tests/unit/llms/s3_vectors/vector_stores/test_s3_vectors_transformation.py
similarity index 99%
rename from tests/test_litellm/llms/s3_vectors/vector_stores/test_s3_vectors_transformation.py
rename to tests/unit/llms/s3_vectors/vector_stores/test_s3_vectors_transformation.py
index 781e92ea7d9..c39887d86ce 100644
--- a/tests/test_litellm/llms/s3_vectors/vector_stores/test_s3_vectors_transformation.py
+++ b/tests/unit/llms/s3_vectors/vector_stores/test_s3_vectors_transformation.py
@@ -55,10 +55,6 @@ def _search_kwargs(**overrides):
 
 
 class TestS3VectorsVectorStoreConfig:
-    def test_init(self):
-        config = S3VectorsVectorStoreConfig()
-        assert config is not None
-
     def test_get_supported_openai_params(self):
         config = S3VectorsVectorStoreConfig()
         params = config.get_supported_openai_params("test-model")
diff --git a/tests/test_litellm/llms/sap/test_sap_fetch_creds.py b/tests/unit/llms/sap/test_sap_fetch_creds.py
similarity index 100%
rename from tests/test_litellm/llms/sap/test_sap_fetch_creds.py
rename to tests/unit/llms/sap/test_sap_fetch_creds.py
diff --git a/tests/test_litellm/llms/scaleway/test_scaleway_audio_transcription_transformation.py b/tests/unit/llms/scaleway/test_scaleway_audio_transcription_transformation.py
similarity index 100%
rename from tests/test_litellm/llms/scaleway/test_scaleway_audio_transcription_transformation.py
rename to tests/unit/llms/scaleway/test_scaleway_audio_transcription_transformation.py
diff --git a/tests/test_litellm/llms/snowflake/test_snowflake_native_endpoints.py b/tests/unit/llms/snowflake/test_snowflake_native_endpoints.py
similarity index 100%
rename from tests/test_litellm/llms/snowflake/test_snowflake_native_endpoints.py
rename to tests/unit/llms/snowflake/test_snowflake_native_endpoints.py
diff --git a/tests/test_litellm/llms/soniox/test_soniox_provider_registration.py b/tests/unit/llms/soniox/test_soniox_provider_registration.py
similarity index 100%
rename from tests/test_litellm/llms/soniox/test_soniox_provider_registration.py
rename to tests/unit/llms/soniox/test_soniox_provider_registration.py
diff --git a/tests/test_litellm/llms/stability/image_generation/test_stability_image_generation.py b/tests/unit/llms/stability/image_generation/test_stability_image_generation.py
similarity index 93%
rename from tests/test_litellm/llms/stability/image_generation/test_stability_image_generation.py
rename to tests/unit/llms/stability/image_generation/test_stability_image_generation.py
index c5b3c8fbdc5..c5a78603f9c 100644
--- a/tests/test_litellm/llms/stability/image_generation/test_stability_image_generation.py
+++ b/tests/unit/llms/stability/image_generation/test_stability_image_generation.py
@@ -10,10 +10,7 @@ from unittest.mock import MagicMock
 import httpx
 import pytest
 
-from litellm.llms.stability.image_generation import (
-    StabilityImageGenerationConfig,
-    get_stability_image_generation_config,
-)
+from litellm.llms.stability.image_generation import StabilityImageGenerationConfig
 from litellm.types.llms.stability import (
     OPENAI_SIZE_TO_STABILITY_ASPECT_RATIO,
     STABILITY_GENERATION_MODELS,
@@ -266,20 +263,6 @@ class TestStabilityImageGenerationConfig:
         assert "filtered" in str(exc_info.value).lower()
 
 
-class TestFactoryFunction:
-    """Test the factory function"""
-
-    def test_get_stability_image_generation_config(self):
-        """Test that factory returns correct config type"""
-        config = get_stability_image_generation_config("stability/sd3")
-        assert isinstance(config, StabilityImageGenerationConfig)
-
-    def test_factory_returns_config_for_any_model(self):
-        """Test that factory works for any model name"""
-        config = get_stability_image_generation_config("stability/custom-model")
-        assert isinstance(config, StabilityImageGenerationConfig)
-
-
 class TestOpenAISizeMapping:
     """Test the size to aspect ratio mapping"""
 
diff --git a/tests/test_litellm/llms/tencent/chat/test_tencent_chat_transformation.py b/tests/unit/llms/tencent/chat/test_tencent_chat_transformation.py
similarity index 100%
rename from tests/test_litellm/llms/tencent/chat/test_tencent_chat_transformation.py
rename to tests/unit/llms/tencent/chat/test_tencent_chat_transformation.py