move some import of CairoMakie.jl to @setup block

Author: ytdHuang

docs/src/users_guide/steadystate.md

@@ -104,7 +104,7 @@ exp_mc = real(sol_mc.expect[1, :])
 sol_me = mesolve(H, ψ0, tlist, c_ops, e_ops=e_ops, progress_bar=false)
 exp_me = real(sol_me.expect[1, :])
 
-# plot the results
+# plot by CairoMakie.jl
 fig = Figure(size = (500, 350))
 ax = Axis(fig[1, 1], 
     title = L"Decay of Fock state $|10\rangle$ in a thermal environment with $\langle n\rangle=2$",

docs/src/users_guide/time_evolution/mesolve.md

@@ -2,6 +2,9 @@
 
 ```@setup mesolve
 using QuantumToolbox
+
+using CairoMakie
+CairoMakie.enable_only_mime!(MIME"image/svg+xml"())
 ```
 
 ## [Von Neumann equation](@id doc-TE:Von-Neumann-equation)
@@ -123,13 +126,11 @@ sol = mesolve(H, ψ0, tlist, c_ops, e_ops = [sigmaz(), sigmay()])
 We can therefore plot the expectation values:
 
 ```@example mesolve
-using CairoMakie
-CairoMakie.enable_only_mime!(MIME"image/svg+xml"())
-
 times = sol.times
 expt_z = real(sol.expect[1,:])
 expt_y = real(sol.expect[2,:])
 
+# plot by CairoMakie.jl
 fig = Figure(size = (500, 350))
 ax = Axis(fig[1, 1], xlabel = "Time", ylabel = "Expectation values")
 lines!(ax, times, expt_z, label = L"\langle\hat{\sigma}_z\rangle", linestyle = :solid)
@@ -166,7 +167,7 @@ e_ops = [a' * a]
 sol = mesolve(H, ψ0, tlist, c_ops, e_ops=e_ops)
 Num = real(sol.expect[1, :])
 
-# plot the results
+# plot by CairoMakie.jl
 fig = Figure(size = (500, 350))
 ax = Axis(fig[1, 1], 
     title = L"Decay of Fock state $|10\rangle$ in a thermal environment with $\langle n\rangle=2$",
@@ -213,6 +214,7 @@ times = sol.times
 N_atom   = real(sol.expect[1,:])
 N_cavity = real(sol.expect[2,:])
 
+# plot by CairoMakie.jl
 fig = Figure(size = (500, 350))
 ax = Axis(fig[1, 1], xlabel = "Time", ylabel = "Expectation values")
 lines!(ax, times, N_atom, label = "atom excitation probability", linestyle = :solid)

docs/src/users_guide/time_evolution/sesolve.md

@@ -23,6 +23,9 @@ The Schrödinger equation, which governs the time-evolution of closed quantum sy
 
 ```@setup sesolve
 using QuantumToolbox
+
+using CairoMakie
+CairoMakie.enable_only_mime!(MIME"image/svg+xml"())
 ```
 
 For example, the time evolution of a quantum spin-``\frac{1}{2}`` system (initialized in spin-``\uparrow``) with tunneling rate ``0.1`` is calculated, and the expectation values of the Pauli-Z operator ``\hat{\sigma}_z`` is also evaluated, with the following code
@@ -68,12 +71,10 @@ print(size(expt))
 We can therefore plot the expectation values:
 
 ```@example sesolve
-using CairoMakie
-CairoMakie.enable_only_mime!(MIME"image/svg+xml"())
-
 expt_z = real(expt[1,:])
 expt_y = real(expt[2,:])
 
+# plot by CairoMakie.jl
 fig = Figure(size = (500, 350))
 ax = Axis(fig[1, 1], xlabel = "Time", ylabel = "Expectation values")
 lines!(ax, times, expt_z, label = L"\langle\hat{\sigma}_z\rangle", linestyle = :solid)

docs/src/users_guide/time_evolution/solution.md

@@ -2,6 +2,9 @@
 
 ```@setup TE-solution
 using QuantumToolbox
+
+using CairoMakie
+CairoMakie.enable_only_mime!(MIME"image/svg+xml"())
 ```
 
 ## [Solution](@id doc-TE:Solution)
@@ -61,9 +64,7 @@ nothing # hide
 we can plot the resulting expectation values:
 
 ```@example TE-solution
-using CairoMakie
-CairoMakie.enable_only_mime!(MIME"image/svg+xml"())
-
+# plot by CairoMakie.jl
 fig = Figure(size = (500, 350))
 ax = Axis(fig[1, 1], xlabel = L"t")
 lines!(ax, times, expt1, label = L"\langle 0 | \rho(t) | 0 \rangle")