Add tests for complex float promotion

JuliaMath · Mar 6, 2023 · 3cbb2bb · 3cbb2bb
1 parent 20de5e7
commit 3cbb2bb
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -228,6 +228,16 @@ end
     end
 end
 
+@testset "Complex float promotion for backwards real FFTs" begin
+    for x in (rand(-5:5, 3), rand(-5:5, 3, 4), rand(-5:5, 3, 4, 5))
+        N = ndims(x)
+        complex_x = complex.(x)
+        d = 2 * size(x, 1) - 1
+        @test irfft(x, d) ≈ irfft(complex.(x), d) ≈ irfft(complex.(float.(x)), d)
+        @test brfft(x, d) ≈ brfft(complex.(x), d) ≈ brfft(complex.(float.(x)), d)
+    end
+end
+
 @testset "ChainRules" begin
     @testset "shift functions" begin
         for x in (randn(3), randn(3, 4), randn(3, 4, 5))