Update src/testers.jl

oxinabox · mzgubic · oxinabox · commit c94ccd88bcdc · 2021-06-16T17:04:37.000+01:00
Co-authored-by: Miha Zgubic &lt;mzgubic@users.noreply.github.com&gt;
diff --git a/src/testers.jl b/src/testers.jl
@@ -250,7 +250,7 @@ function test_rrule(
         )
         @test_msg(
             "The pullback should return 1 cotangent for the primal and each primal input.",
-            length(ad_cotangents) == 1 + length(args)
+            length(ad_cotangents) == length(primals)
         )
 
         # Correctness testing via finite differencing.

Original file line number	Diff line number	Diff line change
`@@ -250,7 +250,7 @@ function test_rrule(`
`250`	`250`	`)`
`251`	`251`	`@test_msg(`
`252`	`252`	`"The pullback should return 1 cotangent for the primal and each primal input.",`
`253`		`- length(ad_cotangents) == 1 + length(args)`
	`253`	`+ length(ad_cotangents) == length(primals)`
`254`	`254`	`)`
`255`	`255`
`256`	`256`	`# Correctness testing via finite differencing.`