From 63c9f7057a3ab683da807ccdca2cf05c81ece075 Mon Sep 17 00:00:00 2001 From: Matthew Barber Date: Thu, 7 Dec 2023 10:23:54 +0000 Subject: [PATCH] Fix `rfft`, `rfftn` and `ihfft` input dtypes when testing --- array_api_tests/test_fft.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/array_api_tests/test_fft.py b/array_api_tests/test_fft.py index 39d96d6c..97740d47 100644 --- a/array_api_tests/test_fft.py +++ b/array_api_tests/test_fft.py @@ -198,7 +198,7 @@ def test_ifftn(x, data): @given( - x=xps.arrays(dtype=xps.complex_dtypes(), shape=fft_shapes_strat), + x=xps.arrays(dtype=xps.floating_dtypes(), shape=fft_shapes_strat), data=st.data(), ) def test_rfft(x, data): @@ -231,7 +231,7 @@ def test_irfft(x, data): @given( - x=xps.arrays(dtype=xps.complex_dtypes(), shape=fft_shapes_strat), + x=xps.arrays(dtype=xps.floating_dtypes(), shape=fft_shapes_strat), data=st.data(), ) def test_rfftn(x, data): @@ -279,7 +279,7 @@ def test_hfft(x, data): @given( - x=xps.arrays(dtype=hh.all_floating_dtypes(), shape=fft_shapes_strat), + x=xps.arrays(dtype=xps.floating_dtypes(), shape=fft_shapes_strat), data=st.data(), ) def test_ihfft(x, data):