handle single and array JSON schemas in test suite

guidance-ai · Dec 7, 2024 · f3c36e2 · f3c36e2
1 parent 8d7517b
commit f3c36e2
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/sample_parser/src/json_schema_testsuite.rs b/sample_parser/src/json_schema_testsuite.rs
@@ -222,8 +222,11 @@ fn main() {
 
     for arg in &args[1..] {
         let schema_file = read_file_to_string(arg);
-        let val: Vec<JsonTest> =
-            serde_json::from_str(&schema_file).expect("Invalid JSON in schema");
+        let val: Vec<JsonTest> = if schema_file.starts_with("{") {
+            vec![serde_json::from_str(&schema_file).expect("Invalid JSON in single schema")]
+        } else {
+            serde_json::from_str(&schema_file).expect("Invalid JSON in array schema")
+        };
         for (idx, t) in val.iter().enumerate() {
             stats.num_tests += 1;
             print!("Running test: {} ({}) #{} ", arg, t.description, idx);