From 798d53ba12c093c554d87ab76e010df0be1227c6 Mon Sep 17 00:00:00 2001
From: M Alghazwi <m.ghazwi@gmail.com>
Date: Thu, 30 Jan 2025 13:47:26 +0100
Subject: [PATCH] add uniform recursion tests

---
 proof-input/src/params.rs            |  4 +--
 proof-input/src/recursion/mod.rs     |  1 +
 proof-input/src/recursion/uniform.rs | 48 ++++++++++++++++++++++++++++
 3 files changed, 51 insertions(+), 2 deletions(-)
 create mode 100644 proof-input/src/recursion/uniform.rs

diff --git a/proof-input/src/params.rs b/proof-input/src/params.rs
index ade2ea7..5ef4358 100644
--- a/proof-input/src/params.rs
+++ b/proof-input/src/params.rs
@@ -10,9 +10,9 @@ use plonky2_poseidon2::poseidon2_hash::poseidon2::Poseidon2Hash;
 
 // test types
 pub const D: usize = 2;
-pub type C = Poseidon2GoldilocksConfig;
+pub type C = PoseidonGoldilocksConfig;
 pub type F = <C as GenericConfig<D>>::F; // this is the goldilocks field
-pub type HF = Poseidon2Hash;
+pub type HF = PoseidonHash;
 
 // hardcoded default params for generating proof input
 const DEFAULT_MAX_DEPTH: usize = 32; // depth of big tree (slot tree depth, includes block tree depth)
diff --git a/proof-input/src/recursion/mod.rs b/proof-input/src/recursion/mod.rs
index 1c2b066..d91a328 100644
--- a/proof-input/src/recursion/mod.rs
+++ b/proof-input/src/recursion/mod.rs
@@ -4,3 +4,4 @@ pub mod cyclic_recursion;
 pub mod tree1;
 pub mod tree2;
 mod hybrid;
+mod uniform;
\ No newline at end of file
diff --git a/proof-input/src/recursion/uniform.rs b/proof-input/src/recursion/uniform.rs
new file mode 100644
index 0000000..9301744
--- /dev/null
+++ b/proof-input/src/recursion/uniform.rs
@@ -0,0 +1,48 @@
+// some tests for approach 2 of the tree recursion
+
+#[cfg(test)]
+mod tests {
+    use plonky2::iop::witness::{PartialWitness};
+    use plonky2::plonk::circuit_builder::CircuitBuilder;
+    use plonky2::plonk::circuit_data::{CircuitConfig};
+    use plonky2::plonk::config::{ GenericConfig};
+    use plonky2::plonk::proof::{ProofWithPublicInputs};
+    use codex_plonky2_circuits::circuits::sample_cells::SampleCircuit;
+    use crate::params::{F, D, C, HF};
+    use crate::gen_input::gen_testing_circuit_input;
+    use crate::params::Params;
+    use codex_plonky2_circuits::recursion::uniform::{tree::TreeRecursion};
+
+    #[test]
+    fn test_uniform_recursion() -> anyhow::Result<()> {
+
+        let config = CircuitConfig::standard_recursion_config();
+        let mut sampling_builder = CircuitBuilder::<F, D>::new(config);
+
+        //------------ sampling inner circuit ----------------------
+        // Circuit that does the sampling - 100 samples
+        let mut params = Params::default();
+        params.input_params.n_samples = 100;
+        params.circuit_params.n_samples = 100;
+        let one_circ_input = gen_testing_circuit_input::<F,D>(&params.input_params);
+        let samp_circ = SampleCircuit::<F,D,HF>::new(params.circuit_params);
+        let inner_tar = samp_circ.sample_slot_circuit_with_public_input(&mut sampling_builder)?;
+        // get generate a sampling proof
+        let mut pw = PartialWitness::<F>::new();
+        samp_circ.sample_slot_assign_witness(&mut pw,&inner_tar,&one_circ_input)?;
+        let inner_data = sampling_builder.build::<C>();
+        println!("sampling circuit degree bits = {:?}", inner_data.common.degree_bits());
+        let inner_proof = inner_data.prove(pw)?;
+
+        let proofs: Vec<ProofWithPublicInputs<F, C, D>> = (0..4).map(|i| inner_proof.clone()).collect();
+
+        // ------------------- tree --------------------
+
+        let mut tree = TreeRecursion::<F,D,C,HF>::build(inner_data.common.clone())?;
+
+        let root = tree.prove_tree(&proofs, inner_data.verifier_data())?;
+
+        Ok(())
+    }
+
+}
\ No newline at end of file