Merge pull request privacy-scaling-explorations#138 from input-output…

…-hk/dev-fix/add-test-vectors-to-poseidon-duplex Resolve "Add test vectors to poseidon duplex"
input-output-hk · Mar 7, 2024 · 10623d5 · 10623d5
2 parents b2752df + adbc67b
commit 10623d5
Show file tree

Hide file tree

Showing 5 changed files with 1,214 additions and 0 deletions.
diff --git a/halo2_proofs/Cargo.toml b/halo2_proofs/Cargo.toml
@@ -106,3 +106,6 @@ required-features = ["dev-graph"]
 
 [[example]]
 name = "emit-duplex-pattern"
+
+[[example]]
+name = "random-duplex-test-vectors"
diff --git a/halo2_proofs/examples/random-duplex-test-vectors.rs b/halo2_proofs/examples/random-duplex-test-vectors.rs
@@ -0,0 +1,5 @@
+use halo2_proofs::poseidon::duplex::test_vector_tools::print_random_duplex_test_vectors;
+
+fn main() {
+    print_random_duplex_test_vectors(5);
+}
diff --git a/halo2_proofs/src/poseidon/duplex.rs b/halo2_proofs/src/poseidon/duplex.rs
@@ -709,3 +709,54 @@ mod test_complex_pattern {
         duplex.finish();
     }
 }
+
+pub mod test_vector_tools;
+
+#[cfg(test)]
+mod test_vectors;
+
+#[cfg(test)]
+mod check_test_vectors {
+    use super::test_vector_tools::check_duplex_test_vector;
+    use super::test_vectors;
+
+    #[test]
+    fn check_fp_test_vectors() {
+        test_vectors::fp_a1_s1()
+            .into_iter()
+            .enumerate()
+            .for_each(|(i, tv)| check_duplex_test_vector(tv, &format!("fp_a1_s1[{i}]")));
+        test_vectors::fp_a1_s2()
+            .into_iter()
+            .enumerate()
+            .for_each(|(i, tv)| check_duplex_test_vector(tv, &format!("fp_a1_s2[{i}]")));
+        test_vectors::fp_a2_s1()
+            .into_iter()
+            .enumerate()
+            .for_each(|(i, tv)| check_duplex_test_vector(tv, &format!("fp_a2_s1[{i}]")));
+        test_vectors::fp_a2_s2()
+            .into_iter()
+            .enumerate()
+            .for_each(|(i, tv)| check_duplex_test_vector(tv, &format!("fp_a2_s2[{i}]")));
+    }
+
+    #[test]
+    fn check_fq_test_vectors() {
+        test_vectors::fq_a1_s1()
+            .into_iter()
+            .enumerate()
+            .for_each(|(i, tv)| check_duplex_test_vector(tv, &format!("fq_a1_s1[{i}]")));
+        test_vectors::fq_a1_s2()
+            .into_iter()
+            .enumerate()
+            .for_each(|(i, tv)| check_duplex_test_vector(tv, &format!("fq_a1_s2[{i}]")));
+        test_vectors::fq_a2_s1()
+            .into_iter()
+            .enumerate()
+            .for_each(|(i, tv)| check_duplex_test_vector(tv, &format!("fq_a2_s1[{i}]")));
+        test_vectors::fq_a2_s2()
+            .into_iter()
+            .enumerate()
+            .for_each(|(i, tv)| check_duplex_test_vector(tv, &format!("fq_a2_s2[{i}]")));
+    }
+}