use crate::{fields::FieldElement, rand::SeedableRng};
use rand::{Rng, rngs::StdRng};
fn can_invert<F: FieldElement>() {
let mut a = F::one();
for _ in 0..10000 {
assert_eq!(a * a.inverse().unwrap(), F::one());
a = a + F::one();
}
a = -F::one();
for _ in 0..10000 {
assert_eq!(a * a.inverse().unwrap(), F::one());
a = a - F::one();
}
assert_eq!(F::zero().inverse(), None);
}
fn rand_element_eval<F: FieldElement, R: Rng>(rng: &mut R) {
for _ in 0..100 {
let a = F::random(rng);
let b = F::random(rng);
let c = F::random(rng);
let d = F::random(rng);
assert_eq!((a + b) * (c + d), (a * c) + (b * c) + (a * d) + (b * d));
}
}
fn rand_element_squaring<F: FieldElement, R: Rng>(rng: &mut R) {
for _ in 0..100 {
let a = F::random(rng);
assert!(a * a == a.squared());
}
let mut cur = F::zero();
for _ in 0..100 {
assert_eq!(cur.squared(), cur * cur);
cur = cur + F::one();
}
}
fn rand_element_addition_and_negation<F: FieldElement, R: Rng>(rng: &mut R) {
for _ in 0..100 {
let a = F::random(rng);
assert_eq!(a + (-a), F::zero());
}
for _ in 0..100 {
let mut a = F::random(rng);
let r = F::random(rng);
let mut b = a + r;
for _ in 0..10 {
let r = F::random(rng);
a = a + r;
b = b + r;
let r = F::random(rng);
a = a - r;
b = b - r;
let r = F::random(rng);
a = a + (-(-r));
b = b + (-(-r));
let r = F::random(rng);
a = a - r;
b = b + (-r);
let r = F::random(rng);
a = a + (-r);
b = b - r;
}
b = b - r;
assert_eq!(a, b);
}
}
fn rand_element_inverse<F: FieldElement, R: Rng>(rng: &mut R) {
for _ in 0..10000 {
let a = F::random(rng);
assert!(a.inverse().unwrap() * a == F::one());
let b = F::random(rng);
assert_eq!((a * b) * (a.inverse().unwrap()), b);
}
}
fn rand_element_multiplication<F: FieldElement, R: Rng>(rng: &mut R) {
for _ in 0..250 {
let a = F::random(rng);
let b = F::random(rng);
let c = F::random(rng);
assert_eq!((a * b) * c, a * (b * c));
}
}
pub fn field_trials<F: FieldElement>() {
can_invert::<F>();
assert_eq!(-F::zero(), F::zero());
assert_eq!(-F::one() + F::one(), F::zero());
assert_eq!(F::zero() - F::zero(), F::zero());
let seed = [
0, 0, 0, 0, 0, 0, 64, 13, 0, 0, 0, 0, 0, 0, 176, 2, 0, 0, 0, 0, 0, 0, 0, 13, 0, 0, 0, 0, 0, 0, 96, 7u8, ];
let mut rng = StdRng::from_seed(seed);
rand_element_squaring::<F, StdRng>(&mut rng);
rand_element_addition_and_negation::<F, StdRng>(&mut rng);
rand_element_multiplication::<F, StdRng>(&mut rng);
rand_element_inverse::<F, StdRng>(&mut rng);
rand_element_eval::<F, StdRng>(&mut rng);
}