mirror of
https://github.com/JuliaFEM/JuliaFEM.jl.git
synced 2026-09-10 13:17:42 +00:00
Jukka Aho
43 lines
1.3 KiB
Julia
43 lines
1.3 KiB
Julia
# This file is a part of JuliaFEM.
|
|
# License is MIT: see https://github.com/JuliaFEM/JuliaFEM.jl/blob/master/LICENSE.md
|
|
|
|
module AssemblyTests
|
|
|
|
using JuliaFEM.Test
|
|
using JuliaFEM.Core: Quad4, Seg2, FieldSet, Field, HeatProblem
|
|
using JuliaFEM.Core: Assembly, assemble!
|
|
|
|
"""assemble a simple two element problem and solve"""
|
|
function test_assembly()
|
|
info("create elements")
|
|
el1 = Quad4([1, 2, 3, 4])
|
|
el1["geometry"] = Vector[[0.0, 0.0], [1.0, 0.0], [1.0, 1.0], [0.0, 1.0]]
|
|
el1["temperature thermal conductivity"] = 6.0
|
|
el1["temperature load"] = 12.0
|
|
el1["density"] = 36.0
|
|
el2 = Seg2([1, 2])
|
|
el2["geometry"] = Vector[[0.0, 0.0], [1.0, 0.0]]
|
|
# Boundary load, linear ramp 0 -> 600 at time 0 -> 1
|
|
el2["temperature flux"] = ((0.0 => 0.0), (1.0 => 600.0))
|
|
info("element created")
|
|
|
|
problem = HeatProblem()
|
|
info("problem created. pushing elements")
|
|
push!(problem, el1)
|
|
push!(problem, el2)
|
|
|
|
info("creating assembly from equations")
|
|
assembly = Assembly()
|
|
assemble!(assembly, problem, 1.0)
|
|
info("solving")
|
|
|
|
free_dofs = [1, 2]
|
|
A = full(assembly.stiffness_matrix)[free_dofs, free_dofs]
|
|
b = full(assembly.force_vector)[free_dofs]
|
|
u = A \ b
|
|
info("solution u=$u")
|
|
@test isapprox(u, [101.0, 101.0])
|
|
end
|
|
|
|
end
|