mirror of
https://github.com/BerriAI/litellm.git
synced 2026-08-28 05:25:59 +00:00
5f864c83ce
* fix: zero out crash-class basedpyright rules across litellm/ * feat(lint): add LIT009 banning inert type: ignore comments * docs: require bracketed rule and reason on every suppression * chore(lint): ratchet budgets down and zero crash-class pyright limits * fix: narrow auto router routelayer through a local before calling * test: add regression tests for crash-class fixes * fix: drop dead AZURE_AD_TOKEN lookups and word-bound the type-ignore regex
41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
import pytest
|
|
|
|
pytest.importorskip("diskcache")
|
|
|
|
from litellm.caching.disk_cache import DiskCache
|
|
|
|
|
|
@pytest.fixture
|
|
def cache(tmp_path):
|
|
return DiskCache(disk_cache_dir=str(tmp_path))
|
|
|
|
|
|
def test_increment_cache_starts_from_zero_when_key_missing(cache):
|
|
assert cache.increment_cache("counter", 3) == 3
|
|
assert cache.get_cache("counter") == 3
|
|
|
|
|
|
def test_increment_cache_adds_to_existing_int(cache):
|
|
cache.set_cache("counter", 7)
|
|
assert cache.increment_cache("counter", 5) == 12
|
|
assert cache.get_cache("counter") == 12
|
|
|
|
|
|
def test_increment_cache_treats_non_int_cached_value_as_zero(cache):
|
|
cache.set_cache("counter", "not-a-number")
|
|
assert cache.increment_cache("counter", 4) == 4
|
|
assert cache.get_cache("counter") == 4
|
|
|
|
|
|
async def test_async_increment_starts_from_zero_when_key_missing(cache):
|
|
assert await cache.async_increment("counter", 2) == 2
|
|
|
|
|
|
async def test_async_increment_adds_to_existing_int(cache):
|
|
await cache.async_set_cache("counter", 10)
|
|
assert await cache.async_increment("counter", 5) == 15
|
|
|
|
|
|
async def test_async_increment_treats_non_int_cached_value_as_zero(cache):
|
|
await cache.async_set_cache("counter", "corrupt")
|
|
assert await cache.async_increment("counter", 9) == 9
|