Add test demonstrating broken multi-arg code-gen

brianguenter · Dec 28, 2024 · db95edd · db95edd
1 parent c089bd7
commit db95edd
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/test/FDTests.jl b/test/FDTests.jl
@@ -2164,3 +2164,17 @@ end
     @test isnan(h([0.0, 2.0])[1])
     @test isapprox(h([1.1, 2.0])[1], 0.11532531756323319)
 end
+
+@testitem "multiarg code generation" begin
+    using FastDifferentiation: FastDifferentiation as FD
+
+    x = FD.make_variables(:x, 3)
+    y = FD.make_variables(:y, 3)
+    f = x .* y
+    f_callable = FD.make_function(f, x, y)
+
+    x_val = ones(3)
+    y_val = ones(3)
+    f_val = f_callable(x_val, y_val)
+    @test f_val ≈ ones(3)
+end