Mercurial > repos > public > sbplib_julia
view test/LazyTensors/tuple_manipulation_test.jl @ 2015:5c2448d6a201 feature/grids/geometry_functions tip
Structure tests a bit more
author | Jonatan Werpers <jonatan@werpers.com> |
---|---|
date | Fri, 09 May 2025 15:57:38 +0200 |
parents | 471a948cd2b2 |
children |
line wrap: on
line source
using Test using Diffinitive.LazyTensors @testset "split_index" begin @test LazyTensors.split_index(2,1,2,2, 1,2,3,4,5,6) == ((1,2,:,5,6),(3,4)) @test LazyTensors.split_index(2,3,2,2, 1,2,3,4,5,6) == ((1,2,:,:,:,5,6),(3,4)) @test LazyTensors.split_index(3,1,1,2, 1,2,3,4,5,6) == ((1,2,3,:,5,6),(4,)) @test LazyTensors.split_index(3,2,1,2, 1,2,3,4,5,6) == ((1,2,3,:,:,5,6),(4,)) @test LazyTensors.split_index(1,1,2,3, 1,2,3,4,5,6) == ((1,:,4,5,6),(2,3)) @test LazyTensors.split_index(1,2,2,3, 1,2,3,4,5,6) == ((1,:,:,4,5,6),(2,3)) @test LazyTensors.split_index(0,1,3,3, 1,2,3,4,5,6) == ((:,4,5,6),(1,2,3)) @test LazyTensors.split_index(3,1,3,0, 1,2,3,4,5,6) == ((1,2,3,:),(4,5,6)) split_index_static(::Val{dim_before}, ::Val{dim_view}, ::Val{dim_index}, ::Val{dim_after}, I...) where {dim_before,dim_view,dim_index,dim_after} = LazyTensors.split_index(dim_before, dim_view, dim_index, dim_after, I...) @inferred split_index_static(Val(2),Val(3),Val(2),Val(2),1,2,3,2,2,4) end @testset "split_tuple" begin @testset "general" begin @test LazyTensors.split_tuple((),()) == () @test LazyTensors.split_tuple((),(0,)) == ((),) @test LazyTensors.split_tuple((1,), (1,)) == tuple((1,)) @test LazyTensors.split_tuple((1,2), (1,1)) == tuple((1,),(2,)) @test LazyTensors.split_tuple((1,2), (0,1,1)) == tuple((),(1,),(2,)) @test LazyTensors.split_tuple((1,2), (1,0,1)) == tuple((1,),(),(2,)) @test LazyTensors.split_tuple((1,2), (1,1,0)) == tuple((1,),(2,),()) @test LazyTensors.split_tuple((1,2,3,4), (2,0,1,1)) == tuple((1,2),(),(3,),(4,)) err_msg = "length(t) must equal sum(szs)" @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((), (2,)) @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((2,), ()) @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((1,), (2,)) @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((1,2), (1,2)) @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((1,2), (1)) split_tuple_static(t, ::Val{SZS}) where {SZS} = LazyTensors.split_tuple(t,SZS) @inferred split_tuple_static((1,2,3,4,5,6), Val((3,1,2))) @inferred split_tuple_static((1,2,3,4),Val((3,1))) @inferred split_tuple_static((1,2,true,4),Val((3,1))) @inferred split_tuple_static((1,2,3,4,5,6),Val((3,2,1))) @inferred split_tuple_static((1,true,3),Val((1,1,1))) end end @testset "sizes_to_ranges" begin @test LazyTensors.sizes_to_ranges((1,)) == (1:1,) @test LazyTensors.sizes_to_ranges((2,)) == (1:2,) @test LazyTensors.sizes_to_ranges((2,3)) == (1:2,3:5) @test LazyTensors.sizes_to_ranges((3,2,4)) == (1:3,4:5,6:9) @test LazyTensors.sizes_to_ranges((0,2)) == (1:0,1:2) @test LazyTensors.sizes_to_ranges((2,0)) == (1:2,2:1) @test LazyTensors.sizes_to_ranges((2,0,3)) == (1:2,2:1,3:5) end @testset "concatenate_tuples" begin @test LazyTensors.concatenate_tuples(()) == () @test LazyTensors.concatenate_tuples((1,)) == (1,) @test LazyTensors.concatenate_tuples((1,), ()) == (1,) @test LazyTensors.concatenate_tuples((),(1,)) == (1,) @test LazyTensors.concatenate_tuples((1,2,3),(4,5)) == (1,2,3,4,5) @test LazyTensors.concatenate_tuples((1,2,3),(4,5),(6,7)) == (1,2,3,4,5,6,7) end @testset "left_pad_tuple" begin @test LazyTensors.left_pad_tuple((1,2), 0, 2) == (1,2) @test LazyTensors.left_pad_tuple((1,2), 0, 3) == (0,1,2) @test LazyTensors.left_pad_tuple((3,2), 1, 6) == (1,1,1,1,3,2) err_msg = "Can't pad tuple of length 2 to 0 elements" @test_throws DomainError(0, err_msg) LazyTensors.left_pad_tuple((1,2), 0, 0) == (1,2) end @testset "right_pad_tuple" begin @test LazyTensors.right_pad_tuple((1,2), 0, 2) == (1,2) @test LazyTensors.right_pad_tuple((1,2), 0, 3) == (1,2,0) @test LazyTensors.right_pad_tuple((3,2), 1, 6) == (3,2,1,1,1,1) err_msg = "Can't pad tuple of length 2 to 0 elements" @test_throws DomainError(0,err_msg) LazyTensors.right_pad_tuple((1,2), 0, 0) == (1,2) end