Skip to content

Commit adaf3bd

Browse files
Update input generation for TestFft2 and TestFftn
1 parent 4f81237 commit adaf3bd

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

dpnp/tests/test_fft.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -380,7 +380,8 @@ def setup_method(self):
380380
@pytest.mark.parametrize("norm", [None, "forward", "backward", "ortho"])
381381
@pytest.mark.parametrize("order", ["C", "F"])
382382
def test_fft2(self, dtype, axes, norm, order):
383-
a_np = generate_random_numpy_array((2, 3, 4), dtype)
383+
x = generate_random_numpy_array((2, 3, 4), dtype)
384+
a_np = numpy.array(x, order=order)
384385
a = dpnp.array(a_np)
385386

386387
result = dpnp.fft.fft2(a, axes=axes, norm=norm)
@@ -442,7 +443,8 @@ def setup_method(self):
442443
@pytest.mark.parametrize("norm", [None, "backward", "forward", "ortho"])
443444
@pytest.mark.parametrize("order", ["C", "F"])
444445
def test_fftn(self, dtype, axes, norm, order):
445-
a_np = generate_random_numpy_array((2, 3, 4, 5), dtype)
446+
x = generate_random_numpy_array((2, 3, 4, 5), dtype)
447+
a_np = numpy.array(x, order=order)
446448
a = dpnp.array(a_np)
447449

448450
result = dpnp.fft.fftn(a, axes=axes, norm=norm)

0 commit comments

Comments
 (0)