22 lines
487 B
Python
22 lines
487 B
Python
import asyncio
|
|
|
|
import pytest
|
|
from great_ai.external.async_lru import alru_cache
|
|
|
|
pytestmark = pytest.mark.asyncio
|
|
|
|
|
|
async def test_cache_clear(check_lru, loop):
|
|
@alru_cache()
|
|
async def coro(val):
|
|
return val
|
|
|
|
inputs = [1, 2, 3]
|
|
coros = [coro(v) for v in inputs]
|
|
ret = await asyncio.gather(*coros)
|
|
assert ret == inputs
|
|
check_lru(coro, hits=0, misses=3, cache=3, tasks=0)
|
|
|
|
coro.cache_clear()
|
|
|
|
check_lru(coro, hits=0, misses=0, cache=0, tasks=0)
|