

def test_keysort(self):
    N = 1000
    rnd = numpy.random.randint(N, size=N)
    for dtype1 in ('S6', 'b1', 'i1', 'i8', 'u4', 'u8', 'f4', 'f8'):
        for dtype2 in ('u4', 'i8'):
            a = numpy.array(rnd, dtype1)
            b = numpy.array(rnd, dtype2)
            c = a.copy()
            d = c.argsort()
            e = c[d]
            f = b[d]
            indexesextension.keysort(a, b)
            self.assertTrue((a == e).all())
            self.assertTrue((b == f).all())

dataset/ETHPy150Open PyTables/PyTables/tables/tests/test_aux.py TestAuxiliaryFunctions.test_keysort/original