added test for parent

FluxML · Aug 7, 2023 · f0e0caf · f0e0caf
1 parent 6129613
commit f0e0caf
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/test/lib/array.jl b/test/lib/array.jl
@@ -96,3 +96,13 @@ end
  end
  end
 end
+
+@testset "parent" begin
+ @testset "$constructor" for constructor in [LowerTriangular, UpperTriangular]
+ x = randn(2, 2)
+ y, pb = Zygote.pullback(x) do x
+ sum(parent(constructor(2 .* x)))
+ end
+ @test first(pb(one(y))) ≈ constructor(2 * ones(2, 2))
+ end
+end