Fix rfft, rfftn and ihfft input dtypes when testing

honno · honno · commit 63c9f7057a3a · 2023-12-07T10:23:54.000Z
diff --git 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):

Original file line number	Diff line number	Diff line change
`@@ -198,7 +198,7 @@ def test_ifftn(x, data):`
`198`	`198`
`199`	`199`
`200`	`200`	`@given(`
`201`		`- x=xps.arrays(dtype=xps.complex_dtypes(), shape=fft_shapes_strat),`
	`201`	`+ x=xps.arrays(dtype=xps.floating_dtypes(), shape=fft_shapes_strat),`
`202`	`202`	`data=st.data(),`
`203`	`203`	`)`
`204`	`204`	`def test_rfft(x, data):`
`@@ -231,7 +231,7 @@ def test_irfft(x, data):`
`231`	`231`
`232`	`232`
`233`	`233`	`@given(`
`234`		`- x=xps.arrays(dtype=xps.complex_dtypes(), shape=fft_shapes_strat),`
	`234`	`+ x=xps.arrays(dtype=xps.floating_dtypes(), shape=fft_shapes_strat),`
`235`	`235`	`data=st.data(),`
`236`	`236`	`)`
`237`	`237`	`def test_rfftn(x, data):`
`@@ -279,7 +279,7 @@ def test_hfft(x, data):`
`279`	`279`
`280`	`280`
`281`	`281`	`@given(`
`282`		`- x=xps.arrays(dtype=hh.all_floating_dtypes(), shape=fft_shapes_strat),`
	`282`	`+ x=xps.arrays(dtype=xps.floating_dtypes(), shape=fft_shapes_strat),`
`283`	`283`	`data=st.data(),`
`284`	`284`	`)`
`285`	`285`	`def test_ihfft(x, data):`