Add Test Case

onyx-dot-app · Nov 24, 2024 · 86d8666 · 86d8666
1 parent 8abcde9
commit 86d8666
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/backend/tests/unit/danswer/llm/answering/stream_processing/test_citation_processing.py b/backend/tests/unit/danswer/llm/answering/stream_processing/test_citation_processing.py
@@ -374,6 +374,17 @@ def process_text(
             "The code demonstrates variable assignment.",
             [],
         ),
+        (
+            "Citation as a single token",
+            [
+                "Here is some text",
+                "[1]",
+                ".",
+                " Some other text",
+            ],
+            "Here is some text[[1]](https://0.com). Some other text",
+            ["doc_0"],
+        ),
     ],
 )
 def test_citation_extraction(