MultivariateLinearRegressionTest.scala

// Wei Chen - Multivariate Linear Regression Test
// 2016-06-04

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

class MultivariateLinearRegressionSuite extends AnyFunSuite {

    val mlr = new MultivariateLinearRegression()
    
    test("MultivariateLinearRegression Test : Clear") {
        assert(mlr.clear())
    }

    test("MultivariateLinearRegression Test : Linear Data") {
        assert(mlr.clear())
        assert(mlr.config(Map[String, Double]()))
        assert(mlr.train(LABELED_LINEAR_DATA.map(yx => yx._1.toDouble -> yx._2)))
        val result = mlr.predict(UNLABELED_LINEAR_DATA)
        val nResult = result.map(v => if (v > 0) 1.0 else -1.0)
        assert(arraysimilar(nResult, LABEL_LINEAR_DATA.map(_.toDouble), 0.9))
    }

    test("MultivariateLinearRegression Test : Nonlinear Data - WRONG") {
        assert(mlr.clear())
        assert(mlr.config(Map[String, Double]()))
        assert(mlr.train(LABELED_NONLINEAR_DATA.map(yx => yx._1.toDouble -> yx._2)))
        val result = mlr.predict(UNLABELED_NONLINEAR_DATA)
        assert(!arraysimilar(result, LABEL_NONLINEAR_DATA.map(_.toDouble), 0.45))
    }

    test("MultivariateLinearRegression Test : Invalid Config & Data") {
        assert(mlr.clear())
        assert(!mlr.config(Map("limit" -> "test")))
        assert(!mlr.train(Array((1, Array(1, 2)), (1, Array()))))
    }
}