throw errors for non-supporting EnrSpace functions

Author: ytdHuang

src/negativity.jl

@@ -71,18 +71,19 @@ Return the partial transpose of a density matrix ``\rho``, where `mask` is an ar
 - `ρ_pt::QuantumObject`: The density matrix with the selected subsystems transposed.
 """
 function partial_transpose(ρ::QuantumObject{Operator}, mask::Vector{Bool})
-    if length(mask) != length(ρ.dimensions)
-        throw(ArgumentError("The length of \`mask\` should be equal to the length of \`ρ.dims\`."))
-    end
-    return _partial_transpose(ρ, mask)
-end
+    any(s -> s isa EnrSpace, QO.dimensions.to) && throw(ArgumentError("partial_transpose does not support EnrSpace"))
+    
+    (length(mask) != length(ρ.dimensions)) && throw(ArgumentError("The length of \`mask\` should be equal to the length of \`ρ.dims\`."))
 
-# for dense matrices
-function _partial_transpose(ρ::QuantumObject{Operator}, mask::Vector{Bool})
     isa(ρ.dimensions, GeneralDimensions) &&
         (get_dimensions_to(ρ) != get_dimensions_from(ρ)) &&
         throw(ArgumentError("Invalid partial transpose for dims = $(_get_dims_string(ρ.dimensions))"))
 
+    return _partial_transpose(ρ, mask)
+end
+
+# for dense matrices
+function _partial_transpose(ρ::QuantumObject{Operator}, mask::Vector{Bool})
     mask2 = [1 + Int(i) for i in mask]
     # mask2 has elements with values equal to 1 or 2
     #   1 - the subsystem don't need to be transposed
@@ -107,10 +108,6 @@ function _partial_transpose(
     ρ::QuantumObject{Operator,DimsType,<:AbstractSparseArray},
     mask::Vector{Bool},
 ) where {DimsType<:AbstractDimensions}
-    isa(ρ.dimensions, GeneralDimensions) &&
-        (get_dimensions_to(ρ) != get_dimensions_from(ρ)) &&
-        throw(ArgumentError("Invalid partial transpose for dims = $(_get_dims_string(ρ.dimensions))"))
-
     M, N = size(ρ)
     dimsTuple = Tuple(dimensions_to_dims(get_dimensions_to(ρ)))
     colptr = ρ.data.colptr

src/qobj/arithmetic_and_attributes.jl

@@ -505,6 +505,8 @@ Quantum Object:   type=Operator()   dims=[2]   size=(2, 2)   ishermitian=true
 ```
 """
 function ptrace(QO::QuantumObject{Ket}, sel::Union{AbstractVector{Int},Tuple})
+    any(s -> s isa EnrSpace, QO.dimensions.to) && throw(ArgumentError("ptrace does not support EnrSpace"))
+
     _non_static_array_warning("sel", sel)
 
     if length(sel) == 0 # return full trace for empty sel
@@ -527,6 +529,8 @@ end
 ptrace(QO::QuantumObject{Bra}, sel::Union{AbstractVector{Int},Tuple}) = ptrace(QO', sel)
 
 function ptrace(QO::QuantumObject{Operator}, sel::Union{AbstractVector{Int},Tuple})
+    any(s -> s isa EnrSpace, QO.dimensions.to) && throw(ArgumentError("ptrace does not support EnrSpace"))
+
     # TODO: support for special cases when some of the subsystems have same `to` and `from` space
     isa(QO.dimensions, GeneralDimensions) &&
         (get_dimensions_to(QO) != get_dimensions_from(QO)) &&
@@ -714,6 +718,8 @@ function SparseArrays.permute(
     A::QuantumObject{ObjType},
     order::Union{AbstractVector{Int},Tuple},
 ) where {ObjType<:Union{Ket,Bra,Operator}}
+    any(s -> s isa EnrSpace, A.dimensions.to) && throw(ArgumentError("permute does not support EnrSpace"))
+
     (length(order) != length(A.dimensions)) &&
         throw(ArgumentError("The order list must have the same length as the number of subsystems (A.dims)"))