def test_cluster_is_inside(self):
clusters = (Cluster(16, [16]), Cluster(4, [4, 5, 6, 7, 8]),
Cluster(0, [0, 1, 2, 3]),
Cluster(9, [9, 10, 11, 12, 13, 14, 15]))
not_in_cluster = Cluster(17, [17, 16])
in_cluster = Cluster(0, [0, 1, 2, 3])
clustering = Clustering(clusters)
self.assertEqual(clustering.cluster_index(not_in_cluster), -1)
self.assertEqual(clustering.cluster_index(in_cluster), 2)
self.assertEqual(clustering.cluster_is_inside(not_in_cluster), False)
self.assertEqual(clustering.cluster_is_inside(in_cluster), True)
def test_cluster_is_inside(self):
clusters =(
Cluster(16,[16]),
Cluster(4,[4,5,6,7,8]),
Cluster(0,[0,1,2,3]),
Cluster(9,[9,10,11,12,13,14,15])
)
not_in_cluster= Cluster(17,[17,16])
in_cluster = Cluster(0,[0,1,2,3])
clusterization = Clustering(clusters)
self.assertEqual(clusterization.cluster_index(not_in_cluster),-1)
self.assertEqual(clusterization.cluster_index(in_cluster),2)
self.assertEqual(clusterization.cluster_is_inside(not_in_cluster),False)
self.assertEqual(clusterization.cluster_is_inside(in_cluster),True)
