@@ -381,16 +381,16 @@ import WaterLily: ×
381381 # stress tensor
382382 u₂ = zeros (N,N,2 ) |> f
383383 u₃ = zeros (N,N,N,3 ) |> f
384- @test GPUArrays. @allowscalar all (WaterLily.∇²u (CartesianIndex (N÷ 2 ,N÷ 2 ),u₂) .≈ 0 )
385- @test GPUArrays. @allowscalar all (WaterLily.∇²u (CartesianIndex (N÷ 2 ,N÷ 2 ,N÷ 2 ),u₃) .≈ 0 )
384+ @test GPUArrays. @allowscalar all (WaterLily. S (CartesianIndex (N÷ 2 ,N÷ 2 ),u₂) .≈ 0 )
385+ @test GPUArrays. @allowscalar all (WaterLily. S (CartesianIndex (N÷ 2 ,N÷ 2 ,N÷ 2 ),u₃) .≈ 0 )
386386 apply! ((i,x)-> x[i],u₂) # uniform gradient
387387 apply! ((i,x)-> x[i],u₃)
388- @test GPUArrays. @allowscalar all (WaterLily.∇²u (CartesianIndex (N÷ 2 ,N÷ 2 ),u₂) .≈ SA[2 0 ; 0 2 ])
389- @test GPUArrays. @allowscalar all (WaterLily.∇²u (CartesianIndex (N÷ 2 ,N÷ 2 ,N÷ 2 ),u₃) .≈ SA[2 0 0 ; 0 2 0 ; 0 0 2 ])
388+ @test GPUArrays. @allowscalar all (WaterLily. S (CartesianIndex (N÷ 2 ,N÷ 2 ),u₂) .≈ SA[2 0 ; 0 2 ])
389+ @test GPUArrays. @allowscalar all (WaterLily. S (CartesianIndex (N÷ 2 ,N÷ 2 ,N÷ 2 ),u₃) .≈ SA[2 0 0 ; 0 2 0 ; 0 0 2 ])
390390 apply! ((i,x)-> x[i% 2 + 1 ],u₂) # shear
391391 apply! ((i,x)-> x[i% 3 + 1 ],u₃)
392- @test GPUArrays. @allowscalar all (WaterLily.∇²u (CartesianIndex (N÷ 2 ,N÷ 2 ),u₂) .≈ SA[0 2 ; 2 0 ])
393- @test GPUArrays. @allowscalar all (WaterLily.∇²u (CartesianIndex (N÷ 2 ,N÷ 2 ,N÷ 2 ),u₃) .≈ SA[0 1 1 ; 1 0 1 ; 1 1 0 ])
392+ @test GPUArrays. @allowscalar all (WaterLily. S (CartesianIndex (N÷ 2 ,N÷ 2 ),u₂) .≈ SA[0 2 ; 2 0 ])
393+ @test GPUArrays. @allowscalar all (WaterLily. S (CartesianIndex (N÷ 2 ,N÷ 2 ,N÷ 2 ),u₃) .≈ SA[0 1 1 ; 1 0 1 ; 1 1 0 ])
394394 # viscous force
395395 u₂ .= 0 ; u₃ .= 0
396396 @test all (WaterLily. viscous_force (u₂,1.0 ,df₂,body) .≈ 0 )
0 commit comments