Fix: Test all metrics cannot be done with 1 sentence for the CIDEr-D …

…metric.
Labbeti · Nov 15, 2023 · 7a23b4f · 7a23b4f
1 parent e270325
commit 7a23b4f
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/tests/test_all.py b/tests/test_all.py
@@ -1,7 +1,6 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 
-import platform
 import unittest
 
 from unittest import TestCase
@@ -11,15 +10,14 @@
 
 class TestAll(TestCase):
     def test_example_1(self) -> None:
-        cands = ["many green trees are in two sides of a curved green river."]
-        mrefs = [
+        cands: list[str] = ["a man is speaking", "rain falls"]
+        mrefs: list[list[str]] = [
             [
-                "many green trees are in two sides of a curved green river .",
-                "many green trees are in two sides of a curved green river .",
-                "many green trees are in two sides of a curved green river .",
-                "many green trees are in two sides of a curved green river .",
-                "many green trees are in two sides of a curved green river .",
-            ]
+                "a man speaks.",
+                "someone speaks.",
+                "a man is speaking while a bird is chirping in the background",
+            ],
+            ["rain is falling hard on a surface"],
         ]
 
         _ = evaluate(cands, mrefs, metrics="all")