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))