def test_sort_component(self):
x = np.random.random((5, 12))
t = Table(x, {'F': (3, 3), 'v': (3, )}, ['random test data'])
unsort = t.get('4_F')
t.sort_by('4_F')
sort = t.get('4_F')
assert np.all(np.sort(unsort, 0) == sort)
def test_sort(self):
t = Table(np.array([[
0,
1,
], [
2,
1,
]]), {'v': (2, )}, ['test data'])
t.add('s', np.array(['b', 'a']))
t.sort_by('s')
assert np.all(t.get('1_v') == np.array([2, 0]).reshape(2, 1))
def test_sort_scalar(self):
x = np.random.random((5, 13))
t = Table(x, {
'F': (3, 3),
'v': (3, ),
's': (1, )
}, ['random test data'])
unsort = t.get('s')
t.sort_by('s')
sort = t.get('s')
assert np.all(np.sort(unsort, 0) == sort)
def test_sort_revert(self):
x = np.random.random((5, 12))
t = Table(x, {'F': (3, 3), 'v': (3, )}, ['random test data'])
sort = t.sort_by('F[1,0]', ascending=False).get('F')[:, 1, 0]
assert np.all(np.sort(sort, 0) == sort[::-1])