diff options
Diffstat (limited to 'tests/test_rr.py')
-rw-r--r-- | tests/test_rr.py | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/tests/test_rr.py b/tests/test_rr.py index 14a4b07..cff18bf 100644 --- a/tests/test_rr.py +++ b/tests/test_rr.py @@ -11,14 +11,18 @@ def choice(seq): return random.choice(seq) +class RRTestCache(RRCache): + def __init__(self, maxsize, choice=choice, missing=None, getsizeof=None): + RRCache.__init__(self, maxsize, choice=choice, + missing=missing, getsizeof=getsizeof) + + class RRCacheTest(unittest.TestCase, CacheTestMixin): - def cache(self, maxsize, choice=choice, missing=None, getsizeof=None): - return RRCache(maxsize, choice=choice, missing=missing, - getsizeof=getsizeof) + Cache = RRTestCache - def test_choice(self): - cache = self.cache(maxsize=2, choice=min) + def test_rr_choice(self): + cache = RRCache(maxsize=2, choice=min) self.assertEqual(min, cache.choice) cache[1] = 1 |