json: add test for type: [array, null] fix

ggerganov · Jun 22, 2024 · 9676f22 · 9676f22
1 parent 9c2cc11
commit 9676f22
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/tests/test-grammar-integration.cpp b/tests/test-grammar-integration.cpp
@@ -773,6 +773,29 @@ static void test_json_schema() {
         }
     );
 
+    test_schema(
+        "",
+        // Schema
+        R"""(
+            {
+                "type": ["array", "null"],
+                "items": { "type": "string" }
+            }
+        )""",
+        // Passing strings
+        {
+            "null",
+            "[]",
+            "[\"foo\"]",
+            "[\"foo\", \"bar\"]",
+        },
+        // Failing strings
+        {
+            "",
+            "\"foo\"",
+            "[\"foo\", 42]",
+        }
+    );
 
     test_schema(
         "min+max items",