scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

test: move safe_deep_copy import to module scope

Browse source 
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
Milan 2026-05-08 15:19:57 +03:00
parent b75b090a34
commit 23c27fc305
No known key found for this signature in database
1 changed files with 2 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/local_testing/test_router_fallback_handlers.py
View file

@ -15,6 +15,7 @@ from unittest.mock import AsyncMock, MagicMock, patch
import litellm
from litellm import Router
from litellm.integrations.custom_logger import CustomLogger
from litellm.litellm_core_utils.core_helpers import safe_deep_copy
from typing import Any, Dict, List
from litellm.router_utils.fallback_event_handlers import (
@ -329,10 +330,6 @@ async def test_fallback_kwargs_not_mutated():
`safe_deep_copy` call is removed, only a shallow reference is passed and
`safe_deep_copy` is never called — causing this test to fail.
"""
from litellm.litellm_core_utils.core_helpers import (
safe_deep_copy as _real_safe_deep_copy,
)
router = Router(
model_list=[
{
@ -396,7 +393,7 @@ async def test_fallback_kwargs_not_mutated():
with patch(
"litellm.router_utils.fallback_event_handlers.safe_deep_copy",
wraps=_real_safe_deep_copy,
wraps=safe_deep_copy,
) as mock_sdc:
with pytest.raises(Exception):
await run_async_fallback(