Mercurial > repos > public > sbplib_julia

comparison test/testLazyTensors.jl @ 457:8fb6a5611c7a feature/inflated_tensormapping

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add tests and docs for flatten_tuple
author Jonatan Werpers <jonatan@werpers.com>
date Wed, 21 Oct 2020 20:10:27 +0200
parents 8f4c31e06689
children 41f9cb6ee5a7
comparison
equal deleted inserted replaced
456:8f4c31e06689 457:8fb6a5611c7a
280 B[1,:,2] + B[2,:,2] + B[3,:,2] atol=5e-13 280 B[1,:,2] + B[2,:,2] + B[3,:,2] atol=5e-13
281 @test B̃*v ≈ B[1,:,1]*v[1,1] + B[2,:,1]*v[2,1] + B[3,:,1]*v[3,1] + 281 @test B̃*v ≈ B[1,:,1]*v[1,1] + B[2,:,1]*v[2,1] + B[3,:,1]*v[3,1] +
282 B[1,:,2]v[1,2] + B[2,:,2]*v[2,2] + B[3,:,2]*v[3,2] atol=5e-13 282 B[1,:,2]v[1,2] + B[2,:,2]*v[2,2] + B[3,:,2]*v[3,2] atol=5e-13
283 283
284 284
285 @inferred (B̃*v)[2] 285 # TODO:
286 # @inferred (B̃*v)[2]
286 end 287 end
287 288
288 289
289 @testset "IdentityMapping" begin 290 @testset "IdentityMapping" begin
290 @test IdentityMapping{Float64}((4,5)) isa IdentityMapping{T,2} where T 291 @test IdentityMapping{Float64}((4,5)) isa IdentityMapping{T,2} where T
345 346
346 @tullio IAIv[a,b,c,d] := Ã[c,i]*v[a,b,i,d] 347 @tullio IAIv[a,b,c,d] := Ã[c,i]*v[a,b,i,d]
347 @test tm*v ≈ IAIv rtol=1e-14 348 @test tm*v ≈ IAIv rtol=1e-14
348 349
349 @inferred LazyTensors.split_index(tm,1,1,1,1) 350 @inferred LazyTensors.split_index(tm,1,1,1,1)
351
350 @inferred (tm*v)[1,1,1,1] 352 @inferred (tm*v)[1,1,1,1]
351 353
352 end 354 end
353 355
354 @testset "slice_tuple" begin 356 @testset "slice_tuple" begin
356 @test LazyTensors.slice_tuple((1,2,3,4,5,6),Val(2), Val(5)) == (2,3,4,5) 358 @test LazyTensors.slice_tuple((1,2,3,4,5,6),Val(2), Val(5)) == (2,3,4,5)
357 @test LazyTensors.slice_tuple((1,2,3,4,5,6),Val(1), Val(3)) == (1,2,3) 359 @test LazyTensors.slice_tuple((1,2,3,4,5,6),Val(1), Val(3)) == (1,2,3)
358 @test LazyTensors.slice_tuple((1,2,3,4,5,6),Val(4), Val(6)) == (4,5,6) 360 @test LazyTensors.slice_tuple((1,2,3,4,5,6),Val(4), Val(6)) == (4,5,6)
359 end 361 end
360 362
361 end 363 @testset "flatten_tuple" begin
364 @test LazyTensors.flatten_tuple((1,)) == (1,)
365 @test LazyTensors.flatten_tuple((1,2,3,4,5,6)) == (1,2,3,4,5,6)
366 @test LazyTensors.flatten_tuple((1,2,(3,4),5,6)) == (1,2,3,4,5,6)
367 @test LazyTensors.flatten_tuple((1,2,(3,(4,5)),6)) == (1,2,3,4,5,6)
368 @test LazyTensors.flatten_tuple(((1,2),(3,4),(5,),6)) == (1,2,3,4,5,6)
369 end
370
371 end