From 73b1250385990a44707ef790af282e45dbf72bdb Mon Sep 17 00:00:00 2001
From: Huai-Ming Yu <hmyuus@gmail.com>
Date: Thu, 23 Jan 2025 23:00:13 +0800
Subject: [PATCH 1/4] add tests for predefined states

---
 test/{test_parameterization.jl => test_scheme.jl} | 11 +++++++++++
 1 file changed, 11 insertions(+)
 rename test/{test_parameterization.jl => test_scheme.jl} (88%)

diff --git a/test/test_parameterization.jl b/test/test_scheme.jl
similarity index 88%
rename from test/test_parameterization.jl
rename to test/test_scheme.jl
index e4ff8b1..eaf2d73 100644
--- a/test/test_parameterization.jl
+++ b/test/test_scheme.jl
@@ -80,11 +80,22 @@ function test_kraus()
     evolve(GeneralScheme(; probe=rho0, param=channel))
     evolve(GeneralScheme(; probe=psi, param=channel))
 end
+
+function test_state()
+    @test PlusState() == [1.0, 1.0]/sqrt(2)
+    @test MinusState() == [1.0, -1.0]/sqrt(2)
+    @test BellState() == BellState(1) == [1.0, 0.0, 0.0, 1.0]/sqrt(2)
+    @test BellState(2) == [1.0, 0.0, 0.0, -1.0]/sqrt(2)
+    @test BellState(3) == [0.0, 1.0, 1.0, 0.0]/sqrt(2)
+    @test BellState(4) == [0.0, 1.0, -1.0, 0.0]/sqrt(2)
+end  # function test_state
+
 function test_parameterization()
     test_lindblad(dyn_method=:Ode)
     test_lindblad(dyn_method=:Expm)
     test_lindblad_pure()
     test_kraus()
+    test_state()
 end  # function test_parameterization
 
 test_parameterization()
\ No newline at end of file

From 85a76639b9f6999f1645d2a874853ce9a2b66634 Mon Sep 17 00:00:00 2001
From: Huai-Ming Yu <hmyuus@gmail.com>
Date: Thu, 23 Jan 2025 23:01:53 +0800
Subject: [PATCH 2/4] remove unused apis

---
 lib/QuanEstimationBase/src/Common/Common.jl | 12 ------------
 1 file changed, 12 deletions(-)

diff --git a/lib/QuanEstimationBase/src/Common/Common.jl b/lib/QuanEstimationBase/src/Common/Common.jl
index f25174b..65d0012 100644
--- a/lib/QuanEstimationBase/src/Common/Common.jl
+++ b/lib/QuanEstimationBase/src/Common/Common.jl
@@ -21,22 +21,10 @@ function repeat_copy(scheme, N)
     [deepcopy(scheme) for _ = 1:N]
 end
 
-function filterZeros!(x::Matrix{T}) where {T<:Complex}
-    x[abs.(x).<eps()] .= zero(T)
-    x
-end
-function filterZeros!(x)
-    filterZeros!.(x)
-end
-
 function filterZeros(x::AbstractVecOrMat{T}) where {T<:Number}
     [x + 1 ≈ 1 ? zero(T) : x for x in x]
 end
 
-function t2Num(t0, dt, t)
-    Int(round((t - t0) / dt)) + 1
-end
-
 function basis(dim, si, ::T)::Array{T} where {T<:Complex}
     result = zeros(T, dim)
     result[si] = 1.0

From 3ab58e8962d42bc6cd4f4071338d5a9041d7b975 Mon Sep 17 00:00:00 2001
From: Huai-Ming Yu <hmyuus@gmail.com>
Date: Thu, 23 Jan 2025 23:03:08 +0800
Subject: [PATCH 3/4] fix runtest

---
 test/runtests.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/test/runtests.jl b/test/runtests.jl
index 9812a67..560eabc 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -40,8 +40,8 @@ end
     end
 end
 
-@testset "Parameterization" begin
-    include("test_parameterization.jl")
+@testset "Scheme" begin
+    include("test_scheme.jl")
 end
 
 @testset "Resource" begin

From 9a933bbfaaaff682f7ce7f432f7cdb3d5aa1f06a Mon Sep 17 00:00:00 2001
From: Huai-Ming Yu <hmyuus@gmail.com>
Date: Thu, 23 Jan 2025 23:04:14 +0800
Subject: [PATCH 4/4] fix runtest

---
 test/test_scheme.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/test/test_scheme.jl b/test/test_scheme.jl
index eaf2d73..696132e 100644
--- a/test/test_scheme.jl
+++ b/test/test_scheme.jl
@@ -95,7 +95,7 @@ function test_parameterization()
     test_lindblad(dyn_method=:Expm)
     test_lindblad_pure()
     test_kraus()
-    test_state()
 end  # function test_parameterization
 
-test_parameterization()
\ No newline at end of file
+test_parameterization()
+test_state()
\ No newline at end of file