add tests to check tstops

Author: ytdHuang

test/core-test/time_evolution.jl

@@ -84,6 +84,7 @@ end
     ##
 
     @test prob.prob.f.f isa ScaledOperator
+    @test !haskey(prob.prob.kwargs, :tstops) # tstops should not exist for time-independent cases
     @test sum(abs.(sol.expect[1, :] .- amp_rabi .* sin.(Ω_rabi * tlist) .^ 2)) / length(tlist) < 0.1
     @test length(sol.times) == length(tlist)
     @test length(sol.times_states) == 1
@@ -223,6 +224,7 @@ end
     sol_me5 = mesolve(H, ψ0, tlist, progress_bar = Val(false))
 
     @test TESetup.prob_me.prob.f.f isa MatrixOperator
+    @test !haskey(TESetup.prob_me.prob.kwargs, :tstops) # tstops should not exist for time-independent cases
     @test isket(sol_me5.states[1])
     @test length(sol_me.times) == length(tlist)
     @test length(sol_me.times_states) == 1
@@ -435,6 +437,7 @@ end
     expect_mc_states_mean2 = expect.(Ref(e_ops[1]), average_states(sol_mc_states2))
 
     @test prob_mc.prob.f.f isa ScaledOperator
+    @test !haskey(prob_mc.prob.kwargs, :tstops) # tstops should not exist for time-independent cases
     @test sum(abs, sol_mc.expect .- sol_me.expect) / length(tlist) < 0.1
     @test sum(abs, sol_mc2.expect .- sol_me.expect) / length(tlist) < 0.1
     @test sum(abs, average_expect(sol_mc3) .- sol_me.expect) / length(tlist) < 0.1
@@ -1206,11 +1209,11 @@ end
 
 @testitem "Example: Qubit driven by two sequential cosine pulses" begin
     # settings of pulses
-    T = 10 # duration of each pulse
     A1 = rand() # Rabi amplitude of pulse 1
     A2 = rand() # Rabi amplitude of pulse 2
     t1 = 0 # starting time of pulse 1
     t2 = 100 # starting time of pulse 2
+    T = 10 # duration of each pulse
     pulse1(p, t) = ((t1 <= t) && (t <= t1 + T)) ? (0.5 * p.A1 * π / T) * (1 - cos(2π * (t - t1) / T)) : 0
     pulse2(p, t) = ((t2 <= t) && (t <= t2 + T)) ? (0.5 * p.A2 * π / T) * (1 - cos(2π * (t - t2) / T)) : 0
 
@@ -1226,8 +1229,10 @@ end
     tlist = range(0.0, stop = 140.0, length = 10000) # don't change tlist, to check the second pulse contribute correctly under 'tstops=tlist' setting
     e_ops = [sigmax(), sigmaz()]
     params = (A1 = A1, A2 = A2)
-    sol_se = sesolve(H, ψ0, tlist; e_ops = e_ops, params = params, progress_bar = Val(false))
-    sol_me = mesolve(H, ρ0, tlist; e_ops = e_ops, params = params, progress_bar = Val(false))
+    prob_se = sesolveProblem(H, ψ0, tlist; e_ops = e_ops, params = params, progress_bar = Val(false))
+    prob_me = mesolveProblem(H, ρ0, tlist; e_ops = e_ops, params = params, progress_bar = Val(false))
+    sol_se = sesolve(prob_se)
+    sol_me = mesolve(prob_me)
 
     # analytic solution
     function θ(t, t_start, A)
@@ -1244,6 +1249,8 @@ end
     X_analytic = sin.(θlist)
     Z_analytic = cos.(θlist)
 
+    @test prob_se.prob.kwargs[:tstops] == tlist # tstops should be equal to tlist for time-dependent cases
+    @test prob_me.prob.kwargs[:tstops] == tlist # tstops should be equal to tlist for time-dependent cases
     @test all(isapprox.(X_analytic, sol_se.expect[1, :]; atol = 1e-6))
     @test all(isapprox.(X_analytic, sol_me.expect[1, :]; atol = 1e-6))
     @test all(isapprox.(Z_analytic, sol_se.expect[2, :]; atol = 1e-6))