Skip to content

Commit eedba14

Browse files
committed
Get type T correct for test irfft
1 parent 2a423e2 commit eedba14

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

test/testplans.jl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -95,11 +95,11 @@ Base.:*(p::InverseTestPlan, x::AbstractArray) = mul!(similar(x, complex(float(el
9595
mutable struct TestRPlan{T,N,G} <: Plan{T}
9696
region::G
9797
sz::NTuple{N,Int}
98-
pinv::Plan{T}
98+
pinv::Plan{Complex{T}}
9999
TestRPlan{T}(region::G, sz::NTuple{N,Int}) where {T,N,G} = new{T,N,G}(region, sz)
100100
end
101101

102-
mutable struct InverseTestRPlan{T,N,G} <: Plan{T}
102+
mutable struct InverseTestRPlan{T,N,G} <: Plan{Complex{T}}
103103
d::Int
104104
region::G
105105
sz::NTuple{N,Int}
@@ -113,10 +113,10 @@ end
113113
AbstractFFTs.ProjectionStyle(::TestRPlan) = AbstractFFTs.RealProjectionStyle()
114114
AbstractFFTs.ProjectionStyle(p::InverseTestRPlan) = AbstractFFTs.RealInverseProjectionStyle(p.d)
115115

116-
function AbstractFFTs.plan_rfft(x::AbstractArray{T}, region; kwargs...) where {T}
116+
function AbstractFFTs.plan_rfft(x::AbstractArray{T}, region; kwargs...) where {T<:Real}
117117
return TestRPlan{T}(region, size(x))
118118
end
119-
function AbstractFFTs.plan_brfft(x::AbstractArray{T}, d, region; kwargs...) where {T}
119+
function AbstractFFTs.plan_brfft(x::AbstractArray{Complex{T}}, d, region; kwargs...) where {T}
120120
return InverseTestRPlan{T}(d, region, size(x))
121121
end
122122
function AbstractFFTs.plan_inv(p::TestRPlan{T,N}) where {T,N}

0 commit comments

Comments
 (0)