sbplib_julia: test/testLazyTensors.jl annotate

annotate test/testLazyTensors.jl @ 541:62d96e2cd165 refactor/tensor_index_coupling

Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible

author	Jonatan Werpers <jonatan@werpers.com>
date	Thu, 26 Nov 2020 21:35:34 +0100
parents	fc0f942b8792
children	53828d3ed132

rev	line source
180 b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	1 using Test
335 f4e3e71a4ff4 Fix `using` commands to refer to local modules within the Sbplib package/module Jonatan Werpers <jonatan@werpers.com> parents: 333 diff changeset	2 using Sbplib.LazyTensors
f4e3e71a4ff4 Fix `using` commands to refer to local modules within the Sbplib package/module Jonatan Werpers <jonatan@werpers.com> parents: 333 diff changeset	3 using Sbplib.RegionIndices
180 b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	4
452 aeda2698166d Add tullio as a test dependency and add a test for apply Jonatan Werpers <jonatan@werpers.com> parents: 451 diff changeset	5 using Tullio
aeda2698166d Add tullio as a test dependency and add a test for apply Jonatan Werpers <jonatan@werpers.com> parents: 451 diff changeset	6
338 2b0c9b30ea3b Add test sets for each submodule to make the test output nicer Jonatan Werpers <jonatan@werpers.com> parents: 335 diff changeset	7 @testset "LazyTensors" begin
2b0c9b30ea3b Add test sets for each submodule to make the test output nicer Jonatan Werpers <jonatan@werpers.com> parents: 335 diff changeset	8
180 b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	9 @testset "Generic Mapping methods" begin
186 715ff09bb2ce Rename and export types in LazyTensors follow julia conventions Jonatan Werpers <jonatan@werpers.com> parents: 184 diff changeset	10 struct DummyMapping{T,R,D} <: TensorMapping{T,R,D} end
541 62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 536 diff changeset	11 LazyTensors.apply(m::DummyMapping{T,R,D}, v, I::Vararg{Any,R}) where {T,R,D} = :apply
180 b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	12 @test range_dim(DummyMapping{Int,2,3}()) == 2
b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	13 @test domain_dim(DummyMapping{Int,2,3}()) == 3
541 62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 536 diff changeset	14 @test apply(DummyMapping{Int,2,3}(), zeros(Int, (0,0,0)),0,0) == :apply
492 cd509e57f898 Implement method eltype for TensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 486 diff changeset	15 @test eltype(DummyMapping{Int,2,3}()) == Int
cd509e57f898 Implement method eltype for TensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 486 diff changeset	16 @test eltype(DummyMapping{Float64,2,3}()) == Float64
180 b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	17 end
b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	18
b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	19 @testset "Mapping transpose" begin
186 715ff09bb2ce Rename and export types in LazyTensors follow julia conventions Jonatan Werpers <jonatan@werpers.com> parents: 184 diff changeset	20 struct DummyMapping{T,R,D} <: TensorMapping{T,R,D} end
180 b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	21
541 62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 536 diff changeset	22 LazyTensors.apply(m::DummyMapping{T,R,D}, v, I::Vararg{Any,R}) where {T,R,D} = :apply
62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 536 diff changeset	23 LazyTensors.apply_transpose(m::DummyMapping{T,R,D}, v, I::Vararg{Any,D}) where {T,R,D} = :apply_transpose
180 b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	24
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	25 LazyTensors.range_size(m::DummyMapping{T,R,D}) where {T,R,D} = :range_size
7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	26 LazyTensors.domain_size(m::DummyMapping{T,R,D}) where {T,R,D} = :domain_size
180 b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	27
b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	28 m = DummyMapping{Float64,2,3}()
232 a20bb4fac23d Improve tests for LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 210 diff changeset	29 @test m' isa TensorMapping{Float64, 3,2}
180 b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	30 @test m'' == m
541 62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 536 diff changeset	31 @test apply(m',zeros(Float64,(0,0)), 0, 0, 0) == :apply_transpose
62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 536 diff changeset	32 @test apply(m'',zeros(Float64,(0,0,0)), 0, 0) == :apply
62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 536 diff changeset	33 @test apply_transpose(m', zeros(Float64,(0,0,0)), 0, 0) == :apply
180 b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	34
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	35 @test range_size(m') == :domain_size
7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	36 @test domain_size(m') == :range_size
180 b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	37 end
b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	38
b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	39 @testset "TensorApplication" begin
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	40 struct SizeDoublingMapping{T,R,D} <: TensorMapping{T,R,D}
7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	41 domain_size::NTuple{D,Int}
7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	42 end
180 b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	43
541 62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 536 diff changeset	44 LazyTensors.apply(m::SizeDoublingMapping{T,R,D}, v, i::Vararg{Any,R}) where {T,R,D} = (:apply,v,i)
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	45 LazyTensors.range_size(m::SizeDoublingMapping) = 2 .* m.domain_size
7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	46 LazyTensors.domain_size(m::SizeDoublingMapping) = m.domain_size
180 b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	47
b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	48
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	49 m = SizeDoublingMapping{Int, 1, 1}((3,))
180 b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	50 v = [0,1,2]
b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	51 @test m*v isa AbstractVector{Int}
b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	52 @test size(mv) == 2 .size(v)
541 62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 536 diff changeset	53 @test (m*v)[0] == (:apply,v,(0,))
180 b7397ae8afaf Move tensor mappings to a package LazyTensor Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	54 @test mmv isa AbstractVector{Int}
541 62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 536 diff changeset	55 @test (mmv)[1] == (:apply,m*v,(1,))
62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 536 diff changeset	56 @test (mmv)[3] == (:apply,m*v,(3,))
62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 536 diff changeset	57 @test (mmv)[6] == (:apply,m*v,(6,))
191 25d2ef206fe9 Add a few test cases for TensorApplication Jonatan Werpers <jonatan@werpers.com> parents: 186 diff changeset	58 @test_broken BoundsError == (mmv)[0]
25d2ef206fe9 Add a few test cases for TensorApplication Jonatan Werpers <jonatan@werpers.com> parents: 186 diff changeset	59 @test_broken BoundsError == (mmv)[7]
420 3796e296fe64 Add test that triggers bug Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	60 @test_throws MethodError m*m
257 d4cd4882ee9f Improve error messages when multiblying with TensorMappings and add some tests for TensorOperators Jonatan Werpers <jonatan@werpers.com> parents: 237 diff changeset	61
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	62 m = SizeDoublingMapping{Int, 2, 1}((3,))
276 e3611c2181bc Add 2D test Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 275 diff changeset	63 @test_throws MethodError m*ones(Int,2,2)
e3611c2181bc Add 2D test Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 275 diff changeset	64 @test_throws MethodError mmv
257 d4cd4882ee9f Improve error messages when multiblying with TensorMappings and add some tests for TensorOperators Jonatan Werpers <jonatan@werpers.com> parents: 237 diff changeset	65
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	66 m = SizeDoublingMapping{Float64, 2, 2}((3,3))
276 e3611c2181bc Add 2D test Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 275 diff changeset	67 v = ones(3,3)
e3611c2181bc Add 2D test Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 275 diff changeset	68 @test size(mv) == 2 .size(v)
541 62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 536 diff changeset	69 @test (m*v)[1,2] == (:apply,v,(1,2))
257 d4cd4882ee9f Improve error messages when multiblying with TensorMappings and add some tests for TensorOperators Jonatan Werpers <jonatan@werpers.com> parents: 237 diff changeset	70
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	71 struct ScalingOperator{T,D} <: TensorMapping{T,D,D}
257 d4cd4882ee9f Improve error messages when multiblying with TensorMappings and add some tests for TensorOperators Jonatan Werpers <jonatan@werpers.com> parents: 237 diff changeset	72 λ::T
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	73 size::NTuple{D,Int}
257 d4cd4882ee9f Improve error messages when multiblying with TensorMappings and add some tests for TensorOperators Jonatan Werpers <jonatan@werpers.com> parents: 237 diff changeset	74 end
d4cd4882ee9f Improve error messages when multiblying with TensorMappings and add some tests for TensorOperators Jonatan Werpers <jonatan@werpers.com> parents: 237 diff changeset	75
541 62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 536 diff changeset	76 LazyTensors.apply(m::ScalingOperator{T,D}, v, I::Vararg{Any,D}) where {T,D} = m.λ*v[I...]
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	77 LazyTensors.range_size(m::ScalingOperator) = m.size
7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	78 LazyTensors.domain_size(m::ScalingOperator) = m.size
257 d4cd4882ee9f Improve error messages when multiblying with TensorMappings and add some tests for TensorOperators Jonatan Werpers <jonatan@werpers.com> parents: 237 diff changeset	79
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	80 m = ScalingOperator{Int,1}(2,(3,))
278 d6edd37551ea Extend 2D test Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 277 diff changeset	81 v = [1,2,3]
d6edd37551ea Extend 2D test Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 277 diff changeset	82 @test m*v isa AbstractVector
d6edd37551ea Extend 2D test Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 277 diff changeset	83 @test m*v == [2,4,6]
257 d4cd4882ee9f Improve error messages when multiblying with TensorMappings and add some tests for TensorOperators Jonatan Werpers <jonatan@werpers.com> parents: 237 diff changeset	84
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	85 m = ScalingOperator{Int,2}(2,(2,2))
278 d6edd37551ea Extend 2D test Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 277 diff changeset	86 v = [[1 2];[3 4]]
d6edd37551ea Extend 2D test Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 277 diff changeset	87 @test m*v == [[2 4];[6 8]]
541 62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 536 diff changeset	88 @test (m*v)[2,1] == 6
198 b5c9be7f391c Fix up some formatting Jonatan Werpers <jonatan@werpers.com> parents: 191 diff changeset	89 end
200 c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	90
210 2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 208 diff changeset	91 @testset "TensorMapping binary operations" begin
2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 208 diff changeset	92 struct ScalarMapping{T,R,D} <: TensorMapping{T,R,D}
2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 208 diff changeset	93 λ::T
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	94 range_size::NTuple{R,Int}
7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	95 domain_size::NTuple{D,Int}
210 2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 208 diff changeset	96 end
2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 208 diff changeset	97
541 62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 536 diff changeset	98 LazyTensors.apply(m::ScalarMapping{T,R,D}, v, I::Vararg{Any,R}) where {T,R,D} = m.λ*v[I...]
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	99 LazyTensors.range_size(m::ScalarMapping) = m.domain_size
7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	100 LazyTensors.domain_size(m::ScalarMapping) = m.range_size
210 2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 208 diff changeset	101
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	102 A = ScalarMapping{Float64,1,1}(2.0, (3,), (3,))
7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	103 B = ScalarMapping{Float64,1,1}(3.0, (3,), (3,))
210 2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 208 diff changeset	104
2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 208 diff changeset	105 v = [1.1,1.2,1.3]
2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 208 diff changeset	106 for i ∈ eachindex(v)
2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 208 diff changeset	107 @test ((A+B)v)[i] == 2v[i] + 3*v[i]
2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 208 diff changeset	108 end
2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 208 diff changeset	109
2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 208 diff changeset	110 for i ∈ eachindex(v)
2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 208 diff changeset	111 @test ((A-B)v)[i] == 2v[i] - 3*v[i]
2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 208 diff changeset	112 end
2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 208 diff changeset	113
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	114 @test range_size(A+B) == range_size(A) == range_size(B)
7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	115 @test domain_size(A+B) == domain_size(A) == domain_size(B)
210 2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 208 diff changeset	116 end
2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 208 diff changeset	117
200 c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	118 @testset "LazyArray" begin
393 b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	119 @testset "LazyConstantArray" begin
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	120 @test LazyTensors.LazyConstantArray(3,(3,2)) isa LazyArray{Int,2}
325 41c3c25e4e3b LazyTensors: Simplify the LazyElementwiseOperation type by restricting it and introducing a LazyConstantArray to handle scalars. Jonatan Werpers <jonatan@werpers.com> parents: 321 diff changeset	121
393 b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	122 lca = LazyTensors.LazyConstantArray(3.0,(3,2))
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	123 @test eltype(lca) == Float64
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	124 @test ndims(lca) == 2
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	125 @test size(lca) == (3,2)
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	126 @test lca[2] == 3.0
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	127 end
200 c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	128 struct DummyArray{T,D, T1<:AbstractArray{T,D}} <: LazyArray{T,D}
c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	129 data::T1
c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	130 end
c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	131 Base.size(v::DummyArray) = size(v.data)
237 1c6afdcfd657 Regretsies on the CartesianIndex stuff. Use Vararg instead Jonatan Werpers <jonatan@werpers.com> parents: 236 diff changeset	132 Base.getindex(v::DummyArray{T,D}, I::Vararg{Int,D}) where {T,D} = v.data[I...]
200 c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	133
c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	134 # Test lazy operations
c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	135 v1 = [1, 2.3, 4]
c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	136 v2 = [1., 2, 3]
263 b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	137 s = 3.4
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	138 r_add_v = v1 .+ v2
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	139 r_sub_v = v1 .- v2
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	140 r_times_v = v1 .* v2
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	141 r_div_v = v1 ./ v2
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	142 r_add_s = v1 .+ s
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	143 r_sub_s = v1 .- s
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	144 r_times_s = v1 .* s
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	145 r_div_s = v1 ./ s
200 c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	146 @test isa(v1 +̃ v2, LazyArray)
c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	147 @test isa(v1 -̃ v2, LazyArray)
c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	148 @test isa(v1 *̃ v2, LazyArray)
c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	149 @test isa(v1 /̃ v2, LazyArray)
263 b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	150 @test isa(v1 +̃ s, LazyArray)
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	151 @test isa(v1 -̃ s, LazyArray)
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	152 @test isa(v1 *̃ s, LazyArray)
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	153 @test isa(v1 /̃ s, LazyArray)
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	154 @test isa(s +̃ v1, LazyArray)
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	155 @test isa(s -̃ v1, LazyArray)
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	156 @test isa(s *̃ v1, LazyArray)
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	157 @test isa(s /̃ v1, LazyArray)
200 c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	158 for i ∈ eachindex(v1)
263 b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	159 @test (v1 +̃ v2)[i] == r_add_v[i]
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	160 @test (v1 -̃ v2)[i] == r_sub_v[i]
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	161 @test (v1 *̃ v2)[i] == r_times_v[i]
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	162 @test (v1 /̃ v2)[i] == r_div_v[i]
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	163 @test (v1 +̃ s)[i] == r_add_s[i]
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	164 @test (v1 -̃ s)[i] == r_sub_s[i]
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	165 @test (v1 *̃ s)[i] == r_times_s[i]
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	166 @test (v1 /̃ s)[i] == r_div_s[i]
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	167 @test (s +̃ v1)[i] == r_add_s[i]
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	168 @test (s -̃ v1)[i] == -r_sub_s[i]
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	169 @test (s *̃ v1)[i] == r_times_s[i]
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	170 @test (s /̃ v1)[i] == 1/r_div_s[i]
200 c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	171 end
c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	172 @test_throws BoundsError (v1 +̃ v2)[4]
201 6b7019f2cd41 Add/fix test checking that no size assertion is performed when lazy operations and operations on lazy arrays are decorated using @inbounds Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 200 diff changeset	173 v2 = [1., 2, 3, 4]
6b7019f2cd41 Add/fix test checking that no size assertion is performed when lazy operations and operations on lazy arrays are decorated using @inbounds Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 200 diff changeset	174 # Test that size of arrays is asserted when not specified inbounds
486 8082d43103c1 Add TODOs relating to SizeMismatch Jonatan Werpers <jonatan@werpers.com> parents: 485 diff changeset	175 # TODO: Replace these errors with SizeMismatch
207 f85a0b38f3ff Change error type when doing element wise operations on lazy arrays of missmatching dimension Jonatan Werpers <jonatan@werpers.com> parents: 206 diff changeset	176 @test_throws DimensionMismatch v1 +̃ v2
200 c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	177
c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	178 # Test operations on LazyArray
c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	179 v1 = DummyArray([1, 2.3, 4])
201 6b7019f2cd41 Add/fix test checking that no size assertion is performed when lazy operations and operations on lazy arrays are decorated using @inbounds Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 200 diff changeset	180 v2 = [1., 2, 3]
200 c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	181 @test isa(v1 + v2, LazyArray)
c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	182 @test isa(v2 + v1, LazyArray)
c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	183 @test isa(v1 - v2, LazyArray)
c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	184 @test isa(v2 - v1, LazyArray)
c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	185 for i ∈ eachindex(v2)
263 b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	186 @test (v1 + v2)[i] == (v2 + v1)[i] == r_add_v[i]
b577b5f64530 Add lazy elementwise operations for array with scalar Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 257 diff changeset	187 @test (v1 - v2)[i] == -(v2 - v1)[i] == r_sub_v[i]
200 c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	188 end
c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	189 @test_throws BoundsError (v1 + v2)[4]
c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	190 v2 = [1., 2, 3, 4]
c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	191 # Test that size of arrays is asserted when not specified inbounds
486 8082d43103c1 Add TODOs relating to SizeMismatch Jonatan Werpers <jonatan@werpers.com> parents: 485 diff changeset	192 # TODO: Replace these errors with SizeMismatch
207 f85a0b38f3ff Change error type when doing element wise operations on lazy arrays of missmatching dimension Jonatan Werpers <jonatan@werpers.com> parents: 206 diff changeset	193 @test_throws DimensionMismatch v1 + v2
200 c19bfad0e836 Add tests for LazyArrays and lazy operations Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 198 diff changeset	194 end
338 2b0c9b30ea3b Add test sets for each submodule to make the test output nicer Jonatan Werpers <jonatan@werpers.com> parents: 335 diff changeset	195
377 8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 371 diff changeset	196
371 241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index. Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	197 @testset "LazyFunctionArray" begin
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index. Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	198 @test LazyFunctionArray(i->i^2, (3,)) == [1,4,9]
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index. Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	199 @test LazyFunctionArray((i,j)->i*j, (3,2)) == [
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index. Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	200 1 2;
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index. Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	201 2 4;
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index. Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	202 3 6;
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index. Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	203 ]
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index. Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	204
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index. Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	205 @test size(LazyFunctionArray(i->i^2, (3,))) == (3,)
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index. Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	206 @test size(LazyFunctionArray((i,j)->i*j, (3,2))) == (3,2)
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index. Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	207
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index. Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	208 @inferred LazyFunctionArray(i->i^2, (3,))[2]
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index. Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	209
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index. Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	210 @test_throws BoundsError LazyFunctionArray(i->i^2, (3,))[4]
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index. Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	211 @test_throws BoundsError LazyFunctionArray((i,j)->i*j, (3,2))[4,2]
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index. Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	212 @test_throws BoundsError LazyFunctionArray((i,j)->i*j, (3,2))[2,3]
241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index. Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	213
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 338 diff changeset	214 end
371 241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index. Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	215
412 d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 399 diff changeset	216 @testset "TensorMappingComposition" begin
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 399 diff changeset	217 A = rand(2,3)
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 399 diff changeset	218 B = rand(3,4)
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 399 diff changeset	219
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 399 diff changeset	220 Ã = LazyLinearMap(A, (1,), (2,))
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 399 diff changeset	221 B̃ = LazyLinearMap(B, (1,), (2,))
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 399 diff changeset	222
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 399 diff changeset	223 @test Ã∘B̃ isa TensorMappingComposition
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 399 diff changeset	224 @test range_size(Ã∘B̃) == (2,)
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 399 diff changeset	225 @test domain_size(Ã∘B̃) == (4,)
485 4b49f03bdb98 Switch from DimensionMismatch to SizeMismatch for boundschecks on compositions Jonatan Werpers <jonatan@werpers.com> parents: 479 diff changeset	226 @test_throws SizeMismatch B̃∘Ã
417 4c6604b7d990 Add dimension checking in the constructor Jonatan Werpers <jonatan@werpers.com> parents: 416 diff changeset	227
4c6604b7d990 Add dimension checking in the constructor Jonatan Werpers <jonatan@werpers.com> parents: 416 diff changeset	228 # @test @inbounds B̃∘Ã # Should not error even though dimensions don't match. (Since ]test runs with forced boundschecking this is currently not testable 2020-10-16)
412 d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 399 diff changeset	229
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 399 diff changeset	230 v = rand(4)
428 6737e769a1ca Reduce exessive tol in TensorMappingComposition tests Jonatan Werpers <jonatan@werpers.com> parents: 424 diff changeset	231 @test Ã∘B̃v ≈ AB*v rtol=1e-14
412 d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 399 diff changeset	232
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 399 diff changeset	233 v = rand(2)
428 6737e769a1ca Reduce exessive tol in TensorMappingComposition tests Jonatan Werpers <jonatan@werpers.com> parents: 424 diff changeset	234 @test (Ã∘B̃)'v ≈ B'A'*v rtol=1e-14
412 d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 399 diff changeset	235 end
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 399 diff changeset	236
377 8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 371 diff changeset	237 @testset "LazyLinearMap" begin
393 b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	238 # Test a standard matrix-vector product
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	239 # mapping vectors of size 4 to vectors of size 3.
377 8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 371 diff changeset	240 A = rand(3,4)
394 7ad644d112de Expand tests for LazyLinearMap and update its documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 393 diff changeset	241 Ã = LazyLinearMap(A, (1,), (2,))
377 8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 371 diff changeset	242 v = rand(4)
411 4aa59af074ef Add implementation of apply_transpose for LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 399 diff changeset	243 w = rand(3)
377 8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 371 diff changeset	244
392 418cfd945715 Fix bug in range_size and domain_size for LazyLinearMap and expand the test Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 377 diff changeset	245 @test Ã isa LazyLinearMap{T,1,1} where T
418cfd945715 Fix bug in range_size and domain_size for LazyLinearMap and expand the test Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 377 diff changeset	246 @test Ã isa TensorMapping{T,1,1} where T
394 7ad644d112de Expand tests for LazyLinearMap and update its documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 393 diff changeset	247 @test range_size(Ã) == (3,)
7ad644d112de Expand tests for LazyLinearMap and update its documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 393 diff changeset	248 @test domain_size(Ã) == (4,)
392 418cfd945715 Fix bug in range_size and domain_size for LazyLinearMap and expand the test Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 377 diff changeset	249
393 b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	250 @test Ãones(4) ≈ Aones(4) atol=5e-13
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	251 @test Ãv ≈ Av atol=5e-13
411 4aa59af074ef Add implementation of apply_transpose for LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 399 diff changeset	252 @test Ã'w ≈ A'w
377 8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 371 diff changeset	253
399 3b4b1758a8ad Add a check for permuted indecies. Jonatan Werpers <jonatan@werpers.com> parents: 394 diff changeset	254 A = rand(2,3,4)
3b4b1758a8ad Add a check for permuted indecies. Jonatan Werpers <jonatan@werpers.com> parents: 394 diff changeset	255 @test_throws DomainError LazyLinearMap(A, (3,1), (2,))
3b4b1758a8ad Add a check for permuted indecies. Jonatan Werpers <jonatan@werpers.com> parents: 394 diff changeset	256
393 b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	257 # Test more exotic mappings
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	258 B = rand(3,4,2)
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	259 # Map vectors of size 2 to matrices of size (3,4)
394 7ad644d112de Expand tests for LazyLinearMap and update its documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 393 diff changeset	260 B̃ = LazyLinearMap(B, (1,2), (3,))
393 b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	261 v = rand(2)
394 7ad644d112de Expand tests for LazyLinearMap and update its documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 393 diff changeset	262
7ad644d112de Expand tests for LazyLinearMap and update its documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 393 diff changeset	263 @test range_size(B̃) == (3,4)
7ad644d112de Expand tests for LazyLinearMap and update its documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 393 diff changeset	264 @test domain_size(B̃) == (2,)
393 b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	265 @test B̃ isa TensorMapping{T,2,1} where T
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	266 @test B̃*ones(2) ≈ B[:,:,1] + B[:,:,2] atol=5e-13
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	267 @test B̃v ≈ B[:,:,1]v[1] + B[:,:,2]*v[2] atol=5e-13
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	268
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	269 # Map matrices of size (3,2) to vectors of size 4
394 7ad644d112de Expand tests for LazyLinearMap and update its documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 393 diff changeset	270 B̃ = LazyLinearMap(B, (2,), (1,3))
393 b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	271 v = rand(3,2)
394 7ad644d112de Expand tests for LazyLinearMap and update its documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 393 diff changeset	272
7ad644d112de Expand tests for LazyLinearMap and update its documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 393 diff changeset	273 @test range_size(B̃) == (4,)
7ad644d112de Expand tests for LazyLinearMap and update its documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 393 diff changeset	274 @test domain_size(B̃) == (3,2)
393 b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	275 @test B̃ isa TensorMapping{T,1,2} where T
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	276 @test B̃*ones(3,2) ≈ B[1,:,1] + B[2,:,1] + B[3,:,1] +
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	277 B[1,:,2] + B[2,:,2] + B[3,:,2] atol=5e-13
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	278 @test B̃v ≈ B[1,:,1]v[1,1] + B[2,:,1]v[2,1] + B[3,:,1]v[3,1] +
b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	279 B[1,:,2]v[1,2] + B[2,:,2]v[2,2] + B[3,:,2]v[3,2] atol=5e-13
377 8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 371 diff changeset	280
455 b86312d14873 Make split_index type stable Jonatan Werpers <jonatan@werpers.com> parents: 453 diff changeset	281
457 8fb6a5611c7a Add tests and docs for flatten_tuple Jonatan Werpers <jonatan@werpers.com> parents: 456 diff changeset	282 # TODO:
8fb6a5611c7a Add tests and docs for flatten_tuple Jonatan Werpers <jonatan@werpers.com> parents: 456 diff changeset	283 # @inferred (B̃*v)[2]
371 241bd2512c20 Add a LazyFunctionArray that evaluates a function for each index. Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	284 end
377 8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 371 diff changeset	285
429 46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	286
434 648a36ebac99 Change name from LazyIdentity to IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 433 diff changeset	287 @testset "IdentityMapping" begin
648a36ebac99 Change name from LazyIdentity to IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 433 diff changeset	288 @test IdentityMapping{Float64}((4,5)) isa IdentityMapping{T,2} where T
648a36ebac99 Change name from LazyIdentity to IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 433 diff changeset	289 @test IdentityMapping{Float64}((4,5)) isa TensorMapping{T,2,2} where T
441 552189707fbe Add convenience constructor Jonatan Werpers <jonatan@werpers.com> parents: 440 diff changeset	290 @test IdentityMapping{Float64}((4,5)) == IdentityMapping{Float64}(4,5)
429 46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	291
442 e70e47fbfa7c Add another convenience constructor Jonatan Werpers <jonatan@werpers.com> parents: 441 diff changeset	292 @test IdentityMapping(3,2) isa IdentityMapping{Float64,2}
e70e47fbfa7c Add another convenience constructor Jonatan Werpers <jonatan@werpers.com> parents: 441 diff changeset	293
429 46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	294 for sz ∈ [(4,5),(3,),(5,6,4)]
434 648a36ebac99 Change name from LazyIdentity to IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 433 diff changeset	295 I = IdentityMapping{Float64}(sz)
429 46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	296 v = rand(sz...)
46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	297 @test I*v == v
46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	298 @test I'*v == v
46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	299
46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	300 @test range_size(I) == sz
46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	301 @test domain_size(I) == sz
46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	302 end
46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	303
434 648a36ebac99 Change name from LazyIdentity to IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 433 diff changeset	304 I = IdentityMapping{Float64}((4,5))
429 46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	305 v = rand(4,5)
46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	306 @inferred (I*v)[3,2]
440 a57b71343aeb Re-add test for inferred types on transposed identity. Jonatan Werpers <jonatan@werpers.com> parents: 439 diff changeset	307 @inferred (I'*v)[3,2]
429 46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	308 @inferred range_size(I)
453 c1ae837f1a2e Add some type inferense tests. Failing.. Jonatan Werpers <jonatan@werpers.com> parents: 452 diff changeset	309
c1ae837f1a2e Add some type inferense tests. Failing.. Jonatan Werpers <jonatan@werpers.com> parents: 452 diff changeset	310 @inferred range_dim(I)
c1ae837f1a2e Add some type inferense tests. Failing.. Jonatan Werpers <jonatan@werpers.com> parents: 452 diff changeset	311 @inferred domain_dim(I)
479 95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 471 diff changeset	312
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 471 diff changeset	313 Ã = rand(4,2)
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 471 diff changeset	314 A = LazyLinearMap(Ã,(1,),(2,))
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 471 diff changeset	315 I1 = IdentityMapping{Float64}(2)
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 471 diff changeset	316 I2 = IdentityMapping{Float64}(4)
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 471 diff changeset	317 @test A∘I1 == A
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 471 diff changeset	318 @test I2∘A == A
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 471 diff changeset	319 @test I1∘I1 == I1
485 4b49f03bdb98 Switch from DimensionMismatch to SizeMismatch for boundschecks on compositions Jonatan Werpers <jonatan@werpers.com> parents: 479 diff changeset	320 @test_throws SizeMismatch I1∘A
4b49f03bdb98 Switch from DimensionMismatch to SizeMismatch for boundschecks on compositions Jonatan Werpers <jonatan@werpers.com> parents: 479 diff changeset	321 @test_throws SizeMismatch A∘I2
4b49f03bdb98 Switch from DimensionMismatch to SizeMismatch for boundschecks on compositions Jonatan Werpers <jonatan@werpers.com> parents: 479 diff changeset	322 @test_throws SizeMismatch I1∘I2
377 8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 371 diff changeset	323 end
429 46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	324
448 912ae510dec9 Use new name of IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 435 diff changeset	325 @testset "InflatedTensorMapping" begin
451 6cf234eef780 Add tests for constructors and size calculations Jonatan Werpers <jonatan@werpers.com> parents: 450 diff changeset	326 I(sz...) = IdentityMapping(sz...)
452 aeda2698166d Add tullio as a test dependency and add a test for apply Jonatan Werpers <jonatan@werpers.com> parents: 451 diff changeset	327
aeda2698166d Add tullio as a test dependency and add a test for apply Jonatan Werpers <jonatan@werpers.com> parents: 451 diff changeset	328 Ã = rand(4,2)
aeda2698166d Add tullio as a test dependency and add a test for apply Jonatan Werpers <jonatan@werpers.com> parents: 451 diff changeset	329 B̃ = rand(4,2,3)
aeda2698166d Add tullio as a test dependency and add a test for apply Jonatan Werpers <jonatan@werpers.com> parents: 451 diff changeset	330 C̃ = rand(4,2,3)
aeda2698166d Add tullio as a test dependency and add a test for apply Jonatan Werpers <jonatan@werpers.com> parents: 451 diff changeset	331
aeda2698166d Add tullio as a test dependency and add a test for apply Jonatan Werpers <jonatan@werpers.com> parents: 451 diff changeset	332 A = LazyLinearMap(Ã,(1,),(2,))
aeda2698166d Add tullio as a test dependency and add a test for apply Jonatan Werpers <jonatan@werpers.com> parents: 451 diff changeset	333 B = LazyLinearMap(B̃,(1,2),(3,))
aeda2698166d Add tullio as a test dependency and add a test for apply Jonatan Werpers <jonatan@werpers.com> parents: 451 diff changeset	334 C = LazyLinearMap(C̃,(1,),(2,3))
451 6cf234eef780 Add tests for constructors and size calculations Jonatan Werpers <jonatan@werpers.com> parents: 450 diff changeset	335
508 27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	336 @testset "Constructors" begin
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	337 @test InflatedTensorMapping(I(3,2), A, I(4)) isa TensorMapping{Float64, 4, 4}
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	338 @test InflatedTensorMapping(I(3,2), B, I(4)) isa TensorMapping{Float64, 5, 4}
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	339 @test InflatedTensorMapping(I(3), C, I(2,3)) isa TensorMapping{Float64, 4, 5}
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	340 @test InflatedTensorMapping(C, I(2,3)) isa TensorMapping{Float64, 3, 4}
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	341 @test InflatedTensorMapping(I(3), C) isa TensorMapping{Float64, 2, 3}
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	342 @test InflatedTensorMapping(I(3), I(2,3)) isa TensorMapping{Float64, 3, 3}
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	343 end
451 6cf234eef780 Add tests for constructors and size calculations Jonatan Werpers <jonatan@werpers.com> parents: 450 diff changeset	344
508 27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	345 @testset "Range and domain size" begin
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	346 @test range_size(InflatedTensorMapping(I(3,2), A, I(4))) == (3,2,4,4)
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	347 @test domain_size(InflatedTensorMapping(I(3,2), A, I(4))) == (3,2,2,4)
448 912ae510dec9 Use new name of IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 435 diff changeset	348
508 27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	349 @test range_size(InflatedTensorMapping(I(3,2), B, I(4))) == (3,2,4,2,4)
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	350 @test domain_size(InflatedTensorMapping(I(3,2), B, I(4))) == (3,2,3,4)
471 3f3001d1020f Add tests to InflatedTensorMapping where one of before and after are missing Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 466 diff changeset	351
508 27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	352 @test range_size(InflatedTensorMapping(I(3), C, I(2,3))) == (3,4,2,3)
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	353 @test domain_size(InflatedTensorMapping(I(3), C, I(2,3))) == (3,2,3,2,3)
457 8fb6a5611c7a Add tests and docs for flatten_tuple Jonatan Werpers <jonatan@werpers.com> parents: 456 diff changeset	354
508 27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	355 @inferred range_size(InflatedTensorMapping(I(3,2), A, I(4))) == (3,2,4,4)
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	356 @inferred domain_size(InflatedTensorMapping(I(3,2), A, I(4))) == (3,2,2,4)
458 41f9cb6ee5a7 Moar tests and moar type stableness Jonatan Werpers <jonatan@werpers.com> parents: 457 diff changeset	357 end
41f9cb6ee5a7 Moar tests and moar type stableness Jonatan Werpers <jonatan@werpers.com> parents: 457 diff changeset	358
508 27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	359 @testset "Application" begin
529 dfad4c16bedb Fix typo Jonatan Werpers <jonatan@werpers.com> parents: 528 diff changeset	360 # Testing regular application and transposed application with inflation "before", "after" and "before and after".
528 78090ec3074d Add a comment on how the test cases were chosen Jonatan Werpers <jonatan@werpers.com> parents: 523 diff changeset	361 # The inflated tensor mappings are chosen to preserve, reduce and increase the dimension of the result compared to the input.
522 3968210797a9 Run tests for inflated tensor mapping in a loop to make things more readable Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	362 tests = [
3968210797a9 Run tests for inflated tensor mapping in a loop to make things more readable Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	363 (
3968210797a9 Run tests for inflated tensor mapping in a loop to make things more readable Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	364 InflatedTensorMapping(I(3,2), A, I(4)),
526 be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	365 (v-> @tullio res[a,b,c,d] := Ã[c,i]*v[a,b,i,d]), # Expected result of apply
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	366 (v-> @tullio res[a,b,c,d] := Ã[i,c]*v[a,b,i,d]), # Expected result of apply_transpose
522 3968210797a9 Run tests for inflated tensor mapping in a loop to make things more readable Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	367 ),
3968210797a9 Run tests for inflated tensor mapping in a loop to make things more readable Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	368 (
523 390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	369 InflatedTensorMapping(I(3,2), B, I(4)),
390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	370 (v-> @tullio res[a,b,c,d,e] := B̃[c,d,i]*v[a,b,i,e]),
526 be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	371 (v-> @tullio res[a,b,c,d] := B̃[i,j,c]*v[a,b,i,j,d]),
523 390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	372 ),
390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	373 (
390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	374 InflatedTensorMapping(I(3,2), C, I(4)),
390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	375 (v-> @tullio res[a,b,c,d] := C̃[c,i,j]*v[a,b,i,j,d]),
526 be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	376 (v-> @tullio res[a,b,c,d,e] := C̃[i,c,d]*v[a,b,i,e]),
523 390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	377 ),
390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	378 (
522 3968210797a9 Run tests for inflated tensor mapping in a loop to make things more readable Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	379 InflatedTensorMapping(I(3,2), A),
3968210797a9 Run tests for inflated tensor mapping in a loop to make things more readable Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	380 (v-> @tullio res[a,b,c] := Ã[c,i]*v[a,b,i]),
526 be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	381 (v-> @tullio res[a,b,c] := Ã[i,c]*v[a,b,i]),
522 3968210797a9 Run tests for inflated tensor mapping in a loop to make things more readable Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	382 ),
3968210797a9 Run tests for inflated tensor mapping in a loop to make things more readable Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	383 (
523 390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	384 InflatedTensorMapping(I(3,2), B),
390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	385 (v-> @tullio res[a,b,c,d] := B̃[c,d,i]*v[a,b,i]),
526 be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	386 (v-> @tullio res[a,b,c] := B̃[i,j,c]*v[a,b,i,j]),
523 390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	387 ),
390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	388 (
390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	389 InflatedTensorMapping(I(3,2), C),
390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	390 (v-> @tullio res[a,b,c] := C̃[c,i,j]*v[a,b,i,j]),
526 be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	391 (v-> @tullio res[a,b,c,d] := C̃[i,c,d]*v[a,b,i]),
523 390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	392 ),
390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	393 (
522 3968210797a9 Run tests for inflated tensor mapping in a loop to make things more readable Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	394 InflatedTensorMapping(A,I(4)),
523 390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	395 (v-> @tullio res[a,b] := Ã[a,i]*v[i,b]),
526 be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	396 (v-> @tullio res[a,b] := Ã[i,a]*v[i,b]),
523 390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	397 ),
390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	398 (
390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	399 InflatedTensorMapping(B,I(4)),
390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	400 (v-> @tullio res[a,b,c] := B̃[a,b,i]*v[i,c]),
526 be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	401 (v-> @tullio res[a,b] := B̃[i,j,a]*v[i,j,b]),
523 390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	402 ),
390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	403 (
390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	404 InflatedTensorMapping(C,I(4)),
390901860b0a Add some tests to test a wider range of possibilities for inflation Jonatan Werpers <jonatan@werpers.com> parents: 522 diff changeset	405 (v-> @tullio res[a,b] := C̃[a,i,j]*v[i,j,b]),
526 be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	406 (v-> @tullio res[a,b,c] := C̃[i,a,b]*v[i,c]),
522 3968210797a9 Run tests for inflated tensor mapping in a loop to make things more readable Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	407 ),
3968210797a9 Run tests for inflated tensor mapping in a loop to make things more readable Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	408 ]
508 27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	409
526 be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	410 @testset "apply" begin
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	411 for i ∈ 1:length(tests)
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	412 tm = tests[i][1]
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	413 v = rand(domain_size(tm)...)
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	414 true_value = tests[i][2](v)
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	415 @test tm*v ≈ true_value rtol=1e-14
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	416 end
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	417 end
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	418
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	419 @testset "apply_transpose" begin
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	420 for i ∈ 1:length(tests)
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	421 tm = tests[i][1]
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	422 v = rand(range_size(tm)...)
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	423 true_value = tests[i][3](v)
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	424 @test tm'*v ≈ true_value rtol=1e-14
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 524 diff changeset	425 end
522 3968210797a9 Run tests for inflated tensor mapping in a loop to make things more readable Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	426 end
508 27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	427
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	428 @testset "Inference of application" begin
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	429 struct ScalingOperator{T,D} <: TensorMapping{T,D,D}
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	430 λ::T
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	431 size::NTuple{D,Int}
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	432 end
458 41f9cb6ee5a7 Moar tests and moar type stableness Jonatan Werpers <jonatan@werpers.com> parents: 457 diff changeset	433
541 62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 536 diff changeset	434 LazyTensors.apply(m::ScalingOperator{T,D}, v, I::Vararg{Any,D}) where {T,D} = m.λ*v[I...]
508 27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	435 LazyTensors.range_size(m::ScalingOperator) = m.size
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	436 LazyTensors.domain_size(m::ScalingOperator) = m.size
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	437
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	438 tm = InflatedTensorMapping(I(2,3),ScalingOperator(2.0, (3,2)),I(3,4))
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	439 v = rand(domain_size(tm)...)
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	440
541 62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 536 diff changeset	441 @inferred apply(tm,v,1,2,3,2,2,4)
508 27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	442 @inferred (tm*v)[1,2,3,2,2,4]
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	443 end
27e64b3d3efa Restructure tests for InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 500 diff changeset	444 end
453 c1ae837f1a2e Add some type inferense tests. Failing.. Jonatan Werpers <jonatan@werpers.com> parents: 452 diff changeset	445
493 df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 486 diff changeset	446 @testset "InflatedTensorMapping of InflatedTensorMapping" begin
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 486 diff changeset	447 A = ScalingOperator(2.0,(2,3))
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 486 diff changeset	448 itm = InflatedTensorMapping(I(3,2), A, I(4))
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 486 diff changeset	449 @test InflatedTensorMapping(I(4), itm, I(2)) == InflatedTensorMapping(I(4,3,2), A, I(4,2))
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 486 diff changeset	450 @test InflatedTensorMapping(itm, I(2)) == InflatedTensorMapping(I(3,2), A, I(4,2))
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 486 diff changeset	451 @test InflatedTensorMapping(I(4), itm) == InflatedTensorMapping(I(4,3,2), A, I(4))
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 486 diff changeset	452
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 486 diff changeset	453 @test InflatedTensorMapping(I(2), I(2), I(2)) isa InflatedTensorMapping # The constructor should always return its type.
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 486 diff changeset	454 end
429 46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	455 end
448 912ae510dec9 Use new name of IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 435 diff changeset	456
520 fe86ac896377 Start refactoring split index and apply to accomodate future addition of apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	457 @testset "split_index" begin
fe86ac896377 Start refactoring split index and apply to accomodate future addition of apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	458 @test LazyTensors.split_index(Val(2),Val(1),Val(2),Val(2),1,2,3,4,5,6) == ((1,2,:,5,6),(3,4))
fe86ac896377 Start refactoring split index and apply to accomodate future addition of apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	459 @test LazyTensors.split_index(Val(2),Val(3),Val(2),Val(2),1,2,3,4,5,6) == ((1,2,:,:,:,5,6),(3,4))
fe86ac896377 Start refactoring split index and apply to accomodate future addition of apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	460 @test LazyTensors.split_index(Val(3),Val(1),Val(1),Val(2),1,2,3,4,5,6) == ((1,2,3,:,5,6),(4,))
fe86ac896377 Start refactoring split index and apply to accomodate future addition of apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	461 @test LazyTensors.split_index(Val(3),Val(2),Val(1),Val(2),1,2,3,4,5,6) == ((1,2,3,:,:,5,6),(4,))
fe86ac896377 Start refactoring split index and apply to accomodate future addition of apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	462 @test LazyTensors.split_index(Val(1),Val(1),Val(2),Val(3),1,2,3,4,5,6) == ((1,:,4,5,6),(2,3))
fe86ac896377 Start refactoring split index and apply to accomodate future addition of apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	463 @test LazyTensors.split_index(Val(1),Val(2),Val(2),Val(3),1,2,3,4,5,6) == ((1,:,:,4,5,6),(2,3))
fe86ac896377 Start refactoring split index and apply to accomodate future addition of apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	464
fe86ac896377 Start refactoring split index and apply to accomodate future addition of apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	465 @test LazyTensors.split_index(Val(0),Val(1),Val(3),Val(3),1,2,3,4,5,6) == ((:,4,5,6),(1,2,3))
fe86ac896377 Start refactoring split index and apply to accomodate future addition of apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	466 @test LazyTensors.split_index(Val(3),Val(1),Val(3),Val(0),1,2,3,4,5,6) == ((1,2,3,:),(4,5,6))
fe86ac896377 Start refactoring split index and apply to accomodate future addition of apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	467
fe86ac896377 Start refactoring split index and apply to accomodate future addition of apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	468 @inferred LazyTensors.split_index(Val(2),Val(3),Val(2),Val(2),1,2,3,2,2,4)
fe86ac896377 Start refactoring split index and apply to accomodate future addition of apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	469 end
fe86ac896377 Start refactoring split index and apply to accomodate future addition of apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 508 diff changeset	470
456 8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	471 @testset "slice_tuple" begin
8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	472 @test LazyTensors.slice_tuple((1,2,3),Val(1), Val(3)) == (1,2,3)
8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	473 @test LazyTensors.slice_tuple((1,2,3,4,5,6),Val(2), Val(5)) == (2,3,4,5)
8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	474 @test LazyTensors.slice_tuple((1,2,3,4,5,6),Val(1), Val(3)) == (1,2,3)
8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	475 @test LazyTensors.slice_tuple((1,2,3,4,5,6),Val(4), Val(6)) == (4,5,6)
448 912ae510dec9 Use new name of IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 435 diff changeset	476 end
456 8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	477
532 588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	478 @testset "split_tuple" begin
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	479 @testset "2 parts" begin
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	480 @test LazyTensors.split_tuple((),Val(0)) == ((),())
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	481 @test LazyTensors.split_tuple((1,),Val(0)) == ((),(1,))
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	482 @test LazyTensors.split_tuple((1,),Val(1)) == ((1,),())
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	483
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	484 @test LazyTensors.split_tuple((1,2,3,4),Val(0)) == ((),(1,2,3,4))
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	485 @test LazyTensors.split_tuple((1,2,3,4),Val(1)) == ((1,),(2,3,4))
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	486 @test LazyTensors.split_tuple((1,2,3,4),Val(2)) == ((1,2),(3,4))
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	487 @test LazyTensors.split_tuple((1,2,3,4),Val(3)) == ((1,2,3),(4,))
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	488 @test LazyTensors.split_tuple((1,2,3,4),Val(4)) == ((1,2,3,4),())
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	489
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	490 @inferred LazyTensors.split_tuple((1,2,3,4),Val(3))
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	491 end
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	492
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	493 @testset "3 parts" begin
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	494 @test LazyTensors.split_tuple((),Val(0),Val(0)) == ((),(),())
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	495 @test LazyTensors.split_tuple((1,2,3),Val(1), Val(1)) == ((1,),(2,),(3,))
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	496
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	497 @test LazyTensors.split_tuple((1,2,3,4,5,6),Val(1),Val(2)) == ((1,),(2,3),(4,5,6))
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	498 @test LazyTensors.split_tuple((1,2,3,4,5,6),Val(3),Val(2)) == ((1,2,3),(4,5),(6,))
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	499
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	500 @inferred LazyTensors.split_tuple((1,2,3,4,5,6),Val(3),Val(2))
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	501 end
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	502 end
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 526 diff changeset	503
457 8fb6a5611c7a Add tests and docs for flatten_tuple Jonatan Werpers <jonatan@werpers.com> parents: 456 diff changeset	504 @testset "flatten_tuple" begin
8fb6a5611c7a Add tests and docs for flatten_tuple Jonatan Werpers <jonatan@werpers.com> parents: 456 diff changeset	505 @test LazyTensors.flatten_tuple((1,)) == (1,)
8fb6a5611c7a Add tests and docs for flatten_tuple Jonatan Werpers <jonatan@werpers.com> parents: 456 diff changeset	506 @test LazyTensors.flatten_tuple((1,2,3,4,5,6)) == (1,2,3,4,5,6)
8fb6a5611c7a Add tests and docs for flatten_tuple Jonatan Werpers <jonatan@werpers.com> parents: 456 diff changeset	507 @test LazyTensors.flatten_tuple((1,2,(3,4),5,6)) == (1,2,3,4,5,6)
8fb6a5611c7a Add tests and docs for flatten_tuple Jonatan Werpers <jonatan@werpers.com> parents: 456 diff changeset	508 @test LazyTensors.flatten_tuple((1,2,(3,(4,5)),6)) == (1,2,3,4,5,6)
8fb6a5611c7a Add tests and docs for flatten_tuple Jonatan Werpers <jonatan@werpers.com> parents: 456 diff changeset	509 @test LazyTensors.flatten_tuple(((1,2),(3,4),(5,),6)) == (1,2,3,4,5,6)
456 8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	510 end
457 8fb6a5611c7a Add tests and docs for flatten_tuple Jonatan Werpers <jonatan@werpers.com> parents: 456 diff changeset	511
468 a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 458 diff changeset	512
477 79a88269d7d0 Add some tests Jonatan Werpers <jonatan@werpers.com> parents: 476 diff changeset	513 @testset "LazyOuterProduct" begin
470 0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	514 struct ScalingOperator{T,D} <: TensorMapping{T,D,D}
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	515 λ::T
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	516 size::NTuple{D,Int}
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	517 end
468 a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 458 diff changeset	518
541 62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 536 diff changeset	519 LazyTensors.apply(m::ScalingOperator{T,D}, v, I::Vararg{Any,D}) where {T,D} = m.λ*v[I...]
470 0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	520 LazyTensors.range_size(m::ScalingOperator) = m.size
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	521 LazyTensors.domain_size(m::ScalingOperator) = m.size
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	522
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	523 A = ScalingOperator(2.0, (5,))
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	524 B = ScalingOperator(3.0, (3,))
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	525 C = ScalingOperator(5.0, (3,2))
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	526
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	527 AB = LazyOuterProduct(A,B)
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	528 @test AB isa TensorMapping{T,2,2} where T
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	529 @test range_size(AB) == (5,3)
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	530 @test domain_size(AB) == (5,3)
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	531
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	532 v = rand(range_size(AB)...)
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	533 @test ABv == 6v
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	534
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	535 ABC = LazyOuterProduct(A,B,C)
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	536
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	537 @test ABC isa TensorMapping{T,4,4} where T
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	538 @test range_size(ABC) == (5,3,3,2)
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	539 @test domain_size(ABC) == (5,3,3,2)
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	540
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	541 @test A⊗B == AB
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	542 @test A⊗B⊗C == ABC
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	543
477 79a88269d7d0 Add some tests Jonatan Werpers <jonatan@werpers.com> parents: 476 diff changeset	544 A = rand(3,2)
79a88269d7d0 Add some tests Jonatan Werpers <jonatan@werpers.com> parents: 476 diff changeset	545 B = rand(2,4,3)
79a88269d7d0 Add some tests Jonatan Werpers <jonatan@werpers.com> parents: 476 diff changeset	546
79a88269d7d0 Add some tests Jonatan Werpers <jonatan@werpers.com> parents: 476 diff changeset	547 v₁ = rand(2,4,3)
79a88269d7d0 Add some tests Jonatan Werpers <jonatan@werpers.com> parents: 476 diff changeset	548 v₂ = rand(4,3,2)
79a88269d7d0 Add some tests Jonatan Werpers <jonatan@werpers.com> parents: 476 diff changeset	549
79a88269d7d0 Add some tests Jonatan Werpers <jonatan@werpers.com> parents: 476 diff changeset	550 Ã = LazyLinearMap(A,(1,),(2,))
79a88269d7d0 Add some tests Jonatan Werpers <jonatan@werpers.com> parents: 476 diff changeset	551 B̃ = LazyLinearMap(B,(1,),(2,3))
79a88269d7d0 Add some tests Jonatan Werpers <jonatan@werpers.com> parents: 476 diff changeset	552
79a88269d7d0 Add some tests Jonatan Werpers <jonatan@werpers.com> parents: 476 diff changeset	553 ÃB̃ = LazyOuterProduct(Ã,B̃)
79a88269d7d0 Add some tests Jonatan Werpers <jonatan@werpers.com> parents: 476 diff changeset	554 @tullio ABv[i,k] := A[i,j]B[k,l,m]v₁[j,l,m]
79a88269d7d0 Add some tests Jonatan Werpers <jonatan@werpers.com> parents: 476 diff changeset	555 @test ÃB̃*v₁ ≈ ABv
79a88269d7d0 Add some tests Jonatan Werpers <jonatan@werpers.com> parents: 476 diff changeset	556
79a88269d7d0 Add some tests Jonatan Werpers <jonatan@werpers.com> parents: 476 diff changeset	557 B̃Ã = LazyOuterProduct(B̃,Ã)
79a88269d7d0 Add some tests Jonatan Werpers <jonatan@werpers.com> parents: 476 diff changeset	558 @tullio BAv[k,i] := A[i,j]B[k,l,m]v₂[l,m,j]
79a88269d7d0 Add some tests Jonatan Werpers <jonatan@werpers.com> parents: 476 diff changeset	559 @test B̃Ã*v₂ ≈ BAv
468 a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 458 diff changeset	560
491 2dc2eac27f75 Add special methods for Identity mappings Jonatan Werpers <jonatan@werpers.com> parents: 477 diff changeset	561 @testset "Indentity mapping arguments" begin
2dc2eac27f75 Add special methods for Identity mappings Jonatan Werpers <jonatan@werpers.com> parents: 477 diff changeset	562 @test LazyOuterProduct(IdentityMapping(3,2), IdentityMapping(1,2)) == IdentityMapping(3,2,1,2)
2dc2eac27f75 Add special methods for Identity mappings Jonatan Werpers <jonatan@werpers.com> parents: 477 diff changeset	563
2dc2eac27f75 Add special methods for Identity mappings Jonatan Werpers <jonatan@werpers.com> parents: 477 diff changeset	564 Ã = LazyLinearMap(A,(1,),(2,))
2dc2eac27f75 Add special methods for Identity mappings Jonatan Werpers <jonatan@werpers.com> parents: 477 diff changeset	565 @test LazyOuterProduct(IdentityMapping(3,2), Ã) == InflatedTensorMapping(IdentityMapping(3,2),Ã)
2dc2eac27f75 Add special methods for Identity mappings Jonatan Werpers <jonatan@werpers.com> parents: 477 diff changeset	566 @test LazyOuterProduct(Ã, IdentityMapping(3,2)) == InflatedTensorMapping(Ã,IdentityMapping(3,2))
500 1c50b5f1b356 Add a test Jonatan Werpers <jonatan@werpers.com> parents: 498 diff changeset	567
1c50b5f1b356 Add a test Jonatan Werpers <jonatan@werpers.com> parents: 498 diff changeset	568 I1 = IdentityMapping(3,2)
1c50b5f1b356 Add a test Jonatan Werpers <jonatan@werpers.com> parents: 498 diff changeset	569 I2 = IdentityMapping(4)
1c50b5f1b356 Add a test Jonatan Werpers <jonatan@werpers.com> parents: 498 diff changeset	570 @test I1⊗Ã⊗I2 == InflatedTensorMapping(I1, Ã, I2)
491 2dc2eac27f75 Add special methods for Identity mappings Jonatan Werpers <jonatan@werpers.com> parents: 477 diff changeset	571 end
2dc2eac27f75 Add special methods for Identity mappings Jonatan Werpers <jonatan@werpers.com> parents: 477 diff changeset	572
457 8fb6a5611c7a Add tests and docs for flatten_tuple Jonatan Werpers <jonatan@werpers.com> parents: 456 diff changeset	573 end
468 a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 458 diff changeset	574
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 458 diff changeset	575 end

Mercurial > repos > public > sbplib_julia

annotate test/testLazyTensors.jl @ 541:62d96e2cd165 refactor/tensor_index_coupling