# -*- coding: utf-8 -*-

import pytest

from scipy.stats import multivariate_normal
import numpy as np

from hypertools.tools.cluster import cluster

cluster1 = np.random.multivariate_normal(np.zeros(3), np.eye(3), size=100)
cluster2 = np.random.multivariate_normal(np.zeros(3)+100, np.eye(3), size=100)
data = np.vstack([cluster1,cluster2])
labels = cluster(data,n_clusters=2)

def test_cluster_n_clusters():
    assert len(set(labels))==2

def test_cluster_returns_list():
    assert type(labels) is list

@@ -500,8 +500,192 @@
 is list%0A
+%0Adef test_cluster_hdbscan():%0A    # Given well separated clusters this should %22just work%22%0A    hdbscan_labels = cluster(data, cluster='HDBSCAN')%0A    assert len(set(hdbscan_labels)) == 2%0A

Add test for HDBSCAN