@@ -660,7 +660,7 @@ def test_fft_usm_ndarray(self, n):
660
660
x = dpt .linspace (- 1 , 1 , 11 )
661
661
a_usm = dpt .asarray (dpt .sin (x ))
662
662
a_np = dpt .asnumpy (a_usm )
663
- out_shape = a_usm .shape [0 ] // 2 + 1 if n is None else n // 2
663
+ out_shape = a_usm .shape [0 ] // 2 + 1 if n is None else n // 2 + 1
664
664
out_dtype = map_dtype_to_device (dpnp .complex128 , a_usm .sycl_device )
665
665
out = dpt .empty (out_shape , dtype = out_dtype )
666
666
@@ -678,7 +678,7 @@ def test_fft_1D_out(self, dtype, n, norm):
678
678
a = dpnp .asarray (a , dtype = dtype )
679
679
a_np = dpnp .asnumpy (a )
680
680
681
- out_shape = a .shape [0 ] // 2 + 1 if n is None else n // 2
681
+ out_shape = a .shape [0 ] // 2 + 1 if n is None else n // 2 + 1
682
682
out_dtype = dpnp .complex64 if dtype == dpnp .float32 else dpnp .complex128
683
683
out = dpnp .empty (out_shape , dtype = out_dtype )
684
684
@@ -697,7 +697,7 @@ def test_fft_1D_on_2D_array_out(self, dtype, n, axis, norm, order):
697
697
a = dpnp .asarray (a_np )
698
698
699
699
out_shape = list (a .shape )
700
- out_shape [axis ] = a .shape [axis ] // 2 + 1 if n is None else n // 2
700
+ out_shape [axis ] = a .shape [axis ] // 2 + 1 if n is None else n // 2 + 1
701
701
out_shape = tuple (out_shape )
702
702
out_dtype = dpnp .complex64 if dtype == dpnp .float32 else dpnp .complex128
703
703
out = dpnp .empty (out_shape , dtype = out_dtype )
0 commit comments