Defining size on UnionAll types associated with MultiValues

gridap · Jun 16, 2019 · 188adca · 188adca
1 parent 78d4982
commit 188adca
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/src/TensorValues.jl b/src/TensorValues.jl
@@ -1,7 +1,7 @@
 module TensorValues
 
 using StaticArrays
-using Base: @propagate_inbounds
+using Base: @propagate_inbounds, @pure
 
 export MultiValue
 export TensorValue

diff --git a/src/Types.jl b/src/Types.jl
@@ -142,6 +142,12 @@ function size(::Type{MultiValue{S,T,N,L}}) where {S,T,N,L}
   size(A)
 end
 
+function size(::Type{<:MultiValue{S}}) where S
+  _s(Size(S))
+end
+
+@pure _s(s::Size{T}) where T = T
+
 # Custom type printing
 
 function show(io::IO,v::MultiValue)

diff --git a/test/TypesTests.jl b/test/TypesTests.jl
@@ -141,6 +141,7 @@ V = VectorValue{3,Int}
 
 V = VectorValue{3}
 @test length(V) == 3
+@test size(V) == (3,)
 
 # Custom type printing