add show method for QobjEvo

Author: ytdHuang

src/QuantumToolbox.jl

@@ -35,6 +35,7 @@ import SciMLBase:
 import StochasticDiffEq: StochasticDiffEqAlgorithm, SRA1
 import SciMLOperators:
     AbstractSciMLOperator,
+    AddedOperator,
     MatrixOperator,
     ScalarOperator,
     IdentityOperator,

src/qobj/quantum_object.jl

@@ -140,7 +140,7 @@ function Base.show(
     return show(io, MIME("text/plain"), op_data)
 end
 
-function Base.show(io::IO, QO::AbstractQuantumObject)
+function Base.show(io::IO, QO::QuantumObject)
     op_data = QO.data
     println(
         io,

src/qobj/quantum_object_evo.jl

@@ -112,6 +112,28 @@ struct QuantumObjectEvolution{
     end
 end
 
+function Base.show(io::IO, QO::QuantumObjectEvolution)
+    op_data = QO.data
+    println(
+        io,
+        "Quantum Object Evo.:   type=",
+        QO.type,
+        "   dims=",
+        QO.dims,
+        "   size=",
+        size(op_data),
+        "   ishermitian=",
+        ishermitian(op_data),
+        "   isconstant=",
+        isconstant(op_data),
+        "   num_elements=",
+        _get_SciMLOperator_num_elements(op_data),
+    )
+    return show(io, MIME("text/plain"), op_data)
+end
+_get_SciMLOperator_num_elements(A::AddedOperator) = length(A.ops)
+_get_SciMLOperator_num_elements(A::AbstractSciMLOperator) = 1
+
 function QuantumObjectEvolution(data::AbstractSciMLOperator, type::QuantumObjectType, dims::Integer)
     return QuantumObjectEvolution(data, type, SVector{1,Int}(dims))
 end

test/core-test/quantum_objects_evo.jl

@@ -93,27 +93,33 @@
         @test issymmetric(Z) == true
     end
 
-    # TODO: Implement a new show method for QuantumObjectEvolution
-    # @testset "REPL show" begin
-    #     N = 10
-    #     a = QobjEvo(destroy(N))
-
-    #     opstring = sprint((t, s) -> show(t, "text/plain", s), a)
-    #     datastring = sprint((t, s) -> show(t, "text/plain", s), a.data)
-    #     a_dims = a.dims
-    #     a_size = size(a)
-    #     a_isherm = isherm(a)
-    #     @test opstring ==
-    #           "Quantum Object:   type=Operator   dims=$a_dims   size=$a_size   ishermitian=$a_isherm\n$datastring"
-
-    #     a = spre(a)
-    #     opstring = sprint((t, s) -> show(t, "text/plain", s), a)
-    #     datastring = sprint((t, s) -> show(t, "text/plain", s), a.data)
-    #     a_dims = a.dims
-    #     a_size = size(a)
-    #     a_isherm = isherm(a)
-    #     @test opstring == "Quantum Object:   type=SuperOperator   dims=$a_dims   size=$a_size\n$datastring"
-    # end
+    @testset "REPL show" begin
+        N = 10
+        a = destroy(N)
+        coef(p, t) = exp(-1im * t)
+        H = QobjEvo((a' * a, (a, coef)))
+
+        opstring = sprint((t, s) -> show(t, "text/plain", s), H)
+        datastring = sprint((t, s) -> show(t, "text/plain", s), H.data)
+        H_dims = H.dims
+        H_size = size(H)
+        H_isherm = isherm(H)
+        H_isconst = isconstant(H)
+        H_elements = 2
+        @test opstring ==
+              "Quantum Object Evo.:   type=Operator   dims=$H_dims   size=$H_size   ishermitian=$H_isherm   isconstant=$H_isconst   num_elements=$$H_elements\n$datastring"
+
+        L = QobjEvo(spre(a))
+        opstring = sprint((t, s) -> show(t, "text/plain", s), L)
+        datastring = sprint((t, s) -> show(t, "text/plain", s), L.data)
+        L_dims = L.dims
+        L_size = size(L)
+        L_isherm = isherm(L)
+        L_isconst = isconstant(L)
+        H_elements = 1
+        @test opstring ==
+              "Quantum Object Evo.:   type=SuperOperator   dims=$L_dims   size=$L_size   ishermitian=$L_isherm   isconstant=$L_isconst   num_elements=$$L_elements\n$datastring"
+    end
 
     @testset "Type Inference (QuantumObject)" begin
         for T in [ComplexF32, ComplexF64]