logos-co
/
nomos-node
mirror of
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use domain elements instead of roots of unity in tests

This commit is contained in:
Daniel Sanchez Quiros 2024-04-01 14:59:06 +02:00
parent 38fe0e8171
commit 6d438d5a18
1 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
nomos-da/kzgrs/src/common.rs
View File

@ -1,5 +1,6 @@
// std // std
use std::io::BufReader; use std::io::{BufReader, Cursor};
use std::thread::current;
// crates // crates
use ark_bls12_381::{fr::Fr, Bls12_381}; use ark_bls12_381::{fr::Fr, Bls12_381};
use ark_ec::pairing::Pairing; use ark_ec::pairing::Pairing;
@ -8,7 +9,7 @@ use ark_poly::domain::general::GeneralEvaluationDomain;
use ark_poly::evaluations::univariate::Evaluations; use ark_poly::evaluations::univariate::Evaluations;
use ark_poly::univariate::DensePolynomial; use ark_poly::univariate::DensePolynomial;
use ark_poly::{DenseUVPolynomial, EvaluationDomain, Polynomial}; use ark_poly::{DenseUVPolynomial, EvaluationDomain, Polynomial};
use ark_serialize::CanonicalDeserialize; use ark_serialize::{CanonicalDeserialize, CanonicalSerialize};
use thiserror::Error; use thiserror::Error;
// internal // internal
@ -33,10 +34,9 @@ fn bytes_to_evaluations<const CHUNK_SIZE: usize>(
.map(|e| { .map(|e| {
let mut buff = [0u8; 32]; let mut buff = [0u8; 32];
buff[0..31].copy_from_slice(e); buff[0..31].copy_from_slice(e);
Fr::new_unchecked( let bint = BigInteger256::deserialize_uncompressed(BufReader::new(buff.as_slice()))
BigInteger256::deserialize_uncompressed(BufReader::new(buff.as_slice())) .unwrap();
.unwrap(), Fr::new(bint)
)
}) })
.collect(), .collect(),
domain, domain,
@ -63,13 +63,16 @@ pub fn bytes_to_polynomial<const CHUNK_SIZE: usize>(
#[cfg(test)] #[cfg(test)]
mod test { mod test {
use super::{bytes_to_polynomial, KzgRsError}; use super::{bytes_to_evaluations, bytes_to_polynomial, KzgRsError};
use ark_bls12_381::fr::Fr; use ark_bls12_381::fr::Fr;
use ark_ec::pairing::Pairing; use ark_ec::pairing::Pairing;
use ark_ff::{BigInteger, PrimeField}; use ark_ff::{BigInteger, FftField, PrimeField};
use ark_poly::{EvaluationDomain, GeneralEvaluationDomain, Polynomial}; use ark_poly::{EvaluationDomain, GeneralEvaluationDomain, Polynomial};
use ark_serialize::CanonicalSerialize;
use once_cell::sync::Lazy; use once_cell::sync::Lazy;
use rand::{thread_rng, Fill}; use rand::{thread_rng, Fill};
use std::io::{BufWriter, Cursor};
use std::ops::Deref;
const CHUNK_SIZE: usize = 31; const CHUNK_SIZE: usize = 31;
static DOMAIN: Lazy<GeneralEvaluationDomain<Fr>> = static DOMAIN: Lazy<GeneralEvaluationDomain<Fr>> =
@ -80,15 +83,12 @@ mod test {
let mut bytes: [u8; CHUNK_SIZE * N] = [0; CHUNK_SIZE * N]; let mut bytes: [u8; CHUNK_SIZE * N] = [0; CHUNK_SIZE * N];
let mut rng = thread_rng(); let mut rng = thread_rng();
bytes.try_fill(&mut rng).unwrap(); bytes.try_fill(&mut rng).unwrap();
let evals = bytes_to_evaluations::<31>(&bytes, *DOMAIN);
let poly = bytes_to_polynomial::<31>(&bytes, *DOMAIN).unwrap(); let poly = bytes_to_polynomial::<31>(&bytes, *DOMAIN).unwrap();
for (i, e) in (0..100).map(|i| 2u32.pow(i)).enumerate() { for i in (0..32) {
let eval_point = Fr::from(e); let eval_point = DOMAIN.element(i);
let point = poly.evaluate(&eval_point); let point = poly.evaluate(&eval_point);
let bint = point.into_bigint(); assert_eq!(evals[i as usize], point);
assert_eq!(
&bytes[CHUNK_SIZE * i..CHUNK_SIZE * i + CHUNK_SIZE],
&bint.to_bytes_be()
)
} }
} }