fix braille and add small test

src/MPSKit.jl

@@ -120,6 +120,7 @@ include("operators/projection.jl")
 include("operators/timedependence.jl")
 include("operators/multipliedoperator.jl")
 include("operators/lazysum.jl")
+include("operators/utility.jl") # extra functions that need all of the above structs to be defined and fit nowhere else
 
 include("transfermatrix/transfermatrix.jl")
 include("transfermatrix/transfer.jl")

src/operators/abstractmpo.jl

@@ -124,38 +124,6 @@ function Base.show(io::IOContext, mpo::AbstractMPO)
     return nothing
 end
 
-function braille(H::SparseMPO)
-    isfinite = (H isa FiniteMPO) || (H isa FiniteMPOHamiltonian)
-    dash = "🭻"
-    stride = 2 #amount of dashes between braille
-    L = length(H)
-
-    brailles = Vector{Vector{String}}(undef, L)
-    buffer = IOBuffer()
-    for (i, W) in enumerate(H)
-        BlockTensorKit.show_braille(buffer, W)
-        brailles[i] = split(String(take!(buffer)))
-    end
-
-    maxheight = maximum(length.(brailles))
-
-    for i in 1:maxheight
-        line = ""
-        line *= ((i == 1 && !isfinite) ? ("... " * dash) : " ")
-        line *= (i > 1 && !isfinite) ? "    " : ""
-        for (j, braille) in enumerate(brailles)
-            line *= (checkbounds(Bool, braille, i) ? braille[i] :
-                     repeat(" ", length(braille[1])))
-            if j < L
-                line *= repeat(((i == 1) ? dash : " "), stride)
-            end
-        end
-        line *= ((i == 1 && !isfinite) ? (dash * " ...") : " ")
-        println(line)
-    end
-    return nothing
-end
-
 # Linear Algebra
 # --------------
 Base.:+(mpo::AbstractMPO) = scale(mpo, One())

src/operators/utility.jl

@@ -0,0 +1,33 @@
+function braille(io::IO, H::Union{SparseMPO,MPOHamiltonian})
+    isfinite = (H isa FiniteMPO) || (H isa FiniteMPOHamiltonian)
+    dash = "🭻"
+    stride = 2 #amount of dashes between braille
+    L = length(H)
+
+    brailles = Vector{Vector{String}}(undef, L)
+    buffer = IOBuffer()
+    for (i, W) in enumerate(H)
+        BlockTensorKit.show_braille(buffer, W)
+        brailles[i] = split(String(take!(buffer)))
+    end
+
+    maxheight = maximum(length.(brailles))
+
+    for i in 1:maxheight
+        line = ""
+        line *= ((i == 1 && !isfinite) ? ("... " * dash) : " ")
+        line *= (i > 1 && !isfinite) ? "    " : ""
+        for (j, braille) in enumerate(brailles)
+            line *= (checkbounds(Bool, braille, i) ? braille[i] :
+                     repeat(" ", length(braille[1])))
+            if j < L
+                line *= repeat(((i == 1) ? dash : " "), stride)
+            end
+        end
+        line *= ((i == 1 && !isfinite) ? (dash * " ...") : " ")
+        println(io, line)
+    end
+    return nothing
+end
+
+braille(H::Union{SparseMPO,MPOHamiltonian}) = braille(stdout, H)

test/other.jl

@@ -63,4 +63,34 @@ end
     end
 end
 
+@testset "braille" begin
+    # Infinite Hamiltonians and MPOs
+    # -------------------------------
+    H = transverse_field_ising()
+    buffer = IOBuffer()
+    braille(buffer, H)
+    output = String(take!(buffer))
+    check = "... 🭻⎡⠉⢀⎤🭻 ...\n     ⎣⠀⢀⎦ \n"
+    @test output == check
+
+    O = make_time_mpo(H, 1.0, TaylorCluster(3, false, false))
+    braille(buffer, O)
+    output = String(take!(buffer))
+    check = "... 🭻⎡⡏⠉⠒⠔⎤🭻 ...\n     ⎣⡇⠀⠀⡂⎦ \n"
+    @test output == check
+
+    # Finite Hamiltonians and MPOs
+    # ----------------------------
+    H = open_boundary_conditions(H, 4)
+    braille(buffer, H)
+    output = String(take!(buffer))
+    check = " ⎡⠉⠀⎤🭻🭻⎡⠉⢀⎤🭻🭻⎡⠉⢀⎤🭻🭻⎡⡀⠀⎤ \n ⎣⠀⠀⎦  ⎣⠀⢀⎦  ⎣⠀⢀⎦  ⎣⡀⠀⎦ \n"
+    @test output == check
+
+    O = make_time_mpo(H, 1.0, TaylorCluster(3, false, false))
+    braille(buffer, O)
+    output = String(take!(buffer))
+    check = " ⎡⠉⠉⎤🭻🭻⎡⠍⠉⠤⠠⎤🭻🭻⎡⡏⠉⠒⠔⎤🭻🭻⎡⡇⠀⎤ \n ⎣⠀⠀⎦  ⎣⡂⠀⠀⠂⎦  ⎣⡇⠀⠀⡂⎦  ⎣⡇⠀⎦ \n"
+    @test output == check
+end
 end