pyqtgraph/tests/util/test_lru_cache.py

53 lines
1.2 KiB
Python

import warnings
with warnings.catch_warnings():
warnings.simplefilter('ignore')
from pyqtgraph.util.lru_cache import LRUCache
def testLRU():
lru = LRUCache(2, 1)
# check twice
checkLru(lru)
checkLru(lru)
def checkLru(lru):
lru[1] = 1
lru[2] = 2
lru[3] = 3
assert len(lru) == 2
assert set([2, 3]) == set(lru.keys())
assert set([2, 3]) == set(lru.values())
lru[2] = 2
assert set([2, 3]) == set(lru.values())
lru[1] = 1
set([2, 1]) == set(lru.values())
#Iterates from the used in the last access to others based on access time.
assert [(2, 2), (1, 1)] == list(lru.items(accessTime=True))
lru[2] = 2
assert [(1, 1), (2, 2)] == list(lru.items(accessTime=True))
del lru[2]
assert [(1, 1), ] == list(lru.items(accessTime=True))
lru[2] = 2
assert [(1, 1), (2, 2)] == list(lru.items(accessTime=True))
_ = lru[1]
assert [(2, 2), (1, 1)] == list(lru.items(accessTime=True))
_ = lru[2]
assert [(1, 1), (2, 2)] == list(lru.items(accessTime=True))
assert lru.get(2) == 2
assert lru.get(3) is None
assert [(1, 1), (2, 2)] == list(lru.items(accessTime=True))
lru.clear()
assert [] == list(lru.items())