Mercurial > repos > public > sbplib_julia

annotate test/SbpOperators/stencil_test.jl @ 1054:ce16519aa7ff feature/nested_stencils

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Tests for non centered nested stencils (grafted from 0fe46190a81a3c4cb36911d78e223ff9d398bd44)
author Jonatan Werpers <jonatan@werpers.com>
date Thu, 20 Jan 2022 13:35:21 +0100
parents cb0a5f41be02
children 011d3ff65e9f
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 using Sbplib.SbpOperators
732
6114274447f5 Add missing using and imports in test files
Jonatan Werpers <jonatan@werpers.com>
parents: 728
diff changeset
3 import Sbplib.SbpOperators.Stencil
1052
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
4 import Sbplib.SbpOperators.NestedStencil
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
5 import Sbplib.SbpOperators.scale
728
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
6
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
7 @testset "Stencil" begin
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
8 s = Stencil((-2,2), (1.,2.,2.,3.,4.))
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
9 @test s isa Stencil{Float64, 5}
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
10
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
11 @test eltype(s) == Float64
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
12 @test SbpOperators.scale(s, 2) == Stencil((-2,2), (2.,4.,4.,6.,8.))
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
13
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
14 @test Stencil(1,2,3,4; center=1) == Stencil((0, 3),(1,2,3,4))
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
15 @test Stencil(1,2,3,4; center=2) == Stencil((-1, 2),(1,2,3,4))
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
16 @test Stencil(1,2,3,4; center=4) == Stencil((-3, 0),(1,2,3,4))
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
17
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
18 @test CenteredStencil(1,2,3,4,5) == Stencil((-2, 2), (1,2,3,4,5))
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
19 @test_throws ArgumentError CenteredStencil(1,2,3,4)
826
4433be383840 Add stencil constructor to change the type of the weights along with a convert method
Jonatan Werpers <jonatan@werpers.com>
parents: 732
diff changeset
20
4433be383840 Add stencil constructor to change the type of the weights along with a convert method
Jonatan Werpers <jonatan@werpers.com>
parents: 732
diff changeset
21 # Changing the type of the weights
4433be383840 Add stencil constructor to change the type of the weights along with a convert method
Jonatan Werpers <jonatan@werpers.com>
parents: 732
diff changeset
22 @test Stencil{Float64}(Stencil(1,2,3,4,5; center=2)) == Stencil(1.,2.,3.,4.,5.; center=2)
4433be383840 Add stencil constructor to change the type of the weights along with a convert method
Jonatan Werpers <jonatan@werpers.com>
parents: 732
diff changeset
23 @test Stencil{Float64}(CenteredStencil(1,2,3,4,5)) == CenteredStencil(1.,2.,3.,4.,5.)
4433be383840 Add stencil constructor to change the type of the weights along with a convert method
Jonatan Werpers <jonatan@werpers.com>
parents: 732
diff changeset
24 @test Stencil{Int}(Stencil(1.,2.,3.,4.,5.; center=2)) == Stencil(1,2,3,4,5; center=2)
4433be383840 Add stencil constructor to change the type of the weights along with a convert method
Jonatan Werpers <jonatan@werpers.com>
parents: 732
diff changeset
25 @test Stencil{Rational}(Stencil(1.,2.,3.,4.,5.; center=2)) == Stencil(1//1,2//1,3//1,4//1,5//1; center=2)
4433be383840 Add stencil constructor to change the type of the weights along with a convert method
Jonatan Werpers <jonatan@werpers.com>
parents: 732
diff changeset
26
4433be383840 Add stencil constructor to change the type of the weights along with a convert method
Jonatan Werpers <jonatan@werpers.com>
parents: 732
diff changeset
27 @testset "convert" begin
4433be383840 Add stencil constructor to change the type of the weights along with a convert method
Jonatan Werpers <jonatan@werpers.com>
parents: 732
diff changeset
28 @test convert(Stencil{Float64}, Stencil(1,2,3,4,5; center=2)) == Stencil(1.,2.,3.,4.,5.; center=2)
4433be383840 Add stencil constructor to change the type of the weights along with a convert method
Jonatan Werpers <jonatan@werpers.com>
parents: 732
diff changeset
29 @test convert(Stencil{Float64}, CenteredStencil(1,2,3,4,5)) == CenteredStencil(1.,2.,3.,4.,5.)
4433be383840 Add stencil constructor to change the type of the weights along with a convert method
Jonatan Werpers <jonatan@werpers.com>
parents: 732
diff changeset
30 @test convert(Stencil{Int}, Stencil(1.,2.,3.,4.,5.; center=2)) == Stencil(1,2,3,4,5; center=2)
4433be383840 Add stencil constructor to change the type of the weights along with a convert method
Jonatan Werpers <jonatan@werpers.com>
parents: 732
diff changeset
31 @test convert(Stencil{Rational}, Stencil(1.,2.,3.,4.,5.; center=2)) == Stencil(1//1,2//1,3//1,4//1,5//1; center=2)
4433be383840 Add stencil constructor to change the type of the weights along with a convert method
Jonatan Werpers <jonatan@werpers.com>
parents: 732
diff changeset
32 end
728
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
33 end
1052
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
34
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
35 @testset "NestedStencil" begin
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
36
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
37 @testset "Constructors" begin
1053
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
38 s1 = CenteredStencil(-1, 1, 0)
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
39 s2 = CenteredStencil(-1, 0, 1)
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
40 s3 = CenteredStencil( 0,-1, 1)
1052
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
41
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
42 ns = NestedStencil(CenteredStencil(s1,s2,s3))
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
43 @test ns isa NestedStencil{Int,3}
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
44
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
45 @test CenteredNestedStencil(s1,s2,s3) == ns
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
46
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
47 @test NestedStencil(s1,s2,s3, center = 2) == ns
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
48 @test NestedStencil(s1,s2,s3, center = 1) == NestedStencil(Stencil(s1,s2,s3, center=1))
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
49
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
50 @test NestedStencil((-1,1,0),(-1,0,1),(0,-1,1), center=2) == ns
1054
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
51 @test NestedStencil((-1,1,0),(-1,0,1),(0,-1,1), center=1) == NestedStencil(Stencil(
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
52 Stencil(-1, 1, 0; center=1),
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
53 Stencil(-1, 0, 1; center=1),
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
54 Stencil( 0,-1, 1; center=1);
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
55 center=1
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
56 ))
1052
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
57
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
58
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
59 @testset "Error handling" begin
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
60 end
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
61 end
1053
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
62
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
63 @testset "apply" begin
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
64 c = [ 1, 3, 6, 10, 15, 21, 28, 36, 45, 55]
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
65 v = [ 2, 3, 5, 7, 11, 13, 17, 19, 23, 29]
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
66
1054
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
67 # Centered
1053
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
68 ns = NestedStencil((-1,1,0),(-1,0,1),(0,-2,2), center=2)
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
69 @test SbpOperators.apply_inner_stencils(ns, c, 4) == Stencil(4,9,10; center=2)
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
70 @test SbpOperators.apply_inner_stencils_backwards(ns, c, 4) == Stencil(-5,-9,-8; center=2)
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
71
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
72 @test SbpOperators.apply_stencil(ns, c, v, 4) == 4*5 + 9*7 + 10*11
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
73 @test SbpOperators.apply_stencil_backwards(ns, c, v, 4) == -8*5 - 9*7 - 5*11
1054
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
74
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
75 # Non-centered
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
76 ns = NestedStencil((-1,1,0),(-1,0,1),(0,-1,1), center=1)
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
77 @test SbpOperators.apply_inner_stencils(ns, c, 4) == Stencil(5,11,6; center=1)
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
78 @test SbpOperators.apply_inner_stencils_backwards(ns, c, 4) == Stencil(-4,-7,-3; center=1)
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
79
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
80 @test SbpOperators.apply_stencil(ns, c, v, 4) == 5*7 + 11*11 + 6*13
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
81 @test SbpOperators.apply_stencil_backwards(ns, c, v, 4) == -3*3 - 7*5 - 4*7
1053
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
82 end
1052
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
83 end