github/darkfi
1
1
Fork 0
mirror of https://github.com/darkrenaissance/darkfi.git synced 2026-04-28 03:00:18 -04:00
Code Issues Packages Projects Releases Wiki Activity

zk/gadget/less_than: Improve tests.

Browse Source
This commit is contained in:
parazyd
2023-07-14 20:10:49 +02:00
parent 74d1e21fef
commit 69c0d7b4f5
1 changed files with 51 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

133
src/zk/gadget/less_than.rs
View File

@@ -249,7 +249,7 @@ impl<const WINDOW_SIZE: usize, const NUM_OF_BITS: usize, const NUM_OF_WINDOWS: u
#[cfg(test)]
mod tests {
use super::*;
use darkfi_sdk::crypto::poseidon_hash;
use darkfi_sdk::crypto::pasta_prelude::PrimeField;
use halo2_proofs::{
circuit::{floor_planner, Value},
dev::{CircuitLayout, MockProver},
@@ -337,26 +337,20 @@ mod tests {
test_circuit!(3, 64, 22);
let k = 5;
let valid_a_vals = vec![pallas::Base::from(13), pallas::Base::zero(), pallas::Base::one()];
let valid_b_vals = vec![
pallas::Base::from(15),
pallas::Base::from(u64::MAX),
pallas::Base::from(rand::random::<u64>()),
let valid_pairs = [
(pallas::Base::from(13), pallas::Base::from(15)),
(pallas::Base::ZERO, pallas::Base::from(u64::MAX)),
(pallas::Base::ONE, pallas::Base::from(rand::random::<u64>())),
];
let invalid_a_vals = vec![
pallas::Base::from(14),
pallas::Base::from(u64::MAX),
pallas::Base::zero(),
pallas::Base::one(),
pallas::Base::from(u64::MAX),
];
let invalid_b_vals = vec![
pallas::Base::from(11),
pallas::Base::zero(),
pallas::Base::zero(),
pallas::Base::one(),
pallas::Base::from(u64::MAX),
let invalid_pairs = [
(pallas::Base::from(14), pallas::Base::from(11)),
(pallas::Base::from(u64::MAX), pallas::Base::ZERO),
(pallas::Base::ZERO, pallas::Base::ZERO),
(pallas::Base::ONE, pallas::Base::ONE),
(pallas::Base::ONE, pallas::Base::ZERO),
(pallas::Base::from(u64::MAX), pallas::Base::from(u64::MAX)),
(pallas::Base::from(u64::MAX), pallas::Base::from(u64::MAX) + pallas::Base::ONE),
];
use plotters::prelude::*;
@@ -368,26 +362,16 @@ mod tests {
.into_drawing_area();
CircuitLayout::default().render(k, &circuit, &root).unwrap();
for i in 0..valid_a_vals.len() {
let a = valid_a_vals[i];
let b = valid_b_vals[i];
for (a, b) in valid_pairs {
println!("64 bit (valid) {:?} < {:?} check", a, b);
let circuit = LessThanCircuit { a: Value::known(a), b: Value::known(b) };
let prover = MockProver::run(k, &circuit, vec![]).unwrap();
prover.assert_satisfied();
}
for i in 0..invalid_a_vals.len() {
let a = invalid_a_vals[i];
let b = invalid_b_vals[i];
for (a, b) in invalid_pairs {
println!("64 bit (invalid) {:?} < {:?} check", a, b);
let circuit = LessThanCircuit { a: Value::known(a), b: Value::known(b) };
let prover = MockProver::run(k, &circuit, vec![]).unwrap();
assert!(prover.verify().is_err())
}
@@ -398,73 +382,58 @@ mod tests {
test_circuit!(3, 253, 85);
let k = 7;
const HEADSTART: pallas::Base = pallas::Base::from_raw([
11731824086999220879,
11830614503713258191,
737869762948382064,
46116860184273879,
]);
const P: pallas::Base = pallas::Base::from_raw([
11037532056220336122,
2469829653914515739,
0,
4611686018427387904,
]);
const P_MINUS_1: pallas::Base = pallas::Base::from_raw([
11037532056220336121,
2469829653914515739,
0,
4611686018427387904,
0x992d30ed00000000,
0x224698fc094cf91b,
0x0000000000000000,
0x4000000000000000,
]);
let seed = poseidon_hash([pallas::Base::from(3), pallas::Base::from(10)]);
let y = poseidon_hash([seed, pallas::Base::from(1)]);
// 2^253 - 1. This is the maximum we can check.
const MAX_253: pallas::Base = pallas::Base::from_raw([
0xFFFFFFFFFFFFFFFF,
0xFFFFFFFFFFFFFFFF,
0xFFFFFFFFFFFFFFFF,
0x1FFFFFFFFFFFFFFF,
]);
let sigma1 = pallas::Base::from(1);
let sigma2 = pallas::Base::from(1);
let value = pallas::Base::from(100_000_000_000);
let target = sigma1 * value + sigma2 * value * value + HEADSTART;
let valid_a_vals = vec![pallas::Base::from(1), y, P_MINUS_1];
let valid_b_vals = vec![pallas::Base::from(3), target, P];
let seed = poseidon_hash([pallas::Base::from(3), pallas::Base::from(2)]);
let invalid_a_vals = vec![
pallas::Base::from(3),
y,
pallas::Base::from(11),
poseidon_hash([seed, pallas::Base::from(1)]),
pallas::Base::from(0b010000100),
let valid_pairs = [
(pallas::Base::ZERO, pallas::Base::ONE),
(pallas::Base::from(u64::MAX), pallas::Base::from(u64::MAX) + pallas::Base::ONE),
(
pallas::Base::from_u128(u128::MAX),
pallas::Base::from_u128(u128::MAX) + pallas::Base::ONE,
),
(MAX_253 - pallas::Base::from(2), MAX_253 - pallas::Base::ONE),
(MAX_253 - pallas::Base::ONE, MAX_253),
(MAX_253, MAX_253 + pallas::Base::ONE), // <-- Is this legit??
];
let invalid_b_vals = vec![
pallas::Base::from(1),
target,
pallas::Base::from(11),
pallas::Base::from(0b010000011),
let invalid_pairs = [
(pallas::Base::ZERO, pallas::Base::ZERO),
(pallas::Base::ONE, pallas::Base::ZERO),
(P_MINUS_1 - pallas::Base::ONE, P_MINUS_1),
(P_MINUS_1, P_MINUS_1),
(P_MINUS_1, pallas::Base::ZERO),
(MAX_253, MAX_253),
(MAX_253, pallas::Base::ZERO),
(MAX_253, pallas::Base::ONE),
(MAX_253 + pallas::Base::ONE, pallas::Base::ZERO),
(MAX_253 + pallas::Base::ONE, pallas::Base::ONE),
(MAX_253 + pallas::Base::ONE, MAX_253 + pallas::Base::ONE),
(MAX_253 + pallas::Base::ONE, MAX_253 + pallas::Base::from(2)),
];
for i in 0..valid_a_vals.len() {
let a = valid_a_vals[i];
let b = valid_b_vals[i];
for (a, b) in valid_pairs {
println!("253 bit (valid) {:?} < {:?} check", a, b);
let circuit = LessThanCircuit { a: Value::known(a), b: Value::known(b) };
let prover = MockProver::run(k, &circuit, vec![]).unwrap();
prover.assert_satisfied();
}
for i in 0..invalid_a_vals.len() {
let a = invalid_a_vals[i];
let b = invalid_b_vals[i];
for (a, b) in invalid_pairs {
println!("253 bit (invalid) {:?} < {:?} check", a, b);
let circuit = LessThanCircuit { a: Value::known(a), b: Value::known(b) };
let prover = MockProver::run(k, &circuit, vec![]).unwrap();
assert!(prover.verify().is_err())
}