Update tests.py

tests.py

@@ -1,17 +1,26 @@
 test_cases = [
     {
-        "predictions": [0, 0],
-        "references": [1, 1],
-        "result": {"metric_score": 0}
+        "predictions": [0.01, 0.01, 0.9],
+        "references": [2, 0],
     },
     {
-        "predictions": [1, 1],
-        "references": [1, 1],
-        "result": {"metric_score": 1}
+        "predictions": [0.01, 0.01, 0.9],
+        "references": [1, 2],
     },
     {
-        "predictions": [1, 0],
-        "references": [1, 1],
-        "result": {"metric_score": 0.5}
+        "predictions": [0.9, 0.01, 0.01],
+        "references": [0, 0],
+    }
+    {
+        "predictions": [0.9, 0.01, 0.01],
+        "references": [2, 2],
+    }
+    {
+        "predictions": [0.01, 0.9, 0.01],
+        "references": [1, 0],
+    }
+    {
+        "predictions": [0.01, 0.9, 0.01],
+        "references": [0, 2],
     }
 ]