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)