83 lines
2.1 KiB
Python
83 lines
2.1 KiB
Python
|
import asyncio
|
||
|
|
import time
|
||
|
|
|
||
|
|
from async_cache import ALRU, ATTL
|
||
|
|
|
||
|
|
|
||
|
|
@ALRU(maxsize=128)
|
||
|
|
async def func(wait: int):
|
||
|
|
await asyncio.sleep(wait)
|
||
|
|
|
||
|
|
|
||
|
|
class TestClassFunc:
|
||
|
|
@ALRU(maxsize=128)
|
||
|
|
async def obj_func(self, wait: int):
|
||
|
|
await asyncio.sleep(wait)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
@ATTL(maxsize=128, ttl=None, skip_args=1)
|
||
|
|
async def skip_arg_func(arg: int, wait: int):
|
||
|
|
await asyncio.sleep(wait)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
@ALRU(maxsize=128)
|
||
|
|
async def class_func(cls, wait: int):
|
||
|
|
await asyncio.sleep(wait)
|
||
|
|
|
||
|
|
|
||
|
|
def test():
|
||
|
|
t1 = time.time()
|
||
|
|
asyncio.get_event_loop().run_until_complete(func(4))
|
||
|
|
t2 = time.time()
|
||
|
|
asyncio.get_event_loop().run_until_complete(func(4))
|
||
|
|
t3 = time.time()
|
||
|
|
t_first_exec = (t2 - t1) * 1000
|
||
|
|
t_second_exec = (t3 - t2) * 1000
|
||
|
|
print(t_first_exec)
|
||
|
|
print(t_second_exec)
|
||
|
|
assert t_first_exec > 4000
|
||
|
|
assert t_second_exec < 4000
|
||
|
|
|
||
|
|
|
||
|
|
def test_obj_fn():
|
||
|
|
t1 = time.time()
|
||
|
|
obj = TestClassFunc()
|
||
|
|
asyncio.get_event_loop().run_until_complete(obj.obj_func(4))
|
||
|
|
t2 = time.time()
|
||
|
|
asyncio.get_event_loop().run_until_complete(obj.obj_func(4))
|
||
|
|
t3 = time.time()
|
||
|
|
t_first_exec = (t2 - t1) * 1000
|
||
|
|
t_second_exec = (t3 - t2) * 1000
|
||
|
|
print(t_first_exec)
|
||
|
|
print(t_second_exec)
|
||
|
|
assert t_first_exec > 4000
|
||
|
|
assert t_second_exec < 4000
|
||
|
|
|
||
|
|
|
||
|
|
def test_class_fn():
|
||
|
|
t1 = time.time()
|
||
|
|
asyncio.get_event_loop().run_until_complete(TestClassFunc.class_func(4))
|
||
|
|
t2 = time.time()
|
||
|
|
asyncio.get_event_loop().run_until_complete(TestClassFunc.class_func(4))
|
||
|
|
t3 = time.time()
|
||
|
|
t_first_exec = (t2 - t1) * 1000
|
||
|
|
t_second_exec = (t3 - t2) * 1000
|
||
|
|
print(t_first_exec)
|
||
|
|
print(t_second_exec)
|
||
|
|
assert t_first_exec > 4000
|
||
|
|
assert t_second_exec < 4000
|
||
|
|
|
||
|
|
|
||
|
|
def test_skip_args():
|
||
|
|
t1 = time.time()
|
||
|
|
asyncio.get_event_loop().run_until_complete(TestClassFunc.skip_arg_func(5, 4))
|
||
|
|
t2 = time.time()
|
||
|
|
asyncio.get_event_loop().run_until_complete(TestClassFunc.skip_arg_func(6, 4))
|
||
|
|
t3 = time.time()
|
||
|
|
t_first_exec = (t2 - t1) * 1000
|
||
|
|
t_second_exec = (t3 - t2) * 1000
|
||
|
|
print(t_first_exec)
|
||
|
|
print(t_second_exec)
|
||
|
|
assert t_first_exec > 4000
|
||
|
|
assert t_second_exec < 4000
|