sbplib_julia: test/LazyTensors/lazy_tensor_operations

comparison test/LazyTensors/lazy_tensor_operations_test.jl @ 1101:1e8270c18edb feature/lazy_tensors/pretty_printing

Merge default

author	Jonatan Werpers <jonatan@werpers.com>
date	Thu, 12 May 2022 21:52:47 +0200
parents	2c1a0722ddb9 2e606d4c0ab1
children

comparison

equal deleted inserted replaced

-:67969bd7e642
+:1e8270c18edb
 @test range_size(m') == :domain_size
 @test domain_size(m') == :range_size
 end
-@testset "LazyTensorApplication" begin
+@testset "TensorApplication" begin
 m = SizeDoublingMapping{Int, 1, 1}((3,))
 mm = SizeDoublingMapping{Int, 1, 1}((6,))
 v = [0,1,2]
 @test size(m*v) == 2 .*size(v)
 @test (m*v)[1] == (:apply,v,(1,))
 @test_throws RangeSizeMismatch SizeDoublingMapping{Float64,1,1}((2,)) + ScalingTensor(2.0, (2,))
 end
 end
-@testset "LazyTensorComposition" begin
+@testset "TensorComposition" begin
 A = rand(2,3)
 B = rand(3,4)
-Ã = LazyLinearMap(A, (1,), (2,))
+Ã = DenseTensor(A, (1,), (2,))
-B̃ = LazyLinearMap(B, (1,), (2,))
+B̃ = DenseTensor(B, (1,), (2,))
-@test Ã∘B̃ isa LazyTensorComposition
+@test Ã∘B̃ isa TensorComposition
 @test range_size(Ã∘B̃) == (2,)
 @test domain_size(Ã∘B̃) == (4,)
 @test_throws DomainSizeMismatch B̃∘Ã
 # @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)
 v = rand(2)
 @test (Ã∘B̃)'*v ≈ B'*A'*v rtol=1e-14
 @test (Ã∘B̃*ComplexF64[1.,2.,3.,4.])[1] isa ComplexF64
 @test ((Ã∘B̃)'*ComplexF64[1.,2.])[1] isa ComplexF64
-end
+a = 2.
+v = rand(3)
-@testset "InflatedLazyTensor" begin
+@test a*Ã isa TensorComposition
+@test a*Ã == Ã*a
+@test range_size(a*Ã) == range_size(Ã)
+@test domain_size(a*Ã) == domain_size(Ã)
+@test a*Ã*v == a.*A*v
+end
+@testset "InflatedTensor" begin
 I(sz...) = IdentityTensor(sz...)
 Ã = rand(4,2)
 B̃ = rand(4,2,3)
 C̃ = rand(4,2,3)
-A = LazyLinearMap(Ã,(1,),(2,))
+A = DenseTensor(Ã,(1,),(2,))
-B = LazyLinearMap(B̃,(1,2),(3,))
+B = DenseTensor(B̃,(1,2),(3,))
-C = LazyLinearMap(C̃,(1,),(2,3))
+C = DenseTensor(C̃,(1,),(2,3))
 @testset "Constructors" begin
-@test InflatedLazyTensor(I(3,2), A, I(4)) isa LazyTensor{Float64, 4, 4}
+@test InflatedTensor(I(3,2), A, I(4)) isa LazyTensor{Float64, 4, 4}
-@test InflatedLazyTensor(I(3,2), B, I(4)) isa LazyTensor{Float64, 5, 4}
+@test InflatedTensor(I(3,2), B, I(4)) isa LazyTensor{Float64, 5, 4}
-@test InflatedLazyTensor(I(3), C, I(2,3)) isa LazyTensor{Float64, 4, 5}
+@test InflatedTensor(I(3), C, I(2,3)) isa LazyTensor{Float64, 4, 5}
-@test InflatedLazyTensor(C, I(2,3)) isa LazyTensor{Float64, 3, 4}
+@test InflatedTensor(C, I(2,3)) isa LazyTensor{Float64, 3, 4}
-@test InflatedLazyTensor(I(3), C) isa LazyTensor{Float64, 2, 3}
+@test InflatedTensor(I(3), C) isa LazyTensor{Float64, 2, 3}
-@test InflatedLazyTensor(I(3), I(2,3)) isa LazyTensor{Float64, 3, 3}
+@test InflatedTensor(I(3), I(2,3)) isa LazyTensor{Float64, 3, 3}
 end
 @testset "Range and domain size" begin
-@test range_size(InflatedLazyTensor(I(3,2), A, I(4))) == (3,2,4,4)
+@test range_size(InflatedTensor(I(3,2), A, I(4))) == (3,2,4,4)
-@test domain_size(InflatedLazyTensor(I(3,2), A, I(4))) == (3,2,2,4)
+@test domain_size(InflatedTensor(I(3,2), A, I(4))) == (3,2,2,4)
-@test range_size(InflatedLazyTensor(I(3,2), B, I(4))) == (3,2,4,2,4)
+@test range_size(InflatedTensor(I(3,2), B, I(4))) == (3,2,4,2,4)
-@test domain_size(InflatedLazyTensor(I(3,2), B, I(4))) == (3,2,3,4)
+@test domain_size(InflatedTensor(I(3,2), B, I(4))) == (3,2,3,4)
-@test range_size(InflatedLazyTensor(I(3), C, I(2,3))) == (3,4,2,3)
+@test range_size(InflatedTensor(I(3), C, I(2,3))) == (3,4,2,3)
-@test domain_size(InflatedLazyTensor(I(3), C, I(2,3))) == (3,2,3,2,3)
+@test domain_size(InflatedTensor(I(3), C, I(2,3))) == (3,2,3,2,3)
-@inferred range_size(InflatedLazyTensor(I(3,2), A, I(4))) == (3,2,4,4)
+@inferred range_size(InflatedTensor(I(3,2), A, I(4))) == (3,2,4,4)
-@inferred domain_size(InflatedLazyTensor(I(3,2), A, I(4))) == (3,2,2,4)
+@inferred domain_size(InflatedTensor(I(3,2), A, I(4))) == (3,2,2,4)
 end
 @testset "Application" begin
 # Testing regular application and transposed application with inflation "before", "after" and "before and after".
 # The inflated tensor mappings are chosen to preserve, reduce and increase the dimension of the result compared to the input.
-tests = [
+cases = [
 (
-InflatedLazyTensor(I(3,2), A, I(4)),
+InflatedTensor(I(3,2), A, I(4)),
 (v-> @tullio res[a,b,c,d] := Ã[c,i]*v[a,b,i,d]), # Expected result of apply
 (v-> @tullio res[a,b,c,d] := Ã[i,c]*v[a,b,i,d]), # Expected result of apply_transpose
 ),
 (
-InflatedLazyTensor(I(3,2), B, I(4)),
+InflatedTensor(I(3,2), B, I(4)),
 (v-> @tullio res[a,b,c,d,e] := B̃[c,d,i]*v[a,b,i,e]),
 (v-> @tullio res[a,b,c,d] := B̃[i,j,c]*v[a,b,i,j,d]),
 ),
 (
-InflatedLazyTensor(I(3,2), C, I(4)),
+InflatedTensor(I(3,2), C, I(4)),
 (v-> @tullio res[a,b,c,d] := C̃[c,i,j]*v[a,b,i,j,d]),
 (v-> @tullio res[a,b,c,d,e] := C̃[i,c,d]*v[a,b,i,e]),
 ),
 (
-InflatedLazyTensor(I(3,2), A),
+InflatedTensor(I(3,2), A),
 (v-> @tullio res[a,b,c] := Ã[c,i]*v[a,b,i]),
 (v-> @tullio res[a,b,c] := Ã[i,c]*v[a,b,i]),
 ),
 (
-InflatedLazyTensor(I(3,2), B),
+InflatedTensor(I(3,2), B),
 (v-> @tullio res[a,b,c,d] := B̃[c,d,i]*v[a,b,i]),
 (v-> @tullio res[a,b,c] := B̃[i,j,c]*v[a,b,i,j]),
 ),
 (
-InflatedLazyTensor(I(3,2), C),
+InflatedTensor(I(3,2), C),
 (v-> @tullio res[a,b,c] := C̃[c,i,j]*v[a,b,i,j]),
 (v-> @tullio res[a,b,c,d] := C̃[i,c,d]*v[a,b,i]),
 ),
 (
-InflatedLazyTensor(A,I(4)),
+InflatedTensor(A,I(4)),
 (v-> @tullio res[a,b] := Ã[a,i]*v[i,b]),
 (v-> @tullio res[a,b] := Ã[i,a]*v[i,b]),
 ),
 (
-InflatedLazyTensor(B,I(4)),
+InflatedTensor(B,I(4)),
 (v-> @tullio res[a,b,c] := B̃[a,b,i]*v[i,c]),
 (v-> @tullio res[a,b] := B̃[i,j,a]*v[i,j,b]),
 ),
 (
-InflatedLazyTensor(C,I(4)),
+InflatedTensor(C,I(4)),
 (v-> @tullio res[a,b] := C̃[a,i,j]*v[i,j,b]),
 (v-> @tullio res[a,b,c] := C̃[i,a,b]*v[i,c]),
 ),
 ]
-@testset "apply" begin
+@testset "$tm" for (tm, true_apply, true_apply_transpose) ∈ cases
-for i ∈ 1:length(tests)
+v = rand(domain_size(tm)...)
-tm = tests[i][1]
+@test tm*v ≈ true_apply(v) rtol=1e-14
-v = rand(domain_size(tm)...)
-true_value = tests[i][2](v)
+v = rand(range_size(tm)...)
-@test tm*v ≈ true_value rtol=1e-14
+@test tm'*v ≈ true_apply_transpose(v) rtol=1e-14
-end
 end
-@testset "apply_transpose" begin
-for i ∈ 1:length(tests)
-tm = tests[i][1]
-v = rand(range_size(tm)...)
-true_value = tests[i][3](v)
-@test tm'*v ≈ true_value rtol=1e-14
-end
-end
 @testset "application to other type" begin
-tm = InflatedLazyTensor(I(3,2), A, I(4))
+tm = InflatedTensor(I(3,2), A, I(4))
 v = rand(ComplexF64, domain_size(tm)...)
 @test (tm*v)[1,2,3,1] isa ComplexF64
 v = rand(ComplexF64, domain_size(tm')...)
 @test (tm'*v)[1,2,2,1] isa ComplexF64
 end
 @testset "Inference of application" begin
-tm = InflatedLazyTensor(I(2,3),ScalingTensor(2.0, (3,2)),I(3,4))
+tm = InflatedTensor(I(2,3),ScalingTensor(2.0, (3,2)),I(3,4))
 v = rand(domain_size(tm)...)
 @inferred apply(tm,v,1,2,3,2,2,4)
 @inferred (tm*v)[1,2,3,2,2,4]
 end
 end
-@testset "InflatedLazyTensor of InflatedLazyTensor" begin
+@testset "InflatedTensor of InflatedTensor" begin
 A = ScalingTensor(2.0,(2,3))
-itm = InflatedLazyTensor(I(3,2), A, I(4))
+itm = InflatedTensor(I(3,2), A, I(4))
-@test  InflatedLazyTensor(I(4), itm, I(2)) == InflatedLazyTensor(I(4,3,2), A, I(4,2))
+@test  InflatedTensor(I(4), itm, I(2)) == InflatedTensor(I(4,3,2), A, I(4,2))
-@test  InflatedLazyTensor(itm, I(2)) == InflatedLazyTensor(I(3,2), A, I(4,2))
+@test  InflatedTensor(itm, I(2)) == InflatedTensor(I(3,2), A, I(4,2))
-@test  InflatedLazyTensor(I(4), itm) == InflatedLazyTensor(I(4,3,2), A, I(4))
+@test  InflatedTensor(I(4), itm) == InflatedTensor(I(4,3,2), A, I(4))
-@test InflatedLazyTensor(I(2), I(2), I(2)) isa InflatedLazyTensor # The constructor should always return its type.
+@test InflatedTensor(I(2), I(2), I(2)) isa InflatedTensor # The constructor should always return its type.
 end
 @testset "Pretty printing" begin
 cases = [
-InflatedLazyTensor(I(4), ScalingTensor(2., (3,2)), I(2)) => (
+InflatedTensor(I(4), ScalingTensor(2., (3,2)), I(2)) => (
 regular="I(4)⊗ScalingTensor{Float64}(2.0, (3, 2))⊗I(2)",
 compact="I(4)⊗2.0*I(3,2)⊗I(2)"
 )
 ]
 B = rand(2,4,3)
 v₁ = rand(2,4,3)
 v₂ = rand(4,3,2)
-Ã = LazyLinearMap(A,(1,),(2,))
+Ã = DenseTensor(A,(1,),(2,))
-B̃ = LazyLinearMap(B,(1,),(2,3))
+B̃ = DenseTensor(B,(1,),(2,3))
 ÃB̃ = LazyOuterProduct(Ã,B̃)
 @tullio ABv[i,k] := A[i,j]*B[k,l,m]*v₁[j,l,m]
 @test ÃB̃*v₁ ≈ ABv
 @test B̃Ã*v₂ ≈ BAv
 @testset "Indentity mapping arguments" begin
 @test LazyOuterProduct(IdentityTensor(3,2), IdentityTensor(1,2)) == IdentityTensor(3,2,1,2)
-Ã = LazyLinearMap(A,(1,),(2,))
+Ã = DenseTensor(A,(1,),(2,))
-@test LazyOuterProduct(IdentityTensor(3,2), Ã) == InflatedLazyTensor(IdentityTensor(3,2),Ã)
+@test LazyOuterProduct(IdentityTensor(3,2), Ã) == InflatedTensor(IdentityTensor(3,2),Ã)
-@test LazyOuterProduct(Ã, IdentityTensor(3,2)) == InflatedLazyTensor(Ã,IdentityTensor(3,2))
+@test LazyOuterProduct(Ã, IdentityTensor(3,2)) == InflatedTensor(Ã,IdentityTensor(3,2))
 I1 = IdentityTensor(3,2)
 I2 = IdentityTensor(4)
-@test I1⊗Ã⊗I2 == InflatedLazyTensor(I1, Ã, I2)
+@test I1⊗Ã⊗I2 == InflatedTensor(I1, Ã, I2)
 end
 end

Mercurial > repos > public > sbplib_julia

comparison test/LazyTensors/lazy_tensor_operations_test.jl @ 1101:1e8270c18edb feature/lazy_tensors/pretty_printing