mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-31 05:58:33 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

Support cache sizes.

Browse source
This commit is contained in:
Raymond Hettinger 2010-08-14 23:52:08 +00:00
parent a893927491
commit 0f56e90f05
2 changed files with 49 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

48
Lib/test/test_functools.py
View file

@ -482,6 +482,30 @@ class TestLRU(unittest.TestCase):
self.assertEqual(f.hits, 0)
self.assertEqual(f.misses, 1)
# test size zero (which means "never-cache")
f_cnt = 0
@functools.lru_cache(0)
def f():
nonlocal f_cnt
f_cnt += 1
return 20
self.assertEqual(f(), 20)
self.assertEqual(f(), 20)
self.assertEqual(f(), 20)
self.assertEqual(f_cnt, 3)
# test size one
f_cnt = 0
@functools.lru_cache(1)
def f():
nonlocal f_cnt
f_cnt += 1
return 20
self.assertEqual(f(), 20)
self.assertEqual(f(), 20)
self.assertEqual(f(), 20)
self.assertEqual(f_cnt, 1)
def test_lfu(self):
def orig(x, y):
return 3*x+y
@ -503,6 +527,30 @@ class TestLRU(unittest.TestCase):
self.assertEqual(f.hits, 0)
self.assertEqual(f.misses, 1)
# test size zero (which means "never-cache")
f_cnt = 0
@functools.lfu_cache(0)
def f():
nonlocal f_cnt
f_cnt += 1
return 20
self.assertEqual(f(), 20)
self.assertEqual(f(), 20)
self.assertEqual(f(), 20)
self.assertEqual(f_cnt, 3)
# test size one
f_cnt = 0
@functools.lfu_cache(1)
def f():
nonlocal f_cnt
f_cnt += 1
return 20
self.assertEqual(f(), 20)
self.assertEqual(f(), 20)
self.assertEqual(f(), 20)
self.assertEqual(f_cnt, 1)
def test_main(verbose=None):
test_classes = (
TestPartial,