diff --git a/tests/test_litellm/litellm_core_utils/test_embedding_optional_params.py b/tests/test_litellm/litellm_core_utils/test_embedding_optional_params.py
new file mode 100644
index 00000000000..6a044a5680a
--- /dev/null
+++ b/tests/test_litellm/litellm_core_utils/test_embedding_optional_params.py
@@ -0,0 +1,51 @@
+import pytest
+
+import litellm
+
+
+@pytest.mark.parametrize("provider", ["azure", "together_ai"])
+def test_embedding_dimensions_drop_params_for_openai_compatible_provider(provider):
+    previous_drop_params = litellm.drop_params
+    try:
+        litellm.drop_params = False
+        dropped = litellm.utils.get_optional_params_embeddings(
+            model=f"{provider}/dummy-model",
+            custom_llm_provider=provider,
+            dimensions=512,
+            drop_params=True,
+        )
+        assert "dimensions" not in dropped
+
+        litellm.drop_params = True
+        dropped_globally = litellm.utils.get_optional_params_embeddings(
+            model=f"{provider}/dummy-model",
+            custom_llm_provider=provider,
+            dimensions=512,
+        )
+        assert "dimensions" not in dropped_globally
+
+        litellm.drop_params = False
+        preserved = litellm.utils.get_optional_params_embeddings(
+            model=f"{provider}/dummy-model",
+            custom_llm_provider=provider,
+            dimensions=512,
+        )
+        assert preserved["dimensions"] == 512
+
+        litellm.drop_params = True
+        model_supported = litellm.utils.get_optional_params_embeddings(
+            model=f"{provider}/text-embedding-3-small",
+            custom_llm_provider=provider,
+            dimensions=512,
+        )
+        assert model_supported["dimensions"] == 512
+
+        explicitly_allowed = litellm.utils.get_optional_params_embeddings(
+            model=f"{provider}/legacy-model",
+            custom_llm_provider=provider,
+            dimensions=512,
+            allowed_openai_params=["dimensions"],
+        )
+        assert explicitly_allowed["dimensions"] == 512
+    finally:
+        litellm.drop_params = previous_drop_params
diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py
index 438c0e39908..6524353aa48 100644
--- a/tests/test_litellm/test_utils.py
+++ b/tests/test_litellm/test_utils.py
@@ -733,54 +733,6 @@ def test_cohere_embedding_optional_params():
     assert optional_params is not None
 
 
-@pytest.mark.parametrize("provider", ["azure", "together_ai"])
-def test_embedding_dimensions_drop_params_for_openai_compatible_provider(provider):
-    previous_drop_params = litellm.drop_params
-    try:
-        litellm.drop_params = False
-        dropped = litellm.utils.get_optional_params_embeddings(
-            model=f"{provider}/dummy-model",
-            custom_llm_provider=provider,
-            dimensions=512,
-            drop_params=True,
-        )
-        assert "dimensions" not in dropped
-
-        litellm.drop_params = True
-        dropped_globally = litellm.utils.get_optional_params_embeddings(
-            model=f"{provider}/dummy-model",
-            custom_llm_provider=provider,
-            dimensions=512,
-        )
-        assert "dimensions" not in dropped_globally
-
-        litellm.drop_params = False
-        preserved = litellm.utils.get_optional_params_embeddings(
-            model=f"{provider}/dummy-model",
-            custom_llm_provider=provider,
-            dimensions=512,
-        )
-        assert preserved["dimensions"] == 512
-
-        litellm.drop_params = True
-        model_supported = litellm.utils.get_optional_params_embeddings(
-            model=f"{provider}/text-embedding-3-small",
-            custom_llm_provider=provider,
-            dimensions=512,
-        )
-        assert model_supported["dimensions"] == 512
-
-        explicitly_allowed = litellm.utils.get_optional_params_embeddings(
-            model=f"{provider}/legacy-model",
-            custom_llm_provider=provider,
-            dimensions=512,
-            allowed_openai_params=["dimensions"],
-        )
-        assert explicitly_allowed["dimensions"] == 512
-    finally:
-        litellm.drop_params = previous_drop_params
-
-
 def validate_model_cost_values(model_data, exceptions=None):
     """
     Validates that cost values in model data do not exceed 1.