Prepare v0.2.0

1 files changed, 41 insertions, 0 deletions

diff --git a/tests/test_cache.py b/tests/test_cache.py
new file mode 100644
index 0000000..5c32f2b
--- /dev/null
+++ b/tests/test_cache.py
@@ -0,0 +1,41 @@
+import unittest
+
+import cachetools
+import collections
+
+
+@cachetools.cache
+class DictCache(dict):
+    pass
+
+
+@cachetools.cache
+class OrderedDictCache(collections.OrderedDict):
+    pass
+
+
+class CacheTest(unittest.TestCase):
+
+    def test_dict_cache(self):
+        cache = DictCache(maxsize=2)
+
+        cache['a'] = 1
+        cache['b'] = 2
+        cache['c'] = 3
+
+        self.assertEqual(len(cache), 2)
+        self.assertTrue('a' in cache or ('b' in cache and 'c' in cache))
+        self.assertTrue('b' in cache or ('a' in cache and 'c' in cache))
+        self.assertTrue('c' in cache or ('a' in cache and 'b' in cache))
+
+    def test_ordered_dict_cache(self):
+        cache = OrderedDictCache(maxsize=2)
+
+        cache['a'] = 1
+        cache['b'] = 2
+        cache['c'] = 3
+
+        self.assertEqual(len(cache), 2)
+        self.assertNotIn('a', cache)
+        self.assertEqual(cache['b'], 2)
+        self.assertEqual(cache['c'], 3)