From c596edb85d97de460027e1fc28f7c5f7ef3a64f8 Mon Sep 17 00:00:00 2001 From: Evgeni Burovski Date: Wed, 13 Nov 2024 18:55:13 +0100 Subject: [PATCH] MAINT: avoid axis=None and s is not None in test_{fftn,ifftn} --- array_api_tests/test_fft.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/array_api_tests/test_fft.py b/array_api_tests/test_fft.py index 582ff5ee..e7d5e1b8 100644 --- a/array_api_tests/test_fft.py +++ b/array_api_tests/test_fft.py @@ -57,6 +57,9 @@ def draw_s_axes_norm_kwargs(x: Array, data: st.DataObject, *, size_gt_1=False) - s_strat = st.none() | s_strat s = data.draw(s_strat, label="s") + # Using `axes is None and s is not None` is disallowed by the spec + assume(axes is not None or s is None) + norm = data.draw(st.sampled_from(["backward", "ortho", "forward"]), label="norm") kwargs = data.draw( hh.specified_kwargs(