Mercurial > repos > public > sbplib_julia

annotate test/LazyTensors/lazy_tensor_operations_test.jl @ 1836:368999a2e243 refactor/lazy_tensors/elementwise_ops

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add TensorNegation
author Jonatan Werpers <jonatan@werpers.com>
date Thu, 09 Jan 2025 15:32:47 +0100
parents 471a948cd2b2
children 200971c71657
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
1726
471a948cd2b2 Rename project from Sbplib to Diffinitive
Vidar Stiernström <vidar.stiernstrom@gmail.com>
parents: 1641
diff changeset
2 using Diffinitive.LazyTensors
471a948cd2b2 Rename project from Sbplib to Diffinitive
Vidar Stiernström <vidar.stiernstrom@gmail.com>
parents: 1641
diff changeset
3 using Diffinitive.RegionIndices
712
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
1641
fddd3a906535 Change name of dummy mapping to avoid conflict
Jonatan Werpers <jonatan@werpers.com>
parents: 1299
diff changeset
7 struct TransposableDummyMapping{T,R,D} <: LazyTensor{T,R,D} end
1007
f7a718bcb4da Add checking of sizes to LazyTensorBinaryOperation
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
8
1641
fddd3a906535 Change name of dummy mapping to avoid conflict
Jonatan Werpers <jonatan@werpers.com>
parents: 1299
diff changeset
9 LazyTensors.apply(m::TransposableDummyMapping{T,R}, v, I::Vararg{Any,R}) where {T,R} = :apply
fddd3a906535 Change name of dummy mapping to avoid conflict
Jonatan Werpers <jonatan@werpers.com>
parents: 1299
diff changeset
10 LazyTensors.apply_transpose(m::TransposableDummyMapping{T,R,D}, v, I::Vararg{Any,D}) where {T,R,D} = :apply_transpose
1007
f7a718bcb4da Add checking of sizes to LazyTensorBinaryOperation
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
11
1641
fddd3a906535 Change name of dummy mapping to avoid conflict
Jonatan Werpers <jonatan@werpers.com>
parents: 1299
diff changeset
12 LazyTensors.range_size(m::TransposableDummyMapping) = :range_size
fddd3a906535 Change name of dummy mapping to avoid conflict
Jonatan Werpers <jonatan@werpers.com>
parents: 1299
diff changeset
13 LazyTensors.domain_size(m::TransposableDummyMapping) = :domain_size
1007
f7a718bcb4da Add checking of sizes to LazyTensorBinaryOperation
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
14
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
15
1007
f7a718bcb4da Add checking of sizes to LazyTensorBinaryOperation
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
16 struct SizeDoublingMapping{T,R,D} <: LazyTensor{T,R,D}
f7a718bcb4da Add checking of sizes to LazyTensorBinaryOperation
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
17 domain_size::NTuple{D,Int}
f7a718bcb4da Add checking of sizes to LazyTensorBinaryOperation
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
18 end
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
19
1007
f7a718bcb4da Add checking of sizes to LazyTensorBinaryOperation
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
20 LazyTensors.apply(m::SizeDoublingMapping{T,R}, v, i::Vararg{Any,R}) where {T,R} = (:apply,v,i)
f7a718bcb4da Add checking of sizes to LazyTensorBinaryOperation
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
21 LazyTensors.range_size(m::SizeDoublingMapping) = 2 .* m.domain_size
f7a718bcb4da Add checking of sizes to LazyTensorBinaryOperation
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
22 LazyTensors.domain_size(m::SizeDoublingMapping) = m.domain_size
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
23
1007
f7a718bcb4da Add checking of sizes to LazyTensorBinaryOperation
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
24
f7a718bcb4da Add checking of sizes to LazyTensorBinaryOperation
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
25
f7a718bcb4da Add checking of sizes to LazyTensorBinaryOperation
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
26 @testset "Mapping transpose" begin
1641
fddd3a906535 Change name of dummy mapping to avoid conflict
Jonatan Werpers <jonatan@werpers.com>
parents: 1299
diff changeset
27 m = TransposableDummyMapping{Float64,2,3}()
995
1ba8a398af9c Rename types
Jonatan Werpers <jonatan@werpers.com>
parents: 993
diff changeset
28 @test m' isa LazyTensor{Float64, 3,2}
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
29 @test m'' == m
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
30 @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
31 @test apply(m'',zeros(Float64,(0,0,0)), 0, 0) == :apply
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
32 @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
33
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
34 @test range_size(m') == :domain_size
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
35 @test domain_size(m') == :range_size
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
36 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
37
1002
271aa6ae1055 Split out a file for tensor types
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
38
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
39 @testset "TensorApplication" begin
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
40 m = SizeDoublingMapping{Int, 1, 1}((3,))
1005
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
41 mm = SizeDoublingMapping{Int, 1, 1}((6,))
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
42 v = [0,1,2]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
43 @test size(m*v) == 2 .*size(v)
1005
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
44 @test (m*v)[1] == (:apply,v,(1,))
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
45 @test (mm*m*v)[1] == (:apply,m*v,(1,))
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
46 @test (mm*m*v)[3] == (:apply,m*v,(3,))
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
47 @test (mm*m*v)[6] == (:apply,m*v,(6,))
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
48 @test_throws MethodError m*m
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
49
938
4a9a96d51940 Fix dispatch issue when indexing a 1D TensorApplication with a CartiesianIndex
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
50 @test (m*v)[CartesianIndex(2)] == (:apply,v,(2,))
1005
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
51 @test (mm*m*v)[CartesianIndex(2)] == (:apply,m*v,(2,))
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
52
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
53 m = SizeDoublingMapping{Float64, 2, 2}((3,3))
1005
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
54 mm = SizeDoublingMapping{Float64, 2, 2}((6,6))
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
55 v = ones(3,3)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
56 @test size(m*v) == 2 .*size(v)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
57 @test (m*v)[1,2] == (:apply,v,(1,2))
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
58
938
4a9a96d51940 Fix dispatch issue when indexing a 1D TensorApplication with a CartiesianIndex
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
59 @test (m*v)[CartesianIndex(2,3)] == (:apply,v,(2,3))
1005
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
60 @test (mm*m*v)[CartesianIndex(4,3)] == (:apply,m*v,(4,3))
938
4a9a96d51940 Fix dispatch issue when indexing a 1D TensorApplication with a CartiesianIndex
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
61
993
2f9beee56a4c Use ScalingTensor instead of custom test type
Jonatan Werpers <jonatan@werpers.com>
parents: 992
diff changeset
62 m = ScalingTensor(2,(3,))
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
63 v = [1,2,3]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
64 @test m*v isa AbstractVector
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
65 @test m*v == [2,4,6]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
66
993
2f9beee56a4c Use ScalingTensor instead of custom test type
Jonatan Werpers <jonatan@werpers.com>
parents: 992
diff changeset
67 m = ScalingTensor(2,(2,2))
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
68 v = [[1 2];[3 4]]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
69 @test m*v == [[2 4];[6 8]]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
70 @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
71
1005
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
72 @testset "Error on index out of bounds" begin
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
73 m = SizeDoublingMapping{Int, 1, 1}((3,))
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
74 v = [0,1,2]
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
75
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
76 @test_throws BoundsError (m*v)[0]
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
77 @test_throws BoundsError (m*v)[7]
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
78 end
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
79
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
80 @testset "Error on unmatched dimensions" begin
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
81 v = [0,1,2]
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
82 m = SizeDoublingMapping{Int, 2, 1}((3,))
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
83 @test_throws MethodError m*ones(Int,2,2)
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
84 @test_throws MethodError m*m*v
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
85 end
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
86
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
87 @testset "Error on unmatched sizes" begin
1006
d9476fede83d Add check methods for range size
Jonatan Werpers <jonatan@werpers.com>
parents: 1005
diff changeset
88 @test_throws DomainSizeMismatch ScalingTensor(2,(2,))*ones(3)
d9476fede83d Add check methods for range size
Jonatan Werpers <jonatan@werpers.com>
parents: 1005
diff changeset
89 @test_throws DomainSizeMismatch ScalingTensor(2,(2,))*ScalingTensor(2,(3,))*ones(3)
1005
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
90 end
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
91
becd95ba0fce Add bounds checking for lazy tensor application and clea up tests a bit
Jonatan Werpers <jonatan@werpers.com>
parents: 1004
diff changeset
92
977
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
93 @testset "Type calculation" begin
993
2f9beee56a4c Use ScalingTensor instead of custom test type
Jonatan Werpers <jonatan@werpers.com>
parents: 992
diff changeset
94 m = ScalingTensor(2,(3,))
942
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
95 v = [1.,2.,3.]
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
96 @test m*v isa AbstractVector{Float64}
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
97 @test m*v == [2.,4.,6.]
977
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
98 @inferred m*v
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
99 @inferred (m*v)[1]
942
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
100
993
2f9beee56a4c Use ScalingTensor instead of custom test type
Jonatan Werpers <jonatan@werpers.com>
parents: 992
diff changeset
101 m = ScalingTensor(2,(2,2))
942
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
102 v = [[1. 2.];[3. 4.]]
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
103 @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
104 @test (m*v)[2,1] == 6.
977
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
105 @inferred m*v
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
106 @inferred (m*v)[1]
942
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
107
993
2f9beee56a4c Use ScalingTensor instead of custom test type
Jonatan Werpers <jonatan@werpers.com>
parents: 992
diff changeset
108 m = ScalingTensor(2. +2. *im,(3,))
942
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
109 v = [1.,2.,3.]
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
110 @test m*v isa AbstractVector{ComplexF64}
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
111 @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
112 @inferred m*v
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
113 @inferred (m*v)[1]
942
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
114
993
2f9beee56a4c Use ScalingTensor instead of custom test type
Jonatan Werpers <jonatan@werpers.com>
parents: 992
diff changeset
115 m = ScalingTensor(1,(3,))
942
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
116 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
117 @test m*v isa AbstractVector{ComplexF64}
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
118 @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
119 @inferred m*v
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
120 @inferred (m*v)[1]
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
121
993
2f9beee56a4c Use ScalingTensor instead of custom test type
Jonatan Werpers <jonatan@werpers.com>
parents: 992
diff changeset
122 m = ScalingTensor(2., (3,))
977
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
123 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
124 @test m*v isa AbstractVector{Vector{Float64}}
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
125 @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
126 @inferred m*v
043d13ef8898 Fix type calculation of LazyTensorMappingApplication element type
Jonatan Werpers <jonatan@werpers.com>
parents: 957
diff changeset
127 @inferred (m*v)[1]
942
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 712
diff changeset
128 end
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
129 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
130
1836
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
131 @testset "TensorNegation" begin
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
132 A = rand(2,3)
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
133 B = rand(3,4)
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
134
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
135 Ã = DenseTensor(A, (1,), (2,))
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
136 B̃ = DenseTensor(B, (1,), (2,))
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
137
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
138 @test -Ã isa TensorNegation
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
139
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
140 v = rand(3)
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
141 @test (-Ã)*v == -(Ã*v)
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
142
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
143 v = rand(4)
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
144 @test (-B̃)*v == -(B̃*v)
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
145
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
146 v = rand(2)
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
147 @test (-Ã)'*v == -(Ã'*v)
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
148
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
149 v = rand(3)
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
150 @test (-B̃)'*v == -(B̃'*v)
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
151
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
152 @test domain_size(-Ã) == (3,)
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
153 @test domain_size(-B̃) == (4,)
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
154
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
155 @test range_size(-Ã) == (2,)
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
156 @test range_size(-B̃) == (3,)
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1726
diff changeset
157 end
1002
271aa6ae1055 Split out a file for tensor types
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
158
995
1ba8a398af9c Rename types
Jonatan Werpers <jonatan@werpers.com>
parents: 993
diff changeset
159 @testset "LazyTensor binary operations" begin
993
2f9beee56a4c Use ScalingTensor instead of custom test type
Jonatan Werpers <jonatan@werpers.com>
parents: 992
diff changeset
160 A = ScalingTensor(2.0, (3,))
2f9beee56a4c Use ScalingTensor instead of custom test type
Jonatan Werpers <jonatan@werpers.com>
parents: 992
diff changeset
161 B = ScalingTensor(3.0, (3,))
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
162
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
163 v = [1.1,1.2,1.3]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
164 for i ∈ eachindex(v)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
165 @test ((A+B)*v)[i] == 2*v[i] + 3*v[i]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
166 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
167
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
168 for i ∈ eachindex(v)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
169 @test ((A-B)*v)[i] == 2*v[i] - 3*v[i]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
170 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
171
993
2f9beee56a4c Use ScalingTensor instead of custom test type
Jonatan Werpers <jonatan@werpers.com>
parents: 992
diff changeset
172
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
173 @test range_size(A+B) == range_size(A) == range_size(B)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
174 @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
175
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
176 @test ((A+B)*ComplexF64[1.1,1.2,1.3])[3] isa ComplexF64
1007
f7a718bcb4da Add checking of sizes to LazyTensorBinaryOperation
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
177
f7a718bcb4da Add checking of sizes to LazyTensorBinaryOperation
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
178 @testset "Error on unmatched sizes" begin
f7a718bcb4da Add checking of sizes to LazyTensorBinaryOperation
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
179 @test_throws Union{DomainSizeMismatch, RangeSizeMismatch} ScalingTensor(2.0, (3,)) + ScalingTensor(2.0, (4,))
f7a718bcb4da Add checking of sizes to LazyTensorBinaryOperation
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
180
f7a718bcb4da Add checking of sizes to LazyTensorBinaryOperation
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
181 @test_throws DomainSizeMismatch ScalingTensor(2.0, (4,)) + SizeDoublingMapping{Float64,1,1}((2,))
f7a718bcb4da Add checking of sizes to LazyTensorBinaryOperation
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
182 @test_throws DomainSizeMismatch SizeDoublingMapping{Float64,1,1}((2,)) + ScalingTensor(2.0, (4,))
f7a718bcb4da Add checking of sizes to LazyTensorBinaryOperation
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
183 @test_throws RangeSizeMismatch ScalingTensor(2.0, (2,)) + SizeDoublingMapping{Float64,1,1}((2,))
f7a718bcb4da Add checking of sizes to LazyTensorBinaryOperation
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
184 @test_throws RangeSizeMismatch SizeDoublingMapping{Float64,1,1}((2,)) + ScalingTensor(2.0, (2,))
f7a718bcb4da Add checking of sizes to LazyTensorBinaryOperation
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
185 end
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
186 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
187
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
188
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
189 @testset "TensorComposition" begin
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
190 A = rand(2,3)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
191 B = rand(3,4)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
192
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
193 Ã = DenseTensor(A, (1,), (2,))
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
194 B̃ = DenseTensor(B, (1,), (2,))
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
195
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
196 @test Ã∘B̃ isa TensorComposition
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
197 @test range_size(Ã∘B̃) == (2,)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
198 @test domain_size(Ã∘B̃) == (4,)
1006
d9476fede83d Add check methods for range size
Jonatan Werpers <jonatan@werpers.com>
parents: 1005
diff changeset
199 @test_throws DomainSizeMismatch B̃∘Ã
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
200
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
201 # @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
202
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
203 v = rand(4)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
204 @test Ã∘B̃*v ≈ A*B*v rtol=1e-14
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 v = rand(2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
207 @test (Ã∘B̃)'*v ≈ B'*A'*v rtol=1e-14
943
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
208
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
209 @test (Ã∘B̃*ComplexF64[1.,2.,3.,4.])[1] isa ComplexF64
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
210 @test ((Ã∘B̃)'*ComplexF64[1.,2.])[1] isa ComplexF64
1084
2e606d4c0ab1 Add support for multiplying a LazyTensor with a scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1017
diff changeset
211
2e606d4c0ab1 Add support for multiplying a LazyTensor with a scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1017
diff changeset
212 a = 2.
2e606d4c0ab1 Add support for multiplying a LazyTensor with a scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1017
diff changeset
213 v = rand(3)
2e606d4c0ab1 Add support for multiplying a LazyTensor with a scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1017
diff changeset
214 @test a*Ã isa TensorComposition
2e606d4c0ab1 Add support for multiplying a LazyTensor with a scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1017
diff changeset
215 @test a*Ã == Ã*a
2e606d4c0ab1 Add support for multiplying a LazyTensor with a scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1017
diff changeset
216 @test range_size(a*Ã) == range_size(Ã)
2e606d4c0ab1 Add support for multiplying a LazyTensor with a scalar
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1017
diff changeset
217 @test domain_size(a*Ã) == domain_size(Ã)
1299
4c0bc52e170f Change to approximate equality in test using floats
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1158
diff changeset
218 @test a*Ã*v ≈ a.*A*v rtol=1e-14
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
219 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
220
992
bc384aaade30 Add a bunch of todos and add a ScalingTensor
Jonatan Werpers <jonatan@werpers.com>
parents: 977
diff changeset
221
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
222 @testset "InflatedTensor" begin
995
1ba8a398af9c Rename types
Jonatan Werpers <jonatan@werpers.com>
parents: 993
diff changeset
223 I(sz...) = IdentityTensor(sz...)
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
224
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
225 Ã = rand(4,2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
226 B̃ = rand(4,2,3)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
227 C̃ = rand(4,2,3)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
228
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
229 A = DenseTensor(Ã,(1,),(2,))
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
230 B = DenseTensor(B̃,(1,2),(3,))
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
231 C = DenseTensor(C̃,(1,),(2,3))
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
232
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
233 @testset "Constructors" begin
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
234 @test InflatedTensor(I(3,2), A, I(4)) isa LazyTensor{Float64, 4, 4}
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
235 @test InflatedTensor(I(3,2), B, I(4)) isa LazyTensor{Float64, 5, 4}
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
236 @test InflatedTensor(I(3), C, I(2,3)) isa LazyTensor{Float64, 4, 5}
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
237 @test InflatedTensor(C, I(2,3)) isa LazyTensor{Float64, 3, 4}
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
238 @test InflatedTensor(I(3), C) isa LazyTensor{Float64, 2, 3}
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
239 @test InflatedTensor(I(3), I(2,3)) isa LazyTensor{Float64, 3, 3}
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
240 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
241
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
242 @testset "Range and domain size" begin
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
243 @test range_size(InflatedTensor(I(3,2), A, I(4))) == (3,2,4,4)
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
244 @test domain_size(InflatedTensor(I(3,2), A, I(4))) == (3,2,2,4)
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
245
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
246 @test range_size(InflatedTensor(I(3,2), B, I(4))) == (3,2,4,2,4)
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
247 @test domain_size(InflatedTensor(I(3,2), B, I(4))) == (3,2,3,4)
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
248
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
249 @test range_size(InflatedTensor(I(3), C, I(2,3))) == (3,4,2,3)
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
250 @test domain_size(InflatedTensor(I(3), C, I(2,3))) == (3,2,3,2,3)
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
251
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
252 @inferred range_size(InflatedTensor(I(3,2), A, I(4))) == (3,2,4,4)
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
253 @inferred domain_size(InflatedTensor(I(3,2), A, I(4))) == (3,2,2,4)
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
254 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
255
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
256 @testset "Application" begin
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
257 # 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
258 # The inflated tensor mappings are chosen to preserve, reduce and increase the dimension of the result compared to the input.
1013
56fe037641ef Simplify some table tests
Jonatan Werpers <jonatan@werpers.com>
parents: 1007
diff changeset
259 cases = [
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
260 (
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
261 InflatedTensor(I(3,2), A, I(4)),
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
262 (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
263 (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
264 ),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
265 (
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
266 InflatedTensor(I(3,2), B, I(4)),
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
267 (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
268 (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
269 ),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
270 (
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
271 InflatedTensor(I(3,2), C, I(4)),
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
272 (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
273 (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
274 ),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
275 (
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
276 InflatedTensor(I(3,2), A),
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
277 (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
278 (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
279 ),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
280 (
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
281 InflatedTensor(I(3,2), B),
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
282 (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
283 (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
284 ),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
285 (
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
286 InflatedTensor(I(3,2), C),
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
287 (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
288 (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
289 ),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
290 (
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
291 InflatedTensor(A,I(4)),
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
292 (v-> @tullio res[a,b] := Ã[a,i]*v[i,b]),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
293 (v-> @tullio res[a,b] := Ã[i,a]*v[i,b]),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
294 ),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
295 (
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
296 InflatedTensor(B,I(4)),
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
297 (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
298 (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
299 ),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
300 (
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
301 InflatedTensor(C,I(4)),
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
302 (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
303 (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
304 ),
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
305 ]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
306
1013
56fe037641ef Simplify some table tests
Jonatan Werpers <jonatan@werpers.com>
parents: 1007
diff changeset
307 @testset "$tm" for (tm, true_apply, true_apply_transpose) ∈ cases
56fe037641ef Simplify some table tests
Jonatan Werpers <jonatan@werpers.com>
parents: 1007
diff changeset
308 v = rand(domain_size(tm)...)
56fe037641ef Simplify some table tests
Jonatan Werpers <jonatan@werpers.com>
parents: 1007
diff changeset
309 @test tm*v ≈ true_apply(v) rtol=1e-14
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
310
1013
56fe037641ef Simplify some table tests
Jonatan Werpers <jonatan@werpers.com>
parents: 1007
diff changeset
311 v = rand(range_size(tm)...)
56fe037641ef Simplify some table tests
Jonatan Werpers <jonatan@werpers.com>
parents: 1007
diff changeset
312 @test tm'*v ≈ true_apply_transpose(v) rtol=1e-14
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
313 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
314
943
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
315 @testset "application to other type" begin
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
316 tm = InflatedTensor(I(3,2), A, I(4))
943
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
317
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
318 v = rand(ComplexF64, domain_size(tm)...)
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
319 @test (tm*v)[1,2,3,1] isa ComplexF64
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
320
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
321 v = rand(ComplexF64, domain_size(tm')...)
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
322 @test (tm'*v)[1,2,2,1] isa ComplexF64
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
323 end
fb060e98ac0a Remove more type assertions
Jonatan Werpers <jonatan@werpers.com>
parents: 942
diff changeset
324
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
325 @testset "Inference of application" begin
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
326 tm = InflatedTensor(I(2,3),ScalingTensor(2.0, (3,2)),I(3,4))
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
327 v = rand(domain_size(tm)...)
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 @inferred apply(tm,v,1,2,3,2,2,4)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
330 @inferred (tm*v)[1,2,3,2,2,4]
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
331 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
332 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
333
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
334 @testset "InflatedTensor of InflatedTensor" begin
993
2f9beee56a4c Use ScalingTensor instead of custom test type
Jonatan Werpers <jonatan@werpers.com>
parents: 992
diff changeset
335 A = ScalingTensor(2.0,(2,3))
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
336 itm = InflatedTensor(I(3,2), A, I(4))
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
337 @test InflatedTensor(I(4), itm, I(2)) == InflatedTensor(I(4,3,2), A, I(4,2))
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
338 @test InflatedTensor(itm, I(2)) == InflatedTensor(I(3,2), A, I(4,2))
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
339 @test InflatedTensor(I(4), itm) == InflatedTensor(I(4,3,2), A, I(4))
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
340
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
341 @test InflatedTensor(I(2), I(2), I(2)) isa InflatedTensor # The constructor should always return its type.
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
342 end
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
343 end
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 @testset "LazyOuterProduct" begin
993
2f9beee56a4c Use ScalingTensor instead of custom test type
Jonatan Werpers <jonatan@werpers.com>
parents: 992
diff changeset
346 A = ScalingTensor(2.0, (5,))
2f9beee56a4c Use ScalingTensor instead of custom test type
Jonatan Werpers <jonatan@werpers.com>
parents: 992
diff changeset
347 B = ScalingTensor(3.0, (3,))
2f9beee56a4c Use ScalingTensor instead of custom test type
Jonatan Werpers <jonatan@werpers.com>
parents: 992
diff changeset
348 C = ScalingTensor(5.0, (3,2))
712
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 AB = LazyOuterProduct(A,B)
995
1ba8a398af9c Rename types
Jonatan Werpers <jonatan@werpers.com>
parents: 993
diff changeset
351 @test AB isa LazyTensor{T,2,2} where T
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
352 @test range_size(AB) == (5,3)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
353 @test domain_size(AB) == (5,3)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
354
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
355 v = rand(range_size(AB)...)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
356 @test AB*v == 6*v
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
357
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
358 ABC = LazyOuterProduct(A,B,C)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
359
995
1ba8a398af9c Rename types
Jonatan Werpers <jonatan@werpers.com>
parents: 993
diff changeset
360 @test ABC isa LazyTensor{T,4,4} where T
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
361 @test range_size(ABC) == (5,3,3,2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
362 @test domain_size(ABC) == (5,3,3,2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
363
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
364 @test A⊗B == AB
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
365 @test A⊗B⊗C == ABC
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
366
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
367 A = rand(3,2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
368 B = rand(2,4,3)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
369
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
370 v₁ = rand(2,4,3)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
371 v₂ = rand(4,3,2)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
372
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
373 Ã = DenseTensor(A,(1,),(2,))
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
374 B̃ = DenseTensor(B,(1,),(2,3))
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
375
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
376 ÃB̃ = LazyOuterProduct(Ã,B̃)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
377 @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
378 @test ÃB̃*v₁ ≈ ABv
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
379
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
380 B̃Ã = LazyOuterProduct(B̃,Ã)
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
381 @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
382 @test B̃Ã*v₂ ≈ BAv
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
383
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
384 @testset "Indentity mapping arguments" begin
995
1ba8a398af9c Rename types
Jonatan Werpers <jonatan@werpers.com>
parents: 993
diff changeset
385 @test LazyOuterProduct(IdentityTensor(3,2), IdentityTensor(1,2)) == IdentityTensor(3,2,1,2)
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
386
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
387 Ã = DenseTensor(A,(1,),(2,))
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
388 @test LazyOuterProduct(IdentityTensor(3,2), Ã) == InflatedTensor(IdentityTensor(3,2),Ã)
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
389 @test LazyOuterProduct(Ã, IdentityTensor(3,2)) == InflatedTensor(Ã,IdentityTensor(3,2))
712
de2df1214394 Split testfile for LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
390
995
1ba8a398af9c Rename types
Jonatan Werpers <jonatan@werpers.com>
parents: 993
diff changeset
391 I1 = IdentityTensor(3,2)
1ba8a398af9c Rename types
Jonatan Werpers <jonatan@werpers.com>
parents: 993
diff changeset
392 I2 = IdentityTensor(4)
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1013
diff changeset
393 @test I1⊗Ã⊗I2 == InflatedTensor(I1, Ã, I2)
712
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
1022
bbbc31953367 Introduce an inflate function in lazy tensors and use it in volume_operator and boundary_operator
Jonatan Werpers <jonatan@werpers.com>
parents: 977
diff changeset
396
bbbc31953367 Introduce an inflate function in lazy tensors and use it in volume_operator and boundary_operator
Jonatan Werpers <jonatan@werpers.com>
parents: 977
diff changeset
397 @testset "inflate" begin
1023
52f07c77299d Merge refactor/lazy_tensors
Jonatan Werpers <jonatan@werpers.com>
parents: 1022 1013
diff changeset
398 I = LazyTensors.inflate(IdentityTensor(),(3,4,5,6), 2)
52f07c77299d Merge refactor/lazy_tensors
Jonatan Werpers <jonatan@werpers.com>
parents: 1022 1013
diff changeset
399 @test I isa LazyTensor{Float64, 3,3}
1022
bbbc31953367 Introduce an inflate function in lazy tensors and use it in volume_operator and boundary_operator
Jonatan Werpers <jonatan@werpers.com>
parents: 977
diff changeset
400 @test range_size(I) == (3,5,6)
bbbc31953367 Introduce an inflate function in lazy tensors and use it in volume_operator and boundary_operator
Jonatan Werpers <jonatan@werpers.com>
parents: 977
diff changeset
401 @test domain_size(I) == (3,5,6)
bbbc31953367 Introduce an inflate function in lazy tensors and use it in volume_operator and boundary_operator
Jonatan Werpers <jonatan@werpers.com>
parents: 977
diff changeset
402
1044
f857057e61e6 Merge default
Jonatan Werpers <jonatan@werpers.com>
parents: 1024 1017
diff changeset
403 @test LazyTensors.inflate(ScalingTensor(1., (4,)),(3,4,5,6), 1) == InflatedTensor(IdentityTensor{Float64}(),ScalingTensor(1., (4,)),IdentityTensor(4,5,6))
f857057e61e6 Merge default
Jonatan Werpers <jonatan@werpers.com>
parents: 1024 1017
diff changeset
404 @test LazyTensors.inflate(ScalingTensor(2., (1,)),(3,4,5,6), 2) == InflatedTensor(IdentityTensor(3),ScalingTensor(2., (1,)),IdentityTensor(5,6))
f857057e61e6 Merge default
Jonatan Werpers <jonatan@werpers.com>
parents: 1024 1017
diff changeset
405 @test LazyTensors.inflate(ScalingTensor(3., (6,)),(3,4,5,6), 4) == InflatedTensor(IdentityTensor(3,4,5),ScalingTensor(3., (6,)),IdentityTensor{Float64}())
1022
bbbc31953367 Introduce an inflate function in lazy tensors and use it in volume_operator and boundary_operator
Jonatan Werpers <jonatan@werpers.com>
parents: 977
diff changeset
406
1024
5be17f647018 Add some more tests
Jonatan Werpers <jonatan@werpers.com>
parents: 1023
diff changeset
407 @test_throws BoundsError LazyTensors.inflate(ScalingTensor(1., (4,)),(3,4,5,6), 0)
5be17f647018 Add some more tests
Jonatan Werpers <jonatan@werpers.com>
parents: 1023
diff changeset
408 @test_throws BoundsError LazyTensors.inflate(ScalingTensor(1., (4,)),(3,4,5,6), 5)
1022
bbbc31953367 Introduce an inflate function in lazy tensors and use it in volume_operator and boundary_operator
Jonatan Werpers <jonatan@werpers.com>
parents: 977
diff changeset
409 end