Minor changes

qutip · Apr 8, 2024 · bedaf06 · bedaf06
1 parent ea98ac6
commit bedaf06
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/test/jet.jl b/test/jet.jl
@@ -1,5 +1,3 @@
 @testset "Code quality (JET.jl)" begin
-    if VERSION >= v"1.9"
-        JET.test_package(QuantumToolbox; target_defined_modules=true, ignore_missing_comparison=true)
-    end
+    JET.test_package(QuantumToolbox; target_defined_modules=true, ignore_missing_comparison=true)
 end
diff --git a/test/time_evolution_and_partial_trace.jl b/test/time_evolution_and_partial_trace.jl
@@ -12,6 +12,8 @@
     e_ops = [a_d * a]
     sol = sesolve(H, psi0, t_l, e_ops=e_ops, alg=Vern7(), progress_bar=false)
     @test sum(abs.(sol.expect[1, :] .- sin.(η * t_l) .^ 2)) / length(t_l) < 0.1
+    @test ptrace(sol.states[end], 1) ≈ ptrace(ket2dm(sol.states[end]), 1)
+    @test ptrace(sol.states[end]', 1) ≈ ptrace(sol.states[end], 1)
 
     a = destroy(N)
     a_d = a'