Valid shapes for test_irfftn

honno · honno · commit ef95ba1887d0 · 2023-10-23T13:39:55.000+01:00
diff --git a/array_api_tests/test_fft.py b/array_api_tests/test_fft.py
@@ -232,7 +232,9 @@ def test_rfftn(x, data):
 
 
 @given(
-    x=xps.arrays(dtype=xps.complex_dtypes(), shape=fft_shapes_strat),
+    x=xps.arrays(
+        dtype=xps.complex_dtypes(), shape=fft_shapes_strat.filter(lambda s: s[-1] > 1)
+    ),
     data=st.data(),
 )
 def test_irfftn(x, data):

Original file line number	Diff line number	Diff line change
`@@ -232,7 +232,9 @@ def test_rfftn(x, data):`
`232`	`232`
`233`	`233`
`234`	`234`	`@given(`
`235`		`- x=xps.arrays(dtype=xps.complex_dtypes(), shape=fft_shapes_strat),`
	`235`	`+ x=xps.arrays(`
	`236`	`+ dtype=xps.complex_dtypes(), shape=fft_shapes_strat.filter(lambda s: s[-1] > 1)`
	`237`	`+ ),`
`236`	`238`	`data=st.data(),`
`237`	`239`	`)`
`238`	`240`	`def test_irfftn(x, data):`