1
- import operator
2
1
import unittest
3
2
4
3
from cachetools import LRUCache , cachedmethod , keys
@@ -9,13 +8,13 @@ def __init__(self, cache, count=0):
9
8
self .cache = cache
10
9
self .count = count
11
10
12
- @cachedmethod (operator . attrgetter ( " cache" ) )
11
+ @cachedmethod (lambda self : self . cache )
13
12
def get (self , value ):
14
13
count = self .count
15
14
self .count += 1
16
15
return count
17
16
18
- @cachedmethod (operator . attrgetter ( " cache" ) , key = keys .typedkey )
17
+ @cachedmethod (lambda self : self . cache , key = keys .typedkey )
19
18
def get_typed (self , value ):
20
19
count = self .count
21
20
self .count += 1
@@ -27,7 +26,7 @@ def __init__(self, cache):
27
26
self .cache = cache
28
27
self .count = 0
29
28
30
- @cachedmethod (operator . attrgetter ( " cache" ) , lock = lambda self : self )
29
+ @cachedmethod (lambda self : self . cache , lock = lambda self : self )
31
30
def get (self , value ):
32
31
return self .count
33
32
@@ -42,11 +41,11 @@ class Unhashable:
42
41
def __init__ (self , cache ):
43
42
self .cache = cache
44
43
45
- @cachedmethod (operator . attrgetter ( " cache" ) )
44
+ @cachedmethod (lambda self : self . cache )
46
45
def get_default (self , value ):
47
46
return value
48
47
49
- @cachedmethod (operator . attrgetter ( " cache" ) , key = keys .hashkey )
48
+ @cachedmethod (lambda self : self . cache , key = keys .hashkey )
50
49
def get_hashkey (self , value ):
51
50
return value
52
51
0 commit comments