Increase coverage

Author: dlfivefifty

src/bases/bases.jl

@@ -669,11 +669,17 @@ function diff_layout(::AbstractBasisLayout, a, order::Int; dims...)
     isone(order) ? diff(a) : diff(diff(a), order-1)
 end
 
+function diff_layout(::SubBasisLayout, Vm, order::Int; dims::Integer=1)
+    dims == 1 || error("not implemented")
+    diff(parent(Vm), order)[:,parentindices(Vm)[2]]
+end
+
 function diff_layout(::SubBasisLayout, Vm, order...; dims::Integer=1)
     dims == 1 || error("not implemented")
     diff(parent(Vm), order...)[:,parentindices(Vm)[2]]
 end
 
+
 function diff_layout(::WeightedBasisLayout{SubBasisLayout}, Vm, order...; dims::Integer=1)
     dims == 1 || error("not implemented")
     w = weight(Vm)
@@ -728,6 +734,11 @@ abslaplacian(A, order...; dims...) = abslaplacian_layout(MemoryLayout(A), A, ord
 abslaplacian_layout(layout, A, order...; dims...) = abslaplacian_axis(axes(A,1), A, order...; dims...)
 abslaplacian_axis(::Inclusion{<:Number}, A, order=1; dims...) = -diff(A, 2order; dims...)
 
+"""
+    abslaplacian(A, k=1)
+
+computes ``Δ^k * A``. 
+"""
 laplacian(A, order...; dims...) = laplacian_layout(MemoryLayout(A), A, order...; dims...)
 laplacian_layout(layout, A, order...; dims...) = laplacian_axis(axes(A,1), A, order...; dims...)
 laplacian_axis(::Inclusion{<:Number}, A, order=1; dims...) = diff(A, 2order; dims...)

src/operators.jl

@@ -219,4 +219,5 @@ abs(Δ::Laplacian{T}) where T = AbsLaplacian{T}(axes(Δ,1), Δ.order)
 -(Δ::Laplacian{<:Any,<:Any,Nothing}) = abs(Δ)
 -(Δ::AbsLaplacian{T,<:Any,Nothing}) where T = Laplacian{T}(Δ.axis)
 
-^(Δ::AbsLaplacian{T}, k::Real) where T = AbsLaplacian{T}(Δ.axis, operatororder(Δ)*k)
+^(Δ::AbsLaplacian{T}, k::Real) where T = AbsLaplacian{T}(Δ.axis, operatororder(Δ)*k)
+^(Δ::AbsLaplacian{T}, k::Integer) where T = AbsLaplacian{T}(Δ.axis, operatororder(Δ)*k)

test/test_splines.jl

@@ -1,6 +1,6 @@
 using ContinuumArrays, LinearAlgebra, Base64, FillArrays, QuasiArrays, BandedMatrices, BlockArrays, Test
 using QuasiArrays: ApplyQuasiArray, ApplyStyle, MemoryLayout, mul, MulQuasiMatrix, Vec
-import LazyArrays: MulStyle, LdivStyle, arguments, applied, apply
+import LazyArrays: MulStyle, LdivStyle, arguments, applied, apply, simplifiable
 import ContinuumArrays: basis, AdjointBasisLayout, ExpansionLayout, BasisLayout, SubBasisLayout, AdjointMappedBasisLayouts, MappedBasisLayout, plan_grid_transform, weaklaplacian
 
 @testset "Splines" begin
@@ -165,10 +165,27 @@ import ContinuumArrays: basis, AdjointBasisLayout, ExpansionLayout, BasisLayout,
         @test typeof(diff(L)) == typeof(diff(L; dims=1)) == typeof(D*L)
         @test_throws ErrorException diff(L; dims=2)
 
+        @test diff(L[:,1:2])[1.1,:] == diff(L)[1.1,1:2]
+
         @test diff(L,0) ≡ L
         @test diff(f,0) ≡ f
-        @test diff(L,2)[1.1,:] == laplacian(L)[1.1,:] == -abslaplacian(L)[1.1,:]
-        @test diff(f,2)[1.1] == laplacian(f)[1.1] == -abslaplacian(f)[1.1]
+        @test diff(L,2)[1.1,:] == laplacian(L)[1.1,:] == -abslaplacian(L)[1.1,:] == laplacian(L,1)[1.1,:] == -abslaplacian(L,1)[1.1,:]
+
+        @test diff(L[:,1:2],2)[1.1,:] == diff(L,2)[1.1,1:2]
+        @test diff(f,2)[1.1] == laplacian(f)[1.1] == laplacian(f,1)[1.1] == -abslaplacian(f)[1.1] == -abslaplacian(f,1)[1.1]
+
+        @test laplacian(L[:,1:2])[1.1,:] == laplacian(L)[1.1,1:2] == -abslaplacian(L[:,1:2])[1.1,:] == -abslaplacian(L)[1.1,1:2]
+
+        Δ = Laplacian(L)
+        @test (Δ * L)[1.1,:] == -(abs(Δ) * L)[1.1,:] == laplacian(L)[1.1,:]
+        @test simplifiable(*, Δ, L) == simplifiable(*, abs(Δ), L) == Val(true)
+        @test -abs(Δ) == Δ
+        @test -Δ == abs(Δ)
+        @test Δ^2 == Δ*Δ
+        @test abs(Δ)^2 == abs(Δ^2) == abs(Δ)^2.0
+        @test simplifiable(*, Δ, Δ) == simplifiable(*, abs(Δ), abs(Δ)) == Val(true)
+        @test summary(Δ) == "Laplacian(Inclusion(1 .. 3))"
+        @test summary(Δ^2) == "Laplacian(Inclusion(1 .. 3), 2)"
 
         M = applied(*, (D*L).args..., [1,2,4])
         @test eltype(materialize(M)) == Float64