rust: Simplify test_gamma_singlet_qed

NNPDF · Feb 3, 2025 · bf5cb42 · bf5cb42
1 parent 79d5437
commit bf5cb42
Showing 1 changed file with 29 additions and 93 deletions.
diff --git a/crates/ekore/src/anomalous_dimensions/unpolarized/spacelike.rs b/crates/ekore/src/anomalous_dimensions/unpolarized/spacelike.rs
@@ -310,100 +310,36 @@ mod tests {
         const N: Complex<f64> = cmplx!(2., 0.);
         let mut c = Cache::new(N);
 
-        assert_approx_eq_cmplx!(
-            f64,
-            gamma_singlet_qed(3, 2, &mut c, NF)[3][0][0][0],
-            cmplx!(3.857918949669738, 0.),
-            epsilon = 1e-5
-        );
-        assert_approx_eq_cmplx!(
-            f64,
-            gamma_singlet_qed(3, 2, &mut c, NF)[3][0][0][1],
-            cmplx!(0., 0.),
-            epsilon = 1e-5
-        );
-        assert_approx_eq_cmplx!(
-            f64,
-            gamma_singlet_qed(3, 2, &mut c, NF)[3][0][0][2],
-            cmplx!(-290.42193908689745, 0.),
-            epsilon = 1e-5
-        );
-        assert_approx_eq_cmplx!(
-            f64,
-            gamma_singlet_qed(3, 2, &mut c, NF)[3][0][0][3],
-            cmplx!(0., 0.),
-            epsilon = 1e-5
-        );
-        assert_approx_eq_cmplx!(
-            f64,
-            gamma_singlet_qed(3, 2, &mut c, NF)[3][0][1][0],
-            cmplx!(0., 0.),
-            epsilon = 1e-5
-        );
-        assert_approx_eq_cmplx!(
-            f64,
-            gamma_singlet_qed(3, 2, &mut c, NF)[3][0][1][1],
-            cmplx!(0., 0.),
-            epsilon = 1e-5
-        );
-        assert_approx_eq_cmplx!(
-            f64,
-            gamma_singlet_qed(3, 2, &mut c, NF)[3][0][1][2],
-            cmplx!(0., 0.),
-            epsilon = 1e-5
-        );
-        assert_approx_eq_cmplx!(
-            f64,
-            gamma_singlet_qed(3, 2, &mut c, NF)[3][0][1][3],
-            cmplx!(0., 0.),
-            epsilon = 1e-5
-        );
-        assert_approx_eq_cmplx!(
-            f64,
-            gamma_singlet_qed(3, 2, &mut c, NF)[3][0][2][0],
-            cmplx!(-3.859554320251334, 0.),
-            epsilon = 1e-5
-        );
-        assert_approx_eq_cmplx!(
-            f64,
-            gamma_singlet_qed(3, 2, &mut c, NF)[3][0][2][1],
-            cmplx!(0., 0.),
-            epsilon = 1e-5
-        );
-        assert_approx_eq_cmplx!(
-            f64,
-            gamma_singlet_qed(3, 2, &mut c, NF)[3][0][2][2],
-            cmplx!(290.4252052962147, 0.),
-            epsilon = 1e-5
-        );
-        assert_approx_eq_cmplx!(
-            f64,
-            gamma_singlet_qed(3, 2, &mut c, NF)[3][0][2][3],
-            cmplx!(0., 0.),
-            epsilon = 1e-5
-        );
-        assert_approx_eq_cmplx!(
-            f64,
-            gamma_singlet_qed(3, 2, &mut c, NF)[3][0][3][0],
-            cmplx!(0., 0.),
-            epsilon = 1e-5
-        );
-        assert_approx_eq_cmplx!(
-            f64,
-            gamma_singlet_qed(3, 2, &mut c, NF)[3][0][3][1],
-            cmplx!(0., 0.),
-            epsilon = 1e-5
-        );
-        assert_approx_eq_cmplx!(
-            f64,
-            gamma_singlet_qed(3, 2, &mut c, NF)[3][0][3][2],
-            cmplx!(0., 0.),
-            epsilon = 1e-5
-        );
-        assert_approx_eq_cmplx!(
+        assert_approx_eq_cmplx_2d!(
             f64,
-            gamma_singlet_qed(3, 2, &mut c, NF)[3][0][3][3],
-            cmplx!(448.0752570151872, 0.),
+            gamma_singlet_qed(3, 2, &mut c, NF)[3][0],
+            [
+                [
+                    cmplx!(3.857918949669738, 0.),
+                    cmplx!(0., 0.),
+                    cmplx!(-290.42193908689745, 0.),
+                    cmplx!(0., 0.)
+                ],
+                [
+                    cmplx!(0., 0.),
+                    cmplx!(0., 0.),
+                    cmplx!(0., 0.),
+                    cmplx!(0., 0.)
+                ],
+                [
+                    cmplx!(-3.859554320251334, 0.),
+                    cmplx!(0., 0.),
+                    cmplx!(290.4252052962147, 0.),
+                    cmplx!(0., 0.)
+                ],
+                [
+                    cmplx!(0., 0.),
+                    cmplx!(0., 0.),
+                    cmplx!(0., 0.),
+                    cmplx!(448.0752570151872, 0.)
+                ]
+            ],
+            4,
             epsilon = 1e-5
         );
     }