add test for forwardAD with graph vector

numericalEFT · Sep 25, 2023 · fef35c2 · fef35c2
1 parent 797b87e
commit fef35c2
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/test/computational_graph.jl b/test/computational_graph.jl
@@ -328,6 +328,16 @@ end
         leaf = [5.0, -1.0, 2.0, 0.0, 0.0, 1.0]
         @test eval!(dual1[F0.id], leafmap, leaf) == 240.0
         @test isequiv(dual[F1.id], dual1[F1.id], :id, :weight, :vertices)
+
+        F0_r1 = F1 + F3
+        dual = forwardAD_root([F0, F0_r1])
+        leafmap[dual[g1.id].id] = 4
+        leafmap[dual[g2.id].id] = 5
+        leafmap[dual[g3.id].id] = 6
+        @test eval!(dual[F0.id], leafmap, leaf) == 240.0
+        @test eval!(dual[F0_r1.id], leafmap, leaf) == 60.0
+        @test isequiv(dual[F0.id], dual1[F0.id], :id, :weight)
+        @test isequiv(dual[F1.id], dual1[F1.id], :id, :weight)
     end
 end