def test_key_sorted(self):
t = Table([
Record(x=1, y=2, z=3),
Record(x=2, y=2, z=3),
Record(x=1, y=2, z=4),
Record(x=3, y=3, z=4),
Record(x=1, y=2, z=3),
Record(x=2, y=1, z=3),
])
assert type(t.key_sorted(['x'])) is Table
assert t.key_sorted(['y']) == Table([
Record(x=2, y=1, z=3),
Record(x=1, y=2, z=3),
Record(x=2, y=2, z=3),
Record(x=1, y=2, z=4),
Record(x=1, y=2, z=3),
Record(x=3, y=3, z=4),
])
assert t.key_sorted(['z', 'y']) == Table([
Record(x=2, y=1, z=3),
Record(x=1, y=2, z=3),
Record(x=2, y=2, z=3),
Record(x=1, y=2, z=3),
Record(x=1, y=2, z=4),
Record(x=3, y=3, z=4),
])
assert t.key_sorted(['x', 'y']) == Table([
Record(x=1, y=2, z=3),
Record(x=1, y=2, z=4),
Record(x=1, y=2, z=3),
Record(x=2, y=1, z=3),
Record(x=2, y=2, z=3),
Record(x=3, y=3, z=4),
])
assert t.key_sorted(['z', 'x', 'y']) == Table([
Record(x=1, y=2, z=3),
Record(x=1, y=2, z=3),
Record(x=2, y=1, z=3),
Record(x=2, y=2, z=3),
Record(x=1, y=2, z=4),
Record(x=3, y=3, z=4),
])
