From d774b08766b2a4b95cc776c3e640afe5f8820ff8 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 3 Jul 2025 23:05:03 -0700
Subject: [PATCH] test_arouter_filter_team_based_models

---
 tests/test_litellm/test_router.py | 20 ++++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/tests/test_litellm/test_router.py b/tests/test_litellm/test_router.py
index 69fecfe4ff2..ce3796bc3ba 100644
--- a/tests/test_litellm/test_router.py
+++ b/tests/test_litellm/test_router.py
@@ -82,7 +82,7 @@ def test_router_with_model_info_and_model_group():
 
 
 @pytest.mark.asyncio
-async def test_router_with_tags_and_fallbacks():
+async def test_arouter_with_tags_and_fallbacks():
     """
     If fallback model missing tag, raise error
     """
@@ -239,7 +239,7 @@ async def test_async_router_acreate_file_with_jsonl():
 
 
 @pytest.mark.asyncio
-async def test_router_async_get_healthy_deployments():
+async def test_arouter_async_get_healthy_deployments():
     """
     Test that afile_content returns the correct file content
     """
@@ -268,7 +268,7 @@ async def test_router_async_get_healthy_deployments():
 
 @pytest.mark.asyncio
 @patch("litellm.amoderation")
-async def test_router_amoderation_with_credential_name(mock_amoderation):
+async def test_arouter_amoderation_with_credential_name(mock_amoderation):
     """
     Test that router.amoderation passes litellm_credential_name to the underlying litellm.amoderation call
     """
@@ -298,7 +298,7 @@ async def test_router_amoderation_with_credential_name(mock_amoderation):
     assert call_kwargs["model"] == "text-moderation-stable"
 
 
-def test_router_test_team_model():
+def test_arouter_test_team_model():
     """
     Test that router.test_team_model returns the correct model
     """
@@ -319,7 +319,7 @@ def test_router_test_team_model():
     assert result is not None
 
 
-def test_router_ignore_invalid_deployments():
+def test_arouter_ignore_invalid_deployments():
     """
     Test that router.ignore_invalid_deployments is set to True
     """
@@ -351,7 +351,7 @@ def test_router_ignore_invalid_deployments():
 
 
 @pytest.mark.asyncio
-async def test_router_aretrieve_batch():
+async def test_arouter_aretrieve_batch():
     """
     Test that router.aretrieve_batch returns the correct response
     """
@@ -387,7 +387,7 @@ async def test_router_aretrieve_batch():
 
 
 @pytest.mark.asyncio
-async def test_router_aretrieve_file_content():
+async def test_arouter_aretrieve_file_content():
     """
     Test that router.acreate_file with JSONL file returns the correct response
     """
@@ -426,7 +426,7 @@ async def test_router_aretrieve_file_content():
 
 
 @pytest.mark.asyncio
-async def test_router_filter_team_based_models():
+async def test_arouter_filter_team_based_models():
     """
     Test that router.filter_team_based_models filters out models that are not in the team
     """
@@ -483,7 +483,7 @@ async def test_router_filter_team_based_models():
     assert result is not None
 
 
-def test_router_should_include_deployment():
+def test_arouter_should_include_deployment():
     """
     Test the should_include_deployment method with various scenarios
 
@@ -604,7 +604,7 @@ def test_router_should_include_deployment():
     ), "Should return True when matching model with exact model_name"
 
 
-def test_router_responses_api_bridge():
+def test_arouter_responses_api_bridge():
     """
     Test that router.responses_api_bridge returns the correct response
     """