def setUp(self):
self.n = 100
self.m = 4
self.data = np.random.randn(self.n, self.m)
self.bounds = np.ones(self.m)
self.kdtree = PeriodicKDTree(self.bounds, self.data, leafsize=2)
self.x = np.random.randn(self.m)
self.d = 0.2
self.k = 10
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 setUp(self):
n = 100
m = 4
self.data = np.random.randn(n, m)
self.bounds = np.ones(m)
self.T = PeriodicKDTree(self.bounds, self.data, leafsize=2)
self.x = np.random.randn(m)
self.p = 2.
self.eps = 0
self.d = 0.2
def setUp(self):
self.data = np.array([[0, 0, 0], [0, 0, 1], [0, 1, 0], [0, 1, 1],
[1, 0, 0], [1, 0, 1], [1, 1, 0], [1, 1, 1]])
self.bounds = 1.1 * np.ones(3)
self.kdtree = PeriodicKDTree(self.bounds, self.data)
self.n = self.kdtree.n
self.m = self.kdtree.m
self.x = np.random.randn(3)
self.d = 0.5
self.k = 4
def setUp(self):
self.data = np.array([[0, 0, 0], [0, 0, 1], [0, 1, 0], [0, 1, 1],
[1, 0, 0], [1, 0, 1], [1, 1, 0], [1, 1, 1]])
self.bounds = 1.1 * np.ones(3)
self.kdtree = PeriodicKDTree(self.bounds, self.data)
def setUp(self):
test_small.setUp(self)
self.kdtree = PeriodicKDTree(self.bounds, self.data, leafsize=1)
from periodic_kdtree import PeriodicKDTree, PeriodicCKDTree
m = 3
n = 10000
r = 1000
bounds = np.ones(m)
data = np.concatenate(
(np.random.randn(n // 2, m), np.random.randn(n - n // 2, m) + np.ones(m)))
queries = np.concatenate(
(np.random.randn(r // 2, m), np.random.randn(r - r // 2, m) + np.ones(m)))
print "dimension %d, %d points" % (m, n)
t = time.time()
T1 = PeriodicKDTree(bounds, data)
print "PeriodicKDTree constructed:\t%g" % (time.time() - t)
t = time.time()
T2 = PeriodicCKDTree(bounds, data)
print "PeriodicCKDTree constructed:\t%g" % (time.time() - t)
t = time.time()
w = T1.query(queries)
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
