def test_pack(self): """Test pack.""" columns = self.client.get_slice() packed = iterators.pack(columns) self.assert_(isinstance(packed, types.GeneratorType)) for obj in packed: self.assert_(isinstance(obj, ttypes.ColumnOrSuperColumn))
def _get_batch_args(self, key, columns, consistency=None): """Return a BatchMutation for the given key and columns.""" consistency = consistency or self.consistency if key.is_super(): columns = [SuperColumn(name=key.super_column, columns=columns)] return (key.keyspace, key.key, {key.column_family: tuple(iterators.pack(columns))}, consistency)
def pack(cols): return list(iterators.pack(cols))