size and ndims for applied vec

JuliaArrays · Dec 6, 2023 · 2e96cb4 · 2e96cb4
1 parent 8c7ad30
commit 2e96cb4
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/src/lazyconcat.jl b/src/lazyconcat.jl
@@ -972,4 +972,6 @@ searchsorted(f::Vcat{<:Any,1}, x) = searchsortedfirst(f, x):searchsortedlast(f,x
 ###
 
 @inline applied_eltype(::typeof(vec), a) = eltype(a)
-@inline applied_axes(::typeof(vec), a) = (oneto(length(a)),)
+@inline applied_axes(::typeof(vec), a) = (oneto(length(a)),)
+@inline applied_ndims(::typeof(vec), a) = 1
+@inline applied_size(::typeof(vec), a) = (length(a),)
diff --git a/test/applytests.jl b/test/applytests.jl
@@ -63,6 +63,16 @@ import ArrayLayouts: StridedLayout
         end
     end
 
+    @testset "vec" begin
+        A = zeros(2,2)
+        v = applied(vec, A)
+        w = vec(A)
+        @test size(v) == size(w) == (4,)
+        @test axes(v) == axes(w) == (Base.OneTo(4),)
+        @test ndims(v) == ndims(w) == 1
+        @test eltype(v) == eltype(w) == Float64
+    end
+
     @testset "view" begin
         a = ApplyArray(+,[1,2,3],[3,4,5])
         v = view(a,1:2)