def test_remove_many_with_cache(self): starting_table = { 0: 0, 1: 1, 2: 2, } c = DataMemoryElement(pickle.dumps(starting_table)) s = MemoryKeyValueStore(c) self.assertDictEqual(s._table, starting_table) s.remove_many([0, 2]) self.assertDictEqual(pickle.loads(c.get_bytes()), {1: 1})
def test_remove_many(self): """ Test expected remove_many functionality. """ s = MemoryKeyValueStore() s._table = { 0: 0, 1: 1, 2: 2, } s.remove_many([0, 1]) self.assertDictEqual(s._table, {2: 2})