Mercurial > repos > public > sbplib_julia

annotate test/testLazyTensors.jl @ 417:4c6604b7d990 feature/tensor_composition

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add dimension checking in the constructor
author Jonatan Werpers <jonatan@werpers.com>
date Fri, 16 Oct 2020 20:32:09 +0200
parents ebc9b2383dae
children 98ba9b7effd3
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
338
2b0c9b30ea3b Add test sets for each submodule to make the test output nicer
Jonatan Werpers <jonatan@werpers.com>
parents: 335
diff changeset
5 @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
6
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
7 @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
8 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
9 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
10 @test range_dim(DummyMapping{Int,2,3}()) == 2
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
11 @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
12 @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
13 end
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
14
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
15 @testset "Mapping transpose" begin
186
715ff09bb2ce Rename and export types in LazyTensors follow julia conventions
Jonatan Werpers <jonatan@werpers.com>
parents: 184
diff changeset
16 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
17
321
277dff5b071a LazyTensors/test/runtests.jl: Fix the indexing style of apply methods
Jonatan Werpers <jonatan@werpers.com>
parents: 278
diff changeset
18 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
19 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
20
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
21 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
22 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
23
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
24 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
25 I = Index{Unknown}(0)
232
a20bb4fac23d Improve tests for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 210
diff changeset
26 @test m' isa TensorMapping{Float64, 3,2}
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
27 @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
28 @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
29 @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
30 @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
31
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
32 @test range_size(m') == :domain_size
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
33 @test domain_size(m') == :range_size
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
34 end
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
35
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
36 @testset "TensorApplication" begin
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
37 struct SizeDoublingMapping{T,R,D} <: TensorMapping{T,R,D}
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
38 domain_size::NTuple{D,Int}
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
39 end
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
40
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
41 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
42 LazyTensors.range_size(m::SizeDoublingMapping) = 2 .* m.domain_size
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
43 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
44
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
45
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
46 m = SizeDoublingMapping{Int, 1, 1}((3,))
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
47 v = [0,1,2]
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
48 @test m*v isa AbstractVector{Int}
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
49 @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
50 @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
51 @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
52 @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
53 @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
54 @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
55 @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
56 @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
57 @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
58 @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
59 @test_broken BoundsError == (m*m*v)[0]
25d2ef206fe9 Add a few test cases for TensorApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 186
diff changeset
60 @test_broken BoundsError == (m*m*v)[7]
257
d4cd4882ee9f Improve error messages when multiblying with TensorMappings and add some tests for TensorOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 237
diff changeset
61
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
62 m = SizeDoublingMapping{Int, 2, 1}((3,))
276
e3611c2181bc Add 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 275
diff changeset
63 @test_throws MethodError m*ones(Int,2,2)
e3611c2181bc Add 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 275
diff changeset
64 @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
65
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
66 m = SizeDoublingMapping{Float64, 2, 2}((3,3))
276
e3611c2181bc Add 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 275
diff changeset
67 v = ones(3,3)
e3611c2181bc Add 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 275
diff changeset
68 I = (Index{Lower}(1),Index{Interior}(2));
e3611c2181bc Add 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 275
diff changeset
69 @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
70 @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
71
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
72 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
73 λ::T
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
74 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
75 end
d4cd4882ee9f Improve error messages when multiblying with TensorMappings and add some tests for TensorOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 237
diff changeset
76
321
277dff5b071a LazyTensors/test/runtests.jl: Fix the indexing style of apply methods
Jonatan Werpers <jonatan@werpers.com>
parents: 278
diff changeset
77 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
78 LazyTensors.range_size(m::ScalingOperator) = m.size
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
79 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
80
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
81 m = ScalingOperator{Int,1}(2,(3,))
278
d6edd37551ea Extend 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 277
diff changeset
82 v = [1,2,3]
d6edd37551ea Extend 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 277
diff changeset
83 @test m*v isa AbstractVector
d6edd37551ea Extend 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 277
diff changeset
84 @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
85
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
86 m = ScalingOperator{Int,2}(2,(2,2))
278
d6edd37551ea Extend 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 277
diff changeset
87 v = [[1 2];[3 4]]
d6edd37551ea Extend 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 277
diff changeset
88 @test m*v == [[2 4];[6 8]]
d6edd37551ea Extend 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 277
diff changeset
89 I = (Index{Upper}(2),Index{Lower}(1))
d6edd37551ea Extend 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 277
diff changeset
90 @test (m*v)[I] == 6
198
b5c9be7f391c Fix up some formatting
Jonatan Werpers <jonatan@werpers.com>
parents: 191
diff changeset
91 end
200
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
92
210
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
93 @testset "TensorMapping binary operations" begin
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
94 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
95 λ::T
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
96 range_size::NTuple{R,Int}
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
97 domain_size::NTuple{D,Int}
210
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
98 end
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
99
321
277dff5b071a LazyTensors/test/runtests.jl: Fix the indexing style of apply methods
Jonatan Werpers <jonatan@werpers.com>
parents: 278
diff changeset
100 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
101 LazyTensors.range_size(m::ScalarMapping) = m.domain_size
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
102 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
103
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
104 A = ScalarMapping{Float64,1,1}(2.0, (3,), (3,))
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
105 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
106
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
107 v = [1.1,1.2,1.3]
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
108 for i ∈ eachindex(v)
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
109 @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
110 end
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
111
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
112 for i ∈ eachindex(v)
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
113 @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
114 end
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
115
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
116 @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
117 @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
118 end
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
119
200
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
120 @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
121 @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
122 @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
123
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 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
125 @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
126 @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
127 @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
128 @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
129 end
200
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
130 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
131 data::T1
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
132 end
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
133 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
134 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
135
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
136 # Test lazy operations
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
137 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
138 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
139 s = 3.4
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
140 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
141 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
142 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
143 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
144 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
145 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
146 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
147 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
148 @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
149 @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
150 @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
151 @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
152 @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
153 @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
154 @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
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(s +̃ v1, 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(s -̃ v1, 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(s *̃ v1, 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)
200
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
160 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
161 @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
162 @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
163 @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
164 @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
165 @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
166 @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
167 @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
168 @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
169 @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
170 @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
171 @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
172 @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
173 end
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
174 @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
175 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
176 # Test that size of arrays is asserted when not specified inbounds
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
177 @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
178
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
179 # Test operations on LazyArray
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
180 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
181 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
182 @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
183 @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
184 @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
185 @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
186 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
187 @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
188 @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
189 end
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
190 @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
191 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
192 # Test that size of arrays is asserted when not specified inbounds
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
193 @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
194 end
338
2b0c9b30ea3b Add test sets for each submodule to make the test output nicer
Jonatan Werpers <jonatan@werpers.com>
parents: 335
diff changeset
195
377
8414c2334393 Start implementing LazyLinearMap
Jonatan Werpers <jonatan@werpers.com>
parents: 371
diff changeset
196
371
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
197 @testset "LazyFunctionArray" begin
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
198 @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
199 @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
200 1 2;
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
201 2 4;
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
202 3 6;
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
203 ]
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
204
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
205 @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
206 @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
207
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
208 @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
209
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
210 @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
211 @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
212 @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
213
348
7fe43d902a27 Start trying to change LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 338
diff changeset
214 end
371
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
215
412
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
216 @testset "TensorMappingComposition" begin
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
217 A = rand(2,3)
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
218 B = rand(3,4)
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
219
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
220 Ã = LazyLinearMap(A, (1,), (2,))
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
221 B̃ = LazyLinearMap(B, (1,), (2,))
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
222
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
223 @test Ã∘B̃ isa TensorMappingComposition
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
224 @test range_size(Ã∘B̃) == (2,)
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
225 @test domain_size(Ã∘B̃) == (4,)
417
4c6604b7d990 Add dimension checking in the constructor
Jonatan Werpers <jonatan@werpers.com>
parents: 416
diff changeset
226 @test_throws DimensionMismatch B̃∘Ã
4c6604b7d990 Add dimension checking in the constructor
Jonatan Werpers <jonatan@werpers.com>
parents: 416
diff changeset
227
4c6604b7d990 Add dimension checking in the constructor
Jonatan Werpers <jonatan@werpers.com>
parents: 416
diff changeset
228 # @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
229
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
230 v = rand(4)
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
231 @test Ã∘B̃*v ≈ A*B*v
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
232
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
233 v = rand(2)
416
ebc9b2383dae Update working tests
Jonatan Werpers <jonatan@werpers.com>
parents: 415
diff changeset
234 @test (Ã∘B̃)'*v ≈ B'*A'*v
412
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
235 end
d94891b8dfca Start implementing TensorComposition
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
236
377
8414c2334393 Start implementing LazyLinearMap
Jonatan Werpers <jonatan@werpers.com>
parents: 371
diff changeset
237 @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
238 # 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
239 # mapping vectors of size 4 to vectors of size 3.
377
8414c2334393 Start implementing LazyLinearMap
Jonatan Werpers <jonatan@werpers.com>
parents: 371
diff changeset
240 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
241 Ã = LazyLinearMap(A, (1,), (2,))
377
8414c2334393 Start implementing LazyLinearMap
Jonatan Werpers <jonatan@werpers.com>
parents: 371
diff changeset
242 v = rand(4)
411
4aa59af074ef Add implementation of apply_transpose for LazyLinearMap
Jonatan Werpers <jonatan@werpers.com>
parents: 399
diff changeset
243 w = rand(3)
377
8414c2334393 Start implementing LazyLinearMap
Jonatan Werpers <jonatan@werpers.com>
parents: 371
diff changeset
244
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
245 @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
246 @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
247 @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
248 @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
249
393
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
250 @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
251 @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
252 @test Ã'*w ≈ A'*w
377
8414c2334393 Start implementing LazyLinearMap
Jonatan Werpers <jonatan@werpers.com>
parents: 371
diff changeset
253
399
3b4b1758a8ad Add a check for permuted indecies.
Jonatan Werpers <jonatan@werpers.com>
parents: 394
diff changeset
254 A = rand(2,3,4)
3b4b1758a8ad Add a check for permuted indecies.
Jonatan Werpers <jonatan@werpers.com>
parents: 394
diff changeset
255 @test_throws DomainError LazyLinearMap(A, (3,1), (2,))
3b4b1758a8ad Add a check for permuted indecies.
Jonatan Werpers <jonatan@werpers.com>
parents: 394
diff changeset
256
393
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
257 # 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
258 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
259 # 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
260 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
261 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
262
7ad644d112de Expand tests for LazyLinearMap and update its documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 393
diff changeset
263 @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
264 @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
265 @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
266 @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
267 @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
268
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 392
diff changeset
269 # 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
270 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
271 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
272
7ad644d112de Expand tests for LazyLinearMap and update its documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 393
diff changeset
273 @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
274 @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
275 @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
276 @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
277 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
278 @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
279 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
280
371
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index.
Jonatan Werpers <jonatan@werpers.com>
parents: 348
diff changeset
281 end
377
8414c2334393 Start implementing LazyLinearMap
Jonatan Werpers <jonatan@werpers.com>
parents: 371
diff changeset
282
8414c2334393 Start implementing LazyLinearMap
Jonatan Werpers <jonatan@werpers.com>
parents: 371
diff changeset
283 end