Gecko [ www.smanegeri1koba.sch.id ]

Name	Size	Permission
__init__.py	9.51 KB	-rw-r--r--
test_cache.py	148 B	-rw-r--r--
test_cached.py	8.19 KB	-rw-r--r--
test_cachedmethod.py	6.89 KB	-rw-r--r--
test_fifo.py	1.34 KB	-rw-r--r--
test_func.py	5.26 KB	-rw-r--r--
test_keys.py	4.66 KB	-rw-r--r--
test_lfu.py	1.17 KB	-rw-r--r--
test_lru.py	1.34 KB	-rw-r--r--
test_mru.py	1.87 KB	-rw-r--r--
test_rr.py	831 B	-rw-r--r--
test_tlru.py	7.96 KB	-rw-r--r--
test_ttl.py	5.85 KB	-rw-r--r--

Code Editor : test_cachedmethod.py

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))

cached.cache.clear()
        self.assertEqual(cached.get(1), 2)

def test_typedmethod_lru(self):
        cached = Cached(LRUCache(maxsize=2))

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)

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)

${this.title}

Code Editor : test_cachedmethod.py