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
|
# Copyright 2024 Caleb Connolly
# SPDX-License-Identifier: GPL-3.0-or-later
from pmb.meta import Cache, Wrapper
def test_cache_hits_basic() -> None:
def multiply_2(x: int) -> int:
return x * 2
multiply_2_cached = Cache("x")(multiply_2)
assert isinstance(multiply_2_cached, Wrapper)
assert multiply_2(2) == 4
assert multiply_2_cached(2) == 4
assert multiply_2_cached.misses == 1
assert multiply_2_cached(2) == 4
assert multiply_2_cached.hits == 1
assert multiply_2_cached(3) == 6
assert multiply_2_cached.misses == 2
assert multiply_2_cached(4) == 8
assert multiply_2_cached.misses == 3
assert multiply_2_cached(3) == 6
assert multiply_2_cached.hits == 2
def test_cache_hits_kwargs() -> None:
def multiply_2(x: int, y: int = 2, z: list[int] = []) -> int:
return x * y + sum(z)
multiply_2_cached = Cache("x", "y", "z")(multiply_2)
assert isinstance(multiply_2_cached, Wrapper)
assert multiply_2(2) == 4
assert multiply_2_cached(2) == 4
assert multiply_2_cached.misses == 1
assert multiply_2(2, 3) == multiply_2_cached(2, 3)
assert multiply_2_cached.misses == 2
assert multiply_2(2, 3) == multiply_2_cached(2, 3)
assert multiply_2_cached.hits == 1
assert multiply_2(3, 4, [1, 1]) == 14
assert multiply_2_cached(3, 4, [1, 1]) == 14
assert multiply_2_cached(3, 3, [1, 1]) == 11
assert multiply_2_cached.misses == 4
assert multiply_2_cached(3, 4, [1, 1]) == 14
assert multiply_2_cached.hits == 2
# Should only cache when y=3
multiply_2_cached_y3 = Cache("x", "z", y=3)(multiply_2)
assert multiply_2_cached_y3(1, 1, [1, 1]) == 3
assert multiply_2_cached_y3.misses == 1
assert multiply_2_cached_y3(1, 1, [1, 1]) == 3
assert multiply_2_cached_y3.misses == 2
assert multiply_2_cached_y3(1, 3, [4, 1]) == 8
assert multiply_2_cached_y3.misses == 3
assert multiply_2_cached_y3(1, 3, [4, 1]) == 8
assert multiply_2_cached_y3.hits == 1
def test_build_key() -> None:
def multiply_2(x: int, y: int = 2, z: list[int] = []) -> int:
return x * y + sum(z)
multiply_2_cached = Cache("x", "y", "z")(multiply_2)
key = multiply_2_cached.cache.build_key(multiply_2, 1, 2, [3, 4])
print(f"KEY: {key}")
assert key == "~1~2~[3, 4]~"
multiply_2_cached_y4 = Cache("x", "z", y=4)(multiply_2)
# Key should be None since y != 4
key = multiply_2_cached_y4.cache.build_key(multiply_2, 1, 2, [3, 4])
print(f"Expecting None KEY: {key}")
assert key is None
# Now we expect a real key since y is 4
key = multiply_2_cached_y4.cache.build_key(multiply_2, 1, 4, [3, 4])
print(f"KEY: {key}")
assert key == "~y=(4)~1~[3, 4]~"
|
