Function test_accumulate_rows

bitvm/src/bn254/msm.rs:432–474

()

    #[test]
    fn test_accumulate_rows() {
        let mut prng = ChaCha20Rng::seed_from_u64(1);
        let q = ark_bn254::G1Affine::rand(&mut prng);
        let fq = ark_bn254::Fr::rand(&mut prng);
        let window = WINDOW_G1_MSM as usize;
        let mut prev = ark_bn254::G1Affine::identity();
        let all_rows = accumulate_addition_chain_for_a_scalar_mul(prev, q, fq, window);

        let expected_msm = (q * fq).into_affine();
        let calculated_msm = all_rows[all_rows.len() - 1].0;
        assert_eq!(expected_msm, calculated_msm);

        for (row_out, row_scr, row_hints) in all_rows {
            let tap_len = row_scr.len();
            let scr = script! {
                // [hints, t, scalar]
                for h in &row_hints {
                    {h.push()}
                }
                {G1Affine::push(prev)}
                {Fr::push(fq)}
                {row_scr}
                {G1Affine::push(row_out)}
                {G1Affine::equalverify()}
                {G1Affine::push(prev) }
                {G1Affine::equalverify()}
                OP_TRUE
            };

            let res = execute_script(scr);
            if res.final_stack.len() > 1 {
                for i in 0..res.final_stack.len() {
                    println!("{i:} {:?}", res.final_stack.get(i));
                }
            }
            prev = row_out;
            assert!(res.success);
            println!(
                "accumulate_addition_terms {:?} max_stat {:?}",
                tap_len, res.stats.max_nb_stack_items
            );
        }
    }

    #[test]
    fn test_accumulate_multiple_rows() {