Mercurial > repos > public > sbplib_julia

annotate LazyTensors/test/runtests.jl @ 276:e3611c2181bc boundary_conditions

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add 2D test
author Vidar Stiernström <vidar.stiernstrom@it.uu.se>
date Mon, 06 Jan 2020 11:46:21 +0100
parents 591609cdcd9b
children 21e5c1dd6794
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
184
6945c15a6a7a Rename package LazyTensor to LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 180
diff changeset
2 using LazyTensors
274
11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 263
diff changeset
3 using RegionIndices
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
4
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
5 @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
6 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
7 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
8 @test range_dim(DummyMapping{Int,2,3}()) == 2
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
9 @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
10 @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
11 end
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
12
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
13 @testset "Generic Operator methods" begin
186
715ff09bb2ce Rename and export types in LazyTensors follow julia conventions
Jonatan Werpers <jonatan@werpers.com>
parents: 184
diff changeset
14 struct DummyOperator{T,D} <: TensorOperator{T,D} end
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
15 @test range_size(DummyOperator{Int,2}(), (3,5)) == (3,5)
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
16 @test domain_size(DummyOperator{Float64, 3}(), (3,3,1)) == (3,3,1)
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
17 end
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
18
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
19 @testset "Mapping transpose" begin
186
715ff09bb2ce Rename and export types in LazyTensors follow julia conventions
Jonatan Werpers <jonatan@werpers.com>
parents: 184
diff changeset
20 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
21
275
591609cdcd9b Dispatch TensorMappingTranspose on region indices. Update tests.
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 274
diff changeset
22 LazyTensors.apply(m::DummyMapping{T,R,D}, v, I::NTuple{R,Index{<:Region}}) where {T,R,D} = :apply
591609cdcd9b Dispatch TensorMappingTranspose on region indices. Update tests.
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 274
diff changeset
23 LazyTensors.apply_transpose(m::DummyMapping{T,R,D}, v, I::NTuple{D,Index{<:Region}}) where {T,R,D} = :apply_transpose
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
24
232
a20bb4fac23d Improve tests for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 210
diff changeset
25 LazyTensors.range_size(m::DummyMapping{T,R,D}, domain_size::NTuple{D,Integer}) where {T,R,D} = :range_size
a20bb4fac23d Improve tests for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 210
diff changeset
26 LazyTensors.domain_size(m::DummyMapping{T,R,D}, range_size::NTuple{R,Integer}) where {T,R,D} = :domain_size
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
27
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
28 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
29 I = Index{Unknown}(0)
232
a20bb4fac23d Improve tests for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 210
diff changeset
30 @test m' isa TensorMapping{Float64, 3,2}
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
31 @test m'' == m
275
591609cdcd9b Dispatch TensorMappingTranspose on region indices. Update tests.
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 274
diff changeset
32 @test apply(m',zeros(Float64,(0,0)), (I,I,I)) == :apply_transpose
591609cdcd9b Dispatch TensorMappingTranspose on region indices. Update tests.
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 274
diff changeset
33 @test apply(m'',zeros(Float64,(0,0,0)),(I,I)) == :apply
591609cdcd9b Dispatch TensorMappingTranspose on region indices. Update tests.
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 274
diff changeset
34 @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
35
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
36 @test range_size(m', (0,0)) == :domain_size
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
37 @test domain_size(m', (0,0,0)) == :range_size
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
38 end
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
39
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
40 @testset "TensorApplication" begin
186
715ff09bb2ce Rename and export types in LazyTensors follow julia conventions
Jonatan Werpers <jonatan@werpers.com>
parents: 184
diff changeset
41 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
42
274
11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 263
diff changeset
43 LazyTensors.apply(m::DummyMapping{T,R,D}, v, i::NTuple{R,Index{<:Region}}) where {T,R,D} = (:apply,v,i)
237
1c6afdcfd657 Regretsies on the CartesianIndex stuff. Use Vararg instead
Jonatan Werpers <jonatan@werpers.com>
parents: 236
diff changeset
44 LazyTensors.apply_transpose(m::DummyMapping{T,R,D}, v, i::NTuple{D,Int}) where {T,R,D} = :apply_transpose
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
45
232
a20bb4fac23d Improve tests for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 210
diff changeset
46 LazyTensors.range_size(m::DummyMapping{T,R,D}, domain_size::NTuple{D,Integer}) where {T,R,D} = 2 .* domain_size
a20bb4fac23d Improve tests for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 210
diff changeset
47 LazyTensors.domain_size(m::DummyMapping{T,R,D}, range_size::NTuple{R,Integer}) where {T,R,D} = range_size.÷2
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
48
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
49
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
50 m = DummyMapping{Int, 1, 1}()
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
51 v = [0,1,2]
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
52 @test m*v isa AbstractVector{Int}
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
53 @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
54 @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
55 @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
56 @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
57 @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
58 @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
59 @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
60 @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
61 @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
62 @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
63 @test_broken BoundsError == (m*m*v)[0]
25d2ef206fe9 Add a few test cases for TensorApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 186
diff changeset
64 @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
65
276
e3611c2181bc Add 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 275
diff changeset
66 m = DummyMapping{Int, 2, 1}()
e3611c2181bc Add 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 275
diff changeset
67 @test_throws MethodError m*ones(Int,2,2)
e3611c2181bc Add 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 275
diff changeset
68 @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
69
276
e3611c2181bc Add 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 275
diff changeset
70 m = DummyMapping{Float64, 2, 2}()
e3611c2181bc Add 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 275
diff changeset
71 v = ones(3,3)
e3611c2181bc Add 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 275
diff changeset
72 I = (Index{Lower}(1),Index{Interior}(2));
e3611c2181bc Add 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 275
diff changeset
73 @test size(m*v) == 2 .*size(v)
e3611c2181bc Add 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 275
diff changeset
74 @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
75
d4cd4882ee9f Improve error messages when multiblying with TensorMappings and add some tests for TensorOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 237
diff changeset
76 struct ScalingOperator{T,D} <: TensorOperator{T,D}
d4cd4882ee9f Improve error messages when multiblying with TensorMappings and add some tests for TensorOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 237
diff changeset
77 λ::T
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
274
11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 263
diff changeset
80 LazyTensors.apply(m::ScalingOperator{T,D}, v, I::Tuple{Index{<:Region}}) where {T,D} = m.λ*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
81
276
e3611c2181bc Add 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 275
diff changeset
82 m = ScalingOperator{Int,1}(2)
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
276
e3611c2181bc Add 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 275
diff changeset
84 @test m*[1,2,3] isa AbstractVector
e3611c2181bc Add 2D test
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 275
diff changeset
85 @test m*[1,2,3] == [2,4,6]
198
b5c9be7f391c Fix up some formatting
Jonatan Werpers <jonatan@werpers.com>
parents: 191
diff changeset
86 end
200
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
87
210
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
88 @testset "TensorMapping binary operations" begin
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
89 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
90 λ::T
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
91 end
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
92
274
11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 263
diff changeset
93 LazyTensors.apply(m::ScalarMapping{T,R,D}, v, I::Tuple{Index{<:Region}}) where {T,R,D} = m.λ*v[I...]
210
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
94 LazyTensors.range_size(m::ScalarMapping, domain_size) = domain_size
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
95 LazyTensors.domain_size(m::ScalarMapping, range_sizes) = range_sizes
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
96
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
97 A = ScalarMapping{Float64,1,1}(2.0)
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
98 B = ScalarMapping{Float64,1,1}(3.0)
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
99
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
100 v = [1.1,1.2,1.3]
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
101 for i ∈ eachindex(v)
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
102 @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
103 end
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
104
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
105 for i ∈ eachindex(v)
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
106 @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
107 end
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
108
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
109 @test range_size(A+B, (3,)) == range_size(A, (3,)) == range_size(B,(3,))
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
110 @test domain_size(A+B, (3,)) == domain_size(A, (3,)) == domain_size(B,(3,))
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
111 end
2aa33d0eef90 Add ability to add and subtract TensorMappings
Jonatan Werpers <jonatan@werpers.com>
parents: 208
diff changeset
112
200
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
113 @testset "LazyArray" begin
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
114 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
115 data::T1
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
116 end
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
117 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
118 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
119
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
120 # Test lazy operations
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
121 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
122 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
123 s = 3.4
b577b5f64530 Add lazy elementwise operations for array with scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 257
diff changeset
124 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
125 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
126 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
127 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
128 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
129 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
130 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
131 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
132 @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
133 @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
134 @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
135 @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
136 @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
137 @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
138 @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
139 @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
140 @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
141 @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
142 @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
143 @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
144 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
145 @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
146 @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
147 @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
148 @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
149 @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
150 @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
151 @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
152 @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
153 @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
154 @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
155 @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
156 @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
157 end
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
158 @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
159 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
160 # 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
161 @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
162
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
163 # Test operations on LazyArray
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
164 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
165 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
166 @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
167 @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
168 @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
169 @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
170 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
171 @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
172 @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
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]
c19bfad0e836 Add tests for LazyArrays and lazy operations
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 198
diff changeset
175 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
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 end