BayesianDecisionTest.scala

// Wei Chen - Bayesian Decision Test
// 2016-06-03

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

class BayesianDecisionSuite extends AnyFunSuite {

    val bayesiandecision = new BayesianDecision()

    test("BayesianDecision Test : Clear") {
        assert(bayesiandecision.clear())
    }

    test("BayesianDecision Test : Linear Data") {
        assert(bayesiandecision.clear())
        assert(bayesiandecision.config(Map[String, Double]()))
        assert(bayesiandecision.train(LABELED_LINEAR_DATA))
        val result = bayesiandecision.predict(UNLABELED_LINEAR_DATA)
        assert(arrayequal(result, LABEL_LINEAR_DATA))
    }

    test("BayesianDecision Test : Nonlinear Data") {
        assert(bayesiandecision.clear())
        assert(bayesiandecision.config(Map[String, Double]()))
        assert(bayesiandecision.train(LABELED_NONLINEAR_DATA))
        val result = bayesiandecision.predict(UNLABELED_NONLINEAR_DATA)
        assert(arrayequal(result, LABEL_NONLINEAR_DATA))
    }

    test("BayesianDecision Test : Invalid Data") {
        assert(bayesiandecision.clear())
        assert(!bayesiandecision.train(Array((1, Array(1, 2)), (1, Array()))))
        assert(bayesiandecision.predict(Array(Array(1.0))).size == 0)
    }
}