mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-14 04:09:09 +00:00
Merge fe722203c6 into d8c0e2c715
This commit is contained in:
commit
ad374147e7
1 changed files with 13 additions and 0 deletions
13
tests/test_budget_tracker_bounds.py
Normal file
13
tests/test_budget_tracker_bounds.py
Normal file
|
|
@ -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
|
||||
Loading…
Add table
Reference in a new issue