Function test_vector

ciphercore-base/src/ops/min_max.rs:257–298

()

    #[test]
    fn test_vector() {
        || -> Result<()> {
            let context = || -> Result<Context> {
                let c = simple_context(|g| {
                    let i1 = g.input(array_type(vec![1, 3, 64], BIT))?;
                    let i2 = g.input(array_type(vec![3, 1, 64], BIT))?;
                    g.create_tuple(vec![
                        g.custom_op(
                            CustomOperation::new(Min {
                                signed_comparison: false,
                            }),
                            vec![i1.clone(), i2.clone()],
                        )?,
                        g.custom_op(
                            CustomOperation::new(Max {
                                signed_comparison: false,
                            }),
                            vec![i1.clone(), i2.clone()],
                        )?,
                    ])
                })?;
                let mapped_c = run_instantiation_pass(c)?;
                Ok(mapped_c.get_context())
            }()?;
            let a = vec![0, 30, 100];
            let b = vec![10, 50, 150];
            let v = random_evaluate(
                context.get_main_graph()?,
                vec![
                    Value::from_flattened_array(&a, UINT64)?,
                    Value::from_flattened_array(&b, UINT64)?,
                ],
            )?
            .to_vector()?;
            let min_a_b = v[0].to_flattened_array_u64(array_type(vec![3, 3], UINT64))?;
            let max_a_b = v[1].to_flattened_array_u64(array_type(vec![3, 3], UINT64))?;
            assert_eq!(min_a_b, vec![0, 10, 10, 0, 30, 50, 0, 30, 100]);
            assert_eq!(max_a_b, vec![10, 30, 100, 50, 50, 100, 150, 150, 150]);
            Ok(())
        }()
        .unwrap();
    }
}