use super::{blocks::Dyn, Deserializable, Digest, Felt, Kernel, ProgramInfo, Serializable};
use crate::{chiplets::hasher, Word};
use proptest::prelude::*;
use rand_utils::prng_array;
#[test]
fn dyn_hash_is_correct() {
let expected_constant =
hasher::merge_in_domain(&[Digest::default(), Digest::default()], Dyn::DOMAIN);
assert_eq!(expected_constant, Dyn::new().hash());
}
proptest! {
#[test]
fn arbitrary_program_info_serialization_works(
kernel_count in prop::num::u8::ANY,
ref seed in any::<[u8; 32]>()
) {
let program_hash = digest_from_seed(*seed);
let kernel: Vec<Digest> = (0..kernel_count)
.scan(*seed, |seed, _| {
*seed = prng_array(*seed);
Some(digest_from_seed(*seed))
})
.collect();
let kernel = Kernel::new(&kernel);
let program_info = ProgramInfo::new(program_hash, kernel);
let bytes = program_info.to_bytes();
let deser = ProgramInfo::read_from_bytes(&bytes).unwrap();
assert_eq!(program_info, deser);
}
}
fn digest_from_seed(seed: [u8; 32]) -> Digest {
let mut digest = Word::default();
digest.iter_mut().enumerate().for_each(|(i, d)| {
*d = <[u8; 8]>::try_from(&seed[i * 8..(i + 1) * 8])
.map(u64::from_le_bytes)
.map(Felt::new)
.unwrap()
});
digest.into()
}