add tests

Author: dlfivefifty

src/classical/jacobi.jl

@@ -482,8 +482,8 @@ end
 broadcastbasis(::typeof(+), w_A::Weighted{<:Any,<:Jacobi}, w_B::Weighted{<:Any,<:Jacobi}) = broadcastbasis(+, convert(WeightedBasis,w_A), convert(WeightedBasis,w_B))
 broadcastbasis(::typeof(+), w_A::Weighted{<:Any,<:Jacobi}, w_B::WeightedJacobi) = broadcastbasis(+, convert(WeightedBasis,w_A), w_B)
 broadcastbasis(::typeof(+), w_A::WeightedJacobi, w_B::Weighted{<:Any,<:Jacobi}) = broadcastbasis(+, w_A, convert(WeightedBasis,w_B))
-broadcastbasis(::typeof(+), A::Jacobi, B::Weighted{<:Any,<:Jacobi}) = A # assume B can be lowered to Legendre... for now
-broadcastbasis(::typeof(+), A::Weighted{<:Any,<:Jacobi}, B::Jacobi) = B # assume B can be lowered to Legendre... for now
+broadcastbasis(::typeof(+), A::Jacobi, B::Weighted{<:Any,<:Jacobi{<:Any,<:Integer}}) = A
+broadcastbasis(::typeof(+), A::Weighted{<:Any,<:Jacobi{<:Any,<:Integer}}, B::Jacobi) = B
 
 function \(w_A::WeightedJacobi, w_B::WeightedJacobi)
     wA,A = w_A.args

test/test_jacobi.jl

@@ -537,4 +537,19 @@ import ClassicalOrthogonalPolynomials: recurrencecoefficients, basis, MulQuasiMa
     @testset "conversion not implemented" begin 
         @test_throws ArgumentError Jacobi(0,0) \ Jacobi(1.1,2.1)
     end
+
+    @testset "broadcastbasis" begin
+        a = Jacobi(1,1) * [1; zeros(∞)]
+        b = Weighted(Jacobi(1,1)) * [1; zeros(∞)]
+        c = Weighted(Jacobi(2,3)) * [1; zeros(∞)]
+
+        @test basis(a+b) == basis(b+a) == basis(a+c) == basis(c+a) == Jacobi(1,1)
+        @test a[0.1]+b[0.1] ≈ (a+b)[0.1] ≈ (b+a)[0.1]
+    end
+
+    @testset "Weighted expand" begin
+        W = Weighted(Jacobi(1,1))
+        @test expand(W, x -> (1-x^2)*exp(x))[0.1] ≈ (1-0.1^2)*exp(0.1)
+        @test grid(W, 5) == grid(W.P, 5)
+    end
 end

test/test_lanczos.jl

@@ -303,16 +303,23 @@ import ClassicalOrthogonalPolynomials: recurrencecoefficients, PaddedColumns, or
         @test jacobimatrix(Q)[1,1] ≈ 1/3
         @test Q[0.5,1:3] ≈ [1, 1.369306393762913, 0.6469364618834543]
     end
-end
-
-@testset "#197" begin
-    @test _emptymaximum(1:5) == 5 
-    @test _emptymaximum(1:0) == 0
-    x = Inclusion(ChebyshevInterval())
-    f = exp.(x)
-    QQ = LanczosPolynomial(f)
-    R = LanczosConversion(QQ.data)
-    v = cache(Zeros(∞))
-    @test (R \ v)[1:500] == zeros(500)
-    @test (R * v)[1:500] == zeros(500)
+
+    @testset "#197" begin
+        @test _emptymaximum(1:5) == 5 
+        @test _emptymaximum(1:0) == 0
+        x = Inclusion(ChebyshevInterval())
+        f = exp.(x)
+        QQ = LanczosPolynomial(f)
+        R = LanczosConversion(QQ.data)
+        v = cache(Zeros(∞))
+        @test (R \ v)[1:500] == zeros(500)
+        @test (R * v)[1:500] == zeros(500)
+    end
+
+    @testset "diff" begin
+        P = Normalized(Legendre())
+        x = axes(P,1)
+        Q = LanczosPolynomial(exp.(x))
+        @test diff(Q)[0.1,3] ≈ -0.1637907411174539
+    end
 end