bug and test fixes

Author: jack-dunham

src/boundaries/corner/tensormacros.jl

@@ -59,7 +59,7 @@ end
 
 function _projectedge!(
     t_dst::T, t_src::T, ma::MA, mb::MB, u::UV, v::UV
-) where {T<:AbsTen{2,2},MA<:AbsTen{1,4},MB<:AbsTen{2,4},UV<:AbsTen{3,1}}
+) where {T<:AbsTen{2,2},MA<:AbsTen{1,4},MB<:AbsTen{4,1},UV<:AbsTen{3,1}}
     @tensoropt (
         k => 2,
         D1 => D,

src/networks.jl

@@ -58,7 +58,7 @@ Base.firstindex(ct::CompositeTensor) = 1
 Base.lastindex(ct::CompositeTensor) = length(ct)
 
 function Base.iterate(ct::CompositeTensor, state=1)
-    state > length(ct) && nothing
+    state > length(ct) && return nothing
     return getindex(ct, state), state + 1
 end
 

test/networks.jl

@@ -8,12 +8,10 @@
 
     t = TensorMap(rand, p * p', e * s * w' * n')
 
-    @test virtualspace(t) == domain(t)
+    @test swapaxes(t) == permute(t, ((1, 2), (4, 3, 6, 5)))
+    @test invertaxes(t) == permute(t, ((1, 2), (5, 6, 3, 4)))
 
-    @test swapaxes(t) == permute(t, (4, 3, 6, 5))
-    @test invertaxes(t) == permutedom(t, (5, 6, 3, 4))
-
-    ct2 = CompositeTensor(t, t')
+    ct = CompositeTensor(t, t')
 
     @test ct[1] == ct[2]'
     @test @constinferred(first(ct)) === parent(@constinferred(last(ct)))
@@ -23,6 +21,6 @@
 
     @test virtualspace(ct, 1) == e * e'
     @test virtualspace(ct, 2) == s * s'
-    @test virtualspace(ct, 3) == w' * n
+    @test virtualspace(ct, 3) == w' * w
     @test virtualspace(ct, 4) == n' * n
 end