test: simpler promote_to testing

EnzymeAD · Nov 8, 2024 · b7a3fa6 · b7a3fa6
1 parent e4d6956
commit b7a3fa6
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/test/complex.jl b/test/complex.jl
@@ -86,3 +86,14 @@ end
     x_concrete = Reactant.to_rarray(x)
     @test @jit(abs.(x_concrete)) ≈ abs.(x)
 end
+
+@testset "promote_to Complex" begin
+    x = 1.0 + 2.0im
+    y = Reactant.ConcreteRNumber(x)
+
+    f = Reactant.compile((y,)) do z
+        z + Reactant.promote_to(Reactant.TracedRNumber{ComplexF64}, 1.0 - 3.0im)
+    end
+
+    @test isapprox(f(y), 2.0 - 1.0im)
+end