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