def test_remove(self): """ Test normal removal. """ s = MemoryKeyValueStore() s._table = { 0: 1, 'a': 'b', } s.remove(0) self.assertDictEqual(s._table, {'a': 'b'})
def test_remove_with_cache(self): """ Test that removal correctly updates the cache element. """ existing_data = { 0: 1, 'a': 'b', } c = DataMemoryElement(pickle.dumps(existing_data)) s = MemoryKeyValueStore(c) self.assertDictEqual(s._table, existing_data) s.remove('a') self.assertDictEqual(s._table, {0: 1}) self.assertDictEqual(pickle.loads(c.get_bytes()), {0: 1})