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

variable-sized tests

Browse Source
This commit is contained in:
Nicholas Ward 2021-09-03 17:15:50 -07:00
parent 4c3f3cda39
commit 1fb7eeb03e
2 changed files with 54 additions and 76 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

112
src/gadgets/permutation.rs
View File

@ -394,79 +394,14 @@ mod tests {
use crate::plonk::circuit_data::CircuitConfig;
use crate::plonk::verifier::verify;
#[test]
fn test_permutation_2x2() -> Result<()> {
fn test_permutation_good(size: usize) -> Result<()> {
type F = CrandallField;
let config = CircuitConfig::large_zk_config();
let pw = PartialWitness::new();
let mut builder = CircuitBuilder::<F, D>::new(config);
let one = F::ONE;
let two = F::from_canonical_usize(2);
let seven = F::from_canonical_usize(7);
let eight = F::from_canonical_usize(8);
let one_two = vec![builder.constant(one), builder.constant(two)];
let seven_eight = vec![builder.constant(seven), builder.constant(eight)];
let a = vec![one_two.clone(), seven_eight.clone()];
let b = vec![seven_eight, one_two];
builder.assert_permutation(a, b);
let data = builder.build();
let proof = data.prove(pw).unwrap();
verify(proof, &data.verifier_only, &data.common)
}
#[test]
fn test_permutation_4x4() -> Result<()> {
type F = CrandallField;
let config = CircuitConfig::large_zk_config();
let pw = PartialWitness::new();
let mut builder = CircuitBuilder::<F, D>::new(config);
let one = F::ONE;
let two = F::from_canonical_usize(2);
let three = F::from_canonical_usize(3);
let four = F::from_canonical_usize(4);
let five = F::from_canonical_usize(5);
let six = F::from_canonical_usize(6);
let seven = F::from_canonical_usize(7);
let eight = F::from_canonical_usize(8);
let one_two = vec![builder.constant(one), builder.constant(two)];
let three_four = vec![builder.constant(three), builder.constant(four)];
let five_six = vec![builder.constant(five), builder.constant(six)];
let seven_eight = vec![builder.constant(seven), builder.constant(eight)];
let a = vec![
one_two.clone(),
three_four.clone(),
five_six.clone(),
seven_eight.clone(),
];
let b = vec![seven_eight, one_two, five_six, three_four];
builder.assert_permutation(a, b);
let data = builder.build();
let proof = data.prove(pw).unwrap();
verify(proof, &data.verifier_only, &data.common)
}
#[test]
fn test_permutation_6x6() -> Result<()> {
type F = CrandallField;
let config = CircuitConfig::large_config();
let pw = PartialWitness::new(config.num_wires);
let mut builder = CircuitBuilder::<F, 4>::new(config);
let lst: Vec<F> = (0..12).map(|n| F::from_canonical_usize(n)).collect();
let lst: Vec<F> = (0..size * 2).map(|n| F::from_canonical_usize(n)).collect();
let a: Vec<Vec<Target>> = lst[..]
.windows(2)
.map(|pair| vec![builder.constant(pair[0]), builder.constant(pair[1])])
@ -481,4 +416,47 @@ mod tests {
verify(proof, &data.verifier_only, &data.common)
}
fn test_permutation_bad(size: usize) -> Result<()> {
type F = CrandallField;
let config = CircuitConfig::large_zk_config();
let pw = PartialWitness::new();
let mut builder = CircuitBuilder::<F, D>::new(config);
let lst1: Vec<F> = (0..size * 2).map(|_| F::rand()).collect();
let lst2: Vec<F> = (0..size * 2).map(|_| F::rand()).collect();
let a: Vec<Vec<Target>> = lst1[..]
.windows(2)
.map(|pair| vec![builder.constant(pair[0]), builder.constant(pair[1])])
.collect();
let b: Vec<Vec<Target>> = lst2[..]
.windows(2)
.map(|pair| vec![builder.constant(pair[0]), builder.constant(pair[1])])
.collect();
builder.assert_permutation(a, b);
let data = builder.build();
let proof = data.prove(pw).unwrap();
verify(proof, &data.verifier_only, &data.common)
}
#[test]
fn test_permutations_good() -> Result<()> {
for n in 2..9 {
test_permutation_good(n).unwrap()
}
Ok(())
}
#[test]
#[should_panic]
fn test_permutations_bad() -> () {
for n in 2..9 {
test_permutation_bad(n).unwrap()
}
}
}

18
src/gates/switch.rs
View File

@ -365,7 +365,7 @@ mod tests {
type F = CrandallField;
type FF = QuarticCrandallField;
const D: usize = 4;
const chunk_size: usize = 4;
const CHUNK_SIZE: usize = 4;
let num_copies = 3;
/// Returns the local wires for a switch gate given the inputs and the switch booleans.
@ -382,11 +382,11 @@ mod tests {
let switch = switch_bools[c];
switches.push(F::from_bool(switch));
let mut first_input_chunk = Vec::with_capacity(chunk_size);
let mut second_input_chunk = Vec::with_capacity(chunk_size);
let mut first_output_chunk = Vec::with_capacity(chunk_size);
let mut second_output_chunk = Vec::with_capacity(chunk_size);
for e in 0..chunk_size {
let mut first_input_chunk = Vec::with_capacity(CHUNK_SIZE);
let mut second_input_chunk = Vec::with_capacity(CHUNK_SIZE);
let mut first_output_chunk = Vec::with_capacity(CHUNK_SIZE);
let mut second_output_chunk = Vec::with_capacity(CHUNK_SIZE);
for e in 0..CHUNK_SIZE {
let first_input = first_inputs[c][e];
let second_input = second_inputs[c][e];
let first_output = if switch { second_input } else { first_input };
@ -406,12 +406,12 @@ mod tests {
v.iter().map(|&x| x.into()).collect::<Vec<_>>()
}
let first_inputs: Vec<Vec<F>> = (0..num_copies).map(|_| F::rand_vec(chunk_size)).collect();
let second_inputs: Vec<Vec<F>> = (0..num_copies).map(|_| F::rand_vec(chunk_size)).collect();
let first_inputs: Vec<Vec<F>> = (0..num_copies).map(|_| F::rand_vec(CHUNK_SIZE)).collect();
let second_inputs: Vec<Vec<F>> = (0..num_copies).map(|_| F::rand_vec(CHUNK_SIZE)).collect();
let switch_bools = vec![true, false, true];
let gate = SwitchGate::<F, D> {
chunk_size,
chunk_size: CHUNK_SIZE,
num_copies,
_phantom: PhantomData,
};