Add ==, isapprox methods to Quantum

bsc-quantic · Apr 5, 2024 · c4bb805 · c4bb805
1 parent a7fe63b
commit c4bb805
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/Quantum.jl b/src/Quantum.jl
@@ -83,6 +83,9 @@ Tenet.TensorNetwork(q::Quantum) = q.tn
 
 Base.copy(q::Quantum) = Quantum(copy(TensorNetwork(q)), copy(q.sites))
 
+Base.:(==)(a::Quantum, b::Quantum) = a.tn == b.tn && a.sites == b.sites
+Base.isapprox(a::Quantum, b::Quantum; kwargs...) = isapprox(a.tn, b.tn; kwargs...) && a.sites == b.sites
+
 function Base.adjoint(qtn::Quantum)
     sites = Iterators.map(qtn.sites) do (site, index)
         site' => index