adapt many tests from Zygote

Author: mcabbott

test/Project.toml

@@ -2,11 +2,14 @@
 ChainRules = "082447d4-558c-5d27-93f4-14fc19e9eca2"
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
 Combinatorics = "861a8166-3701-5b0c-9a16-15d98fcdc6aa"
+Distributed = "8ba89e20-285c-5b6f-9357-94700520ee1b"
+FiniteDifferences = "26cc04aa-876d-5657-8c51-4c34ba976000"
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
 InteractiveUtils = "b77e0a4c-d291-57a0-90e8-8db25a27a240"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
+Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
 StructArrays = "09ab397b-f2b6-538f-b94a-2f83cf4a842a"
 Symbolics = "0c5d862f-8b57-4792-8d23-62f2024744c7"

test/chainrules.jl

@@ -0,0 +1,57 @@
+
+# This file has integration tests for some rules defined in ChainRules.jl,
+# especially those which aim to support higher derivatives, as properly
+# testing those is difficult.
+
+using Diffractor, ChainRulesCore, ForwardDiff
+
+#####
+##### Base/array.jl
+#####
+
+
+
+
+
+
+#####
+##### Base/arraymath.jl
+#####
+
+
+
+
+#####
+##### Base/base.jl
+#####
+
+
+
+
+
+#####
+##### Base/indexing.jl
+#####
+
+
+
+
+#####
+##### Base/mapreduce.jl
+#####
+
+
+
+
+#####
+##### LinearAlgebra/dense.jl
+#####
+
+
+
+
+#####
+##### LinearAlgebra/norm.jl
+#####
+
+

test/runtests.jl

@@ -1,13 +1,29 @@
 using Diffractor
+using Test
+
+@testset verbose=true "ChainRules integration.jl" begin
+    include("chainrules.jl")
+end
+@testset verbose=true "from Zygote" begin
+    include("zygote_features.jl")
+end
+@testset verbose=true "from Zygote's gradcheck.jl" begin
+    include("zygote_gradcheck.jl")
+end
+@testset verbose=true "Unit tests" begin
+
+# The rest of this file is unchanged, except the very end,
+# but IMO we should move these tests to a new file.
+
+# Loading Diffractor: var"'" globally will break many tests above, which use it for adjoint.
+
 using Diffractor: var"'", ∂⃖, DiffractorRuleConfig
 using ChainRules
 using ChainRulesCore
 using ChainRulesCore: ZeroTangent, NoTangent, frule_via_ad, rrule_via_ad
 using Symbolics
 using LinearAlgebra
 
-using Test
-
 # Unit tests
 function tup2(f)
     a, b = ∂⃖{2}()(f, 1)
@@ -214,4 +230,7 @@ z45, delta45 = frule_via_ad(DiffractorRuleConfig(), (0,1), x -> log(exp(x)), 2)
 @test z45 ≈ 2.0
 @test delta45 ≈ 1.0
 
+end
+@testset verbose=true "pseudo-Flux" begin
 include("pinn.jl")
+end