test type stability

Author: ogauthe

test/Project.toml

@@ -1,4 +1,5 @@
 [deps]
+Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
 BlockArrays = "8e7c35d0-a365-5155-bbbb-fb81a777f24e"
 Combinatorics = "861a8166-3701-5b0c-9a16-15d98fcdc6aa"
 EllipsisNotation = "da5c29d0-fa7d-589e-88eb-ea29b0a81949"
@@ -8,16 +9,16 @@ LabelledNumbers = "f856a3a6-4152-4ec4-b2a7-02c1a55d7993"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 NDTensors = "23ae76d9-e61a-49c4-8f12-3f1a16adf9cf"
 Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"
-TensorAlgebra = "68bd88dc-f39d-4e12-b2ca-f046b68fcc6a"
-TensorOperations = "6aa20fa7-93e2-5fca-9bc0-fbd0db3c71a2"
-Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
 SafeTestsets = "1bc83da4-3b8d-516f-aca4-4fe02f6d838f"
 Suppressor = "fd094767-a336-5f1f-9728-57cf17d0bbfb"
+TensorAlgebra = "68bd88dc-f39d-4e12-b2ca-f046b68fcc6a"
+TensorOperations = "6aa20fa7-93e2-5fca-9bc0-fbd0db3c71a2"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
+TestExtras = "5ed8adda-3752-4e41-b88a-e8b09835ee3a"
 
 [compat]
-TensorOperations = "4.1.1"
 Aqua = "0.8.9"
 SafeTestsets = "0.1"
 Suppressor = "0.2"
+TensorOperations = "4.1.1"
 Test = "1.10"

test/test_blockedtuple.jl

@@ -1,6 +1,7 @@
 using Test: @test, @test_throws
 
 using BlockArrays: Block, blocklength, blocklengths, blockedrange, blockisequal, blocks
+using TestExtras: @constinferred
 
 using TensorAlgebra: BlockedTuple
 
@@ -10,20 +11,20 @@ using TensorAlgebra: BlockedTuple
 
   bt = BlockedTuple{divs}(flat)
 
-  @test Tuple(bt) == flat
+  @test @constinferred Tuple(bt) == flat
   @test bt == BlockedTuple((true,), ('a', 2), ("b", 3.0))
   @test BlockedTuple(bt) == bt
   @test blocklength(bt) == 3
   @test blocklengths(bt) == (1, 2, 2)
-  @test blocks(bt) == ((true,), ('a', 2), ("b", 3.0))
+  @test @constinferred blocks(bt) == ((true,), ('a', 2), ("b", 3.0))
 
-  @test bt[1] == true
-  @test bt[2] == 'a'
-  @test bt[Block(1)] == blocks(bt)[1]
-  @test bt[Block(2)] == blocks(bt)[2]
-  @test bt[Block(1):Block(2)] == blocks(bt)[1:2]
-  @test bt[Block(2)[1:2]] == ('a', 2)
-  @test bt[2:4] == ('a', 2, "b")
+  @test @constinferred bt[1] == true
+  @test @constinferred bt[2] == 'a'
+  @test @constinferred bt[Block(1)] == blocks(bt)[1]
+  @test @constinferred bt[Block(2)] == blocks(bt)[2]
+  @test @constinferred bt[Block(1):Block(2)] == blocks(bt)[1:2]
+  @test @constinferred bt[Block(2)[1:2]] == ('a', 2)
+  @test @constinferred bt[2:4] == ('a', 2, "b")
 
   @test firstindex(bt) == 1
   @test lastindex(bt) == 5
@@ -41,11 +42,11 @@ using TensorAlgebra: BlockedTuple
   @test copy(bt) == bt
   @test deepcopy(bt) == bt
 
-  @test map(n -> n + 1, bt) == BlockedTuple{blocklengths(bt)}(Tuple(bt) .+ 1)
+  @test @constinferred map(n -> n + 1, bt) == BlockedTuple{blocklengths(bt)}(Tuple(bt) .+ 1)
   @test bt .+ BlockedTuple((1,), (1, 1), (1, 1)) ==
     BlockedTuple{blocklengths(bt)}(Tuple(bt) .+ 1)
   @test_throws DimensionMismatch bt .+ BlockedTuple((1, 1), (1, 1), (1,))
 
   bt = BlockedTuple((1:2, 1:2), (1:3,))
-  @test length.(bt) == BlockedTuple((2, 2), (3,))
+  @test @constinferred length.(bt) == BlockedTuple((2, 2), (3,))
 end