Mercurial > repos > public > sbplib_julia

annotate test/testLazyTensors.jl @ 493:df566372bb4f feature/avoid_nested_inflated_tensormappings

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Implement constructors to avoid creating nested InflatedTensorMappings
author Jonatan Werpers <jonatan@werpers.com>
date Thu, 05 Nov 2020 13:18:24 +0100
parents 8082d43103c1
children f906f207571c
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
1 using Test
335
f4e3e71a4ff4 Fix `using` commands to refer to local modules within the Sbplib package/module
Jonatan Werpers <jonatan@werpers.com>
parents: 333
diff changeset
2 using Sbplib.LazyTensors
f4e3e71a4ff4 Fix `using` commands to refer to local modules within the Sbplib package/module
Jonatan Werpers <jonatan@werpers.com>
parents: 333
diff changeset
3 using Sbplib.RegionIndices
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
4
452
aeda2698166d Add tullio as a test dependency and add a test for apply
Jonatan Werpers <jonatan@werpers.com>
parents: 451
diff changeset
5 using Tullio
aeda2698166d Add tullio as a test dependency and add a test for apply
Jonatan Werpers <jonatan@werpers.com>
parents: 451
diff changeset
6
338
2b0c9b30ea3b Add test sets for each submodule to make the test output nicer
Jonatan Werpers <jonatan@werpers.com>
parents: 335
diff changeset
7 @testset "LazyTensors" begin
2b0c9b30ea3b Add test sets for each submodule to make the test output nicer
Jonatan Werpers <jonatan@werpers.com>
parents: 335
diff changeset
8
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
9 @testset "Generic Mapping methods" begin
186
715ff09bb2ce Rename and export types in LazyTensors follow julia conventions
Jonatan Werpers <jonatan@werpers.com>
parents: 184
diff changeset
10 struct DummyMapping{T,R,D} <: TensorMapping{T,R,D} end
274
11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 263
diff changeset
11 LazyTensors.apply(m::DummyMapping{T,R,D}, v, i::NTuple{R,Index{<:Region}}) where {T,R,D} = :apply
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
12 @test range_dim(DummyMapping{Int,2,3}()) == 2
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
13 @test domain_dim(DummyMapping{Int,2,3}()) == 3
274
11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 263
diff changeset
14 @test apply(DummyMapping{Int,2,3}(), zeros(Int, (0,0,0)),(Index{Unknown}(0),Index{Unknown}(0))) == :apply
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
15 end
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
16
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
17 @testset "Mapping transpose" begin
186
715ff09bb2ce Rename and export types in LazyTensors follow julia conventions
Jonatan Werpers <jonatan@werpers.com>
parents: 184
diff changeset
18 struct DummyMapping{T,R,D} <: TensorMapping{T,R,D} end
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
19
321
277dff5b071a LazyTensors/test/runtests.jl: Fix the indexing style of apply methods
Jonatan Werpers <jonatan@werpers.com>
parents: 278
diff changeset
20 LazyTensors.apply(m::DummyMapping{T,R,D}, v, I::Vararg{Index{<:Region},R}) where {T,R,D} = :apply
277dff5b071a LazyTensors/test/runtests.jl: Fix the indexing style of apply methods
Jonatan Werpers <jonatan@werpers.com>
parents: 278
diff changeset
21 LazyTensors.apply_transpose(m::DummyMapping{T,R,D}, v, I::Vararg{Index{<:Region},D}) where {T,R,D} = :apply_transpose
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
22
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
23 LazyTensors.range_size(m::DummyMapping{T,R,D}) where {T,R,D} = :range_size
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
24 LazyTensors.domain_size(m::DummyMapping{T,R,D}) where {T,R,D} = :domain_size
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
25
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
26 m = DummyMapping{Float64,2,3}()
275
591609cdcd9b Dispatch TensorMappingTranspose on region indices. Update tests.
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 274
diff changeset
27 I = Index{Unknown}(0)
232
a20bb4fac23d Improve tests for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 210
diff changeset
28 @test m' isa TensorMapping{Float64, 3,2}
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
29 @test m'' == m
321
277dff5b071a LazyTensors/test/runtests.jl: Fix the indexing style of apply methods
Jonatan Werpers <jonatan@werpers.com>
parents: 278
diff changeset
30 @test apply(m',zeros(Float64,(0,0)), I, I, I) == :apply_transpose
277dff5b071a LazyTensors/test/runtests.jl: Fix the indexing style of apply methods
Jonatan Werpers <jonatan@werpers.com>
parents: 278
diff changeset
31 @test apply(m'',zeros(Float64,(0,0,0)), I, I) == :apply
277dff5b071a LazyTensors/test/runtests.jl: Fix the indexing style of apply methods
Jonatan Werpers <jonatan@werpers.com>
parents: 278
diff changeset
32 @test apply_transpose(m', zeros(Float64,(0,0,0)), I, I) == :apply
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
33
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
34 @test range_size(m') == :domain_size
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
35 @test domain_size(m') == :range_size
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
36 end
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
37
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
38 @testset "TensorApplication" begin
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
39 struct SizeDoublingMapping{T,R,D} <: TensorMapping{T,R,D}
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
40 domain_size::NTuple{D,Int}
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
41 end
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
42
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
43 LazyTensors.apply(m::SizeDoublingMapping{T,R,D}, v, i::Vararg{Index{<:Region},R}) where {T,R,D} = (:apply,v,i)
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
44 LazyTensors.range_size(m::SizeDoublingMapping) = 2 .* m.domain_size
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
45 LazyTensors.domain_size(m::SizeDoublingMapping) = m.domain_size
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
46
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
47
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
48 m = SizeDoublingMapping{Int, 1, 1}((3,))
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
49 v = [0,1,2]
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
50 @test m*v isa AbstractVector{Int}
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
51 @test size(m*v) == 2 .*size(v)
274
11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 263
diff changeset
52 @test (m*v)[Index{Upper}(0)] == (:apply,v,(Index{Upper}(0),))
11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 263
diff changeset
53 @test (m*v)[0] == (:apply,v,(Index{Unknown}(0),))
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
54 @test m*m*v isa AbstractVector{Int}
274
11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 263
diff changeset
55 @test (m*m*v)[Index{Upper}(1)] == (:apply,m*v,(Index{Upper}(1),))
11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 263
diff changeset
56 @test (m*m*v)[1] == (:apply,m*v,(Index{Unknown}(1),))
11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 263
diff changeset
57 @test (m*m*v)[Index{Interior}(3)] == (:apply,m*v,(Index{Interior}(3),))
11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 263
diff changeset
58 @test (m*m*v)[3] == (:apply,m*v,(Index{Unknown}(3),))
11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 263
diff changeset
59 @test (m*m*v)[Index{Lower}(6)] == (:apply,m*v,(Index{Lower}(6),))
11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 263
diff changeset
60 @test (m*m*v)[6] == (:apply,m*v,(Index{Unknown}(6),))
191
25d2ef206fe9 Add a few test cases for TensorApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 186
diff changeset
61 @test_broken BoundsError == (m*m*v)[0]
25d2ef206fe9 Add a few test cases for TensorApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 186
diff changeset
62 @test_broken BoundsError == (m*m*v)[7]
420
3796e296fe64 Add test that triggers bug
Jonatan Werpers <jonatan@werpers.com>
parents: 411
diff changeset
63 @test_throws MethodError m*m
257
d4cd4882ee9f Improve error messages when multiblying with TensorMappings and add some tests for TensorOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 237
diff changeset
64
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
65 m = SizeDoublingMapping{Int, 2, 1}((3,))
276
e3611c2181bc Add 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 275
diff changeset
66 @test_throws MethodError m*ones(Int,2,2)
e3611c2181bc Add 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 275
diff changeset
67 @test_throws MethodError m*m*v
257
d4cd4882ee9f Improve error messages when multiblying with TensorMappings and add some tests for TensorOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 237
diff changeset
68
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
69 m = SizeDoublingMapping{Float64, 2, 2}((3,3))
276
e3611c2181bc Add 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 275
diff changeset
70 v = ones(3,3)
e3611c2181bc Add 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 275
diff changeset
71 I = (Index{Lower}(1),Index{Interior}(2));
e3611c2181bc Add 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 275
diff changeset
72 @test size(m*v) == 2 .*size(v)
277
21e5c1dd6794 Add getindex-function for region indices. Update LazyArray tests. NOTE: Not sure if this is the way to go.
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 276
diff changeset
73 @test (m*v)[I] == (:apply,v,I)
257
d4cd4882ee9f Improve error messages when multiblying with TensorMappings and add some tests for TensorOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 237
diff changeset
74
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
75 struct ScalingOperator{T,D} <: TensorMapping{T,D,D}
257
d4cd4882ee9f Improve error messages when multiblying with TensorMappings and add some tests for TensorOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 237
diff changeset
76 λ::T
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
77 size::NTuple{D,Int}
257
d4cd4882ee9f Improve error messages when multiblying with TensorMappings and add some tests for TensorOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 237
diff changeset
78 end
d4cd4882ee9f Improve error messages when multiblying with TensorMappings and add some tests for TensorOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 237
diff changeset
79
321
277dff5b071a LazyTensors/test/runtests.jl: Fix the indexing style of apply methods
Jonatan Werpers <jonatan@werpers.com>
parents: 278
diff changeset
80 LazyTensors.apply(m::ScalingOperator{T,D}, v, I::Vararg{Index,D}) where {T,D} = m.λ*v[I]
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
81 LazyTensors.range_size(m::ScalingOperator) = m.size
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
82 LazyTensors.domain_size(m::ScalingOperator) = m.size
257
d4cd4882ee9f Improve error messages when multiblying with TensorMappings and add some tests for TensorOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 237
diff changeset
83
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
84 m = ScalingOperator{Int,1}(2,(3,))
278
d6edd37551ea Extend 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 277
diff changeset
85 v = [1,2,3]
d6edd37551ea Extend 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 277
diff changeset
86 @test m*v isa AbstractVector
d6edd37551ea Extend 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 277
diff changeset
87 @test m*v == [2,4,6]
257
d4cd4882ee9f Improve error messages when multiblying with TensorMappings and add some tests for TensorOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 237
diff changeset
88
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
89 m = ScalingOperator{Int,2}(2,(2,2))
278
d6edd37551ea Extend 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 277
diff changeset
90 v = [[1 2];[3 4]]
d6edd37551ea Extend 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 277
diff changeset
91 @test m*v == [[2 4];[6 8]]
d6edd37551ea Extend 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 277
diff changeset
92 I = (Index{Upper}(2),Index{Lower}(1))
d6edd37551ea Extend 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 277
diff changeset
93 @test (m*v)[I] == 6
198
b5c9be7f391c Fix up some formatting
Jonatan Werpers <jonatan@werpers.com>
parents: 191
diff changeset
94 end
200
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
95
210
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
96 @testset "TensorMapping binary operations" begin
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
97 struct ScalarMapping{T,R,D} <: TensorMapping{T,R,D}
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
98 λ::T
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
99 range_size::NTuple{R,Int}
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
100 domain_size::NTuple{D,Int}
210
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
101 end
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
102
321
277dff5b071a LazyTensors/test/runtests.jl: Fix the indexing style of apply methods
Jonatan Werpers <jonatan@werpers.com>
parents: 278
diff changeset
103 LazyTensors.apply(m::ScalarMapping{T,R,D}, v, I::Vararg{Index{<:Region}}) where {T,R,D} = m.λ*v[I...]
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
104 LazyTensors.range_size(m::ScalarMapping) = m.domain_size
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
105 LazyTensors.domain_size(m::ScalarMapping) = m.range_size
210
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
106
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
107 A = ScalarMapping{Float64,1,1}(2.0, (3,), (3,))
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
108 B = ScalarMapping{Float64,1,1}(3.0, (3,), (3,))
210
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
109
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
110 v = [1.1,1.2,1.3]
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
111 for i ∈ eachindex(v)
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
112 @test ((A+B)*v)[i] == 2*v[i] + 3*v[i]
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
113 end
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
114
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
115 for i ∈ eachindex(v)
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
116 @test ((A-B)*v)[i] == 2*v[i] - 3*v[i]
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
117 end
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
118
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
119 @test range_size(A+B) == range_size(A) == range_size(B)
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
120 @test domain_size(A+B) == domain_size(A) == domain_size(B)
210
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
121 end
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
122
200
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
123 @testset "LazyArray" begin
393
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
124 @testset "LazyConstantArray" begin
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
125 @test LazyTensors.LazyConstantArray(3,(3,2)) isa LazyArray{Int,2}
325
41c3c25e4e3b LazyTensors: Simplify the LazyElementwiseOperation type by restricting it and introducing a LazyConstantArray to handle scalars.
Jonatan Werpers <jonatan@werpers.com>
parents: 321
diff changeset
126
393
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
127 lca = LazyTensors.LazyConstantArray(3.0,(3,2))
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
128 @test eltype(lca) == Float64
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
129 @test ndims(lca) == 2
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
130 @test size(lca) == (3,2)
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
131 @test lca[2] == 3.0
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
132 end
200
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
133 struct DummyArray{T,D, T1<:AbstractArray{T,D}} <: LazyArray{T,D}
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
134 data::T1
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
135 end
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
136 Base.size(v::DummyArray) = size(v.data)
237
1c6afdcfd657 Regretsies on the CartesianIndex stuff. Use Vararg instead
Jonatan Werpers <jonatan@werpers.com>
parents: 236
diff changeset
137 Base.getindex(v::DummyArray{T,D}, I::Vararg{Int,D}) where {T,D} = v.data[I...]
200
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
138
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
139 # Test lazy operations
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
140 v1 = [1, 2.3, 4]
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
141 v2 = [1., 2, 3]
263
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
142 s = 3.4
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
143 r_add_v = v1 .+ v2
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
144 r_sub_v = v1 .- v2
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
145 r_times_v = v1 .* v2
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
146 r_div_v = v1 ./ v2
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
147 r_add_s = v1 .+ s
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
148 r_sub_s = v1 .- s
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
149 r_times_s = v1 .* s
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
150 r_div_s = v1 ./ s
200
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
151 @test isa(v1 +̃ v2, LazyArray)
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
152 @test isa(v1 -̃ v2, LazyArray)
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
153 @test isa(v1 *̃ v2, LazyArray)
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
154 @test isa(v1 /̃ v2, LazyArray)
263
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
155 @test isa(v1 +̃ s, LazyArray)
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
156 @test isa(v1 -̃ s, LazyArray)
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
157 @test isa(v1 *̃ s, LazyArray)
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
158 @test isa(v1 /̃ s, LazyArray)
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
159 @test isa(s +̃ v1, LazyArray)
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
160 @test isa(s -̃ v1, LazyArray)
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
161 @test isa(s *̃ v1, LazyArray)
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
162 @test isa(s /̃ v1, LazyArray)
200
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
163 for i ∈ eachindex(v1)
263
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
164 @test (v1 +̃ v2)[i] == r_add_v[i]
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
165 @test (v1 -̃ v2)[i] == r_sub_v[i]
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
166 @test (v1 *̃ v2)[i] == r_times_v[i]
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
167 @test (v1 /̃ v2)[i] == r_div_v[i]
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
168 @test (v1 +̃ s)[i] == r_add_s[i]
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
169 @test (v1 -̃ s)[i] == r_sub_s[i]
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
170 @test (v1 *̃ s)[i] == r_times_s[i]
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
171 @test (v1 /̃ s)[i] == r_div_s[i]
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
172 @test (s +̃ v1)[i] == r_add_s[i]
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
173 @test (s -̃ v1)[i] == -r_sub_s[i]
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
174 @test (s *̃ v1)[i] == r_times_s[i]
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
175 @test (s /̃ v1)[i] == 1/r_div_s[i]
200
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
176 end
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
177 @test_throws BoundsError (v1 +̃ v2)[4]
201
6b7019f2cd41 Add/fix test checking that no size assertion is performed when lazy operations and operations on lazy arrays are decorated using @inbounds
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 200
diff changeset
178 v2 = [1., 2, 3, 4]
6b7019f2cd41 Add/fix test checking that no size assertion is performed when lazy operations and operations on lazy arrays are decorated using @inbounds
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 200
diff changeset
179 # Test that size of arrays is asserted when not specified inbounds
486
8082d43103c1 Add TODOs relating to SizeMismatch
Jonatan Werpers <jonatan@werpers.com>
parents: 485
diff changeset
180 # TODO: Replace these errors with SizeMismatch
207
f85a0b38f3ff Change error type when doing element wise operations on lazy arrays of missmatching dimension
Jonatan Werpers <jonatan@werpers.com>
parents: 206
diff changeset
181 @test_throws DimensionMismatch v1 +̃ v2
200
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
182
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
183 # Test operations on LazyArray
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
184 v1 = DummyArray([1, 2.3, 4])
201
6b7019f2cd41 Add/fix test checking that no size assertion is performed when lazy operations and operations on lazy arrays are decorated using @inbounds
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 200
diff changeset
185 v2 = [1., 2, 3]
200
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
186 @test isa(v1 + v2, LazyArray)
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
187 @test isa(v2 + v1, LazyArray)
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
188 @test isa(v1 - v2, LazyArray)
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
189 @test isa(v2 - v1, LazyArray)
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
190 for i ∈ eachindex(v2)
263
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
191 @test (v1 + v2)[i] == (v2 + v1)[i] == r_add_v[i]
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
192 @test (v1 - v2)[i] == -(v2 - v1)[i] == r_sub_v[i]
200
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
193 end
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
194 @test_throws BoundsError (v1 + v2)[4]
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
195 v2 = [1., 2, 3, 4]
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
196 # Test that size of arrays is asserted when not specified inbounds
486
8082d43103c1 Add TODOs relating to SizeMismatch
Jonatan Werpers <jonatan@werpers.com>
parents: 485
diff changeset
197 # TODO: Replace these errors with SizeMismatch
207
f85a0b38f3ff Change error type when doing element wise operations on lazy arrays of missmatching dimension
Jonatan Werpers <jonatan@werpers.com>
parents: 206
diff changeset
198 @test_throws DimensionMismatch v1 + v2
200
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
199 end
338
2b0c9b30ea3b Add test sets for each submodule to make the test output nicer
Jonatan Werpers <jonatan@werpers.com>
parents: 335
diff changeset
200
377
8414c2334393 Start implementing LazyLinearMap
Jonatan Werpers <jonatan@werpers.com>
parents: 371
diff changeset
201
371
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
202 @testset "LazyFunctionArray" begin
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
203 @test LazyFunctionArray(i->i^2, (3,)) == [1,4,9]
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
204 @test LazyFunctionArray((i,j)->i*j, (3,2)) == [
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
205 1 2;
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
206 2 4;
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
207 3 6;
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
208 ]
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
209
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
210 @test size(LazyFunctionArray(i->i^2, (3,))) == (3,)
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
211 @test size(LazyFunctionArray((i,j)->i*j, (3,2))) == (3,2)
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
212
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
213 @inferred LazyFunctionArray(i->i^2, (3,))[2]
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
214
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
215 @test_throws BoundsError LazyFunctionArray(i->i^2, (3,))[4]
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
216 @test_throws BoundsError LazyFunctionArray((i,j)->i*j, (3,2))[4,2]
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
217 @test_throws BoundsError LazyFunctionArray((i,j)->i*j, (3,2))[2,3]
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
218
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
219 end
371
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
220
412
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
221 @testset "TensorMappingComposition" begin
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
222 A = rand(2,3)
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
223 B = rand(3,4)
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
224
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
225 Ã = LazyLinearMap(A, (1,), (2,))
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
226 B̃ = LazyLinearMap(B, (1,), (2,))
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
227
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
228 @test Ã∘B̃ isa TensorMappingComposition
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
229 @test range_size(Ã∘B̃) == (2,)
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
230 @test domain_size(Ã∘B̃) == (4,)
485
4b49f03bdb98 Switch from DimensionMismatch to SizeMismatch for boundschecks on compositions
Jonatan Werpers <jonatan@werpers.com>
parents: 479
diff changeset
231 @test_throws SizeMismatch B̃∘Ã
417
4c6604b7d990 Add dimension checking in the constructor
Jonatan Werpers <jonatan@werpers.com>
parents: 416
diff changeset
232
4c6604b7d990 Add dimension checking in the constructor
Jonatan Werpers <jonatan@werpers.com>
parents: 416
diff changeset
233 # @test @inbounds B̃∘Ã # Should not error even though dimensions don't match. (Since ]test runs with forced boundschecking this is currently not testable 2020-10-16)
412
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
234
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
235 v = rand(4)
428
6737e769a1ca Reduce exessive tol in TensorMappingComposition tests
Jonatan Werpers <jonatan@werpers.com>
parents: 424
diff changeset
236 @test Ã∘B̃*v ≈ A*B*v rtol=1e-14
412
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
237
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
238 v = rand(2)
428
6737e769a1ca Reduce exessive tol in TensorMappingComposition tests
Jonatan Werpers <jonatan@werpers.com>
parents: 424
diff changeset
239 @test (Ã∘B̃)'*v ≈ B'*A'*v rtol=1e-14
412
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
240 end
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
241
377
8414c2334393 Start implementing LazyLinearMap
Jonatan Werpers <jonatan@werpers.com>
parents: 371
diff changeset
242 @testset "LazyLinearMap" begin
393
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
243 # Test a standard matrix-vector product
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
244 # mapping vectors of size 4 to vectors of size 3.
377
8414c2334393 Start implementing LazyLinearMap
Jonatan Werpers <jonatan@werpers.com>
parents: 371
diff changeset
245 A = rand(3,4)
394
7ad644d112de Expand tests for LazyLinearMap and update its documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 393
diff changeset
246 Ã = LazyLinearMap(A, (1,), (2,))
377
8414c2334393 Start implementing LazyLinearMap
Jonatan Werpers <jonatan@werpers.com>
parents: 371
diff changeset
247 v = rand(4)
411
4aa59af074ef Add implementation of apply_transpose for LazyLinearMap
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
248 w = rand(3)
377
8414c2334393 Start implementing LazyLinearMap
Jonatan Werpers <jonatan@werpers.com>
parents: 371
diff changeset
249
392
418cfd945715 Fix bug in range_size and domain_size for LazyLinearMap and expand the test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 377
diff changeset
250 @test à isa LazyLinearMap{T,1,1} where T
418cfd945715 Fix bug in range_size and domain_size for LazyLinearMap and expand the test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 377
diff changeset
251 @test à isa TensorMapping{T,1,1} where T
394
7ad644d112de Expand tests for LazyLinearMap and update its documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 393
diff changeset
252 @test range_size(Ã) == (3,)
7ad644d112de Expand tests for LazyLinearMap and update its documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 393
diff changeset
253 @test domain_size(Ã) == (4,)
392
418cfd945715 Fix bug in range_size and domain_size for LazyLinearMap and expand the test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 377
diff changeset
254
393
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
255 @test Ã*ones(4) ≈ A*ones(4) atol=5e-13
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
256 @test Ã*v ≈ A*v atol=5e-13
411
4aa59af074ef Add implementation of apply_transpose for LazyLinearMap
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
257 @test Ã'*w ≈ A'*w
377
8414c2334393 Start implementing LazyLinearMap
Jonatan Werpers <jonatan@werpers.com>
parents: 371
diff changeset
258
399
3b4b1758a8ad Add a check for permuted indecies.
Jonatan Werpers <jonatan@werpers.com>
parents: 394
diff changeset
259 A = rand(2,3,4)
3b4b1758a8ad Add a check for permuted indecies.
Jonatan Werpers <jonatan@werpers.com>
parents: 394
diff changeset
260 @test_throws DomainError LazyLinearMap(A, (3,1), (2,))
3b4b1758a8ad Add a check for permuted indecies.
Jonatan Werpers <jonatan@werpers.com>
parents: 394
diff changeset
261
393
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
262 # Test more exotic mappings
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
263 B = rand(3,4,2)
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
264 # Map vectors of size 2 to matrices of size (3,4)
394
7ad644d112de Expand tests for LazyLinearMap and update its documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 393
diff changeset
265 B̃ = LazyLinearMap(B, (1,2), (3,))
393
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
266 v = rand(2)
394
7ad644d112de Expand tests for LazyLinearMap and update its documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 393
diff changeset
267
7ad644d112de Expand tests for LazyLinearMap and update its documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 393
diff changeset
268 @test range_size(B̃) == (3,4)
7ad644d112de Expand tests for LazyLinearMap and update its documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 393
diff changeset
269 @test domain_size(B̃) == (2,)
393
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
270 @test B̃ isa TensorMapping{T,2,1} where T
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
271 @test B̃*ones(2) ≈ B[:,:,1] + B[:,:,2] atol=5e-13
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
272 @test B̃*v ≈ B[:,:,1]*v[1] + B[:,:,2]*v[2] atol=5e-13
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
273
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
274 # Map matrices of size (3,2) to vectors of size 4
394
7ad644d112de Expand tests for LazyLinearMap and update its documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 393
diff changeset
275 B̃ = LazyLinearMap(B, (2,), (1,3))
393
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
276 v = rand(3,2)
394
7ad644d112de Expand tests for LazyLinearMap and update its documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 393
diff changeset
277
7ad644d112de Expand tests for LazyLinearMap and update its documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 393
diff changeset
278 @test range_size(B̃) == (4,)
7ad644d112de Expand tests for LazyLinearMap and update its documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 393
diff changeset
279 @test domain_size(B̃) == (3,2)
393
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
280 @test B̃ isa TensorMapping{T,1,2} where T
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
281 @test B̃*ones(3,2) ≈ B[1,:,1] + B[2,:,1] + B[3,:,1] +
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
282 B[1,:,2] + B[2,:,2] + B[3,:,2] atol=5e-13
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
283 @test B̃*v ≈ B[1,:,1]*v[1,1] + B[2,:,1]*v[2,1] + B[3,:,1]*v[3,1] +
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
284 B[1,:,2]v[1,2] + B[2,:,2]*v[2,2] + B[3,:,2]*v[3,2] atol=5e-13
377
8414c2334393 Start implementing LazyLinearMap
Jonatan Werpers <jonatan@werpers.com>
parents: 371
diff changeset
285
455
b86312d14873 Make split_index type stable
Jonatan Werpers <jonatan@werpers.com>
parents: 453
diff changeset
286
457
8fb6a5611c7a Add tests and docs for flatten_tuple
Jonatan Werpers <jonatan@werpers.com>
parents: 456
diff changeset
287 # TODO:
8fb6a5611c7a Add tests and docs for flatten_tuple
Jonatan Werpers <jonatan@werpers.com>
parents: 456
diff changeset
288 # @inferred (B̃*v)[2]
371
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
289 end
377
8414c2334393 Start implementing LazyLinearMap
Jonatan Werpers <jonatan@werpers.com>
parents: 371
diff changeset
290
429
46acb2560451 Start implementing LazyIdentity
Jonatan Werpers <jonatan@werpers.com>
parents: 411
diff changeset
291
434
648a36ebac99 Change name from LazyIdentity to IdentityMapping
Jonatan Werpers <jonatan@werpers.com>
parents: 433
diff changeset
292 @testset "IdentityMapping" begin
648a36ebac99 Change name from LazyIdentity to IdentityMapping
Jonatan Werpers <jonatan@werpers.com>
parents: 433
diff changeset
293 @test IdentityMapping{Float64}((4,5)) isa IdentityMapping{T,2} where T
648a36ebac99 Change name from LazyIdentity to IdentityMapping
Jonatan Werpers <jonatan@werpers.com>
parents: 433
diff changeset
294 @test IdentityMapping{Float64}((4,5)) isa TensorMapping{T,2,2} where T
441
552189707fbe Add convenience constructor
Jonatan Werpers <jonatan@werpers.com>
parents: 440
diff changeset
295 @test IdentityMapping{Float64}((4,5)) == IdentityMapping{Float64}(4,5)
429
46acb2560451 Start implementing LazyIdentity
Jonatan Werpers <jonatan@werpers.com>
parents: 411
diff changeset
296
442
e70e47fbfa7c Add another convenience constructor
Jonatan Werpers <jonatan@werpers.com>
parents: 441
diff changeset
297 @test IdentityMapping(3,2) isa IdentityMapping{Float64,2}
e70e47fbfa7c Add another convenience constructor
Jonatan Werpers <jonatan@werpers.com>
parents: 441
diff changeset
298
429
46acb2560451 Start implementing LazyIdentity
Jonatan Werpers <jonatan@werpers.com>
parents: 411
diff changeset
299 for sz ∈ [(4,5),(3,),(5,6,4)]
434
648a36ebac99 Change name from LazyIdentity to IdentityMapping
Jonatan Werpers <jonatan@werpers.com>
parents: 433
diff changeset
300 I = IdentityMapping{Float64}(sz)
429
46acb2560451 Start implementing LazyIdentity
Jonatan Werpers <jonatan@werpers.com>
parents: 411
diff changeset
301 v = rand(sz...)
46acb2560451 Start implementing LazyIdentity
Jonatan Werpers <jonatan@werpers.com>
parents: 411
diff changeset
302 @test I*v == v
46acb2560451 Start implementing LazyIdentity
Jonatan Werpers <jonatan@werpers.com>
parents: 411
diff changeset
303 @test I'*v == v
46acb2560451 Start implementing LazyIdentity
Jonatan Werpers <jonatan@werpers.com>
parents: 411
diff changeset
304
46acb2560451 Start implementing LazyIdentity
Jonatan Werpers <jonatan@werpers.com>
parents: 411
diff changeset
305 @test range_size(I) == sz
46acb2560451 Start implementing LazyIdentity
Jonatan Werpers <jonatan@werpers.com>
parents: 411
diff changeset
306 @test domain_size(I) == sz
46acb2560451 Start implementing LazyIdentity
Jonatan Werpers <jonatan@werpers.com>
parents: 411
diff changeset
307 end
46acb2560451 Start implementing LazyIdentity
Jonatan Werpers <jonatan@werpers.com>
parents: 411
diff changeset
308
434
648a36ebac99 Change name from LazyIdentity to IdentityMapping
Jonatan Werpers <jonatan@werpers.com>
parents: 433
diff changeset
309 I = IdentityMapping{Float64}((4,5))
429
46acb2560451 Start implementing LazyIdentity
Jonatan Werpers <jonatan@werpers.com>
parents: 411
diff changeset
310 v = rand(4,5)
46acb2560451 Start implementing LazyIdentity
Jonatan Werpers <jonatan@werpers.com>
parents: 411
diff changeset
311 @inferred (I*v)[3,2]
440
a57b71343aeb Re-add test for inferred types on transposed identity.
Jonatan Werpers <jonatan@werpers.com>
parents: 439
diff changeset
312 @inferred (I'*v)[3,2]
429
46acb2560451 Start implementing LazyIdentity
Jonatan Werpers <jonatan@werpers.com>
parents: 411
diff changeset
313 @inferred range_size(I)
453
c1ae837f1a2e Add some type inferense tests. Failing..
Jonatan Werpers <jonatan@werpers.com>
parents: 452
diff changeset
314
c1ae837f1a2e Add some type inferense tests. Failing..
Jonatan Werpers <jonatan@werpers.com>
parents: 452
diff changeset
315 @inferred range_dim(I)
c1ae837f1a2e Add some type inferense tests. Failing..
Jonatan Werpers <jonatan@werpers.com>
parents: 452
diff changeset
316 @inferred domain_dim(I)
479
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping.
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 471
diff changeset
317
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping.
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 471
diff changeset
318 Ã = rand(4,2)
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping.
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 471
diff changeset
319 A = LazyLinearMap(Ã,(1,),(2,))
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping.
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 471
diff changeset
320 I1 = IdentityMapping{Float64}(2)
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping.
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 471
diff changeset
321 I2 = IdentityMapping{Float64}(4)
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping.
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 471
diff changeset
322 @test A∘I1 == A
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping.
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 471
diff changeset
323 @test I2∘A == A
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping.
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 471
diff changeset
324 @test I1∘I1 == I1
485
4b49f03bdb98 Switch from DimensionMismatch to SizeMismatch for boundschecks on compositions
Jonatan Werpers <jonatan@werpers.com>
parents: 479
diff changeset
325 @test_throws SizeMismatch I1∘A
4b49f03bdb98 Switch from DimensionMismatch to SizeMismatch for boundschecks on compositions
Jonatan Werpers <jonatan@werpers.com>
parents: 479
diff changeset
326 @test_throws SizeMismatch A∘I2
4b49f03bdb98 Switch from DimensionMismatch to SizeMismatch for boundschecks on compositions
Jonatan Werpers <jonatan@werpers.com>
parents: 479
diff changeset
327 @test_throws SizeMismatch I1∘I2
377
8414c2334393 Start implementing LazyLinearMap
Jonatan Werpers <jonatan@werpers.com>
parents: 371
diff changeset
328 end
429
46acb2560451 Start implementing LazyIdentity
Jonatan Werpers <jonatan@werpers.com>
parents: 411
diff changeset
329
448
912ae510dec9 Use new name of IdentityMapping
Jonatan Werpers <jonatan@werpers.com>
parents: 435
diff changeset
330 @testset "InflatedTensorMapping" begin
451
6cf234eef780 Add tests for constructors and size calculations
Jonatan Werpers <jonatan@werpers.com>
parents: 450
diff changeset
331 I(sz...) = IdentityMapping(sz...)
452
aeda2698166d Add tullio as a test dependency and add a test for apply
Jonatan Werpers <jonatan@werpers.com>
parents: 451
diff changeset
332
aeda2698166d Add tullio as a test dependency and add a test for apply
Jonatan Werpers <jonatan@werpers.com>
parents: 451
diff changeset
333 Ã = rand(4,2)
aeda2698166d Add tullio as a test dependency and add a test for apply
Jonatan Werpers <jonatan@werpers.com>
parents: 451
diff changeset
334 B̃ = rand(4,2,3)
aeda2698166d Add tullio as a test dependency and add a test for apply
Jonatan Werpers <jonatan@werpers.com>
parents: 451
diff changeset
335 C̃ = rand(4,2,3)
aeda2698166d Add tullio as a test dependency and add a test for apply
Jonatan Werpers <jonatan@werpers.com>
parents: 451
diff changeset
336
aeda2698166d Add tullio as a test dependency and add a test for apply
Jonatan Werpers <jonatan@werpers.com>
parents: 451
diff changeset
337 A = LazyLinearMap(Ã,(1,),(2,))
aeda2698166d Add tullio as a test dependency and add a test for apply
Jonatan Werpers <jonatan@werpers.com>
parents: 451
diff changeset
338 B = LazyLinearMap(B̃,(1,2),(3,))
aeda2698166d Add tullio as a test dependency and add a test for apply
Jonatan Werpers <jonatan@werpers.com>
parents: 451
diff changeset
339 C = LazyLinearMap(C̃,(1,),(2,3))
451
6cf234eef780 Add tests for constructors and size calculations
Jonatan Werpers <jonatan@werpers.com>
parents: 450
diff changeset
340
6cf234eef780 Add tests for constructors and size calculations
Jonatan Werpers <jonatan@werpers.com>
parents: 450
diff changeset
341 @test InflatedTensorMapping(I(3,2), A, I(4)) isa TensorMapping{Float64, 4, 4}
6cf234eef780 Add tests for constructors and size calculations
Jonatan Werpers <jonatan@werpers.com>
parents: 450
diff changeset
342 @test InflatedTensorMapping(I(3,2), B, I(4)) isa TensorMapping{Float64, 5, 4}
6cf234eef780 Add tests for constructors and size calculations
Jonatan Werpers <jonatan@werpers.com>
parents: 450
diff changeset
343 @test InflatedTensorMapping(I(3), C, I(2,3)) isa TensorMapping{Float64, 4, 5}
466
6eb3f7eb08d6 Add additional tests for InflatedTensorMapping constructors
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 458
diff changeset
344 @test InflatedTensorMapping(C, I(2,3)) isa TensorMapping{Float64, 3, 4}
6eb3f7eb08d6 Add additional tests for InflatedTensorMapping constructors
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 458
diff changeset
345 @test InflatedTensorMapping(I(3), C) isa TensorMapping{Float64, 2, 3}
6eb3f7eb08d6 Add additional tests for InflatedTensorMapping constructors
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 458
diff changeset
346 @test InflatedTensorMapping(I(3), I(2,3)) isa TensorMapping{Float64, 3, 3}
451
6cf234eef780 Add tests for constructors and size calculations
Jonatan Werpers <jonatan@werpers.com>
parents: 450
diff changeset
347
6cf234eef780 Add tests for constructors and size calculations
Jonatan Werpers <jonatan@werpers.com>
parents: 450
diff changeset
348 @test range_size(InflatedTensorMapping(I(3,2), A, I(4))) == (3,2,4,4)
6cf234eef780 Add tests for constructors and size calculations
Jonatan Werpers <jonatan@werpers.com>
parents: 450
diff changeset
349 @test domain_size(InflatedTensorMapping(I(3,2), A, I(4))) == (3,2,2,4)
6cf234eef780 Add tests for constructors and size calculations
Jonatan Werpers <jonatan@werpers.com>
parents: 450
diff changeset
350
6cf234eef780 Add tests for constructors and size calculations
Jonatan Werpers <jonatan@werpers.com>
parents: 450
diff changeset
351 @test range_size(InflatedTensorMapping(I(3,2), B, I(4))) == (3,2,4,2,4)
6cf234eef780 Add tests for constructors and size calculations
Jonatan Werpers <jonatan@werpers.com>
parents: 450
diff changeset
352 @test domain_size(InflatedTensorMapping(I(3,2), B, I(4))) == (3,2,3,4)
6cf234eef780 Add tests for constructors and size calculations
Jonatan Werpers <jonatan@werpers.com>
parents: 450
diff changeset
353
6cf234eef780 Add tests for constructors and size calculations
Jonatan Werpers <jonatan@werpers.com>
parents: 450
diff changeset
354 @test range_size(InflatedTensorMapping(I(3), C, I(2,3))) == (3,4,2,3)
6cf234eef780 Add tests for constructors and size calculations
Jonatan Werpers <jonatan@werpers.com>
parents: 450
diff changeset
355 @test domain_size(InflatedTensorMapping(I(3), C, I(2,3))) == (3,2,3,2,3)
6cf234eef780 Add tests for constructors and size calculations
Jonatan Werpers <jonatan@werpers.com>
parents: 450
diff changeset
356
6cf234eef780 Add tests for constructors and size calculations
Jonatan Werpers <jonatan@werpers.com>
parents: 450
diff changeset
357 @inferred range_size(InflatedTensorMapping(I(3,2), A, I(4))) == (3,2,4,4)
6cf234eef780 Add tests for constructors and size calculations
Jonatan Werpers <jonatan@werpers.com>
parents: 450
diff changeset
358 @inferred domain_size(InflatedTensorMapping(I(3,2), A, I(4))) == (3,2,2,4)
6cf234eef780 Add tests for constructors and size calculations
Jonatan Werpers <jonatan@werpers.com>
parents: 450
diff changeset
359
471
3f3001d1020f Add tests to InflatedTensorMapping where one of before and after are missing
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 466
diff changeset
360 # Test InflatedTensorMapping mapping w. before and after
452
aeda2698166d Add tullio as a test dependency and add a test for apply
Jonatan Werpers <jonatan@werpers.com>
parents: 451
diff changeset
361 tm = InflatedTensorMapping(I(3,2), A, I(4))
aeda2698166d Add tullio as a test dependency and add a test for apply
Jonatan Werpers <jonatan@werpers.com>
parents: 451
diff changeset
362 v = rand(domain_size(tm)...)
aeda2698166d Add tullio as a test dependency and add a test for apply
Jonatan Werpers <jonatan@werpers.com>
parents: 451
diff changeset
363 @tullio IAIv[a,b,c,d] := Ã[c,i]*v[a,b,i,d]
aeda2698166d Add tullio as a test dependency and add a test for apply
Jonatan Werpers <jonatan@werpers.com>
parents: 451
diff changeset
364 @test tm*v ≈ IAIv rtol=1e-14
471
3f3001d1020f Add tests to InflatedTensorMapping where one of before and after are missing
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 466
diff changeset
365 @inferred LazyTensors.split_index(tm,1,1,1,1)
448
912ae510dec9 Use new name of IdentityMapping
Jonatan Werpers <jonatan@werpers.com>
parents: 435
diff changeset
366
471
3f3001d1020f Add tests to InflatedTensorMapping where one of before and after are missing
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 466
diff changeset
367 # Test InflatedTensorMapping mapping w. before
3f3001d1020f Add tests to InflatedTensorMapping where one of before and after are missing
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 466
diff changeset
368 tm = InflatedTensorMapping(I(3,2), A)
3f3001d1020f Add tests to InflatedTensorMapping where one of before and after are missing
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 466
diff changeset
369 v = rand(domain_size(tm)...)
3f3001d1020f Add tests to InflatedTensorMapping where one of before and after are missing
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 466
diff changeset
370 @tullio IAIv[a,b,c] := Ã[c,i]*v[a,b,i]
3f3001d1020f Add tests to InflatedTensorMapping where one of before and after are missing
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 466
diff changeset
371 @test tm*v ≈ IAIv rtol=1e-14
3f3001d1020f Add tests to InflatedTensorMapping where one of before and after are missing
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 466
diff changeset
372 @inferred LazyTensors.split_index(tm,1,1,1)
3f3001d1020f Add tests to InflatedTensorMapping where one of before and after are missing
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 466
diff changeset
373
3f3001d1020f Add tests to InflatedTensorMapping where one of before and after are missing
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 466
diff changeset
374 # Test InflatedTensorMapping mapping w. after
3f3001d1020f Add tests to InflatedTensorMapping where one of before and after are missing
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 466
diff changeset
375 tm = InflatedTensorMapping(A,I(4))
3f3001d1020f Add tests to InflatedTensorMapping where one of before and after are missing
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 466
diff changeset
376 v = rand(domain_size(tm)...)
3f3001d1020f Add tests to InflatedTensorMapping where one of before and after are missing
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 466
diff changeset
377 @tullio IAIv[c,d] := Ã[c,i]*v[i,d]
3f3001d1020f Add tests to InflatedTensorMapping where one of before and after are missing
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 466
diff changeset
378 @test tm*v ≈ IAIv rtol=1e-14
3f3001d1020f Add tests to InflatedTensorMapping where one of before and after are missing
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 466
diff changeset
379 @inferred LazyTensors.split_index(tm,1,1)
457
8fb6a5611c7a Add tests and docs for flatten_tuple
Jonatan Werpers <jonatan@werpers.com>
parents: 456
diff changeset
380
458
41f9cb6ee5a7 Moar tests and moar type stableness
Jonatan Werpers <jonatan@werpers.com>
parents: 457
diff changeset
381 struct ScalingOperator{T,D} <: TensorMapping{T,D,D}
41f9cb6ee5a7 Moar tests and moar type stableness
Jonatan Werpers <jonatan@werpers.com>
parents: 457
diff changeset
382 λ::T
41f9cb6ee5a7 Moar tests and moar type stableness
Jonatan Werpers <jonatan@werpers.com>
parents: 457
diff changeset
383 size::NTuple{D,Int}
41f9cb6ee5a7 Moar tests and moar type stableness
Jonatan Werpers <jonatan@werpers.com>
parents: 457
diff changeset
384 end
41f9cb6ee5a7 Moar tests and moar type stableness
Jonatan Werpers <jonatan@werpers.com>
parents: 457
diff changeset
385
41f9cb6ee5a7 Moar tests and moar type stableness
Jonatan Werpers <jonatan@werpers.com>
parents: 457
diff changeset
386 LazyTensors.apply(m::ScalingOperator{T,D}, v, I::Vararg{Index,D}) where {T,D} = m.λ*v[I]
41f9cb6ee5a7 Moar tests and moar type stableness
Jonatan Werpers <jonatan@werpers.com>
parents: 457
diff changeset
387 LazyTensors.range_size(m::ScalingOperator) = m.size
41f9cb6ee5a7 Moar tests and moar type stableness
Jonatan Werpers <jonatan@werpers.com>
parents: 457
diff changeset
388 LazyTensors.domain_size(m::ScalingOperator) = m.size
41f9cb6ee5a7 Moar tests and moar type stableness
Jonatan Werpers <jonatan@werpers.com>
parents: 457
diff changeset
389
41f9cb6ee5a7 Moar tests and moar type stableness
Jonatan Werpers <jonatan@werpers.com>
parents: 457
diff changeset
390 tm = InflatedTensorMapping(I(2,3),ScalingOperator(2.0, (3,2)),I(3,4))
41f9cb6ee5a7 Moar tests and moar type stableness
Jonatan Werpers <jonatan@werpers.com>
parents: 457
diff changeset
391 v = rand(domain_size(tm)...)
41f9cb6ee5a7 Moar tests and moar type stableness
Jonatan Werpers <jonatan@werpers.com>
parents: 457
diff changeset
392
41f9cb6ee5a7 Moar tests and moar type stableness
Jonatan Werpers <jonatan@werpers.com>
parents: 457
diff changeset
393 @inferred LazyTensors.split_index(tm,1,2,3,2,2,4)
41f9cb6ee5a7 Moar tests and moar type stableness
Jonatan Werpers <jonatan@werpers.com>
parents: 457
diff changeset
394 @inferred apply(tm,v,Index{Unknown}.((1,2,3,2,2,4))...)
41f9cb6ee5a7 Moar tests and moar type stableness
Jonatan Werpers <jonatan@werpers.com>
parents: 457
diff changeset
395 @inferred (tm*v)[1,2,3,2,2,4]
453
c1ae837f1a2e Add some type inferense tests. Failing..
Jonatan Werpers <jonatan@werpers.com>
parents: 452
diff changeset
396
493
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 486
diff changeset
397 @testset "InflatedTensorMapping of InflatedTensorMapping" begin
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 486
diff changeset
398 A = ScalingOperator(2.0,(2,3))
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 486
diff changeset
399 itm = InflatedTensorMapping(I(3,2), A, I(4))
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 486
diff changeset
400 @test InflatedTensorMapping(I(4), itm, I(2)) == InflatedTensorMapping(I(4,3,2), A, I(4,2))
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 486
diff changeset
401 @test InflatedTensorMapping(itm, I(2)) == InflatedTensorMapping(I(3,2), A, I(4,2))
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 486
diff changeset
402 @test InflatedTensorMapping(I(4), itm) == InflatedTensorMapping(I(4,3,2), A, I(4))
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 486
diff changeset
403
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 486
diff changeset
404 @test InflatedTensorMapping(I(2), I(2), I(2)) isa InflatedTensorMapping # The constructor should always return its type.
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 486
diff changeset
405 end
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 486
diff changeset
406
429
46acb2560451 Start implementing LazyIdentity
Jonatan Werpers <jonatan@werpers.com>
parents: 411
diff changeset
407 end
448
912ae510dec9 Use new name of IdentityMapping
Jonatan Werpers <jonatan@werpers.com>
parents: 435
diff changeset
408
456
8f4c31e06689 Add docs and tests for slice_tuple
Jonatan Werpers <jonatan@werpers.com>
parents: 455
diff changeset
409 @testset "slice_tuple" begin
8f4c31e06689 Add docs and tests for slice_tuple
Jonatan Werpers <jonatan@werpers.com>
parents: 455
diff changeset
410 @test LazyTensors.slice_tuple((1,2,3),Val(1), Val(3)) == (1,2,3)
8f4c31e06689 Add docs and tests for slice_tuple
Jonatan Werpers <jonatan@werpers.com>
parents: 455
diff changeset
411 @test LazyTensors.slice_tuple((1,2,3,4,5,6),Val(2), Val(5)) == (2,3,4,5)
8f4c31e06689 Add docs and tests for slice_tuple
Jonatan Werpers <jonatan@werpers.com>
parents: 455
diff changeset
412 @test LazyTensors.slice_tuple((1,2,3,4,5,6),Val(1), Val(3)) == (1,2,3)
8f4c31e06689 Add docs and tests for slice_tuple
Jonatan Werpers <jonatan@werpers.com>
parents: 455
diff changeset
413 @test LazyTensors.slice_tuple((1,2,3,4,5,6),Val(4), Val(6)) == (4,5,6)
448
912ae510dec9 Use new name of IdentityMapping
Jonatan Werpers <jonatan@werpers.com>
parents: 435
diff changeset
414 end
456
8f4c31e06689 Add docs and tests for slice_tuple
Jonatan Werpers <jonatan@werpers.com>
parents: 455
diff changeset
415
457
8fb6a5611c7a Add tests and docs for flatten_tuple
Jonatan Werpers <jonatan@werpers.com>
parents: 456
diff changeset
416 @testset "flatten_tuple" begin
8fb6a5611c7a Add tests and docs for flatten_tuple
Jonatan Werpers <jonatan@werpers.com>
parents: 456
diff changeset
417 @test LazyTensors.flatten_tuple((1,)) == (1,)
8fb6a5611c7a Add tests and docs for flatten_tuple
Jonatan Werpers <jonatan@werpers.com>
parents: 456
diff changeset
418 @test LazyTensors.flatten_tuple((1,2,3,4,5,6)) == (1,2,3,4,5,6)
8fb6a5611c7a Add tests and docs for flatten_tuple
Jonatan Werpers <jonatan@werpers.com>
parents: 456
diff changeset
419 @test LazyTensors.flatten_tuple((1,2,(3,4),5,6)) == (1,2,3,4,5,6)
8fb6a5611c7a Add tests and docs for flatten_tuple
Jonatan Werpers <jonatan@werpers.com>
parents: 456
diff changeset
420 @test LazyTensors.flatten_tuple((1,2,(3,(4,5)),6)) == (1,2,3,4,5,6)
8fb6a5611c7a Add tests and docs for flatten_tuple
Jonatan Werpers <jonatan@werpers.com>
parents: 456
diff changeset
421 @test LazyTensors.flatten_tuple(((1,2),(3,4),(5,),6)) == (1,2,3,4,5,6)
456
8f4c31e06689 Add docs and tests for slice_tuple
Jonatan Werpers <jonatan@werpers.com>
parents: 455
diff changeset
422 end
457
8fb6a5611c7a Add tests and docs for flatten_tuple
Jonatan Werpers <jonatan@werpers.com>
parents: 456
diff changeset
423
8fb6a5611c7a Add tests and docs for flatten_tuple
Jonatan Werpers <jonatan@werpers.com>
parents: 456
diff changeset
424 end