Mercurial > repos > public > sbplib_julia

comparison LazyTensors/test/runtests.jl @ 274:11010bb74260 boundary_conditions

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply for TensorMappingBinaryOperation on region indices. Update tests. Update todo
author Vidar Stiernström <vidar.stiernstrom@it.uu.se>
date Mon, 06 Jan 2020 10:54:48 +0100
parents b577b5f64530
children 591609cdcd9b
comparison
equal deleted inserted replaced
273:babc4288e6a6 274:11010bb74260
1 using Test 1 using Test
2 using LazyTensors 2 using LazyTensors
3 using RegionIndices
3 4
4 @testset "Generic Mapping methods" begin 5 @testset "Generic Mapping methods" begin
5 struct DummyMapping{T,R,D} <: TensorMapping{T,R,D} end 6 struct DummyMapping{T,R,D} <: TensorMapping{T,R,D} end
6 LazyTensors.apply(m::DummyMapping{T,R,D}, v, i::NTuple{R,Int}) where {T,R,D} = :apply 7 LazyTensors.apply(m::DummyMapping{T,R,D}, v, i::NTuple{R,Index{<:Region}}) where {T,R,D} = :apply
7 @test range_dim(DummyMapping{Int,2,3}()) == 2 8 @test range_dim(DummyMapping{Int,2,3}()) == 2
8 @test domain_dim(DummyMapping{Int,2,3}()) == 3 9 @test domain_dim(DummyMapping{Int,2,3}()) == 3
9 @test apply(DummyMapping{Int,2,3}(), zeros(Int, (0,0,0)),(0,0)) == :apply 10 @test apply(DummyMapping{Int,2,3}(), zeros(Int, (0,0,0)),(Index{Unknown}(0),Index{Unknown}(0))) == :apply
10 end 11 end
11 12
12 @testset "Generic Operator methods" begin 13 @testset "Generic Operator methods" begin
13 struct DummyOperator{T,D} <: TensorOperator{T,D} end 14 struct DummyOperator{T,D} <: TensorOperator{T,D} end
14 @test range_size(DummyOperator{Int,2}(), (3,5)) == (3,5) 15 @test range_size(DummyOperator{Int,2}(), (3,5)) == (3,5)
36 end 37 end
37 38
38 @testset "TensorApplication" begin 39 @testset "TensorApplication" begin
39 struct DummyMapping{T,R,D} <: TensorMapping{T,R,D} end 40 struct DummyMapping{T,R,D} <: TensorMapping{T,R,D} end
40 41
41 LazyTensors.apply(m::DummyMapping{T,R,D}, v, i::NTuple{R,Int}) where {T,R,D} = (:apply,v,i) 42 LazyTensors.apply(m::DummyMapping{T,R,D}, v, i::NTuple{R,Index{<:Region}}) where {T,R,D} = (:apply,v,i)
42 LazyTensors.apply_transpose(m::DummyMapping{T,R,D}, v, i::NTuple{D,Int}) where {T,R,D} = :apply_transpose 43 LazyTensors.apply_transpose(m::DummyMapping{T,R,D}, v, i::NTuple{D,Int}) where {T,R,D} = :apply_transpose
43 44
44 LazyTensors.range_size(m::DummyMapping{T,R,D}, domain_size::NTuple{D,Integer}) where {T,R,D} = 2 .* domain_size 45 LazyTensors.range_size(m::DummyMapping{T,R,D}, domain_size::NTuple{D,Integer}) where {T,R,D} = 2 .* domain_size
45 LazyTensors.domain_size(m::DummyMapping{T,R,D}, range_size::NTuple{R,Integer}) where {T,R,D} = range_size.÷2 46 LazyTensors.domain_size(m::DummyMapping{T,R,D}, range_size::NTuple{R,Integer}) where {T,R,D} = range_size.÷2
46 47
47 48
48 m = DummyMapping{Int, 1, 1}() 49 m = DummyMapping{Int, 1, 1}()
49 v = [0,1,2] 50 v = [0,1,2]
50 @test m*v isa AbstractVector{Int} 51 @test m*v isa AbstractVector{Int}
51 @test size(m*v) == 2 .*size(v) 52 @test size(m*v) == 2 .*size(v)
52 @test (m*v)[0] == (:apply,v,(0,)) 53 @test (m*v)[Index{Upper}(0)] == (:apply,v,(Index{Upper}(0),))
54 @test (m*v)[0] == (:apply,v,(Index{Unknown}(0),))
53 @test m*m*v isa AbstractVector{Int} 55 @test m*m*v isa AbstractVector{Int}
54 @test (m*m*v)[1] == (:apply,m*v,(1,)) 56 @test (m*m*v)[Index{Upper}(1)] == (:apply,m*v,(Index{Upper}(1),))
55 @test (m*m*v)[3] == (:apply,m*v,(3,)) 57 @test (m*m*v)[1] == (:apply,m*v,(Index{Unknown}(1),))
56 @test (m*m*v)[6] == (:apply,m*v,(6,)) 58 @test (m*m*v)[Index{Interior}(3)] == (:apply,m*v,(Index{Interior}(3),))
59 @test (m*m*v)[3] == (:apply,m*v,(Index{Unknown}(3),))
60 @test (m*m*v)[Index{Lower}(6)] == (:apply,m*v,(Index{Lower}(6),))
61 @test (m*m*v)[6] == (:apply,m*v,(Index{Unknown}(6),))
57 @test_broken BoundsError == (m*m*v)[0] 62 @test_broken BoundsError == (m*m*v)[0]
58 @test_broken BoundsError == (m*m*v)[7] 63 @test_broken BoundsError == (m*m*v)[7]
59 64
60 A = DummyMapping{Int, 2, 1}() 65 A = DummyMapping{Int, 2, 1}()
61 66
64 69
65 struct ScalingOperator{T,D} <: TensorOperator{T,D} 70 struct ScalingOperator{T,D} <: TensorOperator{T,D}
66 λ::T 71 λ::T
67 end 72 end
68 73
69 LazyTensors.apply(m::ScalingOperator{T,D}, v, I::Tuple{Int}) where {T,D} = m.λ*v[I...] 74 LazyTensors.apply(m::ScalingOperator{T,D}, v, I::Tuple{Index{<:Region}}) where {T,D} = m.λ*v[I...]
70 75
71 A = ScalingOperator{Int,1}(2) 76 A = ScalingOperator{Int,1}(2)
72 77
73 @test A*[1,2,3] isa AbstractVector 78 @test A*[1,2,3] isa AbstractVector
74 @test A*[1,2,3] == [2,4,6] 79 @test A*[1,2,3] == [2,4,6]
77 @testset "TensorMapping binary operations" begin 82 @testset "TensorMapping binary operations" begin
78 struct ScalarMapping{T,R,D} <: TensorMapping{T,R,D} 83 struct ScalarMapping{T,R,D} <: TensorMapping{T,R,D}
79 λ::T 84 λ::T
80 end 85 end
81 86
82 LazyTensors.apply(m::ScalarMapping{T,R,D}, v, i) where {T,R,D} = m.λ*v[i] 87 LazyTensors.apply(m::ScalarMapping{T,R,D}, v, I::Tuple{Index{<:Region}}) where {T,R,D} = m.λ*v[I...]
83 LazyTensors.range_size(m::ScalarMapping, domain_size) = domain_size 88 LazyTensors.range_size(m::ScalarMapping, domain_size) = domain_size
84 LazyTensors.domain_size(m::ScalarMapping, range_sizes) = range_sizes 89 LazyTensors.domain_size(m::ScalarMapping, range_sizes) = range_sizes
85 90
86 A = ScalarMapping{Float64,1,1}(2.0) 91 A = ScalarMapping{Float64,1,1}(2.0)
87 B = ScalarMapping{Float64,1,1}(3.0) 92 B = ScalarMapping{Float64,1,1}(3.0)
88 93
89 v = [1.1,1.2,1.3] 94 v = [1.1,1.2,1.3]
90
91 for i ∈ eachindex(v) 95 for i ∈ eachindex(v)
92 @test ((A+B)*v)[i] == 2*v[i] + 3*v[i] 96 @test ((A+B)*v)[i] == 2*v[i] + 3*v[i]
93 end 97 end
94 98
95 for i ∈ eachindex(v) 99 for i ∈ eachindex(v)