Skip to content

Commit 425529e

Browse files
committed
reverse unintended changes
1 parent 88030f5 commit 425529e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/test_fft.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -660,7 +660,7 @@ def test_fft_usm_ndarray(self, n):
660660
x = dpt.linspace(-1, 1, 11)
661661
a_usm = dpt.asarray(dpt.sin(x))
662662
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
664664
out_dtype = map_dtype_to_device(dpnp.complex128, a_usm.sycl_device)
665665
out = dpt.empty(out_shape, dtype=out_dtype)
666666

@@ -678,7 +678,7 @@ def test_fft_1D_out(self, dtype, n, norm):
678678
a = dpnp.asarray(a, dtype=dtype)
679679
a_np = dpnp.asnumpy(a)
680680

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
682682
out_dtype = dpnp.complex64 if dtype == dpnp.float32 else dpnp.complex128
683683
out = dpnp.empty(out_shape, dtype=out_dtype)
684684

@@ -697,7 +697,7 @@ def test_fft_1D_on_2D_array_out(self, dtype, n, axis, norm, order):
697697
a = dpnp.asarray(a_np)
698698

699699
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
701701
out_shape = tuple(out_shape)
702702
out_dtype = dpnp.complex64 if dtype == dpnp.float32 else dpnp.complex128
703703
out = dpnp.empty(out_shape, dtype=out_dtype)

0 commit comments

Comments
 (0)