@@ -5,12 +5,12 @@ use ndarray::prelude::*;
55fn broadcast_1 ( ) {
66 let a_dim = Dim ( [ 2 , 4 , 2 , 2 ] ) ;
77 let b_dim = Dim ( [ 2 , 1 , 2 , 1 ] ) ;
8- let a = ArcArray :: linspace ( 0. , 1. , a_dim. size ( ) ) . reshape ( a_dim) ;
9- let b = ArcArray :: linspace ( 0. , 1. , b_dim. size ( ) ) . reshape ( b_dim) ;
8+ let a = ArcArray :: linspace ( 0. , 1. , a_dim. size ( ) ) . into_shape ( a_dim) . unwrap ( ) ;
9+ let b = ArcArray :: linspace ( 0. , 1. , b_dim. size ( ) ) . into_shape ( b_dim) . unwrap ( ) ;
1010 assert ! ( b. broadcast( a. dim( ) ) . is_some( ) ) ;
1111
1212 let c_dim = Dim ( [ 2 , 1 ] ) ;
13- let c = ArcArray :: linspace ( 0. , 1. , c_dim. size ( ) ) . reshape ( c_dim) ;
13+ let c = ArcArray :: linspace ( 0. , 1. , c_dim. size ( ) ) . into_shape ( c_dim) . unwrap ( ) ;
1414 assert ! ( c. broadcast( 1 ) . is_none( ) ) ;
1515 assert ! ( c. broadcast( ( ) ) . is_none( ) ) ;
1616 assert ! ( c. broadcast( ( 2 , 1 ) ) . is_some( ) ) ;
@@ -31,8 +31,8 @@ fn broadcast_1() {
3131fn test_add ( ) {
3232 let a_dim = Dim ( [ 2 , 4 , 2 , 2 ] ) ;
3333 let b_dim = Dim ( [ 2 , 1 , 2 , 1 ] ) ;
34- let mut a = ArcArray :: linspace ( 0.0 , 1. , a_dim. size ( ) ) . reshape ( a_dim) ;
35- let b = ArcArray :: linspace ( 0.0 , 1. , b_dim. size ( ) ) . reshape ( b_dim) ;
34+ let mut a = ArcArray :: linspace ( 0.0 , 1. , a_dim. size ( ) ) . into_shape ( a_dim) . unwrap ( ) ;
35+ let b = ArcArray :: linspace ( 0.0 , 1. , b_dim. size ( ) ) . into_shape ( b_dim) . unwrap ( ) ;
3636 a += & b;
3737 let t = ArcArray :: from_elem ( ( ) , 1.0f32 ) ;
3838 a += & t;
@@ -43,7 +43,7 @@ fn test_add() {
4343#[ cfg( feature = "std" ) ]
4444fn test_add_incompat ( ) {
4545 let a_dim = Dim ( [ 2 , 4 , 2 , 2 ] ) ;
46- let mut a = ArcArray :: linspace ( 0.0 , 1. , a_dim. size ( ) ) . reshape ( a_dim) ;
46+ let mut a = ArcArray :: linspace ( 0.0 , 1. , a_dim. size ( ) ) . into_shape ( a_dim) . unwrap ( ) ;
4747 let incompat = ArcArray :: from_elem ( 3 , 1.0f32 ) ;
4848 a += & incompat;
4949}
0 commit comments