Fix tests in the same way of #217

NNPDF · Mar 31, 2023 · 47badf2 · 47badf2
1 parent e424970
commit 47badf2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 16 deletions.
diff --git 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
@@ -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)