11extern crate blas_src;
2- use ndarray:: { Array1 , Array2 , ArrayD , linalg :: Dot , Ix1 , Ix2 } ;
2+ use ndarray:: { linalg :: Dot , Array1 , Array2 , ArrayD , Ix1 , Ix2 } ;
33
44#[ test]
5- fn test_arrayd_dot_2d ( ) {
5+ fn test_arrayd_dot_2d ( )
6+ {
67 let mat1 = ArrayD :: from_shape_vec ( vec ! [ 3 , 2 ] , vec ! [ 3.0 ; 6 ] ) . unwrap ( ) ;
78 let mat2 = ArrayD :: from_shape_vec ( vec ! [ 2 , 3 ] , vec ! [ 1.0 ; 6 ] ) . unwrap ( ) ;
89
@@ -21,7 +22,8 @@ fn test_arrayd_dot_2d() {
2122}
2223
2324#[ test]
24- fn test_arrayd_dot_1d ( ) {
25+ fn test_arrayd_dot_1d ( )
26+ {
2527 // Test 1D array dot product
2628 let vec1 = ArrayD :: from_shape_vec ( vec ! [ 3 ] , vec ! [ 1.0 , 2.0 , 3.0 ] ) . unwrap ( ) ;
2729 let vec2 = ArrayD :: from_shape_vec ( vec ! [ 3 ] , vec ! [ 4.0 , 5.0 , 6.0 ] ) . unwrap ( ) ;
@@ -36,7 +38,8 @@ fn test_arrayd_dot_1d() {
3638
3739#[ test]
3840#[ should_panic( expected = "Dot product for ArrayD is only supported for 1D and 2D arrays" ) ]
39- fn test_arrayd_dot_3d ( ) {
41+ fn test_arrayd_dot_3d ( )
42+ {
4043 // Test that 3D arrays are not supported
4144 let arr1 = ArrayD :: from_shape_vec ( vec ! [ 2 , 2 , 2 ] , vec ! [ 1.0 ; 8 ] ) . unwrap ( ) ;
4245 let arr2 = ArrayD :: from_shape_vec ( vec ! [ 2 , 2 , 2 ] , vec ! [ 1.0 ; 8 ] ) . unwrap ( ) ;
@@ -46,7 +49,8 @@ fn test_arrayd_dot_3d() {
4649
4750#[ test]
4851#[ should_panic( expected = "ndarray: inputs 2 × 3 and 4 × 5 are not compatible for matrix multiplication" ) ]
49- fn test_arrayd_dot_incompatible_dims ( ) {
52+ fn test_arrayd_dot_incompatible_dims ( )
53+ {
5054 // Test arrays with incompatible dimensions
5155 let arr1 = ArrayD :: from_shape_vec ( vec ! [ 2 , 3 ] , vec ! [ 1.0 ; 6 ] ) . unwrap ( ) ;
5256 let arr2 = ArrayD :: from_shape_vec ( vec ! [ 4 , 5 ] , vec ! [ 1.0 ; 20 ] ) . unwrap ( ) ;
@@ -55,7 +59,8 @@ fn test_arrayd_dot_incompatible_dims() {
5559}
5660
5761#[ test]
58- fn test_arrayd_dot_matrix_vector ( ) {
62+ fn test_arrayd_dot_matrix_vector ( )
63+ {
5964 // Test matrix-vector multiplication
6065 let mat = ArrayD :: from_shape_vec ( vec ! [ 3 , 2 ] , vec ! [ 1.0 , 2.0 , 3.0 , 4.0 , 5.0 , 6.0 ] ) . unwrap ( ) ;
6166 let vec = ArrayD :: from_shape_vec ( vec ! [ 2 ] , vec ! [ 1.0 , 2.0 ] ) . unwrap ( ) ;
@@ -72,4 +77,4 @@ fn test_arrayd_dot_matrix_vector() {
7277 let expected = mat_2d. dot ( & vec_1d) ;
7378
7479 assert_eq ! ( result. into_dimensionality:: <Ix1 >( ) . unwrap( ) , expected) ;
75- }
80+ }
0 commit comments