diff options
Diffstat (limited to 'tests/test_cache.py')
-rw-r--r-- | tests/test_cache.py | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/tests/test_cache.py b/tests/test_cache.py index 433a733..51536db 100644 --- a/tests/test_cache.py +++ b/tests/test_cache.py @@ -8,3 +8,20 @@ class CacheTest(unittest.TestCase, CacheTestMixin): def cache(self, maxsize, missing=None, getsizeof=None): return Cache(maxsize, missing=missing, getsizeof=getsizeof) + + def test_getsize(self): + # Cache.getsize is deprecated + cache = self.cache(maxsize=3, getsizeof=lambda x: x) + cache.update({1: 1, 2: 2}) + + import warnings + with warnings.catch_warnings(record=True) as w: + warnings.simplefilter("always") + self.assertEqual(1, cache.getsize(1)) + self.assertEqual(1, len(w)) + self.assertEqual(w[0].category, DeprecationWarning) + with warnings.catch_warnings(record=True) as w: + warnings.simplefilter("always") + self.assertEqual(2, cache.getsize(2)) + self.assertEqual(1, len(w)) + self.assertEqual(w[0].category, DeprecationWarning) |