sbplib_julia: test/LazyTensors/lazy_tensor_operations

annotate test/LazyTensors/lazy_tensor_operations_test.jl @ 1878:b12e28a03b2e refactor/lazy_tensors/elementwise_ops

Add missing newline

author	Jonatan Werpers <jonatan@werpers.com>
date	Mon, 27 Jan 2025 17:10:59 +0100
parents	4bd998069053
children

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

Mercurial > repos > public > sbplib_julia

annotate test/LazyTensors/lazy_tensor_operations_test.jl @ 1878:b12e28a03b2e refactor/lazy_tensors/elementwise_ops