def test_cluster_map_clusters_sizes():
rng = np.random.RandomState(42)
nb_clusters = 11
# Generate random indices
indices = [range(rng.randint(1, 10)) for _ in range(nb_clusters)]
cluster_map = ClusterMap()
clusters = [Cluster(indices=indices[i]) for i in range(nb_clusters)]
cluster_map.add_cluster(*clusters)
assert_equal(cluster_map.clusters_sizes(), list(map(len, indices)))
def test_cluster_map_clusters_sizes():
rng = np.random.RandomState(42)
nb_clusters = 11
# Generate random indices
indices = [range(rng.randint(1, 10)) for i in range(nb_clusters)]
cluster_map = ClusterMap()
clusters = [Cluster(indices=indices[i]) for i in range(nb_clusters)]
cluster_map.add_cluster(*clusters)
assert_equal(cluster_map.clusters_sizes(), list(map(len, indices)))
