Support more vcat with Fill (#59)

* Support more vcat with Fill * add tests * Update runtests.jl
JuliaArrays · Feb 3, 2021 · 667656f · 667656f · dlfivefifty · Feb 3, 2021
1 parent 1fea087
commit 667656f
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,6 +1,6 @@
 name = "InfiniteArrays"
 uuid = "4858937d-0d70-526a-a4dd-2d5cb5dd786c"
-version = "0.9.1"
+version = "0.9.2"
 
 [deps]
 ArrayLayouts = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"

diff --git a/src/InfiniteArrays.jl b/src/InfiniteArrays.jl
@@ -100,8 +100,10 @@ for Typ in (:Number, :AbstractVector)
    end
 end
 
-vcat(a::AbstractMatrix, b::AbstractFill{<:Any,2,<:Tuple{OneToInf,OneTo}}) =
-   Vcat(a, b)
+vcat(a::AbstractVector, b::AbstractVector, c::AbstractFill{<:Any,1,<:Tuple{OneToInf}}) = Vcat(vcat(a,b), c)
+vcat(a::AbstractVector, b::AbstractVector, c::AbstractVector, d::AbstractFill{<:Any,1,<:Tuple{OneToInf}}) = Vcat(vcat(a,b,c), d)
+
+vcat(a::AbstractMatrix, b::AbstractFill{<:Any,2,<:Tuple{OneToInf,OneTo}}) = Vcat(a, b)
 
 cat_similar(A, T, shape::Tuple{Infinity}) = zeros(T,∞)
 cat_similar(A::AbstractArray, T, shape::Tuple{Infinity}) =

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -733,6 +733,11 @@ end
         @test [1; 2; 3; zeros(Int,∞)] isa CachedArray
         @test [[1,2]; 3; zeros(Int,∞)] isa CachedArray
         @test [2; [1,2]; 3; zeros(Int,∞)] isa CachedArray
+
+        @test [[1,2]; [3,4]; Zeros(∞)] isa Vcat{<:Any,1,<:Tuple{Array,Zeros}}
+        @test [[1,2]; [3,4]; [5,6]; Zeros(∞)] isa Vcat{<:Any,1,<:Tuple{Array,Zeros}}
+
+        @test [randn(2,2); Zeros(∞,2)] isa Vcat{<:Any,2,<:Tuple{Array,Zeros}}
     end
 
     @testset "sparse print" begin