fix BCs test

test/test_boundary_conditions2D.jl

@@ -1,7 +1,5 @@
 using Test, Suppressor
 using JustRelax
-using ParallelStencil
-@init_parallel_stencil(Threads, Float64, 2)
 
 model = PS_Setup(:cpu, Float64, 2)
 environment!(model)
@@ -10,18 +8,18 @@ environment!(model)
     @suppress begin
     # periodicity
         n = 5 # number of elements
-        Vx, Vy = @rand(n + 1, n + 2), @rand(n + 2, n + 1)
+        Vx, Vy = PTArray(rand(n + 1, n + 2)), PTArray(rand(n + 2, n + 1))
         bcs = FlowBoundaryConditions(;
             no_slip=(left=false, right=false, top=false, bot=false),
             free_slip=(left=false, right=false, top=false, bot=false),
             periodicity=(left=true, right=true, top=true, bot=true),
         )
         flow_bcs!(bcs, Vx, Vy)
 
-        @test @views Vx[:, 1] == Vx[:, end - 1]
-        @test @views Vx[:, end] == Vx[:, 2]
-        @test @views Vy[1, :] == Vy[end - 1, :]
-        @test @views Vy[end, :] == Vy[2, :]
+        @test @views Vx[:  ,   1] == Vx[:      , end - 1]
+        @test @views Vx[:  , end] == Vx[:      ,       2]
+        @test @views Vy[1  ,   :] == Vy[end - 1,       :]
+        @test @views Vy[end,   :] == Vy[2      ,       :]
 
         # free-slip
         bcs = FlowBoundaryConditions(;
@@ -37,6 +35,7 @@ environment!(model)
         @test @views Vy[end, :] == Vy[end - 1, :]
 
         # no-slip
+        Vx, Vy = PTArray(rand(n + 1, n + 2)), PTArray(rand(n + 2, n + 1))
         bcs = FlowBoundaryConditions(;
             no_slip=(left=true, right=true, top=true, bot=true),
             free_slip=(left=false, right=false, top=false, bot=false),
@@ -47,15 +46,17 @@ environment!(model)
         @test sum(!iszero(Vx[end, i]) for i in axes(Vx,2)) == 0
         @test sum(!iszero(Vy[i,   1]) for i in axes(Vy,1)) == 0
         @test sum(!iszero(Vy[i,   1]) for i in axes(Vy,1)) == 0
-        @test @views Vy[1, :]   == -Vy[2, :]
-        @test @views Vy[end, :] == -Vy[end - 1, :]
-        @test @views Vx[:, 1]   == -Vx[:, 2]
-        @test @views Vx[:, end] == -Vx[:, end - 1]
+        @test @views Vy[1  ,   :] == -Vy[2      ,       :]
+        @test @views Vy[end,   :] == -Vy[end - 1,       :]
+        @test @views Vx[:  ,   1] == -Vx[:      ,       2]
+        @test @views Vx[:  , end] == -Vx[:      , end - 1]
 
         # test with StokesArrays
         # periodicity
         ni = 5, 5
         stokes = StokesArrays(ni, ViscoElastic)
+        stokes.V.Vx .= PTArray(rand(n + 1, n + 2))
+        stokes.V.Vy .= PTArray(rand(n + 2, n + 1))
         flow_bcs = FlowBoundaryConditions(;
             no_slip=(left=false, right=false, top=false, bot=false),
             free_slip=(left=false, right=false, top=false, bot=false),
@@ -89,13 +90,13 @@ environment!(model)
         )
         flow_bcs!(stokes, flow_bcs)
 
-        @test @views stokes.V.Vx[1, :] ==
-            stokes.V.Vx[end, :] ==
-            stokes.V.Vy[:, 1] ==
-            stokes.V.Vy[:, end]
-        @test @views stokes.V.Vx[:, 2] == stokes.V.Vx[:, 3] / 3
-        @test @views stokes.V.Vx[:, end-1] == stokes.V.Vx[:, end - 2] / 3
-        @test @views stokes.V.Vy[2, :] == stokes.V.Vy[3, :] / 3
-        @test @views stokes.V.Vy[end-1, :] == stokes.V.Vy[end - 2, :] / 3
+        @test sum(!iszero(stokes.V.Vx[1  , i]) for i in axes(Vx,2)) == 0
+        @test sum(!iszero(stokes.V.Vx[end, i]) for i in axes(Vx,2)) == 0
+        @test sum(!iszero(stokes.V.Vy[i,   1]) for i in axes(Vy,1)) == 0
+        @test sum(!iszero(stokes.V.Vy[i,   1]) for i in axes(Vy,1)) == 0
+        @test @views stokes.V.Vy[1  ,   :] == -stokes.V.Vy[2      ,       :]
+        @test @views stokes.V.Vy[end,   :] == -stokes.V.Vy[end - 1,       :]
+        @test @views stokes.V.Vx[:  ,   1] == -stokes.V.Vx[:      ,       2]
+        @test @views stokes.V.Vx[:  , end] == -stokes.V.Vx[:      , end - 1]
     end
 end