WIP: add tool to generate test cases for BLS operators

Chia-Network · May 4, 2023 · 3ad371c · 3ad371c
1 parent 65d7b58
commit 3ad371c
Show file tree

Hide file tree

Showing 3 changed files with 1,578 additions and 3 deletions.
diff --git a/src/test_ops.rs b/src/test_ops.rs
@@ -203,7 +203,8 @@ fn run_op_test(op: &Opf, args_str: &str, expected: &str, expected_cost: u64) {
     assert_eq!(rest, "");
     let result = op(&mut a, args, 10000000000 as Cost);
     match result {
-        Err(_) => {
+        Err(e) => {
+            println!("Error: {}", e.1);
             assert_eq!(expected, "FAIL");
         }
         Ok(Reduction(cost, ret_value)) => {
@@ -219,7 +220,12 @@ fn run_op_test(op: &Opf, args_str: &str, expected: &str, expected_cost: u64) {
 fn test_ops() {
     use std::fs::read_to_string;
 
-    const TEST_FILES: &[&str; 3] = &["test-core-ops.txt", "test-more-ops.txt", "test-bls-ops.txt"];
+    const TEST_FILES: &[&str; 4] = &[
+        "test-blspy-vectors.txt",
+        "test-core-ops.txt",
+        "test-more-ops.txt",
+        "test-bls-ops.txt",
+    ];
 
     let funs = HashMap::from([
         ("i", op_if as Opf),
@@ -282,7 +288,9 @@ fn test_ops() {
                 continue;
             }
             let (op_name, t) = t.split_once(" ").unwrap();
-            let op = funs.get(op_name).unwrap();
+            let op = funs
+                .get(op_name)
+                .expect(&format!("couldn't find operator \"{op_name}\""));
             let (args, out) = t.split_once("=>").unwrap();
             let (expected, expected_cost) = if out.contains("|") {
                 out.split_once("|").unwrap()