Function test_inline_iterate

ciphercore-base/src/inline/inline_ops.rs:672–721

()

    #[test]
    fn test_inline_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()])?;
                out.set_name("CreateTuple output")?;
                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 = g1.input(scalar_type(BIT))?;
                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)?;
                o2.set_name("Iterate output")?;
                let o3 = g1.tuple_get(o2, 0)?;
                g1.set_output_node(o3.clone())?;
                g1.finalize()?;
            }
            c.set_main_graph(g1.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(g1.clone(), c_out.get_main_graph()?)?;
            // Check names
            let inlined_g = c_out.get_main_graph()?;
            let named_iterate_result = c_out.retrieve_node(inlined_g.clone(), "Iterate output");
            assert!(named_iterate_result.is_ok());
            let named_iterate = named_iterate_result?;
            assert_eq!(named_iterate.get_operation(), Operation::CreateTuple);
            Ok(())
        };
        helper(InlineMode::Simple).unwrap();
        helper(InlineMode::DepthOptimized(DepthOptimizationLevel::Extreme)).unwrap();
    }

    #[test]
    fn test_nested_iterate() {