def test_batch_delete(self): index_store = IndexStore() self.assertEqual(index_store.getidx('a'), 0) self.assertEqual(index_store.getidx('b'), 1) self.assertEqual(index_store.getidx('c'), 2) self.assertEqual(index_store.getidx('d'), 3) with self.assertRaises(KeyError): new_idx_store = index_store.batch_delete_vals(['e', 'c']) new_idx_store = index_store.batch_delete_vals(['b', 'c']) self.assertEqual(new_idx_store.getidx('a'), 0) self.assertEqual(new_idx_store.getidx('c'), 2) self.assertEqual(new_idx_store.getidx('e'), 3) self.assertEqual(index_store.getidx('d'), 3) self.assertEqual(index_store.getidx('c'), 2) self.assertEqual(index_store.getidx('b'), 1) self.assertEqual(index_store.getidx('a'), 0) with self.assertRaises(ValueError): new_idx_store = index_store.batch_delete_idx([5, 1]) new_idx_store = index_store.batch_delete_idx([2, 1]) self.assertEqual(new_idx_store.getidx('a'), 0) self.assertEqual(new_idx_store.getidx('c'), 2) self.assertEqual(new_idx_store.getidx('e'), 3)
def test_batch_delete_extra(self): index_store = IndexStore() self.assertEqual(index_store.getidx('a'), 0) self.assertEqual(index_store.getidx('b'), 1) self.assertEqual(index_store.getidx('c'), 2) self.assertEqual(index_store.getidx('d'), 3) self.assertEqual(index_store.getidx('e'), 4) self.assertEqual(index_store.getidx('f'), 5) del_idxstore = index_store.batch_delete_vals(['b', 'e']) self.assertEqual(list(del_idxstore.items()), [(0, 'a'), (1, 'c'), (2, 'd'), (3, 'f')]) del_idxstore2 = del_idxstore.batch_delete_vals([]) self.assertEqual(list(del_idxstore.items()), list(del_idxstore2.items()))