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,18 +1,32 @@
 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
-
 const fwd = Diffractor.PrimeDerivativeFwd
 const bwd = Diffractor.PrimeDerivativeBack
 
-@testset verbose=true "Diffractor.jl" begin  # overall testset, ensures all tests run
-
 # Unit tests
 function tup2(f)
     a, b = ∂⃖{2}()(f, 1)
@@ -276,7 +290,11 @@ end
     @test_broken gradient(z -> gradient(x -> sum((y -> (x^2*y)).([1,2,3])), z)[1], 5.0) == (12.0,)
 end
 
+
+end
+
+@testset verbose=true "pseudo-Flux" begin
 # Higher order control flow not yet supported (https://github.com/JuliaDiff/Diffractor.jl/issues/24)
-#include("pinn.jl")
+# include("pinn.jl")
+end
 
-end  # overall testset