def run_test_argsort(strings, test_strings, cat): akperm = ak.argsort(strings) aksorted = strings[akperm].to_ndarray() npsorted = np.sort(test_strings) assert((aksorted == npsorted).all()) catperm = ak.argsort(cat) catsorted = cat[catperm].to_ndarray() assert((catsorted == npsorted).all())
def test_compare_argsort(self): # create np version a = np.arange(N) a = a[::-1] iv = np.argsort(a) a = a[iv] # create ak version b = ak.arange(N) b = b[::-1] iv = ak.argsort(b) b = b[iv]
def check_argsort(N): # create np version a = np.arange(N) a = a[::-1] iv = np.argsort(a) a = a[iv] # create ak version b = ak.arange(N) b = b[::-1] iv = ak.argsort(b) b = b[iv] # print(a,b) c = a == b.to_ndarray() # print(type(c),c) return pass_fail(c.all())