def test_graph_str_basic(method): nx.set_node_attributes(graph, attr_str, attr_dict) cluster_object = PRegionsExact() cluster_object.fit_from_networkx(graph, attr_str, n_regions=2, method=method) result = region_list_from_array(cluster_object.labels_) compare_region_lists(result, optimal_clustering)
def test_graph_dict_basic(method): cluster_object = PRegionsExact() cluster_object.fit_from_networkx(graph, attr_dict, n_regions=2, method=method) result = region_list_from_array(cluster_object.labels_) compare_region_lists(result, optimal_clustering)
def test_graph_str_multi_attr(method): nx.set_node_attributes(graph, attr_str, attr_dict) cluster_object = PRegionsExact() cluster_object.fit_from_networkx(graph, double_attr_str, n_regions=2, method=method) result = region_list_from_array(cluster_object.labels_) compare_region_lists(result, optimal_clustering)
def test_graph_dict_multi_attr(method): cluster_object = PRegionsExact() cluster_object.fit_from_networkx(graph, double_attr_dict, n_regions=2, method=method) result = region_list_from_array(cluster_object.labels_) compare_region_lists(result, optimal_clustering)