fix tests

Author: ytdHuang

src/qobj/dimensions.jl

@@ -12,7 +12,8 @@ function Dimensions(dims::Union{AbstractVector{T},NTuple{N,T}}) where {T<:Intege
 
     return Dimensions{L}(SVector{L,AbstractSpace}(Space.(dims)))
 end
-Dimensions(dims::Int) = Dimensions(SVector{1,Int}(dims))
+Dimensions(dims::Int) = Dimensions(Space(dims))
+Dimensions(dims::DimType) where {DimType<:Union{Field,Space}} = Dimensions(SVector{1,AbstractSpace}(dims))
 Dimensions(dims::Any) = throw(
     ArgumentError(
         "The argument dims must be a Tuple or a StaticVector of non-zero length and contain only positive integers.",

src/time_evolution/mcsolve.jl

@@ -299,7 +299,7 @@ function mcsolveEnsembleProblem(
     ensemble_prob = TimeEvolutionProblem(
         EnsembleProblem(prob_mc.prob, prob_func = _prob_func, output_func = _output_func[1], safetycopy = false),
         prob_mc.times,
-        prob_mc.dims,
+        prob_mc._dims,
         (progr = _output_func[2], channel = _output_func[3]),
     )
 
@@ -472,7 +472,7 @@ function mcsolve(
 )
     sol = _mcsolve_solve_ens(ens_prob_mc, alg, ensemble_method, ntraj)
 
-    dims = ens_prob_mc.dims
+    dims = ens_prob_mc._dims
     _sol_1 = sol[:, 1]
     _expvals_sol_1 = _mcsolve_get_expvals(_sol_1)
 

src/time_evolution/mesolve.jl

@@ -179,7 +179,7 @@ end
 function mesolve(prob::TimeEvolutionProblem, alg::OrdinaryDiffEqAlgorithm = Tsit5())
     sol = solve(prob.prob, alg)
 
-    ρt = map(ϕ -> QuantumObject(vec2mat(ϕ), type = Operator, dims = prob.dims), sol.u)
+    ρt = map(ϕ -> QuantumObject(vec2mat(ϕ), type = Operator, dims = prob._dims), sol.u)
 
     return TimeEvolutionSol(
         prob.times,

src/time_evolution/sesolve.jl

@@ -152,7 +152,7 @@ end
 function sesolve(prob::TimeEvolutionProblem, alg::OrdinaryDiffEqAlgorithm = Tsit5())
     sol = solve(prob.prob, alg)
 
-    ψt = map(ϕ -> QuantumObject(ϕ, type = Ket, dims = prob.dims), sol.u)
+    ψt = map(ϕ -> QuantumObject(ϕ, type = Ket, dims = prob._dims), sol.u)
 
     return TimeEvolutionSol(
         prob.times,