def test_cluster(self):
means=[(.6, .5), (.3, .8), (.2, .4)]
sd = 0.05
ps = ClusterSet(means=means, sd=sd)
ac = AddC(kmax=6).batch(ps)
centroids = ac.trim(0.2)
assert len(centroids) == len(means)
for a, b in zip(sorted(centroids), sorted(means)):
assert all_close(a, b, sd) # Tolerance equal to sd...
def test_cluster(self):
means = [(.6, .5), (.3, .8), (.2, .4)]
sd = 0.05
ps = ClusterSet(means=means, sd=sd)
ac = AddC(kmax=6).batch(ps)
centroids = ac.trim(0.2)
assert len(centroids) == len(means)
for a, b in zip(sorted(centroids), sorted(means)):
assert all_close(a, b, sd) # Tolerance equal to sd...
def test_compare_batch_add(self):
ps = PointSet()
ac1 = AddC()
ac2 = AddC()
for p in ps:
ac1 += p
ac2.batch(ps)
for a, b in zip(ac1.centroids, ac2.centroids):
assert all_close(a, b)
for a, b in zip(ac1, ac2):
assert all_close(a.center, b.center)
assert a.count == b.count
assert abs(a.size - b.size) < 1e-8
def test_compare_batch_add(self):
ps = PointSet()
ac1 = AddC()
ac2 = AddC()
for p in ps:
ac1 += p
ac2.batch(ps)
for a, b in zip(ac1.centroids, ac2.centroids):
assert all_close(a, b)
for a, b in zip(ac1, ac2):
assert all_close(a.center, b.center)
assert a.count == b.count
assert abs(a.size - b.size) < 1e-8
def test_batch(self):
ps = PointSet()
ac = AddC()
assert len(ac) == 0
ac.batch(ps)
assert len(ac) == len(ps)
def test_call_and_centroids(self):
ps = PointSet()
ac = AddC(kmax=5)
assert len(ac()) == len(ac.centroids) == 0
ac.batch(ps)
assert len(ac()) == len(ac.centroids) == 5
def test_contains(self):
ps = PointSet(10, 2)
ac = AddC()
assert ps[0] not in ac
ac.batch(ps)
assert ac.centroids[0] in ac
def test_len(self):
ps = PointSet()
ac = AddC(kmax=8)
assert len(ac) == 0
ac.batch(ps)
assert len(ac) == 8
def test_kernel(self):
# For now, we're only testing the gaussian kernel
ps = PointSet()
ac = AddC().batch(ps)
# assert getattr(ac.dist, "__name__") == "gaussian"
assert (ac.fastpair.dist((1, 2, 3, 4, 5), (6, 5, 4, 3, 2)) - 2) < 1e-8
def test_call_and_centroids(self):
ps = PointSet()
ac = AddC(kmax=5)
assert len(ac()) == len(ac.centroids) == 0
ac.batch(ps)
assert len(ac()) == len(ac.centroids) == 5
def test_batch(self):
ps = PointSet()
ac = AddC()
assert len(ac) == 0
ac.batch(ps)
assert len(ac) == len(ps)
def test_add(self):
ps = PointSet()
ac = AddC()
for p in ps:
ac += p
assert len(ac) == len(ps)
def test_init(self):
ac = AddC()
assert len(ac) < 1
def test_npoints(self):
ps = PointSet()
ac = AddC(10).batch(ps)
assert ac.npoints == len(ps)
assert ac.npoints > len(ac)
assert isinstance(ac.npoints, int)
def test_iter(self):
ps = PointSet()
ac = AddC().batch(ps)
assert hasattr(ac, '__iter__')
assert len(ac) == len(ps)
def test_len(self):
ps = PointSet()
ac = AddC(kmax=8)
assert len(ac) == 0
ac.batch(ps)
assert len(ac) == 8
def test_contains(self):
ps = PointSet(10, 2)
ac = AddC()
assert ps[0] not in ac
ac.batch(ps)
assert ac.centroids[0] in ac