Mercurial > repos > public > sbplib_julia

annotate test/ext/sparse_array_kit_test.jl @ 1699:3e9c3986930d feature/lazy_tensors/sparse_conversions

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
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
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1699
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
1 using Test
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
2
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
3 using Sbplib
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
4 using Sbplib.Grids
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
5 using Sbplib.SbpOperators
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
6 using Sbplib.RegionIndices
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
7
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
8 using SparseArrayKit
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
9 using Tokens
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
10 using Tullio
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
11
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
12
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
13 @testset "SparseArray" begin
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
14 g = equidistant_grid((0,0),(1,2), 20,30)
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
15 stencil_set = read_stencil_set(sbp_operators_path()*"standard_diagonal.toml"; order=4)
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
16
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
17
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
18 @testset let Δ = laplace(g, stencil_set), M = SparseArray(Δ)
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
19 @test ndims(M) == 4
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
20 @test size(M) == (20,30,20,30)
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
21
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
22 v = rand(size(g)...)
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
23 @tullio Mv[i,j] := M[i,j,k,l]*v[k,l]
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
24
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
25 @test Mv ≈ Δ*v
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
26 end
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
27
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
28 @testset let dₙ = normal_derivative(g, stencil_set,CartesianBoundary{1,Lower}()), M = SparseArray(dₙ)
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
29 @test ndims(M) == 3
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
30 @test size(M) == (30,20,30)
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
31
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
32 v = rand(size(g)...)
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
33 @tullio Mv[i] := M[i,j,k]*v[j,k]
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
34 @test Mv ≈ dₙ*v
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
35 end
3e9c3986930d Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
36 end