format files

Author: ytdHuang

src/qobj/dimensions.jl

@@ -98,7 +98,7 @@ function _get_dims_string(dimensions::GeneralDimensions)
 end
 _get_dims_string(::Nothing) = "nothing" # for EigsolveResult.dimensions = nothing
 
-Base.:(==)(dim1::Dimensions, dim2::Dimensions) = (dim1.to == dim2.to)
+Base.:(==)(dim1::Dimensions, dim2::Dimensions) = dim1.to == dim2.to
 Base.:(==)(dim1::GeneralDimensions, dim2::GeneralDimensions) = (dim1.to == dim2.to) && (dim1.from == dim2.from)
 Base.:(==)(dim1::Dimensions, dim2::GeneralDimensions) = false
 Base.:(==)(dim1::GeneralDimensions, dim2::Dimensions) = false

src/qobj/energy_restricted.jl

@@ -118,19 +118,15 @@ end
 enr_identity(s_enr::EnrSpace) = QuantumObject(Diagonal(ones(ComplexF64, s_enr.size)), Operator(), Dimensions(s_enr))
 
 function enr_fock(
-    dims::Union{AbstractVector{T},NTuple{N,T}}, 
+    dims::Union{AbstractVector{T},NTuple{N,T}},
     excitations::Int,
     state::AbstractVector{T};
     sparse::Union{Bool,Val} = Val(false),
 ) where {T<:Integer,N}
     s_enr = EnrSpace(dims, excitations)
     return enr_fock(s_enr, state, sparse = sparse)
 end
-function enr_fock(
-    s_enr::EnrSpace,
-    state::AbstractVector{T};
-    sparse::Union{Bool,Val} = Val(false),
-) where {T<:Integer}
+function enr_fock(s_enr::EnrSpace, state::AbstractVector{T}; sparse::Union{Bool,Val} = Val(false)) where {T<:Integer}
     if getVal(sparse)
         array = sparsevec([s_enr.state2idx[[state...]]], [1.0 + 0im], s_enr.size)
     else
@@ -141,10 +137,7 @@ function enr_fock(
     return QuantumObject(array, Ket(), s_enr)
 end
 
-function enr_destroy(
-    dims::Union{AbstractVector{T},NTuple{N,T}}, 
-    excitations::Int
-) where {T<:Integer,N}
+function enr_destroy(dims::Union{AbstractVector{T},NTuple{N,T}}, excitations::Int) where {T<:Integer,N}
     s_enr = EnrSpace(dims, excitations)
     return enr_destroy(s_enr)
 end
@@ -171,7 +164,11 @@ function enr_destroy(s_enr::EnrSpace{N}) where {N}
     return a_ops
 end
 
-function enr_thermal_dm(dims::Union{AbstractVector{T1},NTuple{N,T1}}, excitations::Int, n::Union{T2,AbstractVector{T2}}) where {T1<:Integer,T2<:Real,N}
+function enr_thermal_dm(
+    dims::Union{AbstractVector{T1},NTuple{N,T1}},
+    excitations::Int,
+    n::Union{T2,AbstractVector{T2}},
+) where {T1<:Integer,T2<:Real,N}
     s_enr = EnrSpace(dims, excitations)
     return enr_thermal_dm(s_enr, n)
 end

src/qobj/quantum_object_base.jl

@@ -239,7 +239,8 @@ get_dimensions_from(
 ) where {ObjType<:Union{SuperOperator,OperatorBra,OperatorKet}} = A.dimensions.to
 
 # this creates a list of Space(1), it is used to generate `from` for Ket, and `to` for Bra
-space_one_list(dimensions::NTuple{N,AbstractSpace}) where {N} = ntuple(i -> Space(1), Val(sum(_get_dims_length, dimensions)))
+space_one_list(dimensions::NTuple{N,AbstractSpace}) where {N} =
+    ntuple(i -> Space(1), Val(sum(_get_dims_length, dimensions)))
 _get_dims_length(::Space) = 1
 _get_dims_length(::EnrSpace{N}) where {N} = N
 

test/core-test/enr_state_operator.jl

@@ -8,7 +8,7 @@
         I_s = qeye(D1) ⊗ qeye(D2)
         size_s = prod(dims_s)
         space_s = (Space(D1), Space(D2))
-        
+
         # EnrSpace
         dims_enr = (2, 2, 3)
         excitations = 3
@@ -34,7 +34,8 @@
         for b in basis_list
             ρ_enr_compound += tensor(b', I_enr) * ρ_tot * tensor(b, I_enr)
         end
-        new_dims2 = GeneralDimensions((space_s..., Space(1), Space(1), Space(1)), (space_s..., Space(1), Space(1), Space(1)))
+        new_dims2 =
+            GeneralDimensions((space_s..., Space(1), Space(1), Space(1)), (space_s..., Space(1), Space(1), Space(1)))
         ρ_s_compound = Qobj(zeros(ComplexF64, size_s, size_s), dims = new_dims2)
         basis_list = [enr_fock(space_enr, space_enr.idx2state[idx]) for idx in 1:space_enr.size]
         for b in basis_list
@@ -72,7 +73,7 @@
         c_ops_enr = (√γ * a_enr,)
         sol_enr = mesolve(H_enr, ψ0_enr, tlist, c_ops_enr; e_ops = [sz_enr], progress_bar = Val(false))
         ρ_ss_enr = steadystate(H_enr, c_ops_enr)
-        
+
         # check mesolve result
         @test all(isapprox.(sol_JC.expect, sol_enr.expect, atol = 1e-4))