Mercurial > repos > public > sbplib_julia

annotate test/SbpOperators/stencil_test.jl @ 1057:0728e84af0dc feature/nested_stencils

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add length method for stencils (grafted from 0be29e65521e8a5183c81d90c1163fe7c93dfffe)
author Jonatan Werpers <jonatan@werpers.com>
date Mon, 07 Feb 2022 14:34:03 +0100
parents df498ce0cf52
children 54e36688dab8
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
1057
0728e84af0dc Add length method for stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1055
diff changeset
12
0728e84af0dc Add length method for stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1055
diff changeset
13 @test length(s) == 5
0728e84af0dc Add length method for stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1055
diff changeset
14 @test length(Stencil((-1,2), (1,2,3,4))) == 4
0728e84af0dc Add length method for stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1055
diff changeset
15
728
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
16 @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
17
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
18 @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
19 @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
20 @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
21
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
22 @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
23 @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
24
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 # 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
26 @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
27 @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
28 @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
29 @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
30
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 @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
32 @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
33 @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
34 @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
35 @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
36 end
728
45966c77cb20 Split tests for SbpOperators over several files
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
37 end
1052
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
38
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
39 @testset "NestedStencil" begin
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
40
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
41 @testset "Constructors" begin
1053
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
42 s1 = CenteredStencil(-1, 1, 0)
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
43 s2 = CenteredStencil(-1, 0, 1)
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
44 s3 = CenteredStencil( 0,-1, 1)
1052
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
45
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
46 ns = NestedStencil(CenteredStencil(s1,s2,s3))
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
47 @test ns isa NestedStencil{Int,3}
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
48
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
49 @test CenteredNestedStencil(s1,s2,s3) == ns
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
50
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
51 @test NestedStencil(s1,s2,s3, center = 2) == ns
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
52 @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
53
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
54 @test NestedStencil((-1,1,0),(-1,0,1),(0,-1,1), center=2) == ns
1055
df498ce0cf52 Add methods for CenteredNestedStencil with tuple input
Jonatan Werpers <jonatan@werpers.com>
parents: 1054
diff changeset
55 @test CenteredNestedStencil((-1,1,0),(-1,0,1),(0,-1,1)) == ns
1054
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
56 @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
57 Stencil(-1, 1, 0; center=1),
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
58 Stencil(-1, 0, 1; center=1),
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
59 Stencil( 0,-1, 1; center=1);
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
60 center=1
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
61 ))
1052
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
62
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
63
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
64 @testset "Error handling" begin
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
65 end
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
66 end
1053
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
67
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
68 @testset "apply" begin
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
69 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
70 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
71
1054
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
72 # Centered
1053
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
73 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
74 @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
75 @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
76
cb0a5f41be02 Add some tests and fix some bugs
Jonatan Werpers <jonatan@werpers.com>
parents: 1052
diff changeset
77 @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
78 @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
79
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
80 # Non-centered
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
81 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
82 @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
83 @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
84
ce16519aa7ff Tests for non centered nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 1053
diff changeset
85 @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
86 @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
87 end
1052
ca718fd4e816 Start implementing nested stencils
Jonatan Werpers <jonatan@werpers.com>
parents: 826
diff changeset
88 end