class SpectralClusteringTestCase(unittest.TestCase): def test_spectral_clustering(self): self.graph_type = clustering_config.get_graph_type() self.vertex_list = clustering_config.get_vertex_list() self.edges_list = clustering_config.get_edges_list() self.number_of_clusters = clustering_config.get_number_of_clusters() self.test_cluster_labels = clustering_config.get_cluster_label() self.create_graph() self.find_clusters() self.assertItemsEqual(self.spc.cluster_labels,self.test_cluster_labels,msg='Spectral Clustering Failure') def create_graph(self): self.graph = Graph(graph_type=self.graph_type) # add each vertex for vertex in self.vertex_list: self.graph.add_vertex(vertex) # add each edge for edge in self.edges_list: src,dst,weight = edge self.graph.add_edge(src,dst,weight) def find_clusters(self): # find clusters self.spc = SpectralClustering(self.graph) self.spc.get_clusters(self.number_of_clusters)
def find_clusters(self): # find clusters self.spc = SpectralClustering(self.graph) self.spc.get_clusters(self.number_of_clusters)