avoid stack overflow in different sizes

JuliaApproximation · Nov 13, 2023 · a9e6774 · a9e6774
1 parent a7b5a81
commit a9e6774
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/src/chebyshevtransform.jl b/src/chebyshevtransform.jl
@@ -2,6 +2,8 @@
 
 abstract type ChebyshevPlan{T} <: Plan{T} end
 
+*(P::ChebyshevPlan{T}, x::AbstractArray{T}) where T = error("Plan applied to wrong size array")
+
 size(P::ChebyshevPlan) = isdefined(P, :plan) ? size(P.plan) : (0,)
 length(P::ChebyshevPlan) = isdefined(P, :plan) ? length(P.plan) : 0
 
@@ -151,6 +153,8 @@ end
  ldiv!(_prod_size(size(y), d), y)
 end
 
+
+
 function *(P::ChebyshevTransformPlan{T,1,K,true,N}, x::AbstractArray{T,N}) where {T,K,N}
  isempty(x) && return x
 

diff --git a/test/chebyshevtests.jl b/test/chebyshevtests.jl
@@ -462,4 +462,9 @@ using FastTransforms, Test
  @test F \ (F*X) ≈ F * (F\X) ≈ X
  end
  end
+
+ @testset "incompatible shapes" begin
+ @test_throws ErrorException plan_chebyshevtransform(randn(5)) * randn(5,5)
+ @test_throws ErrorException plan_ichebyshevtransform(randn(5)) * randn(5,5)
+ end
 end