add more tests for CReport

Aeolun · Jul 19, 2016 · 093e8fc · 093e8fc
1 parent 074dcf7
commit 093e8fc
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/tests/Phpml/Metric/ClassificationReportTest.php b/tests/Phpml/Metric/ClassificationReportTest.php
@@ -27,4 +27,24 @@ public function testClassificationReportGenerateWithStringLabels()
         $this->assertEquals($support, $report->getSupport(), '', 0.01);
         $this->assertEquals($average, $report->getAverage(), '', 0.01);
     }
+
+    public function testClassificationReportGenerateWithNumericLabels()
+    {
+        $labels = [0, 1, 2, 2, 2];
+        $predicted = [0, 0, 2, 2, 1];
+
+        $report = new ClassificationReport($labels, $predicted);
+
+        $precision = [0 => 0.5, 1 => 0.0, 2 => 1.0];
+        $recall = [0 => 1.0, 1 => 0.0, 2 => 0.67];
+        $f1score = [0 => 0.67, 1 => 0.0, 2 => 0.80];
+        $support = [0 => 1, 1 => 1, 2 => 3];
+        $average = ['precision' => 0.75, 'recall' => 0.83, 'f1score' => 0.73];
+
+        $this->assertEquals($precision, $report->getPrecision(), '', 0.01);
+        $this->assertEquals($recall, $report->getRecall(), '', 0.01);
+        $this->assertEquals($f1score, $report->getF1score(), '', 0.01);
+        $this->assertEquals($support, $report->getSupport(), '', 0.01);
+        $this->assertEquals($average, $report->getAverage(), '', 0.01);
+    }
 }