Skip to content

Commit c6a171c

Browse files
committed
Fix input dtypes for rfft-like tests
1 parent 671e07e commit c6a171c

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

array_api_tests/test_fft.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,7 @@ def test_ifftn(x, data):
176176

177177

178178
@given(
179-
x=xps.arrays(dtype=xps.complex_dtypes(), shape=fft_shapes_strat),
179+
x=xps.arrays(dtype=xps.floating_dtypes(), shape=fft_shapes_strat),
180180
data=st.data(),
181181
)
182182
def test_rfft(x, data):
@@ -189,7 +189,7 @@ def test_rfft(x, data):
189189

190190

191191
@given(
192-
x=xps.arrays(dtype=xps.complex_dtypes(), shape=fft_shapes_strat),
192+
x=xps.arrays(dtype=xps.floating_dtypes(), shape=fft_shapes_strat),
193193
data=st.data(),
194194
)
195195
def test_irfft(x, data):
@@ -202,7 +202,7 @@ def test_irfft(x, data):
202202

203203

204204
@given(
205-
x=xps.arrays(dtype=xps.complex_dtypes(), shape=fft_shapes_strat),
205+
x=xps.arrays(dtype=xps.floating_dtypes(), shape=fft_shapes_strat),
206206
data=st.data(),
207207
)
208208
def test_rfftn(x, data):
@@ -215,7 +215,7 @@ def test_rfftn(x, data):
215215

216216

217217
@given(
218-
x=xps.arrays(dtype=xps.complex_dtypes(), shape=fft_shapes_strat),
218+
x=xps.arrays(dtype=xps.floating_dtypes(), shape=fft_shapes_strat),
219219
data=st.data(),
220220
)
221221
def test_irfftn(x, data):

0 commit comments

Comments
 (0)