diff --git a/tests/local_testing/test_get_model_file.py b/tests/local_testing/test_get_model_file.py index bbcc9d5864d..e36cf0c24cd 100644 --- a/tests/local_testing/test_get_model_file.py +++ b/tests/local_testing/test_get_model_file.py @@ -17,16 +17,9 @@ def test_get_model_cost_map(): def test_get_local_model_cost_map(): - """Test that we can load the local model cost map""" - from pathlib import Path + """Test that load_local_model_cost_map returns a valid non-empty dict.""" + from litellm.litellm_core_utils.get_model_cost_map import GetModelCostMap - # Test loading from project root (development scenario) - project_root = Path(__file__).parent.parent.parent - model_cost_map_path = project_root / "model_prices_and_context_window.json" - - with open(model_cost_map_path, "r") as f: - print("inside local model cost map") - content = json.load(f) - print("content", content) - assert content is not None - assert len(content) > 0 + content = GetModelCostMap.load_local_model_cost_map() + assert isinstance(content, dict) + assert len(content) > 0 diff --git a/tests/test_litellm/test_deepseek_model_metadata.py b/tests/test_litellm/test_deepseek_model_metadata.py index 13e17667ad4..03c9488f758 100644 --- a/tests/test_litellm/test_deepseek_model_metadata.py +++ b/tests/test_litellm/test_deepseek_model_metadata.py @@ -52,33 +52,33 @@ class TestDeepSeekModelCostEntries: """Verify that provider-prefixed DeepSeek entries contain the same capability flags as their bare-name counterparts in the JSON files.""" - def test_deepseek_chat_supports_response_schema_in_backup(self): + def test_deepseek_chat_supports_response_schema(self): data = _load_model_cost_json() entry = data.get("deepseek/deepseek-chat", {}) assert entry.get("supports_response_schema") is True - def test_deepseek_reasoner_supports_response_schema_in_backup(self): + def test_deepseek_reasoner_supports_response_schema(self): data = _load_model_cost_json() entry = data.get("deepseek/deepseek-reasoner", {}) assert entry.get("supports_response_schema") is True - def test_deepseek_chat_supports_system_messages_in_backup(self): + def test_deepseek_chat_supports_system_messages(self): data = _load_model_cost_json() entry = data.get("deepseek/deepseek-chat", {}) assert entry.get("supports_system_messages") is True - def test_deepseek_reasoner_supports_system_messages_in_backup(self): + def test_deepseek_reasoner_supports_system_messages(self): data = _load_model_cost_json() entry = data.get("deepseek/deepseek-reasoner", {}) assert entry.get("supports_system_messages") is True - def test_deepseek_chat_max_input_tokens_matches_bare_in_backup(self): + def test_deepseek_chat_max_input_tokens_matches_bare(self): data = _load_model_cost_json() bare = data.get("deepseek-chat", {}) prefixed = data.get("deepseek/deepseek-chat", {}) assert prefixed.get("max_input_tokens") == bare.get("max_input_tokens") - def test_deepseek_reasoner_max_output_tokens_matches_bare_in_backup(self): + def test_deepseek_reasoner_max_output_tokens_matches_bare(self): data = _load_model_cost_json() bare = data.get("deepseek-reasoner", {}) prefixed = data.get("deepseek/deepseek-reasoner", {})