HDBSCANTest.scala

// Wei Chen - HDBSCAN Test
// 2016-11-12

import com.scalaml.TestData._
import com.scalaml.general.MatrixFunc._
import com.scalaml.algorithm.HDBSCAN
import org.scalatest.funsuite.AnyFunSuite

class HDBSCANSuite extends AnyFunSuite {

    val hdbscan = new HDBSCAN()
    test("HDBSCAN Test : Clustering Tiny Data") {
        assert(hdbscan.clear())
        assert(hdbscan.config(Map("k" -> 2, "limit" -> 2)))
        val result = hdbscan.cluster(UNLABELED_TINY_DATA)
        assert(arrayequal(result, LABEL_TINY_DATA))
    }

    test("HDBSCAN Test : Clustering Small Data") {
        assert(hdbscan.clear())
        assert(hdbscan.config(Map("k" -> 2, "limit" -> 2)))
        val result = hdbscan.cluster(UNLABELED_SMALL_DATA)
        assert(arrayequal(result, LABEL_SMALL_DATA))
    }

    test("HDBSCAN Test : Clustering Large Data") {
        assert(hdbscan.clear())
        assert(hdbscan.config(Map("k" -> 2, "limit" -> 2)))
        val result = hdbscan.cluster(UNLABELED_LARGE_DATA)
        assert(arrayequal(result, LABEL_LARGE_DATA))
    }

    test("HDBSCAN Test : Invalid Config") {
        assert(hdbscan.clear())
        assert(!hdbscan.config(Map("limit" -> "test")))
    }
}