test operators

tjjarvinen · Dec 3, 2024 · c86babe · c86babe
1 parent 182ff86
commit c86babe
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 2 deletions.
diff --git a/src/submodules/QuantumSystem/operators.jl b/src/submodules/QuantumSystem/operators.jl
@@ -3,8 +3,8 @@ abstract type GridFixedOperator{N} <: AbstractOperator{N} end
 abstract type AbstractCompositeOperator{N} <: AbstractOperator{N} end
 abstract type AbstractHamiltonOperator <: AbstractOperator{1} end
 
-HelmholtzKernel.get_elementgrid(::AbstractOperator) = missing
-HelmholtzKernel.get_elementgrid(ao::GridFixedOperator) = ao.elementgrid
+get_elementgrid(::AbstractOperator) = missing
+get_elementgrid(ao::GridFixedOperator) = ao.elementgrid
 Base.size(::AbstractOperator) = missing
 Base.size(ao::GridFixedOperator) = size(get_elementgrid(ao))
 Base.length(::AbstractOperator{N}) where N = N

diff --git a/test/test-quantumsystem.jl b/test/test-quantumsystem.jl
@@ -33,4 +33,35 @@ using Unitful
 
     normalize!(psi)
     @test braket(psi, psi) ≈ 1.0
+end
+
+
+@testset "Operators" begin
+    ev = ElementVector(0.0u"pm", 1.5u"pm", 3.0u"pm")
+    ego = ElementGridVectorLobatto(ev, 24)
+    ega = ElementGridArray(ego, ego, ego)
+
+    r = position_operator(ega)
+    p = momentum_operator(ega)
+
+    psi = particle_in_box(ega, 1, 2, 3)
+
+    psi1 = r[1] * psi
+    psi2 = p[1] * psi
+
+    @test braket(psi, r[1], psi) ≈ 1.5u"pm"
+    @test braket(psi, r[2], psi) ≈ 1.5u"pm"
+    @test braket(psi, r[3], psi) ≈ 1.5u"pm"
+
+    r2 = dot(r, r)
+
+    @test dimension(r2) == dimension(r)^2
+
+    @test braket(psi, -r2, psi) < 0.0u"pm^2"
+
+    lo = LaplaceOperator()
+    @test braket(psi, lo, psi) < 0.0u"pm^-2"
+
+    grad = gradient_operator(ega)
+    @test length( braket(psi, grad, psi) ) == 3
 end