increase coverage

Author: dlfivefifty

Project.toml

@@ -1,6 +1,6 @@
 name = "InfiniteLinearAlgebra"
 uuid = "cde9dba0-b1de-11e9-2c62-0bab9446c55c"
-version = "0.10"
+version = "0.10.1"
 
 [deps]
 ArrayLayouts = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"

src/InfiniteLinearAlgebra.jl

@@ -114,12 +114,15 @@ function chop(A::AbstractMatrix{T}, tol::Real=zero(real(T))) where T
     return A
 end
 
-pad(c::AbstractVector{T}, ax::Union{OneTo,OneToInf}) where T = [c; Zeros{T}(length(ax)-length(c))]
+pad(c::AbstractVector{T}, ax::Union{OneTo,OneToInf}) where T = Vcat(c, Zeros{T}(length(ax)-length(c)))
+pad(c::AbstractVector{T}, n::Int) where T = Vcat(c, Zeros{T}(n-length(c)))
 
 _colon2axes(::Tuple{}, bx::Tuple{}) = ()
 _colon2axes(ax::Tuple, bx::Tuple{Colon, Vararg{Any}}) = (first(ax), _colon2axes(tail(ax), tail(bx))...)
+_colon2axes(ax::Tuple, bx::Tuple{Union{Integer,Infinity}, Vararg{Any}}) = (oneto(first(bx)), _colon2axes(tail(ax), tail(bx))...)
 _colon2axes(ax::Tuple, bx::Tuple) = (first(bx), _colon2axes(tail(ax), tail(bx))...)
 pad(c, ax...) = PaddedArray(c, _colon2axes(axes(c), ax))
+pad(c, ax::Colon...) = c
 
 pad(c::Transpose, ax, bx) = transpose(pad(parent(c), bx, ax))
 pad(c::Adjoint, ax, bx) = adjoint(pad(parent(c), bx, ax))

test/runtests.jl

@@ -38,9 +38,14 @@ end
 
     @test pad(1:3, 5) == [1:3; 0; 0]
     @test pad(1:3, oneto(∞)) isa Vcat
+    @test pad(1:3, :) ≡ 1:3
     X = Matrix(reshape(1:6, 3, 2))
-    P = pad(X, oneto(3), oneto(∞))
-    @test P isa PaddedArray
+    # TODO: replace ≈ with ==
+    @test pad(X, oneto(3), ∞) ≈ pad(X, oneto(3), oneto(∞)) ≈ pad(X, 3, oneto(∞)) ≈ pad(X, 3, ∞) ≈ pad(X, :, ∞)
+    @test pad(X, oneto(3), oneto(∞)) isa PaddedArray
+    @test pad(X, :, oneto(∞)) isa PaddedArray
+    @test pad(X, :, :) isa Matrix
+    @test pad(X, oneto(10), :) isa PaddedArray
     P = pad(BlockVec(X), blockedrange(Fill(3,∞)))
     @test P isa BlockVec
     @test MemoryLayout(P) isa PaddedColumns