Mercurial > repos > public > sbplib_julia

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

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