Mercurial > repos > public > sbplib_julia
diff test/testLazyTensors.jl @ 457:8fb6a5611c7a feature/inflated_tensormapping
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 |
line wrap: on
line diff
--- a/test/testLazyTensors.jl Wed Oct 21 19:53:36 2020 +0200 +++ b/test/testLazyTensors.jl Wed Oct 21 20:10:27 2020 +0200 @@ -282,7 +282,8 @@ B[1,:,2]v[1,2] + B[2,:,2]*v[2,2] + B[3,:,2]*v[3,2] atol=5e-13 - @inferred (B̃*v)[2] + # TODO: + # @inferred (B̃*v)[2] end @@ -347,6 +348,7 @@ @test tm*v ≈ IAIv rtol=1e-14 @inferred LazyTensors.split_index(tm,1,1,1,1) + @inferred (tm*v)[1,1,1,1] end @@ -358,4 +360,12 @@ @test LazyTensors.slice_tuple((1,2,3,4,5,6),Val(4), Val(6)) == (4,5,6) end +@testset "flatten_tuple" begin + @test LazyTensors.flatten_tuple((1,)) == (1,) + @test LazyTensors.flatten_tuple((1,2,3,4,5,6)) == (1,2,3,4,5,6) + @test LazyTensors.flatten_tuple((1,2,(3,4),5,6)) == (1,2,3,4,5,6) + @test LazyTensors.flatten_tuple((1,2,(3,(4,5)),6)) == (1,2,3,4,5,6) + @test LazyTensors.flatten_tuple(((1,2),(3,4),(5,),6)) == (1,2,3,4,5,6) end + +end