size and \ for JacobiTransformPlan (#164)

* size and \ for JacobiTransformPlan

* Update test_legendre.jl

Author: dlfivefifty

Project.toml

@@ -1,7 +1,7 @@
 name = "ClassicalOrthogonalPolynomials"
 uuid = "b30e2e7b-c4ee-47da-9d5f-2c5c27239acd"
 authors = ["Sheehan Olver <[email protected]>"]
-version = "0.12"
+version = "0.12.1"
 
 [deps]
 ArrayLayouts = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"

src/classical/jacobi.jl

@@ -75,8 +75,10 @@ struct JacobiTransformPlan{T, CHEB2JAC, DCT} <: Plan{T}
 end
 
 JacobiTransformPlan(c2l, ct) = JacobiTransformPlan{promote_type(eltype(c2l),eltype(ct)),typeof(c2l),typeof(ct)}(c2l, ct)
+size(P::JacobiTransformPlan, k...) = size(P.chebtransform, k...)
 
 *(P::JacobiTransformPlan, x::AbstractArray) = P.cheb2jac*(P.chebtransform*x)
+\(P::JacobiTransformPlan, x::AbstractArray) = P.chebtransform\(P.cheb2jac\x)
 
 
 include("legendre.jl")

test/test_legendre.jl

@@ -55,13 +55,15 @@ import QuasiArrays: MulQuasiArray
 
         X = randn(10, 11)
         F1 = plan_transform(P, X, 1)
+        @test size(F1) == (10,11)
         p1 = plan_transform(P, X[:,1])
         @test F1*X ≈ hcat([p1 * X[:,j] for j = 1:size(X,2)]...)
         F2 = plan_transform(P, X, 2)
         p2 = plan_transform(P, X[1,:])
         @test F2*X ≈ vcat([(p2 * X[k,:])' for k = 1:size(X,1)]...)
         F = plan_transform(P, X)
         @test F*X ≈ F2*(F1*X)
+        @test F\(F*X) ≈ X
     end
 
     @testset "operators" begin