Fix index dimension mismatch in TensorNetwork construction

src/TensorNetwork.jl

@@ -45,14 +45,19 @@ struct TensorNetwork <: AbstractTensorNetwork
         tensormap = IdDict{Tensor,Vector{Symbol}}(tensor => inds(tensor) for tensor in tensors)
 
         indexmap = reduce(tensors; init=Dict{Symbol,Vector{Tensor}}()) do dict, tensor
-            # TODO check for inconsistent dimensions?
             for index in inds(tensor)
                 # TODO use lambda? `Tensor[]` might be reused
                 push!(get!(dict, index, Tensor[]), tensor)
             end
             dict
         end
 
+        # Check for inconsistent index dimensions
+        for ind in keys(indexmap)
+            dims = map(tensor -> size(tensor, ind), indexmap[ind])
+            length(unique(dims)) == 1 || throw(DimensionMismatch("Index $(ind) has inconsistent dimension: $(dims)"))
+        end
+
         return new(indexmap, tensormap, CachedField{Vector{Tensor}}())
     end
 end

test/TensorNetwork_test.jl

@@ -23,7 +23,7 @@
         @testset "TensorNetwork with tensors of different dimensions" begin
             tensor1 = Tensor(zeros(2, 2), (:i, :j))
             tensor2 = Tensor(zeros(3, 3), (:j, :k))
-            @test_skip @test_throws DimensionMismatch tn = TensorNetwork([tensor1, tensor2])
+            @test_throws DimensionMismatch tn = TensorNetwork([tensor1, tensor2])
         end
     end