From 3bc4fb771024ab0aaf0f9ead64d394d1de9d5eb0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Niccol=C3=B2=20Laurenti?= Date: Tue, 9 Jan 2024 20:40:40 +0100 Subject: [PATCH] Fix tests --- src/eko/beta.py | 8 ++++---- src/eko/couplings.py | 2 +- tests/eko/test_beta.py | 19 ++++++++++++------- 3 files changed, 17 insertions(+), 12 deletions(-) diff --git a/src/eko/beta.py b/src/eko/beta.py index 6d8b77ec2..d35af42e2 100644 --- a/src/eko/beta.py +++ b/src/eko/beta.py @@ -32,7 +32,7 @@ def beta_qcd_as2(nf): @nb.njit(cache=True) -def beta_qed_aem2(nf, nl=3): +def beta_qed_aem2(nf, nl): r"""Compute the first coefficient of the QED beta function. Implements Eq. (7) of :cite:`Surguladze:1996hx`. @@ -82,7 +82,7 @@ def beta_qcd_as3(nf): @nb.njit(cache=True) -def beta_qed_aem3(nf, nl=3): +def beta_qed_aem3(nf, nl): r"""Compute the second coefficient of the QED beta function. Implements Eq. (7) of :cite:`Surguladze:1996hx`. @@ -244,7 +244,7 @@ def beta_qcd(k, nf): @nb.njit(cache=True) -def beta_qed(k, nf, nl=3): +def beta_qed(k, nf, nl): r"""Compute value of a beta_qed coefficients. Parameters @@ -293,7 +293,7 @@ def b_qcd(k, nf): @nb.njit(cache=True) -def b_qed(k, nf, nl=3): +def b_qed(k, nf, nl): r"""Compute b_qed coefficient. Parameters diff --git a/src/eko/couplings.py b/src/eko/couplings.py index bd6e658db..bdc653832 100644 --- a/src/eko/couplings.py +++ b/src/eko/couplings.py @@ -322,7 +322,7 @@ def couplings_expanded_alphaem_running( ) res_aem += ( -couplings_ref[1] ** 2 - * b_qed((1, 2), nf) + * b_qed((1, 2), nf, nl) * np.log(1 + beta0_qed * couplings_ref[0] * lmu) ) return np.array([res_as, res_aem]) diff --git a/tests/eko/test_beta.py b/tests/eko/test_beta.py index cd64c2b85..8c6be7c5c 100644 --- a/tests/eko/test_beta.py +++ b/tests/eko/test_beta.py @@ -28,7 +28,7 @@ def test_beta_aem2(): """Test first beta function coefficient""" # from hep-ph/9803211 np.testing.assert_approx_equal( - beta.beta_qed_aem2(5), -4.0 / 3 * (3 + 3 * (2 * 4 / 9 + 3 * 1 / 9)) + beta.beta_qed_aem2(5, 3), -4.0 / 3 * (3 + 3 * (2 * 4 / 9 + 3 * 1 / 9)) ) @@ -43,7 +43,7 @@ def test_beta_aem3(): """Test second beta function coefficient""" # from hep-ph/9803211 np.testing.assert_approx_equal( - beta.beta_qed_aem3(5), -4.0 * (3 + 3 * (2 * 16 / 81 + 3 * 1 / 81)) + beta.beta_qed_aem3(5, 3), -4.0 * (3 + 3 * (2 * 16 / 81 + 3 * 1 / 81)) ) @@ -66,21 +66,26 @@ def test_beta_as5(): def test_beta(): """beta-wrapper""" nf = 3 + nl = 3 np.testing.assert_allclose(beta.beta_qcd((2, 0), nf), beta.beta_qcd_as2(nf)) np.testing.assert_allclose(beta.beta_qcd((3, 0), nf), beta.beta_qcd_as3(nf)) np.testing.assert_allclose(beta.beta_qcd((4, 0), nf), beta.beta_qcd_as4(nf)) np.testing.assert_allclose(beta.beta_qcd((5, 0), nf), beta.beta_qcd_as5(nf)) np.testing.assert_allclose(beta.beta_qcd((2, 1), nf), beta.beta_qcd_as2aem1(nf)) - np.testing.assert_allclose(beta.beta_qed((0, 2), nf), beta.beta_qed_aem2(nf)) - np.testing.assert_allclose(beta.beta_qed((0, 3), nf), beta.beta_qed_aem3(nf)) - np.testing.assert_allclose(beta.beta_qed((1, 2), nf), beta.beta_qed_aem2as1(nf)) + np.testing.assert_allclose( + beta.beta_qed((0, 2), nf, nl), beta.beta_qed_aem2(nf, nl) + ) + np.testing.assert_allclose( + beta.beta_qed((0, 3), nf, nl), beta.beta_qed_aem3(nf, nl) + ) + np.testing.assert_allclose(beta.beta_qed((1, 2), nf, nl), beta.beta_qed_aem2as1(nf)) with pytest.raises(ValueError): beta.beta_qcd((6, 0), 3) with pytest.raises(ValueError): - beta.beta_qed((0, 4), 3) + beta.beta_qed((0, 4), 3, 3) def test_b(): """b-wrapper""" np.testing.assert_allclose(beta.b_qcd((2, 0), 3), 1.0) - np.testing.assert_allclose(beta.b_qed((0, 2), 3), 1.0) + np.testing.assert_allclose(beta.b_qed((0, 2), 3, 3), 1.0)