Function test_gemm_ones

tests/math.rs:98–106

()

    // Test GEMM with one matrix
    #[test]
    fn test_gemm_ones() {
        let matrix_a = one_matrix(VECTOR_SIZE);
        let matrix_b = one_matrix(VECTOR_SIZE);
        let mut result_matrix = zero_matrix(VECTOR_SIZE);
        gemm::gemm(&matrix_a, &matrix_b, &mut result_matrix);
        let expected_result: Vec<[f32; VECTOR_SIZE]> =
            vec![[VECTOR_SIZE as f32; VECTOR_SIZE]; VECTOR_SIZE];
        assert_eq!(result_matrix, expected_result);
    }
}