@@ -2541,17 +2541,17 @@ fn test_remove_index_oob3() {
25412541}
25422542
25432543#[ test]
2544- fn test_split_re_im_view ( ) {
2544+ fn test_split_complex_view ( ) {
25452545 let a = Array3 :: from_shape_fn ( ( 3 , 4 , 5 ) , |( i, j, k) | {
25462546 Complex :: < f32 > :: new ( i as f32 * j as f32 , k as f32 )
25472547 } ) ;
2548- let Complex { re, im } = a. view ( ) . split_re_im ( ) ;
2548+ let Complex { re, im } = a. view ( ) . split_complex ( ) ;
25492549 assert_relative_eq ! ( re. sum( ) , 90. ) ;
25502550 assert_relative_eq ! ( im. sum( ) , 120. ) ;
25512551}
25522552
25532553#[ test]
2554- fn test_split_re_im_view_roundtrip ( ) {
2554+ fn test_split_complex_view_roundtrip ( ) {
25552555 let a_re = Array3 :: from_shape_fn ( ( 3 , 1 , 5 ) , |( i, j, _k) | {
25562556 i * j
25572557 } ) ;
@@ -2561,49 +2561,49 @@ fn test_split_re_im_view_roundtrip() {
25612561 let a = Array3 :: from_shape_fn ( ( 3 , 1 , 5 ) , |( i, j, k) | {
25622562 Complex :: new ( a_re[ [ i, j, k] ] , a_im[ [ i, j, k] ] )
25632563 } ) ;
2564- let Complex { re, im } = a. view ( ) . split_re_im ( ) ;
2564+ let Complex { re, im } = a. view ( ) . split_complex ( ) ;
25652565 assert_eq ! ( a_re, re) ;
25662566 assert_eq ! ( a_im, im) ;
25672567}
25682568
25692569#[ test]
2570- fn test_split_re_im_view_mut ( ) {
2570+ fn test_split_complex_view_mut ( ) {
25712571 let eye_scalar = Array2 :: < u32 > :: eye ( 4 ) ;
25722572 let eye_complex = Array2 :: < Complex < u32 > > :: eye ( 4 ) ;
25732573 let mut a = Array2 :: < Complex < u32 > > :: zeros ( ( 4 , 4 ) ) ;
2574- let Complex { mut re, im } = a. view_mut ( ) . split_re_im ( ) ;
2574+ let Complex { mut re, im } = a. view_mut ( ) . split_complex ( ) ;
25752575 re. assign ( & eye_scalar) ;
25762576 assert_eq ! ( im. sum( ) , 0 ) ;
25772577 assert_eq ! ( a, eye_complex) ;
25782578}
25792579
25802580#[ test]
2581- fn test_split_re_im_zerod ( ) {
2581+ fn test_split_complex_zerod ( ) {
25822582 let mut a = Array0 :: from_elem ( ( ) , Complex :: new ( 42 , 32 ) ) ;
2583- let Complex { re, im } = a. view ( ) . split_re_im ( ) ;
2583+ let Complex { re, im } = a. view ( ) . split_complex ( ) ;
25842584 assert_eq ! ( re. get( ( ) ) , Some ( & 42 ) ) ;
25852585 assert_eq ! ( im. get( ( ) ) , Some ( & 32 ) ) ;
2586- let cmplx = a. view_mut ( ) . split_re_im ( ) ;
2586+ let cmplx = a. view_mut ( ) . split_complex ( ) ;
25872587 cmplx. re . assign_to ( cmplx. im ) ;
25882588 assert_eq ! ( a. get( ( ) ) . unwrap( ) . im, 42 ) ;
25892589}
25902590
25912591#[ test]
2592- fn test_split_re_im_permuted ( ) {
2592+ fn test_split_complex_permuted ( ) {
25932593 let a = Array3 :: from_shape_fn ( ( 3 , 4 , 5 ) , |( i, j, k) | {
25942594 Complex :: new ( i * k + j, k)
25952595 } ) ;
25962596 let permuted = a. view ( ) . permuted_axes ( [ 1 , 0 , 2 ] ) ;
2597- let Complex { re, im } = permuted. split_re_im ( ) ;
2597+ let Complex { re, im } = permuted. split_complex ( ) ;
25982598 assert_eq ! ( re. get( ( 3 , 2 , 4 ) ) . unwrap( ) , & 11 ) ;
25992599 assert_eq ! ( im. get( ( 3 , 2 , 4 ) ) . unwrap( ) , & 4 ) ;
26002600}
26012601
26022602#[ test]
2603- fn test_split_re_im_invert_axis ( ) {
2603+ fn test_split_complex_invert_axis ( ) {
26042604 let mut a = Array :: from_shape_fn ( ( 2 , 3 , 2 ) , |( i, j, k) | Complex :: new ( i as f64 + j as f64 , i as f64 + k as f64 ) ) ;
26052605 a. invert_axis ( Axis ( 1 ) ) ;
2606- let cmplx = a. view ( ) . split_re_im ( ) ;
2606+ let cmplx = a. view ( ) . split_complex ( ) ;
26072607 assert_eq ! ( cmplx. re, a. mapv( |z| z. re) ) ;
26082608 assert_eq ! ( cmplx. im, a. mapv( |z| z. im) ) ;
26092609}
0 commit comments