slightly modyfing base

src/base.jl

@@ -146,6 +146,17 @@ function MPS(O::MPO)
     ψ
 end  
 
+function Base.randn(::Type{MPS{T}}, D::Int, rank::Union{Vector, NTuple}) where {T}
+    L = length(rank)
+    ψ = MPS(T, L)
+    ψ[1] = randn(T, 1, rank[1], D)
+    for i ∈ 2:(L-1)
+        ψ[i] = randn(T, D, rank[i], D)
+    end
+    ψ[end] = randn(T, D, rank[end], 1)
+    ψ
+end
+
 function Base.randn(::Type{MPS{T}}, L::Int, D::Int, d::Int) where {T}
     ψ = MPS(T, L)
     ψ[1] = randn(T, 1, d, D)

test/base.jl

@@ -23,6 +23,27 @@ T = ComplexF64
     @test ϕ ≈ ψ
 
     show(ψ)
+
+    dims = (3, 2, 5, 4) 
+    @info "Veryfing ψ of arbitrary rank" dims
+
+    ψ = randn(MPS{T}, D, dims)
+    @test verify_bonds(ψ) == nothing
+
+    @test ψ == ψ
+    @test ψ ≈ ψ
+
+    @test length(ψ) == length(dims)
+    @test size(ψ) == (length(dims), )
+    @test eltype(ψ) == ComplexF64
+    @test rank(ψ) == dims
+    @test bond_dimension(ψ) ≈ D
+
+    ϕ = copy(ψ) 
+    @test ϕ == ψ
+    @test ϕ ≈ ψ
+
+    show(ψ)
 end
 
 @testset "Random MPO" begin

test/runtests.jl

@@ -22,11 +22,11 @@ if CUDA.functional() && CUDA.has_cutensor() && false
 end
 
 push!(my_tests,
-#    "base.jl",
+    "base.jl",
 #    "contractions.jl",
 #    "compressions.jl",
 #    "ising.jl",
-    "search.jl"
+#    "search.jl"
 )
 
 for my_test in my_tests

test/search.jl

@@ -76,7 +76,7 @@ end
         β = get_prop(ig, :β)
         rank = get_prop(ig, :rank)
 
-        @info "Generating Gibbs state - |ρ>" d L rank β ϵ
+        @info "Generating Gibbs state - |ρ>" L rank β ϵ
 
         ψ = ones(rank...)