Add tests

Author: albertomercurio

src/qobj/eigsolve.jl

@@ -129,7 +129,7 @@ function LinearAlgebra.mul!(y::AbstractVector, A::EigsolveInverseMap, x::Abstrac
     return copyto!(y, solve!(A.linsolve).u)
 end
 
-function _update_schur_eigs!(Hₘ, Uₘ, Uₘᵥ, f, k, m, β, sorted_vals, sortby, rev)
+function _update_schur_eigs!(Hₘ, Uₘ, Uₘᵥ, f, k, β, sorted_vals, sortby, rev)
     copyto!(Uₘ, Hₘ)
     F = schur!(Uₘ)
 
@@ -230,7 +230,7 @@ function _eigsolve(
 
     M = typeof(cache0)
 
-    _update_schur_eigs!(Hₘ, Uₘ, Uₘᵥ, f, k, m, β, sorted_vals, sortby, rev)
+    _update_schur_eigs!(Hₘ, Uₘ, Uₘᵥ, f, k, β, sorted_vals, sortby, rev)
 
     numops = m
     iter = 0
@@ -256,7 +256,7 @@ function _eigsolve(
 
         # println( A * Vₘ ≈ Vₘ * M(Hₘ) + qₘ * M(transpose(βeₘ)) )     # SHOULD BE TRUE
 
-        _update_schur_eigs!(Hₘ, Uₘ, Uₘᵥ, f, k, m, β, sorted_vals, sortby, rev)
+        _update_schur_eigs!(Hₘ, Uₘ, Uₘᵥ, f, k, β, sorted_vals, sortby, rev)
 
         numops += m - k - 1
         iter += 1

test/ext-test/cpu/arbitrary_precision/Project.toml

@@ -0,0 +1,3 @@
+[deps]
+GenericSchur = "c145ed77-6b09-5dd9-b285-bf645a82121e"
+QuantumToolbox = "6c2fb7c5-b903-41d2-bc5e-5a7c320b9fab"

test/ext-test/cpu/arbitrary_precision/arbitrary_precision.jl

@@ -0,0 +1,57 @@
+@testset "Arbitrary Precision" verbose=true begin
+    N = 20
+    Δ = 1.0
+    U = 0.1
+    F = 0.1
+    κ = 0.1
+
+    a = destroy(N)
+    ψ0 = fock(N, 1)
+    H = Δ * a' * a + U / 2 * a' * a' * a * a + F * (a + a')
+
+    c_ops = [sqrt(κ) * a]
+
+    tlist = range(0, 40, 100)
+
+    a_big = SparseMatrixCSC{Complex{BigFloat}}(a)
+    ψ0_big = Vector{Complex{BigFloat}}(ψ0)
+    H_big = Δ * a_big' * a_big + U / 2 * a_big' * a_big' * a_big * a_big + F * (a_big + a_big')
+
+    c_ops_big = [sqrt(BigFloat(κ)) * a_big]
+
+    @testset "Arbitrary Precision (sesolve)" begin
+        sol = sesolve(H, ψ0, tlist; progress_bar = Val(false))
+        sol_big = sesolve(H_big, ψ0_big, tlist; progress_bar = Val(false))
+
+        # Test all fidelities are close to 1
+        @test all((x) -> isapprox(fidelity(x[1], x[2]), 1; atol = 1e-7), zip(sol.states, sol_big.states))
+
+        @inferred sesolve(H_big, ψ0_big, tlist; progress_bar = Val(false))
+    end
+
+    @testset "Arbitrary Precision (mesolve)" begin
+        sol = mesolve(H, ψ0, tlist, c_ops; progress_bar = Val(false))
+        sol_big = mesolve(H_big, ψ0_big, tlist, c_ops_big; progress_bar = Val(false))
+
+        # Test all fidelities are close to 1
+        @test all((x) -> isapprox(hilbert_dist(x[1], x[2]), 0; atol = 1e-10), zip(sol.states, sol_big.states))
+
+        @inferred mesolve(H_big, ψ0_big, tlist, c_ops_big; progress_bar = Val(false))
+    end
+
+    @testset "Arbitrary Precision (eigenvalues)" begin
+        L = liouvillian(H, c_ops)
+        L_big = liouvillian(H_big, c_ops_big)
+
+        vals, vecs = eigenstates(L; sparse = true, eigvals = 6, krylovdim = 20)
+        vals_big, vecs_big = eigenstates(L_big; sparse = true, eigvals = 6, krylovdim = 20)
+
+        # Align eigenvalues
+        idxs = [findmin(abs.(vals_big .- val))[2] for val in vals]
+
+        @test vals ≈ vals_big[idxs]
+        @test all(zip(vecs, vecs_big[idxs])) do (v, v_big)
+            return isapprox(abs(dot(v.data, v_big.data)), 1; atol = 1e-7)
+        end
+    end
+end

test/runtests.jl

@@ -2,7 +2,8 @@ using Test
 using TestItemRunner
 using Pkg
 
-const GROUP_LIST = String["All", "Core", "Code-Quality", "AutoDiff_Ext", "Makie_Ext", "CUDA_Ext"]
+const GROUP_LIST =
+    String["All", "Core", "Code-Quality", "AutoDiff_Ext", "Makie_Ext", "CUDA_Ext", "Arbitrary_Precision_Ext"]
 
 const GROUP = get(ENV, "GROUP", "All")
 (GROUP in GROUP_LIST) || throw(ArgumentError("Unknown GROUP = $GROUP\nThe allowed groups are: $GROUP_LIST\n"))
@@ -81,3 +82,17 @@ if (GROUP == "CUDA_Ext")
 
     include(joinpath(testdir, "ext-test", "gpu", "cuda_ext.jl"))
 end
+
+if (GROUP == "Arbitrary_Precision_Ext")
+    Pkg.activate("ext-test/cpu/arbitrary_precision")
+    Pkg.develop(PackageSpec(path = dirname(@__DIR__)))
+    Pkg.update()
+
+    using QuantumToolbox
+    using SparseArrays
+    using GenericSchur
+
+    QuantumToolbox.about()
+
+    include(joinpath(testdir, "ext-test", "cpu", "arbitrary_precision", "arbitrary_precision.jl"))
+end