Skip to content

Commit 1da21e7

Browse files
committed
remove irrelevant cases from test
1 parent 1f4acc3 commit 1da21e7

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

mkl_fft/tests/test_fftnd.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -267,17 +267,15 @@ def test_s_axes_out(dtype, s, axes, func):
267267
@requires_numpy_2
268268
@pytest.mark.parametrize("dtype", [complex, float])
269269
@pytest.mark.parametrize("axes", [(1, 2, 3), (-1, -2, -3), [2, 1, 3]])
270-
@pytest.mark.parametrize("func", ["fftn", "ifftn", "rfftn"])
270+
@pytest.mark.parametrize("func", ["fftn", "ifftn"])
271271
def test_s_none_vs_s_full(dtype, axes, func):
272272
shape = (2, 30, 20, 10)
273-
if dtype is complex and func != "rfftn":
273+
if dtype is complex:
274274
x = np.random.random(shape) + 1j * np.random.random(shape)
275275
else:
276276
x = np.random.random(shape)
277277

278278
implied_s = [shape[ax] for ax in axes]
279-
if func == "irfftn":
280-
implied_s[-1] = 2 * (implied_s[-1] - 1)
281279

282280
r1 = getattr(np.fft, func)(x, axes=axes)
283281
r2 = getattr(mkl_fft, func)(x, axes=axes)

0 commit comments

Comments
 (0)