Added size on MultiValue type

gridap · May 19, 2019 · c8844a7 · c8844a7
1 parent 0febd15
commit c8844a7
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/Types.jl b/src/Types.jl
@@ -137,6 +137,11 @@ end
 
 length(::Type{MultiValue{S,T,N,L}}) where {S,T,N,L} = L
 
+function size(::Type{MultiValue{S,T,N,L}}) where {S,T,N,L}
+  A = SArray{S,T,N,L}
+  size(A)
+end
+
 # Custom type printing
 
 function show(io::IO,v::MultiValue)

diff --git a/test/TypesTests.jl b/test/TypesTests.jl
@@ -133,6 +133,12 @@ a = ones(Int,3)
 b = convert(VectorValue{3,Int},a)
 @test isa(b,VectorValue{3,Int})
 
+# Misc operations on the type itself
+
+V = VectorValue{3,Int}
+@test length(V) == 3
+@test size(V) == (3,)
+
 # Custom type printing
 
 s = "TensorValues.MultiValue{Tuple{3,2},Float64,2,6}(1.0, 2.0, 3.0, 4.0, 5.0, 6.0)"