def test_cache(): items_to_cache = set((('a', 'bla'), ('a', 'foo'), ('b', 'foo'))) cache = Cache(items_to_cache) data = [{'bla': 1, 'xyz': 2}, {'bla': 3, 'xyz': 4}] # Should be cached cache.add('a', data) # Should not be cached cache.add('b', data) a_bla = cache.retrieve('a.bla') assert a_bla == [1, 3] b_bla = cache.retrieve('b.bla') assert b_bla == []
def test_cache_no_columns(): items_to_cache = set() cache = Cache(items_to_cache) data = [{'bla': 1, 'xyz': 2}, {'bla': 3, 'xyz': 4}] # Should not be cached cache.add('a', data) a_bla = cache.retrieve('a.bla') assert a_bla == []