def test_dict(method): value_dict = dataframe_to_dict(gdf, attr_str) cluster_object = PRegionsExact() cluster_object.fit_from_dict(neighbors_dict, value_dict, n_regions=2, method=method) result = region_list_from_array(cluster_object.labels_) compare_region_lists(result, optimal_clustering)
def test_dict_multi_attr(method): cluster_object = PRegionsExact() cluster_object.fit_from_dict(neighbors_dict, double_attr_dict, n_regions=2, method=method) obtained = region_list_from_array(cluster_object.labels_) compare_region_lists(obtained, optimal_clustering)