diff options
Diffstat (limited to 'tests/test_mru.py')
-rw-r--r-- | tests/test_mru.py | 50 |
1 files changed, 50 insertions, 0 deletions
diff --git a/tests/test_mru.py b/tests/test_mru.py new file mode 100644 index 0000000..d11dba4 --- /dev/null +++ b/tests/test_mru.py @@ -0,0 +1,50 @@ +import unittest + +from cachetools import MRUCache + +from . import CacheTestMixin + + +class MRUCacheTest(unittest.TestCase, CacheTestMixin): + + Cache = MRUCache + + def test_evict__writes_only(self): + cache = MRUCache(maxsize=2) + + cache[1] = 1 + cache[2] = 2 + cache[3] = 3 # Evicts 1 because nothing's been used yet + + assert len(cache) == 2 + assert 1 not in cache, "Wrong key was evicted. Should have been '1'." + assert 2 in cache + assert 3 in cache + + def test_evict__with_access(self): + cache = MRUCache(maxsize=2) + + cache[1] = 1 + cache[2] = 2 + cache[1] + cache[2] + cache[3] = 3 # Evicts 2 + assert 2 not in cache, "Wrong key was evicted. Should have been '2'." + assert 1 in cache + assert 3 in cache + + def test_evict__with_delete(self): + cache = MRUCache(maxsize=2) + + cache[1] = 1 + cache[2] = 2 + del cache[2] + cache[3] = 3 # Doesn't evict anything because we just deleted 2 + + assert 2 not in cache + assert 1 in cache + + cache[4] = 4 # Should evict 1 as we just accessed it with __contains__ + assert 1 not in cache + assert 3 in cache + assert 4 in cache |