no assumption on AbstractBlockTuple parameters

Author: ogauthe

src/blockedtuple.jl

@@ -38,7 +38,9 @@ Base.length(bt::AbstractBlockTuple) = length(Tuple(bt))
 Base.lastindex(bt::AbstractBlockTuple) = length(bt)
 
 function Base.map(f, bt::AbstractBlockTuple)
-  return unspecify_type_parameters(typeof(bt)){blocklengths(bt)}(map(f, Tuple(bt)))
+  BL = blocklengths(bt)
+  # use Val to preserve compile time knowledge of BL
+  return unspecify_type_parameters(typeof(bt))(map(f, Tuple(bt)), Val(BL))
 end
 
 # Broadcast interface
@@ -57,7 +59,7 @@ end
 function Base.copy(
   bc::Broadcast.Broadcasted{AbstractBlockTupleBroadcastStyle{BlockLengths,BT}}
 ) where {BlockLengths,BT}
-  return BT{BlockLengths}(bc.f.((Tuple.(bc.args))...))
+  return BT(bc.f.((Tuple.(bc.args))...), Val(BlockLengths))
 end
 
 # BlockArrays interface
@@ -96,6 +98,10 @@ tuplemortar(tt::Tuple{Vararg{Tuple}}) = BlockedTuple{length.(tt)}(flatten_tuples
 function BlockedTuple(flat::Tuple, BlockLengths::Tuple{Vararg{Int}})
   return BlockedTuple{BlockLengths}(flat)
 end
+function BlockedTuple(flat::Tuple, ::Val{BlockLengths}) where {BlockLengths}
+  # use Val to preserve compile time knowledge of BL
+  return BlockedTuple{BlockLengths}(flat)
+end
 BlockedTuple(bt::AbstractBlockTuple) = BlockedTuple{blocklengths(bt)}(Tuple(bt))
 
 # Base interface

test/test_blockedtuple.jl

@@ -42,11 +42,11 @@ using TensorAlgebra: BlockedTuple, tuplemortar
   bt = tuplemortar(((1,), (4, 2), (5, 3)))
   @test Tuple(bt) == (1, 4, 2, 5, 3)
   @test blocklengths(bt) == (1, 2, 2)
-  @test deepcopy(bt) == bt
+  @test (@constinferred deepcopy(bt)) == bt
 
   @test (@constinferred map(n -> n + 1, bt)) ==
     BlockedTuple{blocklengths(bt)}(Tuple(bt) .+ 1)
-  @test bt .+ tuplemortar(((1,), (1, 1), (1, 1))) ==
+  @test (@constinferred bt .+ tuplemortar(((1,), (1, 1), (1, 1)))) ==
     BlockedTuple{blocklengths(bt)}(Tuple(bt) .+ 1)
   @test_throws DimensionMismatch bt .+ tuplemortar(((1, 1), (1, 1), (1,)))