Mercurial > repos > public > sbplib_julia
annotate test/ext/sparse_arrays_test.jl @ 1726:471a948cd2b2 rename_module
Rename project from Sbplib to Diffinitive
author | Vidar Stiernström <vidar.stiernstrom@gmail.com> |
---|---|
date | Sat, 07 Sep 2024 12:11:53 -0700 |
parents | 7274ec5050a0 |
children |
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 |
1726
471a948cd2b2
Rename project from Sbplib to Diffinitive
Vidar Stiernström <vidar.stiernstrom@gmail.com>
parents:
1723
diff
changeset
|
3 using Diffinitive |
471a948cd2b2
Rename project from Sbplib to Diffinitive
Vidar Stiernström <vidar.stiernstrom@gmail.com>
parents:
1723
diff
changeset
|
4 using Diffinitive.Grids |
471a948cd2b2
Rename project from Sbplib to Diffinitive
Vidar Stiernström <vidar.stiernstrom@gmail.com>
parents:
1723
diff
changeset
|
5 using Diffinitive.SbpOperators |
1699
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
6 |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
7 using SparseArrays |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
8 using Tokens |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
9 |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
10 |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
11 @testset "SparseArray" begin |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
12 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
|
13 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
|
14 |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
15 |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
16 @testset let Δ = laplace(g, stencil_set), M = sparse(Δ) |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
17 @test ndims(M) == 2 |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
18 @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
|
19 |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
20 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
|
21 |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
22 Mv = M*reshape(v,:) |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
23 @test Mv ≈ reshape(Δ*v,:) |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
24 end |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
25 |
1722
1dd64b46ca2a
Fix types used for boundary identifiers in tests
Vidar Stiernström <vidar.stiernstrom@gmail.com>
parents:
1699
diff
changeset
|
26 @testset let dₙ = normal_derivative(g, stencil_set,CartesianBoundary{1,LowerBoundary}()), M = sparse(dₙ) |
1699
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
27 @test ndims(M) == 2 |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
28 @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
|
29 |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
30 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
|
31 Mv = M*reshape(v,:) |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
32 @test Mv ≈ reshape(dₙ*v,:) |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
33 end |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
34 end |