From 3ead7c300440bc1d79273dcff8e7c950a3b0f77d Mon Sep 17 00:00:00 2001
From: Alessandro Candido <candido.ale@gmail.com>
Date: Fri, 10 Mar 2023 12:56:58 +0100
Subject: [PATCH] Fix tests in the same way of #217

---
 tests/eko/kernels/test_kernels_QEDns.py | 21 ++++++---------------
 tests/eko/test_couplings.py             |  2 +-
 2 files changed, 7 insertions(+), 16 deletions(-)

diff --git a/tests/eko/kernels/test_kernels_QEDns.py b/tests/eko/kernels/test_kernels_QEDns.py
index bb6208363..f683c5a36 100644
--- a/tests/eko/kernels/test_kernels_QEDns.py
+++ b/tests/eko/kernels/test_kernels_QEDns.py
@@ -1,11 +1,13 @@
-import warnings
+from math import nan
 
 import numpy as np
 import pytest
 
 from eko import basis_rotation as br
-from eko import beta
+from eko.couplings import Couplings
+from eko.io.types import CouplingEvolutionMethod, CouplingsRef
 from eko.kernels import non_singlet_qed as ns
+from eko.quantities.heavy_quarks import QuarkMassScheme
 from ekore.anomalous_dimensions.unpolarized import space_like as ad
 
 methods = [
@@ -112,17 +114,6 @@ def test_zero_true_gamma():
                         )
 
 
-from math import nan
-
-from eko.couplings import Couplings, couplings_mod_ev
-from eko.io.types import (
-    CouplingEvolutionMethod,
-    CouplingsRef,
-    EvolutionMethod,
-    MatchingScales,
-    QuarkMassSchemes,
-)
-
 alpharef = (0.118, 0.00781)
 masses = [m**2 for m in (2.0, 4.5, 175.0)]
 muref = 91.0
@@ -154,7 +145,7 @@ def test_ode():
                     order,
                     evmod,
                     masses,
-                    hqm_scheme=QuarkMassSchemes.POLE,
+                    hqm_scheme=QuarkMassScheme.POLE,
                     thresholds_ratios=[1.0, 1.0, 1.0],
                 )
                 a0 = sc.a_s(mu2_0)
@@ -244,7 +235,7 @@ def test_ode_true_gamma():
                     order,
                     evmod,
                     masses,
-                    hqm_scheme=QuarkMassSchemes.POLE,
+                    hqm_scheme=QuarkMassScheme.POLE,
                     thresholds_ratios=[1.0, 1.0, 1.0],
                 )
                 a0 = sc.a_s(mu2_0)
diff --git a/tests/eko/test_couplings.py b/tests/eko/test_couplings.py
index de4538a3d..d441dfc98 100644
--- a/tests/eko/test_couplings.py
+++ b/tests/eko/test_couplings.py
@@ -92,7 +92,7 @@ def test_init(self):
                 evmod,
                 masses,
                 hqm_scheme=QuarkMassScheme.POLE,
-                thresholds_ratios=MatchingScales(c=1.0, b=1.0, t=1.0),
+                thresholds_ratios=MatchingScales([1.0, 1.0, 1.0]),
             )
         with pytest.raises(ValueError):
             coup2 = copy.deepcopy(couplings)