Mercurial > repos > public > sbplib_julia

annotate test/SbpOperators/boundaryops/boundary_restriction_test.jl @ 1652:65b2d2c72fbc feature/sbp_operators/laplace_curvilinear

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add boundary restriction operator for mapped grid
author Jonatan Werpers <jonatan@werpers.com>
date Wed, 26 Jun 2024 12:54:29 +0200
parents 43aaf710463e
children f3d7e2d7a43f
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
728
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
1 using Test
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
2
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
3 using Sbplib.SbpOperators
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
4 using Sbplib.Grids
732
6114274447f5 Add missing using and imports in test files
Jonatan Werpers <jonatan@werpers.com>
parents: 728
diff changeset
5 using Sbplib.LazyTensors
934
af670581b464 Add tests for constructing boundary operators from stencil sets
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 926
diff changeset
6 using Sbplib.RegionIndices
989
7bf3121c6864 Add type StencilSet
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 961
diff changeset
7 using Sbplib.SbpOperators: BoundaryOperator, Stencil
728
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
8
1652
65b2d2c72fbc Add boundary restriction operator for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1529
diff changeset
9 using StaticArrays
65b2d2c72fbc Add boundary restriction operator for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1529
diff changeset
10
728
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
11 @testset "boundary_restriction" begin
1018
5ec49dd2c7c4 Reintroduce read_stencil_set
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 989
diff changeset
12 stencil_set = read_stencil_set(sbp_operators_path()*"standard_diagonal.toml"; order = 4)
772
bea2feebbeca Fix boundaryops tests
Jonatan Werpers <jonatan@werpers.com>
parents: 732
diff changeset
13 e_closure = parse_stencil(stencil_set["e"]["closure"])
1529
43aaf710463e Change to signature of equidistant_grid to same style as many array methods.
Jonatan Werpers <jonatan@werpers.com>
parents: 1282
diff changeset
14 g_1D = equidistant_grid(0.0, 1.0, 11)
43aaf710463e Change to signature of equidistant_grid to same style as many array methods.
Jonatan Werpers <jonatan@werpers.com>
parents: 1282
diff changeset
15 g_2D = equidistant_grid((0.0, 0.0), (1.0,1.0), 11, 15)
728
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
16
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
17 @testset "boundary_restriction" begin
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
18 @testset "1D" begin
1282
11b08b242e48 Make equdistant_grid return an EquidistantGrid for the 1d Case
Jonatan Werpers <jonatan@werpers.com>
parents: 1281
diff changeset
19 e_l = boundary_restriction(g_1D,stencil_set,Lower())
11b08b242e48 Make equdistant_grid return an EquidistantGrid for the 1d Case
Jonatan Werpers <jonatan@werpers.com>
parents: 1281
diff changeset
20 @test e_l == BoundaryOperator(g_1D,Stencil{Float64}(e_closure),Lower())
728
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
21 @test e_l isa BoundaryOperator{T,Lower} where T
995
1ba8a398af9c Rename types
Jonatan Werpers <jonatan@werpers.com>
parents: 961
diff changeset
22 @test e_l isa LazyTensor{T,0,1} where T
728
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
23
1282
11b08b242e48 Make equdistant_grid return an EquidistantGrid for the 1d Case
Jonatan Werpers <jonatan@werpers.com>
parents: 1281
diff changeset
24 e_r = boundary_restriction(g_1D,stencil_set,Upper())
11b08b242e48 Make equdistant_grid return an EquidistantGrid for the 1d Case
Jonatan Werpers <jonatan@werpers.com>
parents: 1281
diff changeset
25 @test e_r == BoundaryOperator(g_1D,Stencil{Float64}(e_closure),Upper())
728
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
26 @test e_r isa BoundaryOperator{T,Upper} where T
995
1ba8a398af9c Rename types
Jonatan Werpers <jonatan@werpers.com>
parents: 961
diff changeset
27 @test e_r isa LazyTensor{T,0,1} where T
728
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
28 end
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
29
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
30 @testset "2D" begin
1281
1cc45207817e Fix tests for boundary operator and boundary restriction
Jonatan Werpers <jonatan@werpers.com>
parents: 1040
diff changeset
31 e_w = boundary_restriction(g_2D,stencil_set,CartesianBoundary{1,Upper}())
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 995
diff changeset
32 @test e_w isa InflatedTensor
995
1ba8a398af9c Rename types
Jonatan Werpers <jonatan@werpers.com>
parents: 961
diff changeset
33 @test e_w isa LazyTensor{T,1,2} where T
728
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
34 end
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
35 end
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
36
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
37 @testset "Application" begin
1652
65b2d2c72fbc Add boundary restriction operator for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1529
diff changeset
38 @testset "EquidistantGrid" begin
1281
1cc45207817e Fix tests for boundary operator and boundary restriction
Jonatan Werpers <jonatan@werpers.com>
parents: 1040
diff changeset
39 e_l, e_r = boundary_restriction.(Ref(g_1D), Ref(stencil_set), boundary_identifiers(g_1D))
1282
11b08b242e48 Make equdistant_grid return an EquidistantGrid for the 1d Case
Jonatan Werpers <jonatan@werpers.com>
parents: 1281
diff changeset
40 v = eval_on(g_1D,x->1+x^2)
728
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
41 u = fill(3.124)
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
42
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
43 @test (e_l*v)[] == v[1]
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
44 @test (e_r*v)[] == v[end]
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
45 @test (e_r*v)[1] == v[end]
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
46 end
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
47
1652
65b2d2c72fbc Add boundary restriction operator for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1529
diff changeset
48 @testset "TensorGrid" begin
1281
1cc45207817e Fix tests for boundary operator and boundary restriction
Jonatan Werpers <jonatan@werpers.com>
parents: 1040
diff changeset
49 e_w, e_e, e_s, e_n = boundary_restriction.(Ref(g_2D), Ref(stencil_set), boundary_identifiers(g_2D))
728
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
50 v = rand(11, 15)
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
51 u = fill(3.124)
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
52
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
53 @test e_w*v == v[1,:]
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
54 @test e_e*v == v[end,:]
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
55 @test e_s*v == v[:,1]
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
56 @test e_n*v == v[:,end]
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
57 end
1652
65b2d2c72fbc Add boundary restriction operator for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1529
diff changeset
58
65b2d2c72fbc Add boundary restriction operator for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1529
diff changeset
59 @testset "MappedGrid" begin
65b2d2c72fbc Add boundary restriction operator for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1529
diff changeset
60 c = Chart(unitsquare()) do (ξ,η)
65b2d2c72fbc Add boundary restriction operator for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1529
diff changeset
61 @SVector[2ξ + η*(1-η), 3η+(1+η/2)*ξ^2]
65b2d2c72fbc Add boundary restriction operator for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1529
diff changeset
62 end
65b2d2c72fbc Add boundary restriction operator for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1529
diff changeset
63 Grids.jacobian(c::typeof(c), (ξ,η)) = @SMatrix[2 1-2η; (2+η)*ξ 3+ξ^2/2]
65b2d2c72fbc Add boundary restriction operator for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1529
diff changeset
64
65b2d2c72fbc Add boundary restriction operator for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1529
diff changeset
65 mg = equidistant_grid(c, 10,13)
65b2d2c72fbc Add boundary restriction operator for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1529
diff changeset
66
65b2d2c72fbc Add boundary restriction operator for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1529
diff changeset
67 e_w, e_e, e_s, e_n = boundary_restriction.(Ref(mg), Ref(stencil_set), boundary_identifiers(mg))
65b2d2c72fbc Add boundary restriction operator for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1529
diff changeset
68 v = rand(10, 13)
65b2d2c72fbc Add boundary restriction operator for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1529
diff changeset
69
65b2d2c72fbc Add boundary restriction operator for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1529
diff changeset
70 @test e_w*v == v[1,:]
65b2d2c72fbc Add boundary restriction operator for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1529
diff changeset
71 @test e_e*v == v[end,:]
65b2d2c72fbc Add boundary restriction operator for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1529
diff changeset
72 @test e_s*v == v[:,1]
65b2d2c72fbc Add boundary restriction operator for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1529
diff changeset
73 @test e_n*v == v[:,end]
65b2d2c72fbc Add boundary restriction operator for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1529
diff changeset
74 end
728
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
75 end
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
76 end