mirror of
https://github.com/JuliaFEM/JuliaFEM.jl.git
synced 2026-08-31 08:16:23 +00:00
Jukka Aho
51 lines
1.5 KiB
Julia
51 lines
1.5 KiB
Julia
# This file is a part of JuliaFEM.
|
|
# License is MIT: see https://github.com/JuliaFEM/JuliaFEM.jl/blob/master/LICENSE.md
|
|
|
|
using FactCheck
|
|
using JuliaFEM: Element, Basis, FieldSet
|
|
|
|
# prototype element
|
|
type MockElement <: Element
|
|
connectivity :: Array{Int, 1}
|
|
basis :: Basis
|
|
fields :: Dict{Symbol, FieldSet}
|
|
end
|
|
function MockElement(connectivity)
|
|
h(xi) = [
|
|
(1-xi[1])*(1-xi[2])/4
|
|
(1+xi[1])*(1-xi[2])/4
|
|
(1+xi[1])*(1+xi[2])/4
|
|
(1-xi[1])*(1+xi[2])/4]
|
|
dh(xi) = [
|
|
-(1-xi[2])/4.0 -(1-xi[1])/4.0
|
|
(1-xi[2])/4.0 -(1+xi[1])/4.0
|
|
(1+xi[2])/4.0 (1+xi[1])/4.0
|
|
-(1+xi[2])/4.0 (1-xi[1])/4.0]
|
|
basis = Basis(h, dh)
|
|
MockElement(connectivity, basis, Dict())
|
|
end
|
|
JuliaFEM.get_number_of_basis_functions(el::Type{MockElement}) = 4
|
|
JuliaFEM.get_element_dimension(el::Type{MockElement}) = 2
|
|
|
|
|
|
using JuliaFEM: test_element
|
|
facts("test test_element against mock element") do
|
|
test_element(MockElement)
|
|
end
|
|
|
|
|
|
using JuliaFEM: new_fieldset!, add_field!, Field, get_fieldset
|
|
facts("test adding fieldsets and fields to element") do
|
|
el = MockElement([1, 2, 3, 4])
|
|
fieldset = new_fieldset!(el, "geometry")
|
|
field1 = Field(0.0, [0.0, 0.0, 0.0, 0.0])
|
|
add_field!(el, "geometry", field1)
|
|
field2 = Field(1.0, [1.0, 1.0, 1.0, 1.0])
|
|
add_field!(fieldset, field2)
|
|
fields = get_fieldset(el, "geometry")
|
|
@fact length(fields) --> 2
|
|
@fact fields[1] --> field1
|
|
@fact fields[2] --> field2
|
|
end
|
|
|