dot tests

Author: dlfivefifty

src/classical/legendre.jl

@@ -40,6 +40,7 @@ singularities(d::Inclusion{T,<:Interval}) where T = LegendreWeight{T}()[affine(d
 singularities(::AbstractFillLayout, P) = LegendreWeight{eltype(P)}()
 
 _basis(::LegendreWeight{T}) where T = Legendre{T}()
+_basis(v::SubQuasiArray) = view(_basis(parent(v)), parentindices(v)[1], :)
 
 struct Legendre{T} <: AbstractJacobi{T} end
 Legendre() = Legendre{Float64}()
@@ -171,7 +172,7 @@ _sum(p::SubQuasiArray{T,1,Legendre{T},<:Tuple{Inclusion,Int}}, ::Colon) where T
 # dot
 ###
 
-_dot(::Inclusion{<:Any,<:ChebyshevInterval}, a, b) = __dot(MemoryLayout(a), MemoryLayout(b), a, b)
+_dot(::Inclusion{<:Any,<:AbstractInterval}, a, b) = __dot(MemoryLayout(a), MemoryLayout(b), a, b)
 function __dot(::ExpansionLayout, ::ExpansionLayout, a, b)
     P,c = basis(a),coefficients(a)
     Q,d = basis(b),coefficients(b)

test/runtests.jl

@@ -59,15 +59,13 @@ include("test_roots.jl")
 end
 
 @testset "basis" begin
-     x = Inclusion(ChebyshevInterval())
-     @test sum(x) == 2.0
-     sum(x .^ 2)
-     sum(exp.(x))
-     @test dot(x, x) ≈ 2/3
-     dot(x.^2, x.^2)
-     dot(exp.(x), x.^2)
-     dot(x, exp.(x)) ≈ dot(exp.(x), x)
-     
-     Inclusion(1 .. 2)
-
+    for x in (Inclusion(ChebyshevInterval()), Inclusion(1 .. 2))
+        @test sum(x) == last(x)-first(x)
+        # sum(x .^ 2)
+        # sum(exp.(x))
+        @test dot(x, x) ≈ sum(expand(x .^2))
+        @test dot(x.^2, x.^2) ≈ sum(expand(x .^4))
+        @test dot(exp.(x), x.^2) ≈ sum(expand(x .^2 .* exp.(x)))
+        @test dot(x, exp.(x)) ≈ dot(exp.(x), x)
+    end
 end