Function test_emulate_external_hash

bitvm/src/chunk/helpers.rs:204–250

()

    #[test]
    #[allow(clippy::needless_range_loop)]
    fn test_emulate_external_hash() {
        fn emulate_extern_hash_fps_scripted(msgs: Vec<ark_bn254::Fq>) -> [u8; 64] {
            assert!(
                msgs.len() == 4
                    || msgs.len() == 2
                    || msgs.len() == 12
                    || msgs.len() == 14
                    || msgs.len() == 6
                    || msgs.len() == 8
            );

            let scr = script! {
                for i in 0..msgs.len() {
                    {Fq::push(msgs[i])}
                }
                if msgs.len() == 4 {
                    {hash_fp4()}
                } else if msgs.len() == 2 {
                    {hash_fp2()}
                } else if msgs.len() == 6 {
                    {hash_fp6()}
                } else if msgs.len() == 14 {
                    {hash_fp14()}
                }
                {unpack_limbs_to_nibbles()}
            };
            let exec_result = execute_script(scr.clone());
            let mut arr = [0u8; 64];
            for i in 0..exec_result.final_stack.len() {
                let v = exec_result.final_stack.get(i);
                if v.is_empty() {
                    arr[i] = 0;
                } else {
                    arr[i] = v[0];
                }
            }
            arr
        }

        let mut prng = ChaCha20Rng::seed_from_u64(0);
        let _f = ark_bn254::Fq12::rand(&mut prng);

        let ps = vec![ark_bn254::Fq::ONE + ark_bn254::Fq::ONE; 14];
        let res = emulate_extern_hash_fps_scripted(ps.clone());
        let res2 = extern_hash_fps(ps);
        assert_eq!(res, res2);
    }
}