sbplib_julia: test/testLazyTensors.jl comparison

Add tullio as a test dependency and add a test for apply

author	Jonatan Werpers <jonatan@werpers.com>
date	Mon, 19 Oct 2020 22:34:58 +0200
parents	6cf234eef780
children	c1ae837f1a2e

comparison

equal deleted inserted replaced

-:6cf234eef780
+:aeda2698166d
 using Test
 using Sbplib.LazyTensors
 using Sbplib.RegionIndices
+using Tullio
 @testset "LazyTensors" begin
 @testset "Generic Mapping methods" begin
 struct DummyMapping{T,R,D} <: TensorMapping{T,R,D} end
 @inferred range_size(I)
 end
 @testset "InflatedTensorMapping" begin
 I(sz...) = IdentityMapping(sz...)
-A = LazyLinearMap(rand(4,2),(1,),(2,))
-B = LazyLinearMap(rand(4,2,3),(1,2),(3,))
+Ã = rand(4,2)
-C = LazyLinearMap(rand(4,2,3),(1,),(2,3))
+B̃ = rand(4,2,3)
+C̃ = rand(4,2,3)
+A = LazyLinearMap(Ã,(1,),(2,))
+B = LazyLinearMap(B̃,(1,2),(3,))
+C = LazyLinearMap(C̃,(1,),(2,3))
 @test InflatedTensorMapping(I(3,2), A, I(4)) isa TensorMapping{Float64, 4, 4}
 @test InflatedTensorMapping(I(3,2), B, I(4)) isa TensorMapping{Float64, 5, 4}
 @test InflatedTensorMapping(I(3), C, I(2,3)) isa TensorMapping{Float64, 4, 5}
 @test domain_size(InflatedTensorMapping(I(3), C, I(2,3))) == (3,2,3,2,3)
 @inferred range_size(InflatedTensorMapping(I(3,2), A, I(4))) == (3,2,4,4)
 @inferred domain_size(InflatedTensorMapping(I(3,2), A, I(4))) == (3,2,2,4)
+tm = InflatedTensorMapping(I(3,2), A, I(4))
-end
+v = rand(domain_size(tm)...)
-end
+@tullio IAIv[a,b,c,d] := Ã[c,i]*v[a,b,i,d]
+@test tm*v ≈ IAIv rtol=1e-14
+end
+end

Mercurial > repos > public > sbplib_julia