class TestLRU(unittest.TestCase): def setUp(self): """set up: init a LRUCache""" self.cache = LRUCache(3) def test_lru_basic(self): test_value = [0, 1, 2, 3, 4] expected_value = [2, 3, 4] removed_value = [0, 1] for i in test_value: self.cache.add_value(i, i) for i in expected_value: self.assertEqual(self.cache.get_value(i), i) for i in removed_value: self.assertEqual(self.cache.get_value(i), None) def test_ttl(self): cache = LRUCache(3) cache.add_value('hello', 'world', ttl=3) self.assertEqual(cache.get_value('hello'), 'world') sleep(5) self.assertEqual(cache.get_value('hello'), None) def tearDown(self): """Cleaning up after the test"""
def main(): cache = LRUCache(3) print 'add hello -> world with ttl=60 into lru' cache.add_value('hello', 'world', ttl=3) print 'get value from cache: %s -> %s' % ('hello', cache.get_value('hello'))
def setUp(self): """set up: init a LRUCache""" self.cache = LRUCache(3)
def test_ttl(self): cache = LRUCache(3) cache.add_value('hello', 'world', ttl=3) self.assertEqual(cache.get_value('hello'), 'world') sleep(5) self.assertEqual(cache.get_value('hello'), None)