def test_eps_neighborhood(self):
distances = CondensedMatrix([
1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 2., 1., 1., 1., 1.,
1., 3., 1., 1., 1., 1., 2., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
2., 1.
])
dbscan_alg = DBSCANAlgorithm(distances)
self.assertItemsEqual(
dbscan_alg._DBSCANAlgorithm__eps_neighborhood(6, 3),
[0, 1, 2, 3, 4, 5, 7, 8])
self.assertItemsEqual(
dbscan_alg._DBSCANAlgorithm__eps_neighborhood(6, 2),
[0, 1, 3, 4, 5, 7, 8])
self.assertItemsEqual(
dbscan_alg._DBSCANAlgorithm__eps_neighborhood(6, 1), [0, 4, 5, 7])
self.assertItemsEqual(
dbscan_alg._DBSCANAlgorithm__eps_neighborhood(6, 1.), [0, 4, 5, 7])
self.assertItemsEqual(distances.element_neighbors_within_radius(6, 3),
[0, 1, 2, 3, 4, 5, 7, 8])
self.assertItemsEqual(distances.element_neighbors_within_radius(6, 2),
[0, 1, 3, 4, 5, 7, 8])
self.assertItemsEqual(distances.element_neighbors_within_radius(6, 1),
[0, 4, 5, 7])
self.assertItemsEqual(distances.element_neighbors_within_radius(6, 1.),
[0, 4, 5, 7])
def test_eps_neighborhood(self):
distances = CondensedMatrix([ 1., 1., 1., 1., 1., 1., 1., 1.,
1., 1., 1., 1., 2., 1. ,1.,
1., 1., 1., 3., 1., 1.,
1., 1., 2., 1., 1.,
1., 1., 1., 1.,
1., 1., 1.,
1., 2.,
1.])
dbscan_alg = DBSCANAlgorithm(distances)
self.assertItemsEqual(dbscan_alg._DBSCANAlgorithm__eps_neighborhood(6,3),[0, 1, 2, 3, 4, 5, 7, 8])
self.assertItemsEqual(dbscan_alg._DBSCANAlgorithm__eps_neighborhood(6,2),[0, 1, 3, 4, 5, 7, 8])
self.assertItemsEqual(dbscan_alg._DBSCANAlgorithm__eps_neighborhood(6,1),[0, 4, 5, 7])
self.assertItemsEqual(dbscan_alg._DBSCANAlgorithm__eps_neighborhood(6,1.),[0, 4, 5, 7])
self.assertItemsEqual(distances.element_neighbors_within_radius(6,3),[0, 1, 2, 3, 4, 5, 7, 8])
self.assertItemsEqual(distances.element_neighbors_within_radius(6,2),[0, 1, 3, 4, 5, 7, 8])
self.assertItemsEqual(distances.element_neighbors_within_radius(6,1),[0, 4, 5, 7])
self.assertItemsEqual(distances.element_neighbors_within_radius(6,1.),[0, 4, 5, 7])
