def test_batch_one_remove(self): self._perform_batch_insertion() batch = Batch(self.collection, self.connection) batch.add_remove({ 'batch_insert_1': 3 }) batch.execute() self.assertFalse(self.collection.find({'batch_insert_1': 3})) self.assertTrue(self.collection.find({'batch_insert_3': 29}))