changeset 1237:66012b95fa7b refactor/LazyTensors/tuple_manipulation

Improve readability
author Vidar Stiernström <vidar.stiernstrom@it.uu.se>
date Mon, 20 Feb 2023 20:35:52 +0100
parents de6a9635f293
children c92246fb2c04
files test/LazyTensors/tuple_manipulation_test.jl
diffstat 1 files changed, 9 insertions(+), 5 deletions(-) [+]
line wrap: on
line diff
--- a/test/LazyTensors/tuple_manipulation_test.jl	Sun Feb 19 12:14:43 2023 +0100
+++ b/test/LazyTensors/tuple_manipulation_test.jl	Mon Feb 20 20:35:52 2023 +0100
@@ -12,7 +12,11 @@
     @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...)
+    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
 
@@ -66,14 +70,14 @@
     @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)
-
-    @test_throws DomainError(0, "Can't pad tuple of length 2 to 0 elements") LazyTensors.left_pad_tuple((1,2), 0, 0) == (1,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)
-
-    @test_throws DomainError(0, "Can't pad tuple of length 2 to 0 elements") LazyTensors.right_pad_tuple((1,2), 0, 0) == (1,2)
+    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