def test_multiple_items_get(self):
item_count = 16
cache = LRU(max_cache_size=item_count)
for item_id in range(item_count):
cache.add(item_id, self._item_id2item(item_id), "Task1", "flow1")
# let's use the following 5 items, so they get to the front
used_items = [10, 5, 3, 1, 7]
for item_id in used_items:
assert cache.get(item_id, "Task1",
"flow1") == self._item_id2item(item_id)
# add 5 dummy elements
for item_id in range(1, 6):
cache.add(item_id * 100, self._item_id2item(item_id * 100),
"Task1", "flow1")
removed_items = [0, 2, 4, 6, 8]
for item_id in removed_items:
with pytest.raises(CacheMissError):
cache.get(item_id, "Task1", "flow1")
cached_items = list(set(range(item_count)) - set(removed_items))
for item_id in cached_items:
assert cache.get(item_id, "Task1",
"flow1") == self._item_id2item(item_id)
def test_one_item_miss(self):
cache = LRU(max_cache_size=1)
cache.add("item_id1", "item1", "Task1", "flow1")
cache.add("item_id2", "item2", "Task1", "flow1")
with pytest.raises(CacheMissError):
cache.get("item_id1", "Task1", "flow1")
def test_one_item_miss(self):
cache = LRU(max_cache_size=1)
cache.add("item_id1", "item1", "Task1", "flow1")
cache.add("item_id2", "item2", "Task1", "flow1")
with pytest.raises(CacheMissError):
cache.get("item_id1", "Task1", "flow1")
def test_multiple_items_get(self):
item_count = 16
cache = LRU(max_cache_size=item_count)
for item_id in range(item_count):
cache.add(item_id, self._item_id2item(item_id), "Task1", "flow1")
# let's use the following 5 items, so they get to the front
used_items = [10, 5, 3, 1, 7]
for item_id in used_items:
assert cache.get(item_id, "Task1", "flow1") == self._item_id2item(item_id)
# add 5 dummy elements
for item_id in range(1, 6):
cache.add(item_id*100, self._item_id2item(item_id*100), "Task1", "flow1")
removed_items = [0, 2, 4, 6, 8]
for item_id in removed_items:
with pytest.raises(CacheMissError):
cache.get(item_id, "Task1", "flow1")
cached_items = list(set(range(item_count)) - set(removed_items))
for item_id in cached_items:
assert cache.get(item_id, "Task1", "flow1") == self._item_id2item(item_id)
def __init__(self):
self.cache = LRU(max_cache_size=0)
def __init__(self, is_flow):
self.is_flow = is_flow
self.cache = LRU(max_cache_size=0)
def test_multiple_items(self):
item_count = 16
cache = LRU(max_cache_size=item_count)
for item_id in range(item_count):
cache.add(item_id, self._item_id2item(item_id), "Task1", "flow1")
cache.add(item_count, self._item_id2item(item_count), "Task1", "flow1")
with pytest.raises(CacheMissError):
# the very first is removed
cache.get(0, "Task1", "flow1")
for item_id in range(item_count - 1, 0, -1):
assert cache.get(item_id, "Task1", "flow1") == self._item_id2item(item_id)
# re-add 0, so we have 0 - (item_count-1)
cache.add(0, self._item_id2item(0), "Task1", "flow1")
with pytest.raises(CacheMissError):
cache.get(item_count, "Task1", "flow1")
for item_id in range(item_count):
assert cache.get(item_id, "Task1", "flow1") == self._item_id2item(item_id)
def test_two_items(self):
cache = LRU(max_cache_size=2)
cache.add("item_id1", "item1", "Task1", "flow1")
cache.add("item_id2", "item2", "Task1", "flow1")
cache.add("item_id3", "item3", "Task1", "flow1")
with pytest.raises(CacheMissError):
cache.get("item_id1", "Task1", "flow1")
assert cache.get("item_id2", "Task1", "flow1") == "item2"
assert cache.get("item_id3", "Task1", "flow1") == "item3"
# we made "item2" a candidate for next deletion as "item3" was used lastly
cache.add("item_id1", "item1", "Task1", "flow1")
with pytest.raises(CacheMissError):
cache.get("item_id2", "Task1", "flow1")
def test_multiple_items(self):
item_count = 16
cache = LRU(max_cache_size=item_count)
for item_id in range(item_count):
cache.add(item_id, self._item_id2item(item_id), "Task1", "flow1")
cache.add(item_count, self._item_id2item(item_count), "Task1", "flow1")
with pytest.raises(CacheMissError):
# the very first is removed
cache.get(0, "Task1", "flow1")
for item_id in range(item_count - 1, 0, -1):
assert cache.get(item_id, "Task1",
"flow1") == self._item_id2item(item_id)
# re-add 0, so we have 0 - (item_count-1)
cache.add(0, self._item_id2item(0), "Task1", "flow1")
with pytest.raises(CacheMissError):
cache.get(item_count, "Task1", "flow1")
for item_id in range(item_count):
assert cache.get(item_id, "Task1",
"flow1") == self._item_id2item(item_id)
def test_two_items(self):
cache = LRU(max_cache_size=2)
cache.add("item_id1", "item1", "Task1", "flow1")
cache.add("item_id2", "item2", "Task1", "flow1")
cache.add("item_id3", "item3", "Task1", "flow1")
with pytest.raises(CacheMissError):
cache.get("item_id1", "Task1", "flow1")
assert cache.get("item_id2", "Task1", "flow1") == "item2"
assert cache.get("item_id3", "Task1", "flow1") == "item3"
# we made "item2" a candidate for next deletion as "item3" was used lastly
cache.add("item_id1", "item1", "Task1", "flow1")
with pytest.raises(CacheMissError):
cache.get("item_id2", "Task1", "flow1")