add tests

Author: Fe-r-oz

src/steadystate.jl

@@ -417,7 +417,7 @@ function _steadystate_fourier(
 
     v0 = _dense_similar(L_0_mat, n_fourier * N)
     fill!(v0, 0)
-    @allowscalar v0[n_max * N + 1] = weight
+    @allowscalar v0[n_max*N+1] = weight
 
     if !isnothing(solver.Pl)
         kwargs = merge((; kwargs...), (Pl = solver.Pl(M),))

test/ext-test/gpu/cuda_ext.jl

@@ -154,7 +154,6 @@ end
     @test ρ_ss_cpu.data ≈ Array(ρ_ss_gpu_csr.data) atol = 1e-8 * length(ρ_ss_cpu)
 end
 
-
 @testset "CUDA steadystate_fourier" begin
     N = 2
     ωd = 1.0
@@ -163,16 +162,19 @@ end
     Hp = cu(sigmax())
     Hm = cu(sigmax())
     c_ops = [cu(sqrt(0.1) * sigmam())]
-    ρ_list1 = steadystate_fourier(H0, Hp, Hm, ωd, c_ops;
-                                  solver = SteadyStateLinearSolver(),
-                                  n_max = n_max)
-    ρ0 = steadystate_fourier(H0, Hp, Hm, ωd, c_ops;
-                             solver = SSFloquetEffectiveLiouvillian(SteadyStateDirectSolver()),
-                             n_max = n_max)
+    ρ_list1 = steadystate_fourier(H0, Hp, Hm, ωd, c_ops; solver = SteadyStateLinearSolver(), n_max = n_max)
+    ρ0 = steadystate_fourier(
+        H0,
+        Hp,
+        Hm,
+        ωd,
+        c_ops;
+        solver = SSFloquetEffectiveLiouvillian(SteadyStateDirectSolver()),
+        n_max = n_max,
+    )
     @test isapprox(ρ0, ρ_list1[1]; atol = 1e-6)
 end
 
-
 @testset "CUDA ptrace" begin
     g = fock(2, 1)
     e = fock(2, 0)