Mercurial > repos > public > sbplib_julia
comparison test/ext/sparse_array_kit_test.jl @ 1699:3e9c3986930d feature/lazy_tensors/sparse_conversions
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
| author | Jonatan Werpers <jonatan@werpers.com> |
|---|---|
| date | Mon, 02 Sep 2024 15:35:54 +0200 |
| parents | |
| children | 1dd64b46ca2a |
comparison
equal
deleted
inserted
replaced
| 1676:5a1f51b4e3d9 | 1699:3e9c3986930d |
|---|---|
| 1 using Test | |
| 2 | |
| 3 using Sbplib | |
| 4 using Sbplib.Grids | |
| 5 using Sbplib.SbpOperators | |
| 6 using Sbplib.RegionIndices | |
| 7 | |
| 8 using SparseArrayKit | |
| 9 using Tokens | |
| 10 using Tullio | |
| 11 | |
| 12 | |
| 13 @testset "SparseArray" begin | |
| 14 g = equidistant_grid((0,0),(1,2), 20,30) | |
| 15 stencil_set = read_stencil_set(sbp_operators_path()*"standard_diagonal.toml"; order=4) | |
| 16 | |
| 17 | |
| 18 @testset let Δ = laplace(g, stencil_set), M = SparseArray(Δ) | |
| 19 @test ndims(M) == 4 | |
| 20 @test size(M) == (20,30,20,30) | |
| 21 | |
| 22 v = rand(size(g)...) | |
| 23 @tullio Mv[i,j] := M[i,j,k,l]*v[k,l] | |
| 24 | |
| 25 @test Mv ≈ Δ*v | |
| 26 end | |
| 27 | |
| 28 @testset let dₙ = normal_derivative(g, stencil_set,CartesianBoundary{1,Lower}()), M = SparseArray(dₙ) | |
| 29 @test ndims(M) == 3 | |
| 30 @test size(M) == (30,20,30) | |
| 31 | |
| 32 v = rand(size(g)...) | |
| 33 @tullio Mv[i] := M[i,j,k]*v[j,k] | |
| 34 @test Mv ≈ dₙ*v | |
| 35 end | |
| 36 end |