def test_join_on_eq_with_true_dt_with_result_limit(self):
nl = ak.get_config()['numLocales']
lim = (self.N + nl) * self.N
res_size = self.N * self.N
I,J = ak.join_on_eq_with_dt(self.a1,self.a1,self.a1,self.a1,self.dt,"true_dt",result_limit=lim)
self.assertEqual(res_size, I.size)
self.assertEqual(res_size, J.size)
def check_coargsort(N_per_locale):
print(">>> arkouda coargsort")
cfg = ak.get_config()
N = N_per_locale * cfg["numLocales"]
print("numLocales = {}, N = {:,}".format(cfg["numLocales"], N))
check_int(N)
check_float(N)
check_int_float(N)
check_large(N)
def check_coargsort(N_per_locale):
print(">>> arkouda coargsort")
cfg = ak.get_config()
N = N_per_locale * cfg["numLocales"]
print("numLocales = {}, N = {:,}".format(cfg["numLocales"], N))
for algo in ak.SortingAlgorithm:
check_int(N, algo)
check_float(N, algo)
check_int_float(N, algo)
check_large(N, algo)
def test_join_on_eq_with_pos_dt(self):
I, J = ak.join_on_eq_with_dt(self.a2, self.a1, self.t1, self.t2,
self.dt, "pos_dt")
nl = ak.get_config()['numLocales']
self.assertEqual(self.N // nl, I.size)
self.assertEqual(self.N // nl, J.size)
