From 6fc232d143aba399aa0f363fbe4dbd0005befc6f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Dominik=20Fuch=C3=9F?= <dominik.fuchss@kit.edu>
Date: Wed, 21 Aug 2024 13:41:28 +0200
Subject: [PATCH] Some tests

---
 .../calculation/ClassificationMetricsTest.kt  | 82 +++++++++++++++++++
 1 file changed, 82 insertions(+)
 create mode 100644 calculator/src/test/kotlin/edu/kit/kastel/mcse/ardoco/metrics/calculation/ClassificationMetricsTest.kt

diff --git a/calculator/src/test/kotlin/edu/kit/kastel/mcse/ardoco/metrics/calculation/ClassificationMetricsTest.kt b/calculator/src/test/kotlin/edu/kit/kastel/mcse/ardoco/metrics/calculation/ClassificationMetricsTest.kt
new file mode 100644
index 0000000..ca7bcb1
--- /dev/null
+++ b/calculator/src/test/kotlin/edu/kit/kastel/mcse/ardoco/metrics/calculation/ClassificationMetricsTest.kt
@@ -0,0 +1,82 @@
+package edu.kit.kastel.mcse.ardoco.metrics.calculation
+
+import org.junit.jupiter.api.Assertions.assertAll
+import org.junit.jupiter.api.Assertions.assertEquals
+import org.junit.jupiter.api.Test
+import org.junit.jupiter.api.function.Executable
+
+class ClassificationMetricsTest {
+    @Test
+    fun calculatePrecisionTest() {
+        assertAll(
+            Executable { assertEquals(.5, calculatePrecision(10, 10), 1e-3) },
+            Executable { assertEquals(.857, calculatePrecision(6, 1), 1e-3) },
+            Executable { assertEquals(.154, calculatePrecision(10, 55), 1e-3) },
+            Executable { assertEquals(.905, calculatePrecision(210, 22), 1e-3) }
+        )
+    }
+
+    @Test
+    fun calculateRecallTest() {
+        assertAll(
+            Executable { assertEquals(.5, calculateRecall(10, 10), 1e-3) },
+            Executable { assertEquals(.75, calculateRecall(6, 2), 1e-3) },
+            Executable { assertEquals(.154, calculateRecall(10, 55), 1e-3) },
+            Executable { assertEquals(.871, calculateRecall(210, 31), 1e-3) }
+        )
+    }
+
+    @Test
+    fun calculateF1FromPrecisionRecallTest() {
+        assertAll(
+            Executable { assertEquals(1.0, calculateF1(1.0, 1.0), 1e-2) },
+            Executable { assertEquals(0.0, calculateF1(0.0, 1.0), 1e-2) },
+            Executable { assertEquals(0.0, calculateF1(1.0, 0.0), 1e-2) },
+            Executable { assertEquals(0.18, calculateF1(.9, .1), 1e-2) },
+            Executable { assertEquals(0.48, calculateF1(.6, .4), 1e-2) },
+            Executable { assertEquals(0.42, calculateF1(.3, .7), 1e-2) },
+            Executable { assertEquals(0.9, calculateF1(.9, .9), 1e-2) },
+            Executable { assertEquals(0.48, calculateF1(.4, .6), 1e-2) }
+        )
+    }
+
+    @Test
+    fun calculateAccuracyTest() {
+        assertAll(
+            Executable { assertEquals(.5, calculateAccuracy(10, 10, 10, 10), 1e-3) },
+            Executable { assertEquals(.75, calculateAccuracy(6, 1, 2, 3), 1e-3) },
+            Executable { assertEquals(.214, calculateAccuracy(10, 55, 55, 20), 1e-3) },
+            Executable { assertEquals(.967, calculateAccuracy(210, 22, 31, 1337), 1e-3) }
+        )
+    }
+
+    @Test
+    fun calculatePhiCoefficientTest() {
+        assertAll(
+            Executable { assertEquals(.0, calculatePhiCoefficient(10, 10, 10, 10), 1e-3) },
+            Executable { assertEquals(.478, calculatePhiCoefficient(6, 1, 2, 3), 1e-3) },
+            Executable { assertEquals(-.579, calculatePhiCoefficient(10, 55, 55, 20), 1e-3) },
+            Executable {
+                assertEquals(
+                    .869,
+                    calculatePhiCoefficient(210, 22, 31, 1337),
+                    1e-3
+                )
+            },
+            Executable { assertEquals(.0, calculatePhiCoefficient(0, 0, 11, 11), 1e-3) },
+            Executable { assertEquals(.0, calculatePhiCoefficient(11, 0, 11, 0), 1e-3) }
+        )
+    }
+
+    @Test
+    fun calculateSpecificityTest() {
+        assertAll(
+            Executable { assertEquals(.5, calculateSpecificity(1, 1), 1e-3) },
+            Executable { assertEquals(.76, calculateSpecificity(1337, 420), 1e-3) },
+            Executable { assertEquals(.0, calculateSpecificity(0, 20), 1e-3) },
+            Executable { assertEquals(1.0, calculateSpecificity(20, 0), 1e-3) },
+            Executable { assertEquals(1.0, calculateSpecificity(0, 0), 1e-3) },
+            Executable { assertEquals(.375, calculateSpecificity(3, 5), 1e-3) }
+        )
+    }
+}