コード例 #1
0
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())
コード例 #2
0
 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]
コード例 #3
0
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())