From 7a6c1cc2c42bd63a42ad2e4bae4d362cd4daafb7 Mon Sep 17 00:00:00 2001
From: Felix Hekhorn <felix.hekhorn@uni-tuebingen.de>
Date: Fri, 19 Jul 2024 15:15:12 +0300
Subject: [PATCH] Introduce EvoMethods

---
 src/eko/evolution_operator/__init__.py |  6 +++++
 src/eko/kernels/__init__.py            | 32 ++++++++++++++++++++++++++
 src/eko/scale_variations/__init__.py   | 13 ++++++-----
 tests/eko/kernels/test_init.py         | 21 +++++++++++++++++
 4 files changed, 66 insertions(+), 6 deletions(-)
 create mode 100644 tests/eko/kernels/test_init.py

diff --git a/src/eko/evolution_operator/__init__.py b/src/eko/evolution_operator/__init__.py
index 16aa4bb5b..8cd5829bd 100644
--- a/src/eko/evolution_operator/__init__.py
+++ b/src/eko/evolution_operator/__init__.py
@@ -20,6 +20,7 @@
 from .. import basis_rotation as br
 from .. import interpolation, mellin
 from .. import scale_variations as sv
+from ..kernels import ev_method
 from ..kernels import non_singlet as ns
 from ..kernels import non_singlet_qed as qed_ns
 from ..kernels import singlet as s
@@ -780,6 +781,11 @@ def labels(self):
                 labels.extend(br.singlet_unified_labels)
         return labels
 
+    @property
+    def ev_method(self):
+        """Return the evolution method."""
+        return ev_method(self.config["method"])
+
     def quad_ker(self, label, logx, areas):
         """Return partially initialized integrand function.
 
diff --git a/src/eko/kernels/__init__.py b/src/eko/kernels/__init__.py
index 7640727b5..a10998b23 100644
--- a/src/eko/kernels/__init__.py
+++ b/src/eko/kernels/__init__.py
@@ -1 +1,33 @@
 """The solutions to the |DGLAP| equations."""
+
+import enum
+
+
+class EvoMethods(enum.IntEnum):
+    """Enumerate evolution methods."""
+
+    ITERATE_EXACT = enum.auto()
+    ITERATE_EXPANDED = enum.auto()
+    PERTURBATIVE_EXACT = enum.auto()
+    PERTURBATIVE_EXPANDED = enum.auto()
+    TRUNCATED = enum.auto()
+    ORDERED_TRUNCATED = enum.auto()
+    DECOMPOSE_EXACT = enum.auto()
+    DECOMPOSE_EXPANDED = enum.auto()
+
+
+def ev_method(s: EvoMethods) -> EvoMethods:
+    """Return the evolution methods.
+
+    Parameters
+    ----------
+    s :
+        string representation
+
+    Returns
+    -------
+    i :
+        int representation
+
+    """
+    return EvoMethods[s.value.upper().replace("-", "_")]
diff --git a/src/eko/scale_variations/__init__.py b/src/eko/scale_variations/__init__.py
index 9c43f1f9c..21d56a503 100644
--- a/src/eko/scale_variations/__init__.py
+++ b/src/eko/scale_variations/__init__.py
@@ -6,29 +6,30 @@
 
 import enum
 
+from ..io.types import ScaleVariationsMethod
 from . import expanded, exponentiated
 
 
 class Modes(enum.IntEnum):
-    """Enumerate scale Variation modes."""
+    """Enumerate scale variation modes."""
 
     unvaried = enum.auto()
     exponentiated = enum.auto()
     expanded = enum.auto()
 
 
-def sv_mode(s):
+def sv_mode(s: ScaleVariationsMethod) -> Modes:
     """Return the scale variation mode.
 
     Parameters
     ----------
-    s : str
+    s :
         string representation
 
     Returns
     -------
-    enum.IntEnum
-        enum representation
+    i :
+        int representation
 
     """
     if s is not None:
@@ -40,6 +41,6 @@ class ScaleVariationModeMixin:
     """Mixin to cast scale variation mode."""
 
     @property
-    def sv_mode(self):
+    def sv_mode(self) -> Modes:
         """Return the scale variation mode."""
         return sv_mode(self.config["ModSV"])
diff --git a/tests/eko/kernels/test_init.py b/tests/eko/kernels/test_init.py
new file mode 100644
index 000000000..3717de275
--- /dev/null
+++ b/tests/eko/kernels/test_init.py
@@ -0,0 +1,21 @@
+from eko.io.types import EvolutionMethod
+from eko.kernels import EvoMethods, ev_method
+
+
+def test_ev_method():
+    methods = {
+        "iterate-expanded": EvoMethods.ITERATE_EXPANDED,
+        "decompose-expanded": EvoMethods.DECOMPOSE_EXPANDED,
+        "perturbative-expanded": EvoMethods.PERTURBATIVE_EXPANDED,
+        "truncated": EvoMethods.TRUNCATED,
+        "ordered-truncated": EvoMethods.ORDERED_TRUNCATED,
+        "iterate-exact": EvoMethods.ITERATE_EXACT,
+        "decompose-exact": EvoMethods.DECOMPOSE_EXACT,
+        "perturbative-exact": EvoMethods.PERTURBATIVE_EXACT,
+    }
+    assert len(methods.keys()) == len(EvolutionMethod)
+    assert len(methods.keys()) == len(EvoMethods)
+    for s, i in methods.items():
+        j = ev_method(EvolutionMethod(s))
+        assert j == i
+        assert isinstance(j, int)