Function test_cot_f32

datafusion/functions/src/math/cot.rs:162–199

()

    #[test]
    fn test_cot_f32() {
        let array = Arc::new(Float32Array::from(vec![12.1, 30.0, 90.0, -30.0]));
        let arg_fields = vec![Field::new("a", DataType::Float32, false).into()];
        let args = ScalarFunctionArgs {
            args: vec![ColumnarValue::Array(Arc::clone(&array) as ArrayRef)],
            arg_fields,
            number_rows: array.len(),
            return_field: Field::new("f", DataType::Float32, true).into(),
            config_options: Arc::new(ConfigOptions::default()),
        };
        let result = CotFunc::new()
            .invoke_with_args(args)
            .expect("failed to initialize function cot");

        match result {
            ColumnarValue::Array(arr) => {
                let floats = as_float32_array(&arr)
                    .expect("failed to convert result to a Float32Array");

                let expected = Float32Array::from(vec![
                    -1.986_460_4,
                    -0.156_119_96,
                    -0.501_202_8,
                    0.156_119_96,
                ]);

                let eps = 1e-6;
                assert_eq!(floats.len(), 4);
                assert!((floats.value(0) - expected.value(0)).abs() < eps);
                assert!((floats.value(1) - expected.value(1)).abs() < eps);
                assert!((floats.value(2) - expected.value(2)).abs() < eps);
                assert!((floats.value(3) - expected.value(3)).abs() < eps);
            }
            ColumnarValue::Scalar(_) => {
                panic!("Expected an array value")
            }
        }
    }

    #[test]
    fn test_cot_f64() {