feat: Overload exponentiation operator for BioMarkovChain

BioJulia · Jul 29, 2024 · 09f0371 · 09f0371
1 parent 533e53d
commit 09f0371
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/extended.jl b/src/extended.jl
@@ -60,4 +60,10 @@ end
 
 @inline Base.length(bmc::BioMarkovChain) = length(bmc.inits)
 @inline Base.size(bmc::BioMarkovChain) = size(bmc.tpm)
-@inline Base.eltype(bmc::BioMarkovChain) = bmc.alphabet
+@inline Base.eltype(bmc::BioMarkovChain) = bmc.alphabet
+
+## Overload operators
+
+# Base.:(==)(a::BioMarkovChain, b::BioMarkovChain) = a.tpm == b.tpm && a.inits == b.inits
+Base.:(^)(a::BioMarkovChain, n::Int) = BioMarkovChain(a.alphabet, a.tpm^n, a.inits, n)
+