Mercurial > repos > public > sbplib_julia

annotate test/BoundaryConditions/sat_test.jl @ 1482:329720b9ba0d feature/boundary_conditions

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add test for 1D grid
author Vidar Stiernström <vidar.stiernstrom@it.uu.se>
date Mon, 25 Dec 2023 19:23:49 +0100
parents b96858a50e35
children 330c39505a94
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1217
ea2e8254820a Update docstrings and start implementing tests
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
1 using Test
ea2e8254820a Update docstrings and start implementing tests
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
2
ea2e8254820a Update docstrings and start implementing tests
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
3
ea2e8254820a Update docstrings and start implementing tests
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
4 using Sbplib.BoundaryConditions
ea2e8254820a Update docstrings and start implementing tests
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
5 using Sbplib.Grids
ea2e8254820a Update docstrings and start implementing tests
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
6 using Sbplib.LazyTensors
1479
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
7 using Sbplib.SbpOperators
1217
ea2e8254820a Update docstrings and start implementing tests
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
8
1479
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
9 stencil_set = read_stencil_set(sbp_operators_path()*"standard_diagonal.toml"; order = 4)
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
10
1482
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
11 struct MockOp end
1217
ea2e8254820a Update docstrings and start implementing tests
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
12
1482
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
13 function BoundaryConditions.sat_tensors(op::MockOp, g::Grid, bc::DirichletCondition)
1479
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
14 e = boundary_restriction(g, stencil_set, id(bc))
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
15 L = e
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
16 sat_op = e'
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
17 return sat_op, L
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
18 end
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
19
1482
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
20 function BoundaryConditions.sat_tensors(op::MockOp, g::Grid, bc::DirichletCondition, a)
1479
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
21 e = boundary_restriction(g, stencil_set, id(bc))
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
22 L = a*e
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
23 sat_op = e'
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
24 return sat_op, L
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
25 end
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
26
1482
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
27 function BoundaryConditions.sat_tensors(op::MockOp, g::Grid, bc::NeumannCondition)
1479
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
28 e = boundary_restriction(g, stencil_set, id(bc))
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
29 d = normal_derivative(g, stencil_set, id(bc))
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
30 L = d
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
31 sat_op = e'
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
32 return sat_op, L
1217
ea2e8254820a Update docstrings and start implementing tests
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
33 end
ea2e8254820a Update docstrings and start implementing tests
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
34
1479
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
35 @testset "sat" begin
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
36 op = MockOp()
1482
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
37 @testset "1D" begin
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
38 grid = equidistant_grid(11, 0., 1.)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
39 l, r = boundary_identifiers(grid)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
40 u = eval_on(grid, x-> 1. + 2x^2)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
41 dc = DirichletCondition(1.0, l)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
42 g_l = discretize_data(grid, dc)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
43 SAT_l = sat(op, grid, dc)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
44 @test SAT_l(u, g_l) ≈ zeros((size(grid))) atol = 1e-13
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
45
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
46 nc = NeumannCondition(4.0, r)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
47 g_r = discretize_data(grid, nc)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
48 SAT_r = sat(op, grid, nc)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
49 @test SAT_r(u, g_r) ≈ zeros((size(grid))) atol = 1e-13
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
50 end
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
51 @testset "2D" begin
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
52 grid = equidistant_grid((11,13), (0.,0.), (1.,1.))
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
53 W, E, S, N = boundary_identifiers(grid)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
54 u = eval_on(grid, (x,y) -> x+y^2)
1479
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
55
1482
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
56 dc_W = DirichletCondition(1.0, W)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
57 SAT_W = sat(op, grid, dc_W)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
58 g_W = discretize_data(grid, dc_W)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
59 r_W = zeros(size(grid))
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
60 r_W[1,:] .= map(y -> (y^2-1.), range(0., 1., length=13))
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
61 @test SAT_W(u, g_W) ≈ r_W atol = 1e-13
1479
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
62
1482
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
63 dc_E = DirichletCondition(2, E)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
64 SAT_E = sat(op, grid, dc_E, 2.)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
65 g_E = discretize_data(grid, dc_E)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
66 r_E = zeros(size(grid))
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
67 r_E[end,:] .= map(y -> (2*(1. + y^2)-2.), range(0., 1., length=13))
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
68 @test SAT_E(u, g_E) ≈ r_E atol = 1e-13
1479
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
69
1482
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
70 nc_S = NeumannCondition(.0, S)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
71 SAT_S = sat(op, grid, nc_S)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
72 g_S = discretize_data(grid, nc_S)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
73 @test SAT_S(u, g_S) ≈ zeros(size(grid)) atol = 1e-13
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
74
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
75 nc_N = NeumannCondition(2.0, N)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
76 SAT_N = sat(op, grid, nc_N)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
77 g_N = discretize_data(grid, nc_N)
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
78 @test SAT_N(u, g_N) ≈ zeros(size(grid)) atol = 1e-13
329720b9ba0d Add test for 1D grid
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1479
diff changeset
79 end
1479
b96858a50e35 Add tests for SAT
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1395
diff changeset
80 end