Mercurial > repos > public > sbplib_julia

comparison test/LazyTensors/tuple_manipulation_test.jl @ 1232:a8fa8c1137cc refactor/grids

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Merge refactor/LazyTensors/tuple_manipulation
author Jonatan Werpers <jonatan@werpers.com>
date Sun, 19 Feb 2023 22:07:57 +0100
parents de6a9635f293
children 66012b95fa7b
comparison
equal deleted inserted replaced
1222:5f677cd6f0b6 1232:a8fa8c1137cc
1 using Test 1 using Test
2 using Sbplib.LazyTensors 2 using Sbplib.LazyTensors
3 3
4 @testset "split_index" begin 4 @testset "split_index" begin
5 @test LazyTensors.split_index(Val(2),Val(1),Val(2),Val(2),1,2,3,4,5,6) == ((1,2,:,5,6),(3,4)) 5 @test LazyTensors.split_index(2,1,2,2, 1,2,3,4,5,6) == ((1,2,:,5,6),(3,4))
6 @test LazyTensors.split_index(Val(2),Val(3),Val(2),Val(2),1,2,3,4,5,6) == ((1,2,:,:,:,5,6),(3,4)) 6 @test LazyTensors.split_index(2,3,2,2, 1,2,3,4,5,6) == ((1,2,:,:,:,5,6),(3,4))
7 @test LazyTensors.split_index(Val(3),Val(1),Val(1),Val(2),1,2,3,4,5,6) == ((1,2,3,:,5,6),(4,)) 7 @test LazyTensors.split_index(3,1,1,2, 1,2,3,4,5,6) == ((1,2,3,:,5,6),(4,))
8 @test LazyTensors.split_index(Val(3),Val(2),Val(1),Val(2),1,2,3,4,5,6) == ((1,2,3,:,:,5,6),(4,)) 8 @test LazyTensors.split_index(3,2,1,2, 1,2,3,4,5,6) == ((1,2,3,:,:,5,6),(4,))
9 @test LazyTensors.split_index(Val(1),Val(1),Val(2),Val(3),1,2,3,4,5,6) == ((1,:,4,5,6),(2,3)) 9 @test LazyTensors.split_index(1,1,2,3, 1,2,3,4,5,6) == ((1,:,4,5,6),(2,3))
10 @test LazyTensors.split_index(Val(1),Val(2),Val(2),Val(3),1,2,3,4,5,6) == ((1,:,:,4,5,6),(2,3)) 10 @test LazyTensors.split_index(1,2,2,3, 1,2,3,4,5,6) == ((1,:,:,4,5,6),(2,3))
11 11
12 @test LazyTensors.split_index(Val(0),Val(1),Val(3),Val(3),1,2,3,4,5,6) == ((:,4,5,6),(1,2,3)) 12 @test LazyTensors.split_index(0,1,3,3, 1,2,3,4,5,6) == ((:,4,5,6),(1,2,3))
13 @test LazyTensors.split_index(Val(3),Val(1),Val(3),Val(0),1,2,3,4,5,6) == ((1,2,3,:),(4,5,6)) 13 @test LazyTensors.split_index(3,1,3,0, 1,2,3,4,5,6) == ((1,2,3,:),(4,5,6))
14 14
15 @inferred LazyTensors.split_index(Val(2),Val(3),Val(2),Val(2),1,2,3,2,2,4) 15 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...)
16 end 16 @inferred split_index_static(Val(2),Val(3),Val(2),Val(2),1,2,3,2,2,4)
17
18 @testset "slice_tuple" begin
19 @test LazyTensors.slice_tuple((1,2,3),Val(1), Val(3)) == (1,2,3)
20 @test LazyTensors.slice_tuple((1,2,3,4,5,6),Val(2), Val(5)) == (2,3,4,5)
21 @test LazyTensors.slice_tuple((1,2,3,4,5,6),Val(1), Val(3)) == (1,2,3)
22 @test LazyTensors.slice_tuple((1,2,3,4,5,6),Val(4), Val(6)) == (4,5,6)
23 end 17 end
24 18
25 @testset "split_tuple" begin 19 @testset "split_tuple" begin
26 @testset "2 parts" begin 20 @testset "general" begin
27 @test LazyTensors.split_tuple((),Val(0)) == ((),()) 21 @test LazyTensors.split_tuple((),()) == ()
28 @test LazyTensors.split_tuple((1,),Val(0)) == ((),(1,)) 22 @test LazyTensors.split_tuple((),(0,)) == ((),)
29 @test LazyTensors.split_tuple((1,),Val(1)) == ((1,),()) 23 @test LazyTensors.split_tuple((1,), (1,)) == tuple((1,))
24 @test LazyTensors.split_tuple((1,2), (1,1)) == tuple((1,),(2,))
25 @test LazyTensors.split_tuple((1,2), (0,1,1)) == tuple((),(1,),(2,))
26 @test LazyTensors.split_tuple((1,2), (1,0,1)) == tuple((1,),(),(2,))
27 @test LazyTensors.split_tuple((1,2), (1,1,0)) == tuple((1,),(2,),())
28 @test LazyTensors.split_tuple((1,2,3,4), (2,0,1,1)) == tuple((1,2),(),(3,),(4,))
30 29
31 @test LazyTensors.split_tuple((1,2,3,4),Val(0)) == ((),(1,2,3,4)) 30 err_msg = "length(t) must equal sum(szs)"
32 @test LazyTensors.split_tuple((1,2,3,4),Val(1)) == ((1,),(2,3,4)) 31 @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((), (2,))
33 @test LazyTensors.split_tuple((1,2,3,4),Val(2)) == ((1,2),(3,4)) 32 @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((2,), ())
34 @test LazyTensors.split_tuple((1,2,3,4),Val(3)) == ((1,2,3),(4,)) 33 @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((1,), (2,))
35 @test LazyTensors.split_tuple((1,2,3,4),Val(4)) == ((1,2,3,4),()) 34 @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((1,2), (1,2))
35 @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((1,2), (1))
36 36
37 @test LazyTensors.split_tuple((1,2,true,4),Val(3)) == ((1,2,true),(4,)) 37 split_tuple_static(t, ::Val{SZS}) where {SZS} = LazyTensors.split_tuple(t,SZS)
38 38 @inferred split_tuple_static((1,2,3,4,5,6), Val((3,1,2)))
39 @inferred LazyTensors.split_tuple((1,2,3,4),Val(3)) 39 @inferred split_tuple_static((1,2,3,4),Val((3,1)))
40 @inferred LazyTensors.split_tuple((1,2,true,4),Val(3)) 40 @inferred split_tuple_static((1,2,true,4),Val((3,1)))
41 end 41 @inferred split_tuple_static((1,2,3,4,5,6),Val((3,2,1)))
42 42 @inferred split_tuple_static((1,true,3),Val((1,1,1)))
43 @testset "3 parts" begin
44 @test LazyTensors.split_tuple((),Val(0),Val(0)) == ((),(),())
45 @test LazyTensors.split_tuple((1,2,3),Val(1), Val(1)) == ((1,),(2,),(3,))
46 @test LazyTensors.split_tuple((1,true,3),Val(1), Val(1)) == ((1,),(true,),(3,))
47
48 @test LazyTensors.split_tuple((1,2,3,4,5,6),Val(1),Val(2)) == ((1,),(2,3),(4,5,6))
49 @test LazyTensors.split_tuple((1,2,3,4,5,6),Val(3),Val(2)) == ((1,2,3),(4,5),(6,))
50
51 @inferred LazyTensors.split_tuple((1,2,3,4,5,6),Val(3),Val(2))
52 @inferred LazyTensors.split_tuple((1,true,3),Val(1), Val(1))
53 end 43 end
54 end 44 end
55 45
56 @testset "flatten_tuple" begin 46 @testset "sizes_to_ranges" begin
57 @test LazyTensors.flatten_tuple((1,)) == (1,) 47 @test LazyTensors.sizes_to_ranges((1,)) == (1:1,)
58 @test LazyTensors.flatten_tuple((1,2,3,4,5,6)) == (1,2,3,4,5,6) 48 @test LazyTensors.sizes_to_ranges((2,)) == (1:2,)
59 @test LazyTensors.flatten_tuple((1,2,(3,4),5,6)) == (1,2,3,4,5,6) 49 @test LazyTensors.sizes_to_ranges((2,3)) == (1:2,3:5)
60 @test LazyTensors.flatten_tuple((1,2,(3,(4,5)),6)) == (1,2,3,4,5,6) 50 @test LazyTensors.sizes_to_ranges((3,2,4)) == (1:3,4:5,6:9)
61 @test LazyTensors.flatten_tuple(((1,2),(3,4),(5,),6)) == (1,2,3,4,5,6) 51 @test LazyTensors.sizes_to_ranges((0,2)) == (1:0,1:2)
52 @test LazyTensors.sizes_to_ranges((2,0)) == (1:2,2:1)
53 @test LazyTensors.sizes_to_ranges((2,0,3)) == (1:2,2:1,3:5)
54 end
55
56 @testset "concatenate_tuples" begin
57 @test LazyTensors.concatenate_tuples(()) == ()
58 @test LazyTensors.concatenate_tuples((1,)) == (1,)
59 @test LazyTensors.concatenate_tuples((1,), ()) == (1,)
60 @test LazyTensors.concatenate_tuples((),(1,)) == (1,)
61 @test LazyTensors.concatenate_tuples((1,2,3),(4,5)) == (1,2,3,4,5)
62 @test LazyTensors.concatenate_tuples((1,2,3),(4,5),(6,7)) == (1,2,3,4,5,6,7)
62 end 63 end
63 64
64 @testset "left_pad_tuple" begin 65 @testset "left_pad_tuple" begin
65 @test LazyTensors.left_pad_tuple((1,2), 0, 2) == (1,2) 66 @test LazyTensors.left_pad_tuple((1,2), 0, 2) == (1,2)
66 @test LazyTensors.left_pad_tuple((1,2), 0, 3) == (0,1,2) 67 @test LazyTensors.left_pad_tuple((1,2), 0, 3) == (0,1,2)