| Server IP : 185.124.137.96 / Your IP : 216.73.216.187 Web Server : LiteSpeed System : Linux id-dci-web1986.main-hosting.eu 5.14.0-611.26.1.el9_7.x86_64 #1 SMP PREEMPT_DYNAMIC Thu Jan 29 05:24:47 EST 2026 x86_64 User : u686484674 ( 686484674) PHP Version : 8.0.30 Disable Function : system, exec, shell_exec, passthru, mysql_list_dbs, ini_alter, dl, symlink, link, chgrp, leak, popen, apache_child_terminate, virtual, mb_send_mail MySQL : OFF | cURL : ON | WGET : ON | Perl : OFF | Python : OFF | Sudo : OFF | Pkexec : OFF Directory : /./opt/gsutil/third_party/cachetools/tests/ |
Upload File : |
import unittest
from cachetools import LRUCache, cachedmethod, keys
class Cached:
def __init__(self, cache, count=0):
self.cache = cache
self.count = count
@cachedmethod(lambda self: self.cache)
def get(self, value):
count = self.count
self.count += 1
return count
@cachedmethod(lambda self: self.cache, key=keys.typedmethodkey)
def get_typedmethod(self, value):
count = self.count
self.count += 1
return count
class Locked:
def __init__(self, cache):
self.cache = cache
self.count = 0
@cachedmethod(lambda self: self.cache, lock=lambda self: self)
def get(self, value):
return self.count
def __enter__(self):
self.count += 1
def __exit__(self, *exc):
pass
class Unhashable:
def __init__(self, cache):
self.cache = cache
@cachedmethod(lambda self: self.cache)
def get_default(self, value):
return value
@cachedmethod(lambda self: self.cache, key=keys.hashkey)
def get_hashkey(self, value):
return value
# https://github.com/tkem/cachetools/issues/107
def __hash__(self):
raise TypeError("unhashable type")
class CachedMethodTest(unittest.TestCase):
def test_dict(self):
cached = Cached({})
self.assertEqual(cached.get(0), 0)
self.assertEqual(cached.get(1), 1)
self.assertEqual(cached.get(1), 1)
self.assertEqual(cached.get(1.0), 1)
self.assertEqual(cached.get(1.0), 1)
cached.cache.clear()
self.assertEqual(cached.get(1), 2)
def test_typedmethod_dict(self):
cached = Cached(LRUCache(maxsize=2))
self.assertEqual(cached.get_typedmethod(0), 0)
self.assertEqual(cached.get_typedmethod(1), 1)
self.assertEqual(cached.get_typedmethod(1), 1)
self.assertEqual(cached.get_typedmethod(1.0), 2)
self.assertEqual(cached.get_typedmethod(1.0), 2)
self.assertEqual(cached.get_typedmethod(0.0), 3)
self.assertEqual(cached.get_typedmethod(0), 4)
def test_lru(self):
cached = Cached(LRUCache(maxsize=2))
self.assertEqual(cached.get(0), 0)
self.assertEqual(cached.get(1), 1)
self.assertEqual(cached.get(1), 1)
self.assertEqual(cached.get(1.0), 1)
self.assertEqual(cached.get(1.0), 1)
cached.cache.clear()
self.assertEqual(cached.get(1), 2)
def test_typedmethod_lru(self):
cached = Cached(LRUCache(maxsize=2))
self.assertEqual(cached.get_typedmethod(0), 0)
self.assertEqual(cached.get_typedmethod(1), 1)
self.assertEqual(cached.get_typedmethod(1), 1)
self.assertEqual(cached.get_typedmethod(1.0), 2)
self.assertEqual(cached.get_typedmethod(1.0), 2)
self.assertEqual(cached.get_typedmethod(0.0), 3)
self.assertEqual(cached.get_typedmethod(0), 4)
def test_nospace(self):
cached = Cached(LRUCache(maxsize=0))
self.assertEqual(cached.get(0), 0)
self.assertEqual(cached.get(1), 1)
self.assertEqual(cached.get(1), 2)
self.assertEqual(cached.get(1.0), 3)
self.assertEqual(cached.get(1.0), 4)
def test_nocache(self):
cached = Cached(None)
self.assertEqual(cached.get(0), 0)
self.assertEqual(cached.get(1), 1)
self.assertEqual(cached.get(1), 2)
self.assertEqual(cached.get(1.0), 3)
self.assertEqual(cached.get(1.0), 4)
def test_weakref(self):
import weakref
import fractions
import gc
# in Python 3.7, `int` does not support weak references even
# when subclassed, but Fraction apparently does...
class Int(fractions.Fraction):
def __add__(self, other):
return Int(fractions.Fraction.__add__(self, other))
cached = Cached(weakref.WeakValueDictionary(), count=Int(0))
self.assertEqual(cached.get(0), 0)
gc.collect()
self.assertEqual(cached.get(0), 1)
ref = cached.get(1)
self.assertEqual(ref, 2)
self.assertEqual(cached.get(1), 2)
self.assertEqual(cached.get(1.0), 2)
ref = cached.get_typedmethod(1)
self.assertEqual(ref, 3)
self.assertEqual(cached.get_typedmethod(1), 3)
self.assertEqual(cached.get_typedmethod(1.0), 4)
cached.cache.clear()
self.assertEqual(cached.get(1), 5)
def test_locked_dict(self):
cached = Locked({})
self.assertEqual(cached.get(0), 1)
self.assertEqual(cached.get(1), 3)
self.assertEqual(cached.get(1), 3)
self.assertEqual(cached.get(1.0), 3)
self.assertEqual(cached.get(2.0), 7)
def test_locked_nocache(self):
cached = Locked(None)
self.assertEqual(cached.get(0), 0)
self.assertEqual(cached.get(1), 0)
self.assertEqual(cached.get(1), 0)
self.assertEqual(cached.get(1.0), 0)
self.assertEqual(cached.get(1.0), 0)
def test_locked_nospace(self):
cached = Locked(LRUCache(maxsize=0))
self.assertEqual(cached.get(0), 1)
self.assertEqual(cached.get(1), 3)
self.assertEqual(cached.get(1), 5)
self.assertEqual(cached.get(1.0), 7)
self.assertEqual(cached.get(1.0), 9)
def test_unhashable(self):
cached = Unhashable(LRUCache(maxsize=0))
self.assertEqual(cached.get_default(0), 0)
self.assertEqual(cached.get_default(1), 1)
with self.assertRaises(TypeError):
cached.get_hashkey(0)
def test_wrapped(self):
cache = {}
cached = Cached(cache)
self.assertEqual(len(cache), 0)
self.assertEqual(cached.get.__wrapped__(cached, 0), 0)
self.assertEqual(len(cache), 0)
self.assertEqual(cached.get(0), 1)
self.assertEqual(len(cache), 1)
self.assertEqual(cached.get(0), 1)
self.assertEqual(len(cache), 1)
def test_attributes(self):
cache = {}
cached = Cached(cache)
self.assertIs(cached.get.cache(cached), cache)
self.assertIs(cached.get.cache_key, keys.methodkey)
self.assertIs(cached.get.cache_lock, None)
def test_attributes_lock(self):
cache = {}
cached = Locked(cache)
self.assertIs(cached.get.cache(cached), cache)
self.assertIs(cached.get.cache_key, keys.methodkey)
self.assertIs(cached.get.cache_lock(cached), cached)
def test_clear(self):
cache = {}
cached = Cached(cache)
self.assertEqual(cached.get(0), 0)
self.assertEqual(len(cache), 1)
cached.get.cache_clear(cached)
self.assertEqual(len(cache), 0)
def test_clear_locked(self):
cache = {}
cached = Locked(cache)
self.assertEqual(cached.get(0), 1)
self.assertEqual(len(cache), 1)
self.assertEqual(cached.count, 2)
cached.get.cache_clear(cached)
self.assertEqual(len(cache), 0)
self.assertEqual(cached.count, 3)