add tests for waspas

Author: jbytecode

src/main/scala/waspas.scala

@@ -6,7 +6,9 @@ case class WaspasResult(
     normalizedDecisionMatrix: Mat,
     scoresWSM: Vec,
     scoresWPM: Vec,
-    scores: Vec
+    scores: Vec,
+    bestIndex: Int,
+    orderings: VecInt 
 ) extends MCDMResult
 
 def waspas(
@@ -40,4 +42,8 @@ def waspas(
 
   val scores = Matrix.rowsums(Matrix.mul(scoreTables, Matrix.makeRowMatrix(l)))
 
-  WaspasResult(normalizedDecisionMat, scoresWSM, scoresWPM, scores)
+  val bestIndex = scores.zipWithIndex.maxBy(_._1)._2
+
+  val orderings = scores.zipWithIndex.sortBy(_._1).map(_._2)
+
+  WaspasResult(normalizedDecisionMat, scoresWSM, scoresWPM, scores, bestIndex, orderings)

src/test/scala/testwaspas.scala

@@ -44,6 +44,10 @@ class TestWaspas extends munit.FunSuite {
         0.7873956894790834,
         0.7674278741781709)
 
+    val expectedBestIndex = 0
+
+    val expectedOrdering = Array(2, 1, 4, 3, 0)
+
     A.assert(Matrix.elementwise_equal(result.normalizedDecisionMatrix, expectedNormalizedMatrix, 1e-5))
 
     A.assert(Matrix.elementwise_equal(result.scoresWSM, expectedScoresWSM, 1e-5))
@@ -52,5 +56,9 @@ class TestWaspas extends munit.FunSuite {
 
     A.assert(Matrix.elementwise_equal(result.scores, expectedScores, 1e-5))
 
+    A.assertEquals(result.bestIndex, expectedBestIndex)
+
+    expectedOrdering.zip(result.orderings).foreach { case (a, b) => A.assertEquals(a, b) }
+
   }
 }