Mercurial > repos > public > sbplib_julia

annotate test/LazyTensors/lazy_tensor_operations_test.jl @ 977:043d13ef8898 feature/tensormapping_application_promotion

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Fix type calculation of LazyTensorMappingApplication element type
author Jonatan Werpers <jonatan@werpers.com>
date Tue, 15 Mar 2022 21:03:17 +0100
parents 86889fc5b63f
children bc384aaade30 bbbc31953367
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
1 using Test
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
2 using Sbplib.LazyTensors
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
3 using Sbplib.RegionIndices
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
4
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
5 using Tullio
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
6
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
7 @testset "Mapping transpose" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
8 struct DummyMapping{T,R,D} <: TensorMapping{T,R,D} end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
9
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
10 LazyTensors.apply(m::DummyMapping{T,R}, v, I::Vararg{Any,R}) where {T,R} = :apply
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
11 LazyTensors.apply_transpose(m::DummyMapping{T,R,D}, v, I::Vararg{Any,D}) where {T,R,D} = :apply_transpose
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
12
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
13 LazyTensors.range_size(m::DummyMapping) = :range_size
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
14 LazyTensors.domain_size(m::DummyMapping) = :domain_size
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
15
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
16 m = DummyMapping{Float64,2,3}()
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
17 @test m' isa TensorMapping{Float64, 3,2}
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
18 @test m'' == m
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
19 @test apply(m',zeros(Float64,(0,0)), 0, 0, 0) == :apply_transpose
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
20 @test apply(m'',zeros(Float64,(0,0,0)), 0, 0) == :apply
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
21 @test apply_transpose(m', zeros(Float64,(0,0,0)), 0, 0) == :apply
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
22
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
23 @test range_size(m') == :domain_size
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
24 @test domain_size(m') == :range_size
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
25 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
26
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
27 @testset "TensorApplication" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
28 struct SizeDoublingMapping{T,R,D} <: TensorMapping{T,R,D}
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
29 domain_size::NTuple{D,Int}
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
30 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
31
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
32 LazyTensors.apply(m::SizeDoublingMapping{T,R}, v, i::Vararg{Any,R}) where {T,R} = (:apply,v,i)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
33 LazyTensors.range_size(m::SizeDoublingMapping) = 2 .* m.domain_size
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
34 LazyTensors.domain_size(m::SizeDoublingMapping) = m.domain_size
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
35
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
36
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
37 m = SizeDoublingMapping{Int, 1, 1}((3,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
38 v = [0,1,2]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
39 @test size(m*v) == 2 .*size(v)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
40 @test (m*v)[0] == (:apply,v,(0,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
41 @test (m*m*v)[1] == (:apply,m*v,(1,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
42 @test (m*m*v)[3] == (:apply,m*v,(3,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
43 @test (m*m*v)[6] == (:apply,m*v,(6,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
44 @test_broken BoundsError == (m*m*v)[0]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
45 @test_broken BoundsError == (m*m*v)[7]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
46 @test_throws MethodError m*m
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
47
938
4a9a96d51940 Fix dispatch issue when indexing a 1D TensorApplication with a CartiesianIndex
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
48 @test (m*v)[CartesianIndex(2)] == (:apply,v,(2,))
4a9a96d51940 Fix dispatch issue when indexing a 1D TensorApplication with a CartiesianIndex
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
49 @test (m*m*v)[CartesianIndex(2)] == (:apply,m*v,(2,))
4a9a96d51940 Fix dispatch issue when indexing a 1D TensorApplication with a CartiesianIndex
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
50
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
51 m = SizeDoublingMapping{Int, 2, 1}((3,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
52 @test_throws MethodError m*ones(Int,2,2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
53 @test_throws MethodError m*m*v
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
54
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
55 m = SizeDoublingMapping{Float64, 2, 2}((3,3))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
56 v = ones(3,3)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
57 @test size(m*v) == 2 .*size(v)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
58 @test (m*v)[1,2] == (:apply,v,(1,2))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
59
938
4a9a96d51940 Fix dispatch issue when indexing a 1D TensorApplication with a CartiesianIndex
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
60 @test (m*v)[CartesianIndex(2,3)] == (:apply,v,(2,3))
4a9a96d51940 Fix dispatch issue when indexing a 1D TensorApplication with a CartiesianIndex
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
61 @test (m*m*v)[CartesianIndex(4,3)] == (:apply,m*v,(4,3))
4a9a96d51940 Fix dispatch issue when indexing a 1D TensorApplication with a CartiesianIndex
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
62
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
63 struct ScalingOperator{T,D} <: TensorMapping{T,D,D}
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
64 λ::T
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
65 size::NTuple{D,Int}
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
66 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
67
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
68 LazyTensors.apply(m::ScalingOperator{T,D}, v, I::Vararg{Any,D}) where {T,D} = m.λ*v[I...]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
69 LazyTensors.range_size(m::ScalingOperator) = m.size
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
70 LazyTensors.domain_size(m::ScalingOperator) = m.size
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
71
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
72 m = ScalingOperator{Int,1}(2,(3,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
73 v = [1,2,3]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
74 @test m*v isa AbstractVector
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
75 @test m*v == [2,4,6]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
76
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
77 m = ScalingOperator{Int,2}(2,(2,2))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
78 v = [[1 2];[3 4]]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
79 @test m*v == [[2 4];[6 8]]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
80 @test (m*v)[2,1] == 6
942
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
81
977
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
82 @testset "Type calculation" begin
942
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
83 m = ScalingOperator{Int,1}(2,(3,))
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
84 v = [1.,2.,3.]
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
85 @test m*v isa AbstractVector{Float64}
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
86 @test m*v == [2.,4.,6.]
977
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
87 @inferred m*v
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
88 @inferred (m*v)[1]
942
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
89
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
90 m = ScalingOperator{Int,2}(2,(2,2))
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
91 v = [[1. 2.];[3. 4.]]
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
92 @test m*v == [[2. 4.];[6. 8.]]
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
93 @test (m*v)[2,1] == 6.
977
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
94 @inferred m*v
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
95 @inferred (m*v)[1]
942
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
96
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
97 m = ScalingOperator{ComplexF64,1}(2. +2. *im,(3,))
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
98 v = [1.,2.,3.]
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
99 @test m*v isa AbstractVector{ComplexF64}
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
100 @test m*v == [2. + 2. *im, 4. + 4. *im, 6. + 6. *im]
977
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
101 @inferred m*v
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
102 @inferred (m*v)[1]
942
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
103
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
104 m = ScalingOperator{ComplexF64,1}(1,(3,))
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
105 v = [2. + 2. *im, 4. + 4. *im, 6. + 6. *im]
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
106 @test m*v isa AbstractVector{ComplexF64}
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
107 @test m*v == [2. + 2. *im, 4. + 4. *im, 6. + 6. *im]
977
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
108 @inferred m*v
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
109 @inferred (m*v)[1]
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
110
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
111 m = ScalingOperator{Float64,1}(2., (3,))
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
112 v = [[1,2,3], [3,2,1],[1,3,1]]
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
113 @test m*v isa AbstractVector{Vector{Float64}}
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
114 @test m*v == [[2.,4.,6.], [6.,4.,2.],[2.,6.,2.]]
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
115 @inferred m*v
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
116 @inferred (m*v)[1]
942
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
117 end
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
118 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
119
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
120 @testset "TensorMapping binary operations" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
121 struct ScalarMapping{T,R,D} <: TensorMapping{T,R,D}
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
122 λ::T
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
123 range_size::NTuple{R,Int}
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
124 domain_size::NTuple{D,Int}
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
125 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
126
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
127 LazyTensors.apply(m::ScalarMapping{T,R}, v, I::Vararg{Any,R}) where {T,R} = m.λ*v[I...]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
128 LazyTensors.range_size(m::ScalarMapping) = m.domain_size
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
129 LazyTensors.domain_size(m::ScalarMapping) = m.range_size
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
130
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
131 A = ScalarMapping{Float64,1,1}(2.0, (3,), (3,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
132 B = ScalarMapping{Float64,1,1}(3.0, (3,), (3,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
133
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
134 v = [1.1,1.2,1.3]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
135 for i ∈ eachindex(v)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
136 @test ((A+B)*v)[i] == 2*v[i] + 3*v[i]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
137 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
138
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
139 for i ∈ eachindex(v)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
140 @test ((A-B)*v)[i] == 2*v[i] - 3*v[i]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
141 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
142
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
143 @test range_size(A+B) == range_size(A) == range_size(B)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
144 @test domain_size(A+B) == domain_size(A) == domain_size(B)
943
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
145
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
146 @test ((A+B)*ComplexF64[1.1,1.2,1.3])[3] isa ComplexF64
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
147 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
148
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
149
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
150 @testset "TensorMappingComposition" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
151 A = rand(2,3)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
152 B = rand(3,4)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
153
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
154 Ã = LazyLinearMap(A, (1,), (2,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
155 B̃ = LazyLinearMap(B, (1,), (2,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
156
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
157 @test Ã∘B̃ isa TensorMappingComposition
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
158 @test range_size(Ã∘B̃) == (2,)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
159 @test domain_size(Ã∘B̃) == (4,)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
160 @test_throws SizeMismatch B̃∘Ã
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
161
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
162 # @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)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
163
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
164 v = rand(4)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
165 @test Ã∘B̃*v ≈ A*B*v rtol=1e-14
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
166
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
167 v = rand(2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
168 @test (Ã∘B̃)'*v ≈ B'*A'*v rtol=1e-14
943
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
169
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
170 @test (Ã∘B̃*ComplexF64[1.,2.,3.,4.])[1] isa ComplexF64
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
171 @test ((Ã∘B̃)'*ComplexF64[1.,2.])[1] isa ComplexF64
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
172 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
173
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
174 @testset "LazyLinearMap" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
175 # Test a standard matrix-vector product
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
176 # mapping vectors of size 4 to vectors of size 3.
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
177 A = rand(3,4)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
178 Ã = LazyLinearMap(A, (1,), (2,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
179 v = rand(4)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
180 w = rand(3)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
181
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
182 @test à isa LazyLinearMap{T,1,1} where T
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
183 @test à isa TensorMapping{T,1,1} where T
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
184 @test range_size(Ã) == (3,)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
185 @test domain_size(Ã) == (4,)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
186
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
187 @test Ã*ones(4) ≈ A*ones(4) atol=5e-13
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
188 @test Ã*v ≈ A*v atol=5e-13
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
189 @test Ã'*w ≈ A'*w
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
190
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
191 A = rand(2,3,4)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
192 @test_throws DomainError LazyLinearMap(A, (3,1), (2,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
193
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
194 # Test more exotic mappings
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
195 B = rand(3,4,2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
196 # Map vectors of size 2 to matrices of size (3,4)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
197 B̃ = LazyLinearMap(B, (1,2), (3,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
198 v = rand(2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
199
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
200 @test range_size(B̃) == (3,4)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
201 @test domain_size(B̃) == (2,)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
202 @test B̃ isa TensorMapping{T,2,1} where T
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
203 @test B̃*ones(2) ≈ B[:,:,1] + B[:,:,2] atol=5e-13
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
204 @test B̃*v ≈ B[:,:,1]*v[1] + B[:,:,2]*v[2] atol=5e-13
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
205
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
206 # Map matrices of size (3,2) to vectors of size 4
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
207 B̃ = LazyLinearMap(B, (2,), (1,3))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
208 v = rand(3,2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
209
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
210 @test range_size(B̃) == (4,)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
211 @test domain_size(B̃) == (3,2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
212 @test B̃ isa TensorMapping{T,1,2} where T
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
213 @test B̃*ones(3,2) ≈ B[1,:,1] + B[2,:,1] + B[3,:,1] +
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
214 B[1,:,2] + B[2,:,2] + B[3,:,2] atol=5e-13
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
215 @test B̃*v ≈ B[1,:,1]*v[1,1] + B[2,:,1]*v[2,1] + B[3,:,1]*v[3,1] +
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
216 B[1,:,2]v[1,2] + B[2,:,2]*v[2,2] + B[3,:,2]*v[3,2] atol=5e-13
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
217
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
218
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
219 # TODO:
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
220 # @inferred (B̃*v)[2]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
221 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
222
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
223
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
224 @testset "IdentityMapping" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
225 @test IdentityMapping{Float64}((4,5)) isa IdentityMapping{T,2} where T
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
226 @test IdentityMapping{Float64}((4,5)) isa TensorMapping{T,2,2} where T
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
227 @test IdentityMapping{Float64}((4,5)) == IdentityMapping{Float64}(4,5)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
228
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
229 @test IdentityMapping(3,2) isa IdentityMapping{Float64,2}
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
230
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
231 for sz ∈ [(4,5),(3,),(5,6,4)]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
232 I = IdentityMapping{Float64}(sz)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
233 v = rand(sz...)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
234 @test I*v == v
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
235 @test I'*v == v
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
236
943
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
237 v = rand(ComplexF64,sz...)
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
238 @test I*v == v
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
239 @test I'*v == v
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
240
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
241 @test range_size(I) == sz
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
242 @test domain_size(I) == sz
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
243 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
244
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
245 I = IdentityMapping{Float64}((4,5))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
246 v = rand(4,5)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
247 @inferred (I*v)[3,2]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
248 @inferred (I'*v)[3,2]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
249 @inferred range_size(I)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
250
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
251 @inferred range_dim(I)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
252 @inferred domain_dim(I)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
253
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
254 Ã = rand(4,2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
255 A = LazyLinearMap(Ã,(1,),(2,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
256 I1 = IdentityMapping{Float64}(2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
257 I2 = IdentityMapping{Float64}(4)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
258 @test A∘I1 == A
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
259 @test I2∘A == A
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
260 @test I1∘I1 == I1
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
261 @test_throws SizeMismatch I1∘A
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
262 @test_throws SizeMismatch A∘I2
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
263 @test_throws SizeMismatch I1∘I2
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
264 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
265
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
266 @testset "InflatedTensorMapping" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
267 I(sz...) = IdentityMapping(sz...)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
268
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
269 Ã = rand(4,2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
270 B̃ = rand(4,2,3)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
271 C̃ = rand(4,2,3)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
272
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
273 A = LazyLinearMap(Ã,(1,),(2,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
274 B = LazyLinearMap(B̃,(1,2),(3,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
275 C = LazyLinearMap(C̃,(1,),(2,3))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
276
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
277 @testset "Constructors" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
278 @test InflatedTensorMapping(I(3,2), A, I(4)) isa TensorMapping{Float64, 4, 4}
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
279 @test InflatedTensorMapping(I(3,2), B, I(4)) isa TensorMapping{Float64, 5, 4}
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
280 @test InflatedTensorMapping(I(3), C, I(2,3)) isa TensorMapping{Float64, 4, 5}
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
281 @test InflatedTensorMapping(C, I(2,3)) isa TensorMapping{Float64, 3, 4}
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
282 @test InflatedTensorMapping(I(3), C) isa TensorMapping{Float64, 2, 3}
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
283 @test InflatedTensorMapping(I(3), I(2,3)) isa TensorMapping{Float64, 3, 3}
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
284 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
285
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
286 @testset "Range and domain size" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
287 @test range_size(InflatedTensorMapping(I(3,2), A, I(4))) == (3,2,4,4)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
288 @test domain_size(InflatedTensorMapping(I(3,2), A, I(4))) == (3,2,2,4)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
289
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
290 @test range_size(InflatedTensorMapping(I(3,2), B, I(4))) == (3,2,4,2,4)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
291 @test domain_size(InflatedTensorMapping(I(3,2), B, I(4))) == (3,2,3,4)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
292
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
293 @test range_size(InflatedTensorMapping(I(3), C, I(2,3))) == (3,4,2,3)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
294 @test domain_size(InflatedTensorMapping(I(3), C, I(2,3))) == (3,2,3,2,3)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
295
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
296 @inferred range_size(InflatedTensorMapping(I(3,2), A, I(4))) == (3,2,4,4)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
297 @inferred domain_size(InflatedTensorMapping(I(3,2), A, I(4))) == (3,2,2,4)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
298 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
299
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
300 @testset "Application" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
301 # Testing regular application and transposed application with inflation "before", "after" and "before and after".
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
302 # The inflated tensor mappings are chosen to preserve, reduce and increase the dimension of the result compared to the input.
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
303 tests = [
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
304 (
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
305 InflatedTensorMapping(I(3,2), A, I(4)),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
306 (v-> @tullio res[a,b,c,d] := Ã[c,i]*v[a,b,i,d]), # Expected result of apply
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
307 (v-> @tullio res[a,b,c,d] := Ã[i,c]*v[a,b,i,d]), # Expected result of apply_transpose
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
308 ),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
309 (
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
310 InflatedTensorMapping(I(3,2), B, I(4)),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
311 (v-> @tullio res[a,b,c,d,e] := B̃[c,d,i]*v[a,b,i,e]),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
312 (v-> @tullio res[a,b,c,d] := B̃[i,j,c]*v[a,b,i,j,d]),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
313 ),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
314 (
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
315 InflatedTensorMapping(I(3,2), C, I(4)),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
316 (v-> @tullio res[a,b,c,d] := C̃[c,i,j]*v[a,b,i,j,d]),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
317 (v-> @tullio res[a,b,c,d,e] := C̃[i,c,d]*v[a,b,i,e]),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
318 ),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
319 (
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
320 InflatedTensorMapping(I(3,2), A),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
321 (v-> @tullio res[a,b,c] := Ã[c,i]*v[a,b,i]),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
322 (v-> @tullio res[a,b,c] := Ã[i,c]*v[a,b,i]),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
323 ),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
324 (
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
325 InflatedTensorMapping(I(3,2), B),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
326 (v-> @tullio res[a,b,c,d] := B̃[c,d,i]*v[a,b,i]),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
327 (v-> @tullio res[a,b,c] := B̃[i,j,c]*v[a,b,i,j]),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
328 ),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
329 (
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
330 InflatedTensorMapping(I(3,2), C),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
331 (v-> @tullio res[a,b,c] := C̃[c,i,j]*v[a,b,i,j]),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
332 (v-> @tullio res[a,b,c,d] := C̃[i,c,d]*v[a,b,i]),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
333 ),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
334 (
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
335 InflatedTensorMapping(A,I(4)),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
336 (v-> @tullio res[a,b] := Ã[a,i]*v[i,b]),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
337 (v-> @tullio res[a,b] := Ã[i,a]*v[i,b]),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
338 ),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
339 (
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
340 InflatedTensorMapping(B,I(4)),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
341 (v-> @tullio res[a,b,c] := B̃[a,b,i]*v[i,c]),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
342 (v-> @tullio res[a,b] := B̃[i,j,a]*v[i,j,b]),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
343 ),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
344 (
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
345 InflatedTensorMapping(C,I(4)),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
346 (v-> @tullio res[a,b] := C̃[a,i,j]*v[i,j,b]),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
347 (v-> @tullio res[a,b,c] := C̃[i,a,b]*v[i,c]),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
348 ),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
349 ]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
350
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
351 @testset "apply" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
352 for i ∈ 1:length(tests)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
353 tm = tests[i][1]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
354 v = rand(domain_size(tm)...)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
355 true_value = tests[i][2](v)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
356 @test tm*v ≈ true_value rtol=1e-14
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
357 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
358 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
359
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
360 @testset "apply_transpose" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
361 for i ∈ 1:length(tests)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
362 tm = tests[i][1]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
363 v = rand(range_size(tm)...)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
364 true_value = tests[i][3](v)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
365 @test tm'*v ≈ true_value rtol=1e-14
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
366 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
367 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
368
943
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
369 @testset "application to other type" begin
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
370 tm = InflatedTensorMapping(I(3,2), A, I(4))
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
371
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
372 v = rand(ComplexF64, domain_size(tm)...)
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
373 @test (tm*v)[1,2,3,1] isa ComplexF64
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
374
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
375 v = rand(ComplexF64, domain_size(tm')...)
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
376 @test (tm'*v)[1,2,2,1] isa ComplexF64
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
377 end
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
378
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
379 @testset "Inference of application" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
380 struct ScalingOperator{T,D} <: TensorMapping{T,D,D}
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
381 λ::T
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
382 size::NTuple{D,Int}
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
383 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
384
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
385 LazyTensors.apply(m::ScalingOperator{T,D}, v, I::Vararg{Any,D}) where {T,D} = m.λ*v[I...]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
386 LazyTensors.range_size(m::ScalingOperator) = m.size
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
387 LazyTensors.domain_size(m::ScalingOperator) = m.size
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
388
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
389 tm = InflatedTensorMapping(I(2,3),ScalingOperator(2.0, (3,2)),I(3,4))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
390 v = rand(domain_size(tm)...)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
391
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
392 @inferred apply(tm,v,1,2,3,2,2,4)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
393 @inferred (tm*v)[1,2,3,2,2,4]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
394 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
395 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
396
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
397 @testset "InflatedTensorMapping of InflatedTensorMapping" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
398 A = ScalingOperator(2.0,(2,3))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
399 itm = InflatedTensorMapping(I(3,2), A, I(4))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
400 @test InflatedTensorMapping(I(4), itm, I(2)) == InflatedTensorMapping(I(4,3,2), A, I(4,2))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
401 @test InflatedTensorMapping(itm, I(2)) == InflatedTensorMapping(I(3,2), A, I(4,2))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
402 @test InflatedTensorMapping(I(4), itm) == InflatedTensorMapping(I(4,3,2), A, I(4))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
403
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
404 @test InflatedTensorMapping(I(2), I(2), I(2)) isa InflatedTensorMapping # The constructor should always return its type.
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
405 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
406 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
407
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
408 @testset "split_index" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
409 @test LazyTensors.split_index(Val(2),Val(1),Val(2),Val(2),1,2,3,4,5,6) == ((1,2,:,5,6),(3,4))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
410 @test LazyTensors.split_index(Val(2),Val(3),Val(2),Val(2),1,2,3,4,5,6) == ((1,2,:,:,:,5,6),(3,4))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
411 @test LazyTensors.split_index(Val(3),Val(1),Val(1),Val(2),1,2,3,4,5,6) == ((1,2,3,:,5,6),(4,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
412 @test LazyTensors.split_index(Val(3),Val(2),Val(1),Val(2),1,2,3,4,5,6) == ((1,2,3,:,:,5,6),(4,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
413 @test LazyTensors.split_index(Val(1),Val(1),Val(2),Val(3),1,2,3,4,5,6) == ((1,:,4,5,6),(2,3))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
414 @test LazyTensors.split_index(Val(1),Val(2),Val(2),Val(3),1,2,3,4,5,6) == ((1,:,:,4,5,6),(2,3))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
415
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
416 @test LazyTensors.split_index(Val(0),Val(1),Val(3),Val(3),1,2,3,4,5,6) == ((:,4,5,6),(1,2,3))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
417 @test LazyTensors.split_index(Val(3),Val(1),Val(3),Val(0),1,2,3,4,5,6) == ((1,2,3,:),(4,5,6))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
418
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
419 @inferred LazyTensors.split_index(Val(2),Val(3),Val(2),Val(2),1,2,3,2,2,4)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
420 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
421
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
422 @testset "slice_tuple" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
423 @test LazyTensors.slice_tuple((1,2,3),Val(1), Val(3)) == (1,2,3)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
424 @test LazyTensors.slice_tuple((1,2,3,4,5,6),Val(2), Val(5)) == (2,3,4,5)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
425 @test LazyTensors.slice_tuple((1,2,3,4,5,6),Val(1), Val(3)) == (1,2,3)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
426 @test LazyTensors.slice_tuple((1,2,3,4,5,6),Val(4), Val(6)) == (4,5,6)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
427 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
428
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
429 @testset "split_tuple" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
430 @testset "2 parts" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
431 @test LazyTensors.split_tuple((),Val(0)) == ((),())
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
432 @test LazyTensors.split_tuple((1,),Val(0)) == ((),(1,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
433 @test LazyTensors.split_tuple((1,),Val(1)) == ((1,),())
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
434
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
435 @test LazyTensors.split_tuple((1,2,3,4),Val(0)) == ((),(1,2,3,4))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
436 @test LazyTensors.split_tuple((1,2,3,4),Val(1)) == ((1,),(2,3,4))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
437 @test LazyTensors.split_tuple((1,2,3,4),Val(2)) == ((1,2),(3,4))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
438 @test LazyTensors.split_tuple((1,2,3,4),Val(3)) == ((1,2,3),(4,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
439 @test LazyTensors.split_tuple((1,2,3,4),Val(4)) == ((1,2,3,4),())
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
440
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
441 @test LazyTensors.split_tuple((1,2,true,4),Val(3)) == ((1,2,true),(4,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
442
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
443 @inferred LazyTensors.split_tuple((1,2,3,4),Val(3))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
444 @inferred LazyTensors.split_tuple((1,2,true,4),Val(3))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
445 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
446
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
447 @testset "3 parts" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
448 @test LazyTensors.split_tuple((),Val(0),Val(0)) == ((),(),())
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
449 @test LazyTensors.split_tuple((1,2,3),Val(1), Val(1)) == ((1,),(2,),(3,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
450 @test LazyTensors.split_tuple((1,true,3),Val(1), Val(1)) == ((1,),(true,),(3,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
451
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
452 @test LazyTensors.split_tuple((1,2,3,4,5,6),Val(1),Val(2)) == ((1,),(2,3),(4,5,6))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
453 @test LazyTensors.split_tuple((1,2,3,4,5,6),Val(3),Val(2)) == ((1,2,3),(4,5),(6,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
454
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
455 @inferred LazyTensors.split_tuple((1,2,3,4,5,6),Val(3),Val(2))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
456 @inferred LazyTensors.split_tuple((1,true,3),Val(1), Val(1))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
457 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
458 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
459
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
460 @testset "flatten_tuple" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
461 @test LazyTensors.flatten_tuple((1,)) == (1,)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
462 @test LazyTensors.flatten_tuple((1,2,3,4,5,6)) == (1,2,3,4,5,6)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
463 @test LazyTensors.flatten_tuple((1,2,(3,4),5,6)) == (1,2,3,4,5,6)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
464 @test LazyTensors.flatten_tuple((1,2,(3,(4,5)),6)) == (1,2,3,4,5,6)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
465 @test LazyTensors.flatten_tuple(((1,2),(3,4),(5,),6)) == (1,2,3,4,5,6)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
466 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
467
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
468
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
469 @testset "LazyOuterProduct" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
470 struct ScalingOperator{T,D} <: TensorMapping{T,D,D}
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
471 λ::T
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
472 size::NTuple{D,Int}
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
473 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
474
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
475 LazyTensors.apply(m::ScalingOperator{T,D}, v, I::Vararg{Any,D}) where {T,D} = m.λ*v[I...]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
476 LazyTensors.range_size(m::ScalingOperator) = m.size
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
477 LazyTensors.domain_size(m::ScalingOperator) = m.size
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
478
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
479 A = ScalingOperator(2.0, (5,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
480 B = ScalingOperator(3.0, (3,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
481 C = ScalingOperator(5.0, (3,2))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
482
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
483 AB = LazyOuterProduct(A,B)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
484 @test AB isa TensorMapping{T,2,2} where T
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
485 @test range_size(AB) == (5,3)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
486 @test domain_size(AB) == (5,3)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
487
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
488 v = rand(range_size(AB)...)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
489 @test AB*v == 6*v
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
490
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
491 ABC = LazyOuterProduct(A,B,C)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
492
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
493 @test ABC isa TensorMapping{T,4,4} where T
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
494 @test range_size(ABC) == (5,3,3,2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
495 @test domain_size(ABC) == (5,3,3,2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
496
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
497 @test A⊗B == AB
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
498 @test A⊗B⊗C == ABC
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
499
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
500 A = rand(3,2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
501 B = rand(2,4,3)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
502
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
503 v₁ = rand(2,4,3)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
504 v₂ = rand(4,3,2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
505
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
506 Ã = LazyLinearMap(A,(1,),(2,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
507 B̃ = LazyLinearMap(B,(1,),(2,3))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
508
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
509 ÃB̃ = LazyOuterProduct(Ã,B̃)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
510 @tullio ABv[i,k] := A[i,j]*B[k,l,m]*v₁[j,l,m]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
511 @test ÃB̃*v₁ ≈ ABv
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
512
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
513 B̃Ã = LazyOuterProduct(B̃,Ã)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
514 @tullio BAv[k,i] := A[i,j]*B[k,l,m]*v₂[l,m,j]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
515 @test B̃Ã*v₂ ≈ BAv
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
516
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
517 @testset "Indentity mapping arguments" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
518 @test LazyOuterProduct(IdentityMapping(3,2), IdentityMapping(1,2)) == IdentityMapping(3,2,1,2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
519
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
520 Ã = LazyLinearMap(A,(1,),(2,))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
521 @test LazyOuterProduct(IdentityMapping(3,2), Ã) == InflatedTensorMapping(IdentityMapping(3,2),Ã)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
522 @test LazyOuterProduct(Ã, IdentityMapping(3,2)) == InflatedTensorMapping(Ã,IdentityMapping(3,2))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
523
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
524 I1 = IdentityMapping(3,2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
525 I2 = IdentityMapping(4)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
526 @test I1⊗Ã⊗I2 == InflatedTensorMapping(I1, Ã, I2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
527 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
528
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
529 end