Function test_nested_iterate

ciphercore-base/src/inline/inline_ops.rs:724–784

()

    #[test]
    fn test_nested_iterate() {
        let helper = |mode| -> Result<()> {
            let c = create_context()?;

            let g0 = c.create_graph()?;
            {
                let i01 = g0.input(scalar_type(BIT))?;
                let i02 = g0.input(scalar_type(BIT))?;
                let sum12 = g0.add(i01.clone(), i02.clone())?;
                let out = g0.create_tuple(vec![sum12.clone(), sum12.clone()])?;
                g0.set_output_node(out)?;
                g0.finalize()?;
            }
            let g1 = c.create_graph()?;
            {
                let i21 = g1.input(scalar_type(BIT))?;
                let i22 = g1.input(scalar_type(BIT))?;
                let i23 = i21.clone();
                let o1 = g1.tuple_get(g1.call(g0.clone(), vec![i21, i22])?, 0)?;
                let input_vec = g1.repeat(i23, 5)?;
                let o2 = g1.iterate(g0.clone(), o1, input_vec)?;
                g1.set_output_node(o2.clone())?;
                g1.finalize()?;
            }
            let g2 = c.create_graph()?;
            {
                let i21 = g2.input(scalar_type(BIT))?;
                let i22 = g2.input(scalar_type(BIT))?;
                let i23 = g2.input(scalar_type(BIT))?;
                let o1 = g2.tuple_get(g2.call(g0.clone(), vec![i21.clone(), i22.clone()])?, 0)?;
                let input_vec = g2.create_vector(
                    scalar_type(BIT),
                    vec![
                        i21.clone(),
                        i22.clone(),
                        i23.clone(),
                        o1.clone(),
                        o1.clone(),
                        o1.clone(),
                    ],
                )?;
                let o2 = g2.iterate(g1.clone(), o1, input_vec)?;
                g2.set_output_node(o2.clone())?;
                g2.finalize()?;
            }
            c.set_main_graph(g2.clone())?;
            c.finalize()?;
            let c_out = inline_operations(
                c.clone(),
                InlineConfig {
                    default_mode: mode,
                    ..Default::default()
                },
            )?;
            assert_eq!(c_out.get_graphs().len(), 1);
            verify_on_all_inputs(g2.clone(), c_out.get_main_graph()?)?;
            Ok(())
        };