Mercurial > repos > public > sbplib_julia

comparison test/LazyTensors/tuple_manipulation_test.jl @ 1231:de6a9635f293 refactor/LazyTensors/tuple_manipulation

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add docstring for concatenate_tuples
author Jonatan Werpers <jonatan@werpers.com>
date Sun, 19 Feb 2023 12:14:43 +0100
parents 8f4259fbd39c
children 66012b95fa7b
comparison
equal deleted inserted replaced
1230:723a583cef96 1231:de6a9635f293
32 @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((2,), ()) 32 @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((2,), ())
33 @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((1,), (2,)) 33 @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((1,), (2,))
34 @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((1,2), (1,2)) 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)) 35 @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((1,2), (1))
36 36
37
38 split_tuple_static(t, ::Val{SZS}) where {SZS} = LazyTensors.split_tuple(t,SZS) 37 split_tuple_static(t, ::Val{SZS}) where {SZS} = LazyTensors.split_tuple(t,SZS)
39
40 @inferred split_tuple_static((1,2,3,4,5,6), Val((3,1,2))) 38 @inferred split_tuple_static((1,2,3,4,5,6), Val((3,1,2)))
41 @inferred split_tuple_static((1,2,3,4),Val((3,1))) 39 @inferred split_tuple_static((1,2,3,4),Val((3,1)))
42 @inferred split_tuple_static((1,2,true,4),Val((3,1))) 40 @inferred split_tuple_static((1,2,true,4),Val((3,1)))
43 @inferred split_tuple_static((1,2,3,4,5,6),Val((3,2,1))) 41 @inferred split_tuple_static((1,2,3,4,5,6),Val((3,2,1)))
44 @inferred split_tuple_static((1,true,3),Val((1,1,1))) 42 @inferred split_tuple_static((1,true,3),Val((1,1,1)))