mirror of
https://github.com/JuliaFEM/JuliaFEM.jl.git
synced 2026-08-20 20:22:03 +00:00
Jukka Aho
124 lines
4.2 KiB
Julia
124 lines
4.2 KiB
Julia
# This file is a part of JuliaFEM.
|
|
# License is MIT: see https://github.com/JuliaFEM/JuliaFEM.jl/blob/master/LICENSE.md
|
|
|
|
module ElementTests
|
|
|
|
using JuliaFEM.Test
|
|
|
|
using JuliaFEM: Element, Basis, Field, FieldSet, FunctionSpace, test_element
|
|
|
|
""" Prototype element
|
|
|
|
This should always pass test_element if everything is ok.
|
|
"""
|
|
type MockElement <: Element
|
|
connectivity :: Vector{Int}
|
|
basis :: Basis
|
|
fields :: FieldSet
|
|
end
|
|
|
|
function MockElement(connectivity)
|
|
|
|
h(xi) = 1/4*[
|
|
(1-xi[1])*(1-xi[2])
|
|
(1+xi[1])*(1-xi[2])
|
|
(1+xi[1])*(1+xi[2])
|
|
(1-xi[1])*(1+xi[2])]'
|
|
|
|
dh(xi) = 1/4*[
|
|
-(1-xi[2]) (1-xi[2]) (1+xi[2]) -(1+xi[2])
|
|
-(1-xi[1]) -(1+xi[1]) (1+xi[1]) (1-xi[1])]
|
|
|
|
basis = Basis(h, dh)
|
|
MockElement(connectivity, basis, Dict())
|
|
end
|
|
|
|
Base.size(element::Type{MockElement}) = (2, 4)
|
|
|
|
"""test test_element against mock element"""
|
|
function test_mockelement()
|
|
test_element(MockElement)
|
|
end
|
|
|
|
""" test adding fieldsets and fields to element"""
|
|
function test_add_fields_to_element()
|
|
el = MockElement([1, 2, 3, 4])
|
|
#geometry = Field([0.0, 0.0, 0.0, 0.0])
|
|
el["geometry"] = Field([0.0, 0.0, 0.0, 0.0])
|
|
@test el["geometry"][1].time == 0.0
|
|
@test last(el["geometry"]) == [0.0, 0.0, 0.0, 0.0]
|
|
el["geometry"] = Field(Vector[[0.0, 0.0], [1.0, 0.0], [1.0, 1.0], [0.0, 1.0]])
|
|
@test last(el["geometry"])[3] == [1.0, 1.0]
|
|
el["geometry"] = Vector[[0.0, 0.0], [1.0, 0.0], [1.0, 1.0], [0.0, 1.0]]
|
|
@test last(el["geometry"])[3] == [1.0, 1.0]
|
|
el["geometry"] = [0.0 0.0; 1.0 0.0; 1.0 1.0; 0.0 1.0]'
|
|
@test last(el["geometry"])[3] == [1.0, 1.0]
|
|
el["geometry"] = (0.0, [0.0, 0.0, 0.0, 0.0]), (1.0, [1.0, 1.0, 1.0, 1.0])
|
|
field = el["geometry"]
|
|
@test length(field) == 2 # two time steps
|
|
el["boundary flux"] = (0.0, 0.0), (1.0, 6.0)
|
|
end
|
|
|
|
function test_add_fields_to_element_2()
|
|
el = MockElement([1, 2, 3, 4])
|
|
el["data"] = (0.0 => [1, 2], 1.0 => [2, 3])
|
|
@test length(el["data"]) == 2
|
|
@test el["data"][1].time == 0.0
|
|
@test el["data"][2].time == 1.0
|
|
@test last(el["data"][1]) == [1, 2]
|
|
@test last(el["data"][2]) == [2, 3]
|
|
end
|
|
|
|
function test_add_data_to_element_using_push()
|
|
el = MockElement([1, 2, 3, 4])
|
|
el["data"] = [0, 0, 0, 0]
|
|
|
|
push!(el["data"], [1, 2, 3, 4])
|
|
@test length(el["data"]) == 1
|
|
@test length(el["data"][1]) == 2
|
|
@test el["data"][1].time == 0.0
|
|
|
|
push!(el["data"], 1.0 => [2, 3, 4, 5]) # creates new timestep at t=1.0
|
|
push!(el["data"], [3, 4, 5, 6]) # adds new increment data to last timestep
|
|
@test length(el["data"]) == 2
|
|
@test length(el["data"][2]) == 2
|
|
@test el["data"][2].time == 1.0
|
|
|
|
end
|
|
|
|
#=
|
|
facts("interpolation of fields in some function space") do
|
|
|
|
el = MockElement([1, 2, 3, 4])
|
|
fieldset1 = FieldSet("geometry", [Field(0.0, Vector[[0.0, 0.0], [1.0, 0.0], [1.0, 1.0], [0.0, 1.0]])])
|
|
fieldset2 = FieldSet("constant scalar field", [Field(0.0, 1.0)])
|
|
fieldset3 = FieldSet("scalar field", [Field(0.0, [1.0, 2.0, 3.0, 4.0])])
|
|
fieldset4 = FieldSet("vector field 1", [Field(0.0, Vector[[1.0], [2.0], [3.0], [4.0]])])
|
|
fieldset5 = FieldSet("vector field 2", [Field(0.0, Vector[[1.0, 5.0], [2.0, 6.0], [3.0, 7.0], [4.0, 8.0]])])
|
|
fieldset6 = FieldSet("vector field 3", [Field(0.0, Vector[[1.0, 5.0, 9.0], [2.0, 6.0, 10.0], [3.0, 7.0, 11.0], [4.0, 8.0, 12.0]])])
|
|
fieldset7 = FieldSet("tensor field 1", [Field(0.0, Matrix[[1.0 5.0; 9.0 13.0], [2.0 6.0; 10.0 14.0], [3.0 7.0; 11.0 15.0], [4.0 8.0; 12.0 16.0]])])
|
|
|
|
element["geometry"] = fieldset1
|
|
element["constant scalar field"] = fieldset2
|
|
element["scalar field"] = fieldset3
|
|
element["vector field 1"] = fieldset4
|
|
element["vector field 2"] = fieldset5
|
|
element["vector field 3"] = fieldset6
|
|
element["tensor field 1"] = fieldset7
|
|
|
|
xi = [0.0, 0.0]
|
|
t = 0.0
|
|
u = FunctionSpace(element)
|
|
v = FunctionSpace(element)
|
|
|
|
@fact v("constant scalar field", xi, t) --> 1.0
|
|
@fact v("scalar field", xi, t) --> 1/4*(1+2+3+4)
|
|
@fact v("vector field 1", xi, t) --> [1/4*(1+2+3+4)]
|
|
@fact v("vector field 2", xi, t) --> 1/4*[1+2+3+4, 5+6+7+8]
|
|
@fact v("vector field 3", xi, t) --> 1/4*[1+2+3+4, 5+6+7+8, 9+10+11+12]
|
|
@fact v("tensor field 1", xi, t) --> 1/4*[1+2+3+4 5+6+7+8; 9+10+11+12 13+14+15+16]
|
|
end
|
|
=#
|
|
|
|
end
|