1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
import inspect
import threading
import unittest
import wrapt
@wrapt.decorator
def memoize(wrapped, instance, args, kwargs):
if instance is None and inspect.isclass(wrapped):
# Wrapped function is a class and we are creating an
# instance of the class. Don't support this case, just
# return straight away.
return wrapped(*args, **kwargs)
# Retrieve the cache, attaching an empty one if none exists.
cache = wrapped.__dict__.setdefault("_memoize_cache", {})
# Now see if entry is in the cache and if it isn't then call
# the wrapped function to generate it.
key = (args, frozenset(kwargs.items()))
try:
return cache[key]
except KeyError:
result = cache[key] = wrapped(*args, **kwargs)
return result
@memoize
def function1(count, text):
return count * text
class C1:
@memoize
def function1(self, count, text):
return count * text
@memoize
@classmethod
def function2(cls, count, text):
return count * text
@memoize
@staticmethod
def function3(count, text):
return count * text
c1 = C1()
class TestSynchronized(unittest.TestCase):
def test_function(self):
value1 = function1(10, "0123456789")
value2 = function1(10, "0123456789")
self.assertEqual(value1, value2)
self.assertEqual(id(value1), id(value2))
self.assertTrue(hasattr(function1, "_memoize_cache"))
def test_instancemethod(self):
value1 = c1.function1(10, "0123456789")
value2 = c1.function1(10, "0123456789")
self.assertEqual(value1, value2)
self.assertEqual(id(value1), id(value2))
self.assertTrue(hasattr(C1.function1, "_memoize_cache"))
def test_classmethod(self):
value1 = C1.function2(10, "0123456789")
value2 = C1.function2(10, "0123456789")
self.assertEqual(value1, value2)
self.assertEqual(id(value1), id(value2))
self.assertTrue(hasattr(C1.function2, "_memoize_cache"))
def test_staticmethod(self):
value1 = C1.function3(10, "0123456789")
value2 = C1.function3(10, "0123456789")
self.assertEqual(value1, value2)
self.assertEqual(id(value1), id(value2))
self.assertTrue(hasattr(C1.function3, "_memoize_cache"))
if __name__ == "__main__":
unittest.main()
|
