add test and benchmark for MT with PoseidonBN128Hash

plonky2/benches/merkle.rs

@@ -6,6 +6,7 @@ use plonky2::hash::hash_types::RichField;
 use plonky2::hash::keccak::KeccakHash;
 use plonky2::hash::merkle_tree::MerkleTree;
 use plonky2::hash::poseidon::PoseidonHash;
+use plonky2::hash::poseidon_bn128::PoseidonBN128Hash;
 use plonky2::plonk::config::Hasher;
 use tynm::type_name;
 
@@ -31,6 +32,7 @@ pub(crate) fn bench_merkle_tree<F: RichField, H: Hasher<F>>(c: &mut Criterion) {
 fn criterion_benchmark(c: &mut Criterion) {
     bench_merkle_tree::<GoldilocksField, PoseidonHash>(c);
     bench_merkle_tree::<GoldilocksField, KeccakHash<25>>(c);
+    bench_merkle_tree::<GoldilocksField, PoseidonBN128Hash>(c);
 }
 
 criterion_group!(benches, criterion_benchmark);

plonky2/src/hash/merkle_tree.rs

@@ -589,6 +589,7 @@ mod tests {
     use super::*;
     use crate::field::extension::Extendable;
     use crate::hash::merkle_proofs::verify_merkle_proof_to_cap;
+    use crate::hash::poseidon_bn128::PoseidonBN128GoldilocksConfig;
     use crate::plonk::config::{GenericConfig, KeccakGoldilocksConfig, PoseidonGoldilocksConfig};
 
     fn random_data<F: RichField>(n: usize, k: usize) -> Vec<Vec<F>> {
@@ -669,4 +670,19 @@ mod tests {
 
         Ok(())
     }
+
+    #[test]
+    fn test_merkle_trees_poseidonbn128() -> Result<()> {
+        const D: usize = 2;
+        type C = PoseidonBN128GoldilocksConfig;
+        type F = <C as GenericConfig<D>>::F;
+
+        let log_n = 12;
+        let n = 1 << log_n;
+        let leaves = random_data::<F>(n, 7);
+
+        verify_all_leaves::<F, C, D>(leaves, 1)?;
+
+        Ok(())
+    }
 }