@@ -2541,17 +2541,17 @@ fn test_remove_index_oob3() {
2541
2541
}
2542
2542
2543
2543
#[ test]
2544
- fn test_split_re_im_view ( ) {
2544
+ fn test_split_complex_view ( ) {
2545
2545
let a = Array3 :: from_shape_fn ( ( 3 , 4 , 5 ) , |( i, j, k) | {
2546
2546
Complex :: < f32 > :: new ( i as f32 * j as f32 , k as f32 )
2547
2547
} ) ;
2548
- let Complex { re, im } = a. view ( ) . split_re_im ( ) ;
2548
+ let Complex { re, im } = a. view ( ) . split_complex ( ) ;
2549
2549
assert_relative_eq ! ( re. sum( ) , 90. ) ;
2550
2550
assert_relative_eq ! ( im. sum( ) , 120. ) ;
2551
2551
}
2552
2552
2553
2553
#[ test]
2554
- fn test_split_re_im_view_roundtrip ( ) {
2554
+ fn test_split_complex_view_roundtrip ( ) {
2555
2555
let a_re = Array3 :: from_shape_fn ( ( 3 , 1 , 5 ) , |( i, j, _k) | {
2556
2556
i * j
2557
2557
} ) ;
@@ -2561,49 +2561,49 @@ fn test_split_re_im_view_roundtrip() {
2561
2561
let a = Array3 :: from_shape_fn ( ( 3 , 1 , 5 ) , |( i, j, k) | {
2562
2562
Complex :: new ( a_re[ [ i, j, k] ] , a_im[ [ i, j, k] ] )
2563
2563
} ) ;
2564
- let Complex { re, im } = a. view ( ) . split_re_im ( ) ;
2564
+ let Complex { re, im } = a. view ( ) . split_complex ( ) ;
2565
2565
assert_eq ! ( a_re, re) ;
2566
2566
assert_eq ! ( a_im, im) ;
2567
2567
}
2568
2568
2569
2569
#[ test]
2570
- fn test_split_re_im_view_mut ( ) {
2570
+ fn test_split_complex_view_mut ( ) {
2571
2571
let eye_scalar = Array2 :: < u32 > :: eye ( 4 ) ;
2572
2572
let eye_complex = Array2 :: < Complex < u32 > > :: eye ( 4 ) ;
2573
2573
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 ( ) ;
2575
2575
re. assign ( & eye_scalar) ;
2576
2576
assert_eq ! ( im. sum( ) , 0 ) ;
2577
2577
assert_eq ! ( a, eye_complex) ;
2578
2578
}
2579
2579
2580
2580
#[ test]
2581
- fn test_split_re_im_zerod ( ) {
2581
+ fn test_split_complex_zerod ( ) {
2582
2582
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 ( ) ;
2584
2584
assert_eq ! ( re. get( ( ) ) , Some ( & 42 ) ) ;
2585
2585
assert_eq ! ( im. get( ( ) ) , Some ( & 32 ) ) ;
2586
- let cmplx = a. view_mut ( ) . split_re_im ( ) ;
2586
+ let cmplx = a. view_mut ( ) . split_complex ( ) ;
2587
2587
cmplx. re . assign_to ( cmplx. im ) ;
2588
2588
assert_eq ! ( a. get( ( ) ) . unwrap( ) . im, 42 ) ;
2589
2589
}
2590
2590
2591
2591
#[ test]
2592
- fn test_split_re_im_permuted ( ) {
2592
+ fn test_split_complex_permuted ( ) {
2593
2593
let a = Array3 :: from_shape_fn ( ( 3 , 4 , 5 ) , |( i, j, k) | {
2594
2594
Complex :: new ( i * k + j, k)
2595
2595
} ) ;
2596
2596
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 ( ) ;
2598
2598
assert_eq ! ( re. get( ( 3 , 2 , 4 ) ) . unwrap( ) , & 11 ) ;
2599
2599
assert_eq ! ( im. get( ( 3 , 2 , 4 ) ) . unwrap( ) , & 4 ) ;
2600
2600
}
2601
2601
2602
2602
#[ test]
2603
- fn test_split_re_im_invert_axis ( ) {
2603
+ fn test_split_complex_invert_axis ( ) {
2604
2604
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 ) ) ;
2605
2605
a. invert_axis ( Axis ( 1 ) ) ;
2606
- let cmplx = a. view ( ) . split_re_im ( ) ;
2606
+ let cmplx = a. view ( ) . split_complex ( ) ;
2607
2607
assert_eq ! ( cmplx. re, a. mapv( |z| z. re) ) ;
2608
2608
assert_eq ! ( cmplx. im, a. mapv( |z| z. im) ) ;
2609
2609
}
0 commit comments