logos-storage/plonky2
1
0
Fork 0
mirror of https://github.com/logos-storage/plonky2.git synced 2026-01-08 00:33:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

tests for biguint gadget

Browse Source
This commit is contained in:
Nicholas Ward 2021-10-19 14:39:10 -07:00
parent b045afbb8a
commit 649c2e2b52
2 changed files with 45 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
src/gadgets/biguint.rs
View File

@ -1,6 +1,6 @@
use std::marker::PhantomData;
use num::Integer;
use num::{BigUint, Integer};
use crate::field::extension_field::Extendable;
use crate::field::field_types::RichField;
@ -26,6 +26,18 @@ impl BigUintTarget {
}
impl<F: RichField + Extendable<D>, const D: usize> CircuitBuilder<F, D> {
fn constant_biguint(&mut self, value: BigUint) -> BigUintTarget {
let limb_values = value.to_u32_digits();
let mut limbs = Vec::new();
for i in 0..limb_values.len() {
limbs.push(U32Target(
self.constant(F::from_canonical_u32(limb_values[i])),
));
}
BigUintTarget { limbs }
}
fn connect_biguint(&mut self, lhs: BigUintTarget, rhs: BigUintTarget) {
let min_limbs = lhs.num_limbs().min(rhs.num_limbs());
for i in 0..min_limbs {
@ -208,6 +220,36 @@ impl<F: RichField + Extendable<D>, const D: usize> SimpleGenerator<F>
#[cfg(test)]
mod tests {
use anyhow::Result;
use num::{BigUint, FromPrimitive};
use crate::{
field::crandall_field::CrandallField,
iop::witness::PartialWitness,
plonk::{circuit_builder::CircuitBuilder, circuit_data::CircuitConfig, verifier::verify},
};
#[test]
fn test_biguint_add() {}
fn test_biguint_add() -> Result<()> {
let x_value = BigUint::from_u128(22222222222222222222222222222222222222).unwrap();
let y_value = BigUint::from_u128(33333333333333333333333333333333333333).unwrap();
let expected_z_value = &x_value + &y_value;
type F = CrandallField;
let config = CircuitConfig::large_config();
let pw = PartialWitness::new();
let mut builder = CircuitBuilder::<F, 4>::new(config);
let x = builder.constant_biguint(x_value);
let y = builder.constant_biguint(y_value);
let z = builder.add_biguint(x, y);
let expected_z = builder.constant_biguint(expected_z_value);
builder.connect_biguint(z, expected_z);
let data = builder.build();
let proof = data.prove(pw).unwrap();
verify(proof, &data.verifier_only, &data.common)
}
}

3
src/iop/witness.rs
View File

@ -5,8 +5,7 @@ use num::{BigUint, FromPrimitive, Zero};
use crate::field::extension_field::target::ExtensionTarget;
use crate::field::extension_field::{Extendable, FieldExtension};
use crate::field::field_types::{Field, PrimeField};
use crate::gadgets::arithmetic_u32::U32Target;
use crate::field::field_types::Field;
use crate::gadgets::biguint::BigUintTarget;
use crate::hash::hash_types::HashOutTarget;
use crate::hash::hash_types::{HashOut, MerkleCapTarget};