Mercurial > repos > public > sbplib_julia
diff test/testSbpOperators.jl @ 592:4781e759d92f refactor/toml_operator_format
Merge default
author | Jonatan Werpers <jonatan@werpers.com> |
---|---|
date | Wed, 02 Dec 2020 14:20:24 +0100 |
parents | 8e4f86c4bf75 |
children | cc86b920531a e40e7439d1b4 |
line wrap: on
line diff
--- a/test/testSbpOperators.jl Wed Dec 02 14:19:37 2020 +0100 +++ b/test/testSbpOperators.jl Wed Dec 02 14:20:24 2020 +0100 @@ -180,67 +180,144 @@ @test_broken Qinv*(Q*v) ≈ v @test Qinv*v == Qinv'*v end -# -# @testset "BoundaryValue" begin -# op = readOperator(sbp_operators_path()*"d2_4th.txt",sbp_operators_path()*"h_4th.txt") -# g = EquidistantGrid((4,5), (0.0, 0.0), (1.0,1.0)) -# -# e_w = BoundaryValue(op, g, CartesianBoundary{1,Lower}()) -# e_e = BoundaryValue(op, g, CartesianBoundary{1,Upper}()) -# e_s = BoundaryValue(op, g, CartesianBoundary{2,Lower}()) -# e_n = BoundaryValue(op, g, CartesianBoundary{2,Upper}()) -# -# v = zeros(Float64, 4, 5) -# v[:,5] = [1, 2, 3,4] -# v[:,4] = [1, 2, 3,4] -# v[:,3] = [4, 5, 6, 7] -# v[:,2] = [7, 8, 9, 10] -# v[:,1] = [10, 11, 12, 13] -# -# @test e_w isa TensorMapping{T,2,1} where T -# @test e_w' isa TensorMapping{T,1,2} where T -# -# @test domain_size(e_w, (3,2)) == (2,) -# @test domain_size(e_e, (3,2)) == (2,) -# @test domain_size(e_s, (3,2)) == (3,) -# @test domain_size(e_n, (3,2)) == (3,) -# -# @test size(e_w'*v) == (5,) -# @test size(e_e'*v) == (5,) -# @test size(e_s'*v) == (4,) -# @test size(e_n'*v) == (4,) -# -# @test e_w'*v == [10,7,4,1.0,1] -# @test e_e'*v == [13,10,7,4,4.0] -# @test e_s'*v == [10,11,12,13.0] -# @test e_n'*v == [1,2,3,4.0] -# -# g_x = [1,2,3,4.0] -# g_y = [5,4,3,2,1.0] -# -# G_w = zeros(Float64, (4,5)) -# G_w[1,:] = g_y -# -# G_e = zeros(Float64, (4,5)) -# G_e[4,:] = g_y -# -# G_s = zeros(Float64, (4,5)) -# G_s[:,1] = g_x -# -# G_n = zeros(Float64, (4,5)) -# G_n[:,5] = g_x -# -# @test size(e_w*g_y) == (UnknownDim,5) -# @test size(e_e*g_y) == (UnknownDim,5) -# @test size(e_s*g_x) == (4,UnknownDim) -# @test size(e_n*g_x) == (4,UnknownDim) -# -# # These tests should be moved to where they are possible (i.e we know what the grid should be) -# @test_broken e_w*g_y == G_w -# @test_broken e_e*g_y == G_e -# @test_broken e_s*g_x == G_s -# @test_broken e_n*g_x == G_n -# end + +@testset "BoundaryRestrictrion" begin + op = readOperator(sbp_operators_path()*"d2_4th.txt",sbp_operators_path()*"h_4th.txt") + g_1D = EquidistantGrid(11, 0.0, 1.0) + g_2D = EquidistantGrid((11,15), (0.0, 0.0), (1.0,1.0)) + + @testset "Constructors" begin + @testset "1D" begin + e_l = BoundaryRestriction{Lower}(op.eClosure,size(g_1D)[1]) + @test e_l == BoundaryRestriction(g_1D,op.eClosure,Lower()) + @test e_l == boundary_restriction(g_1D,op.eClosure,CartesianBoundary{1,Lower}()) + @test e_l isa TensorMapping{T,0,1} where T + + e_r = BoundaryRestriction{Upper}(op.eClosure,size(g_1D)[1]) + @test e_r == BoundaryRestriction(g_1D,op.eClosure,Upper()) + @test e_r == boundary_restriction(g_1D,op.eClosure,CartesianBoundary{1,Upper}()) + @test e_r isa TensorMapping{T,0,1} where T + end + + @testset "2D" begin + e_w = boundary_restriction(g_2D,op.eClosure,CartesianBoundary{1,Upper}()) + @test e_w isa InflatedTensorMapping + @test e_w isa TensorMapping{T,1,2} where T + end + end + + e_l = boundary_restriction(g_1D, op.eClosure, CartesianBoundary{1,Lower}()) + e_r = boundary_restriction(g_1D, op.eClosure, CartesianBoundary{1,Upper}()) + + e_w = boundary_restriction(g_2D, op.eClosure, CartesianBoundary{1,Lower}()) + e_e = boundary_restriction(g_2D, op.eClosure, CartesianBoundary{1,Upper}()) + e_s = boundary_restriction(g_2D, op.eClosure, CartesianBoundary{2,Lower}()) + e_n = boundary_restriction(g_2D, op.eClosure, CartesianBoundary{2,Upper}()) + + @testset "Sizes" begin + @testset "1D" begin + @test domain_size(e_l) == (11,) + @test domain_size(e_r) == (11,) + + @test range_size(e_l) == () + @test range_size(e_r) == () + end + + @testset "2D" begin + @test domain_size(e_w) == (11,15) + @test domain_size(e_e) == (11,15) + @test domain_size(e_s) == (11,15) + @test domain_size(e_n) == (11,15) + + @test range_size(e_w) == (15,) + @test range_size(e_e) == (15,) + @test range_size(e_s) == (11,) + @test range_size(e_n) == (11,) + end + end + + + @testset "Application" begin + @testset "1D" begin + v = evalOn(g_1D,x->1+x^2) + u = fill(3.124) + @test (e_l*v)[] == v[1] + @test (e_r*v)[] == v[end] + @test (e_r*v)[1] == v[end] + @test e_l'*u == [u[]; zeros(10)] + @test e_r'*u == [zeros(10); u[]] + end + + @testset "2D" begin + v = rand(11, 15) + u = fill(3.124) + + @test e_w*v == v[1,:] + @test e_e*v == v[end,:] + @test e_s*v == v[:,1] + @test e_n*v == v[:,end] + + + g_x = rand(11) + g_y = rand(15) + + G_w = zeros(Float64, (11,15)) + G_w[1,:] = g_y + + G_e = zeros(Float64, (11,15)) + G_e[end,:] = g_y + + G_s = zeros(Float64, (11,15)) + G_s[:,1] = g_x + + G_n = zeros(Float64, (11,15)) + G_n[:,end] = g_x + + @test e_w'*g_y == G_w + @test e_e'*g_y == G_e + @test e_s'*g_x == G_s + @test e_n'*g_x == G_n + end + + @testset "Regions" begin + u = fill(3.124) + @test (e_l'*u)[Index(1,Lower)] == 3.124 + @test (e_l'*u)[Index(2,Lower)] == 0 + @test (e_l'*u)[Index(6,Interior)] == 0 + @test (e_l'*u)[Index(10,Upper)] == 0 + @test (e_l'*u)[Index(11,Upper)] == 0 + + @test (e_r'*u)[Index(1,Lower)] == 0 + @test (e_r'*u)[Index(2,Lower)] == 0 + @test (e_r'*u)[Index(6,Interior)] == 0 + @test (e_r'*u)[Index(10,Upper)] == 0 + @test (e_r'*u)[Index(11,Upper)] == 3.124 + end + end + + @testset "Inferred" begin + v = ones(Float64, 11) + u = fill(1.) + + @inferred apply(e_l, v) + @inferred apply(e_r, v) + + @inferred apply_transpose(e_l, u, 4) + @inferred apply_transpose(e_l, u, Index(1,Lower)) + @inferred apply_transpose(e_l, u, Index(2,Lower)) + @inferred apply_transpose(e_l, u, Index(6,Interior)) + @inferred apply_transpose(e_l, u, Index(10,Upper)) + @inferred apply_transpose(e_l, u, Index(11,Upper)) + + @inferred apply_transpose(e_r, u, 4) + @inferred apply_transpose(e_r, u, Index(1,Lower)) + @inferred apply_transpose(e_r, u, Index(2,Lower)) + @inferred apply_transpose(e_r, u, Index(6,Interior)) + @inferred apply_transpose(e_r, u, Index(10,Upper)) + @inferred apply_transpose(e_r, u, Index(11,Upper)) + end + +end # # @testset "NormalDerivative" begin # op = readOperator(sbp_operators_path()*"d2_4th.txt",sbp_operators_path()*"h_4th.txt")