def test_random_ball_vectorized(): n = 20 m = 5 bounds = np.ones(m) T = PeriodicKDTree(bounds, np.random.randn(n,m)) r = T.query_ball_point(np.random.randn(2,3,m),1) assert_equal(r.shape,(2,3)) assert_(isinstance(r[0,0],list))
def test_random_ball_vectorized(): n = 20 m = 5 bounds = np.ones(m) T = PeriodicKDTree(bounds, np.random.randn(n, m)) r = T.query_ball_point(np.random.randn(2, 3, m), 1) assert_equal(r.shape, (2, 3)) assert_(isinstance(r[0, 0], list))
print "PeriodicKDTree %d lookups:\t%g" % (r, time.time() - t) del w t = time.time() w = T2.query(queries) print "PeriodicCKDTree %d lookups:\t%g" % (r, time.time() - t) del w T3 = PeriodicCKDTree(bounds, data, leafsize=n) t = time.time() w = T3.query(queries) print "flat PeriodicCKDTree %d lookups:\t%g" % (r, time.time() - t) del w t = time.time() w1 = T1.query_ball_point(queries, 0.2) print "PeriodicKDTree %d ball lookups:\t%g" % (r, time.time() - t) t = time.time() w2 = T2.query_ball_point(queries, 0.2) print "PeriodicCKDTree %d ball lookups:\t%g" % (r, time.time() - t) t = time.time() w3 = T3.query_ball_point(queries, 0.2) print "flat PeriodicCKDTree %d ball lookups:\t%g" % (r, time.time() - t) all_good = True for a, b in zip(w1, w2): if sorted(a) != sorted(b): all_good = False for a, b in zip(w1, w3):
print "PeriodicKDTree %d lookups:\t%g" % (r, time.time()-t) del w t = time.time() w = T2.query(queries) print "PeriodicCKDTree %d lookups:\t%g" % (r, time.time()-t) del w T3 = PeriodicCKDTree(bounds,data,leafsize=n) t = time.time() w = T3.query(queries) print "flat PeriodicCKDTree %d lookups:\t%g" % (r, time.time()-t) del w t = time.time() w1 = T1.query_ball_point(queries, 0.2) print "PeriodicKDTree %d ball lookups:\t%g" % (r, time.time()-t) t = time.time() w2 = T2.query_ball_point(queries, 0.2) print "PeriodicCKDTree %d ball lookups:\t%g" % (r, time.time()-t) t = time.time() w3 = T3.query_ball_point(queries, 0.2) print "flat PeriodicCKDTree %d ball lookups:\t%g" % (r, time.time()-t) all_good = True for a, b in zip(w1, w2): if sorted(a) != sorted(b): all_good = False for a, b in zip(w1, w3):