Function half_f16_works

tests/array.rs:567–601

()

#[cfg(feature = "half")]
#[test]
fn half_f16_works() {
    Python::attach(|py| {
        let np = py.eval(c_str!("__import__('numpy')"), None, None).unwrap();
        let locals = [("np", &np)].into_py_dict(py).unwrap();

        let array = py
            .eval(
                c_str!("np.array([[1, 2], [3, 4]], dtype='float16')"),
                None,
                Some(&locals),
            )
            .unwrap()
            .cast_into::<PyArray2<f16>>()
            .unwrap();

        assert_eq!(
            array.readonly().as_array(),
            array![
                [f16::from_f32(1.0), f16::from_f32(2.0)],
                [f16::from_f32(3.0), f16::from_f32(4.0)]
            ]
        );

        array
            .readwrite()
            .as_array_mut()
            .map_inplace(|value| *value *= f16::from_f32(2.0));

        py_run!(
            py,
            array np,
            "assert np.all(array == np.array([[2, 4], [6, 8]], dtype='float16'))"
        );
    });
}

#[cfg(feature = "half")]
#[test]