add tests

Author: dlfivefifty

Project.toml

@@ -4,6 +4,7 @@ version = "1.6.0"
 
 [deps]
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
+ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
@@ -22,6 +23,7 @@ Aqua = "0.8"
 ChainRulesCore = "1"
 ChainRulesTestUtils = "1"
 FiniteDifferences = "0.12"
+ForwardDiff = "0.10"
 LinearAlgebra = "<0.0.1, 1"
 Random = "<0.0.1, 1"
 Test = "<0.0.1, 1"
@@ -33,9 +35,10 @@ Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
 ChainRulesTestUtils = "cdddcdb0-9152-4a09-a978-84456f9df70a"
 FiniteDifferences = "26cc04aa-876d-5657-8c51-4c34ba976000"
+ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 Unitful = "1986cc42-f94f-5a68-af5c-568840ba703d"
 
 [targets]
-test = ["Aqua", "ChainRulesCore", "ChainRulesTestUtils", "FiniteDifferences", "Random", "Test", "Unitful"]
+test = ["Aqua", "ChainRulesCore", "ChainRulesTestUtils", "FiniteDifferences", "ForwardDiff", "Random", "Test", "Unitful"]

ext/AbstractFFTsForwardDiffExt.jl

@@ -8,7 +8,7 @@ import AbstractFFTs: Plan
 for P in (:Plan, :ScaledPlan)  # need ScaledPlan to avoid ambiguities
     @eval begin
         Base.:*(p::AbstractFFTs.$P, x::AbstractArray{DT}) where DT<:Dual = array2dual(DT, p * dual2array(x))
-        Base.:*(p::AbstractFFTs.$P, x::AbstractArray{<:Complex{DT}}) where DT<:Dual = array2dual(DT, p * dual2array(x))
+        Base.:*(p::AbstractFFTs.$P, x::AbstractArray{Complex{DT}}) where DT<:Dual = array2dual(DT, p * dual2array(x))
     end
 end
 

src/AbstractFFTs.jl

@@ -11,6 +11,7 @@ include("TestUtils.jl")
 if !isdefined(Base, :get_extension)
     include("../ext/AbstractFFTsChainRulesCoreExt.jl")
     include("../ext/AbstractFFTsTestExt.jl")
+    include("../ext/AbstractFFTsForwardDiffExt.jl")
 end
 
 end # module

test/TestPlans.jl

@@ -219,7 +219,7 @@ function LinearAlgebra.mul!(y::AbstractArray{<:Complex, N}, p::TestRPlan, x::Abs
     return y
 end
 
-function Base.:*(p::TestRPlan{T}, x::AbstractArray{T}) where T
+function Base.:*(p::TestRPlan{Typ}, x::AbstractArray{Typ}) where Typ
     # create output array
     firstdim = first(p.region)::Int
     d = size(x, firstdim)
@@ -241,7 +241,7 @@ function LinearAlgebra.mul!(y::AbstractArray{<:Real, N}, p::InverseTestRPlan, x:
     real_invdft!(y, x, p.region)
 end
 
-function Base.:*(p::InverseTestRPlan{T}, x::AbstractArray{T}) where T
+function Base.:*(p::InverseTestRPlan{T}, x::AbstractArray{Complex{T}}) where T
     # create output array
     firstdim = first(p.region)::Int
     d = p.d

test/abstractfftsforwarddiff.jl

@@ -52,6 +52,6 @@ using ForwardDiff: Dual, partials, value
     end
 
     c1 = complex.(x1)
-    @test mul!(similar(c1), FFTW.plan_fft(x1), x1) == fft(x1)
-    @test mul!(similar(c1), FFTW.plan_fft(c1), c1) == fft(c1)
+    @test mul!(similar(c1), plan_fft(x1), x1) == fft(x1)
+    @test mul!(similar(c1), plan_fft(c1), c1) == fft(c1)
 end