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