modified: test/Spaces/terrain_warp.jl

CliMA · Jun 17, 2024 · 0b09015 · 0b09015
1 parent dad9d73
commit 0b09015
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/test/Spaces/terrain_warp.jl b/test/Spaces/terrain_warp.jl
@@ -450,6 +450,27 @@ end
             @test yinit123 ≈ y123
             @test yinit123_cpu ≈ y123_cpu
             @test parent(y123_cpu) ≈ Array(parent(y123))
+
+            # test DSS for a Contravariant123Vector
+            y123 = @. Geometry.Contravariant123Vector(y1, y2, y3)
+            y123_cpu = @. Geometry.Contravariant123Vector(y1_cpu, y2_cpu, y3_cpu)
+
+            dss_buffer123 = Spaces.create_dss_buffer(y123)
+            dss_buffer123_cpu = Spaces.create_dss_buffer(y123_cpu)
+
+            # ensure physical velocity is continous across SE boundary for initial state
+            Spaces.weighted_dss!(y123 => dss_buffer123)
+            Spaces.weighted_dss!(y123_cpu => dss_buffer123_cpu)
+
+            yinit123 = copy(y123)
+            yinit123_cpu = copy(y123_cpu)
+
+            Spaces.weighted_dss!(y123, dss_buffer123)
+            Spaces.weighted_dss!(y123_cpu, dss_buffer123_cpu)
+
+            @test yinit123 ≈ y123
+            @test yinit123_cpu ≈ y123_cpu
+            @test parent(y123_cpu) ≈ Array(parent(y123))
         end
     end
 end