Implement size and length to BasisInfo (#17)

JuliaFEM · Jan 30, 2018 · 87ba789 · 87ba789
1 parent 049ef88
commit 87ba789
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/math.jl b/src/math.jl
@@ -146,6 +146,14 @@ type BasisInfo{B<:AbstractBasis,T}
     detJ::T
 end
 
+function length{T<:AbstractBasis}(B::BasisInfo{T})
+    return length(T)
+end
+
+function size{T<:AbstractBasis}(B::BasisInfo{T})
+    return size(T)
+end
+
 """
 Initialization of data type `BasisInfo`.
 

diff --git a/test/test_math.jl b/test/test_math.jl
@@ -53,6 +53,8 @@ end
 
 @testset "test BasisInfo" begin
     B = BasisInfo(Seg2)
+    @test length(B) == 2
+    @test size(B) == (1, 2)
     X = ((0.0,), (1.1))
     xi = (0.0,)
     eval_basis!(B, X, xi)