Copy async_lru for Python 3.7 compatibility
This commit is contained in:
parent
2b7188b1b8
commit
18c6ba62bb
15 changed files with 1321 additions and 0 deletions
38
tests/external/test_cache_info.py
vendored
Normal file
38
tests/external/test_cache_info.py
vendored
Normal file
|
|
@ -0,0 +1,38 @@
|
|||
import asyncio
|
||||
|
||||
import pytest
|
||||
from great_ai.external.async_lru import alru_cache
|
||||
|
||||
pytestmark = pytest.mark.asyncio
|
||||
|
||||
|
||||
async def test_cache_info(check_lru, loop):
|
||||
@alru_cache(maxsize=4)
|
||||
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, maxsize=4)
|
||||
|
||||
coro.cache_clear()
|
||||
|
||||
check_lru(coro, hits=0, misses=0, cache=0, tasks=0, maxsize=4)
|
||||
|
||||
inputs = [1, 1, 1]
|
||||
coros = [coro(v) for v in inputs]
|
||||
ret = await asyncio.gather(*coros)
|
||||
assert ret == inputs
|
||||
check_lru(coro, hits=2, misses=1, cache=1, tasks=0, maxsize=4)
|
||||
|
||||
coro.cache_clear()
|
||||
|
||||
check_lru(coro, hits=0, misses=0, cache=0, tasks=0, maxsize=4)
|
||||
|
||||
inputs = [1, 2, 3, 4] * 2
|
||||
coros = [coro(v) for v in inputs]
|
||||
ret = await asyncio.gather(*coros)
|
||||
assert ret == inputs
|
||||
check_lru(coro, hits=4, misses=4, cache=4, tasks=0, maxsize=4)
|
||||
Loading…
Add table
Add a link
Reference in a new issue