def test_graph_dict_basic():
cluster_object = MaxPRegionsHeu()
cluster_object.fit_from_networkx(graph, attr_dict,
spatially_extensive_attr_dict,
threshold=threshold)
result = region_list_from_array(cluster_object.labels_)
compare_region_lists(result, optimal_clustering)
def test_graph_dict_basic():
cluster_object = MaxPRegionsHeu(random_state=0)
cluster_object.fit_from_networkx(graph, attr_dict,
spatially_extensive_attr_dict,
threshold=threshold)
result = region_list_from_array(cluster_object.labels_)
compare_region_lists(result, optimal_clustering)
def test_graph_dict_multi_attr():
cluster_object = MaxPRegionsHeu(random_state=0)
cluster_object.fit_from_networkx(graph,
double_attr_dict,
double_spatially_extensive_attr_dict,
threshold=double_threshold)
result = region_list_from_array(cluster_object.labels_)
compare_region_lists(result, optimal_clustering)
def test_graph_str_basic():
nx.set_node_attributes(graph, attr_str, attr_dict)
nx.set_node_attributes(graph, spatially_extensive_attr_str,
spatially_extensive_attr_dict)
cluster_object = MaxPRegionsHeu(random_state=0)
cluster_object.fit_from_networkx(graph, attr_str,
spatially_extensive_attr_str,
threshold=threshold)
result = region_list_from_array(cluster_object.labels_)
compare_region_lists(result, optimal_clustering)
def test_graph_str_multi_attr():
nx.set_node_attributes(graph, attr_str, attr_dict)
nx.set_node_attributes(graph, spatially_extensive_attr_str,
spatially_extensive_attr_dict)
cluster_object = MaxPRegionsHeu()
cluster_object.fit_from_networkx(graph,
[attr_str] * 2,
[spatially_extensive_attr_str] * 2,
threshold=double_threshold)
result = region_list_from_array(cluster_object.labels_)
compare_region_lists(result, optimal_clustering)
def test_graph_str_multi_attr():
nx.set_node_attributes(graph, attr_str, attr_dict)
nx.set_node_attributes(graph, spatially_extensive_attr_str,
spatially_extensive_attr_dict)
cluster_object = MaxPRegionsHeu(random_state=0)
cluster_object.fit_from_networkx(graph,
[attr_str] * 2,
[spatially_extensive_attr_str] * 2,
threshold=double_threshold)
result = region_list_from_array(cluster_object.labels_)
compare_region_lists(result, optimal_clustering)