diff --git a/tests/test_budget_tracker_bounds.py b/tests/test_budget_tracker_bounds.py new file mode 100644 index 00000000000..703c1f19c93 --- /dev/null +++ b/tests/test_budget_tracker_bounds.py @@ -0,0 +1,13 @@ +import pytest +from litellm.caching.caching import DualCache + +def test_dual_cache_in_memory_set_get(): + cache = DualCache(in_memory_cache=True) + cache.set_cache(key="user_spend_123", value=45.50, ttl=300) + cached_val = cache.get_cache(key="user_spend_123") + assert cached_val == 45.50 + +def test_dual_cache_miss(): + cache = DualCache(in_memory_cache=True) + val = cache.get_cache(key="non_existent_key_xyz") + assert val is None \ No newline at end of file