Check that the convolutions are the same in test_{fk,grid}_rotations

Author: Radonirinaunimi

pineappl_py/tests/test_fk_table.py

@@ -117,6 +117,7 @@ def test_fktable(
 
     def test_fktable_rotations(
         self,
+        pdf,
         download_objects,
         tmp_path,
         fkname: str = "FKTABLE_CMSTTBARTOT8TEV-TOPDIFF8TEVTOT.pineappl.lz4",
@@ -137,7 +138,19 @@ def test_fktable_rotations(
         # check that the FK table can be loaded properly
         path = f"{tmp_path}/rotated_fktable.pineappl.lz4"
         fk.write_lz4(path)
-        _ = FkTable.read(path)
+        fk_mod = FkTable.read(path)
+
+        # check that the convolutions are the same
+        np.testing.assert_allclose(
+            fk.convolve(
+                pdg_convs=fk.convolutions,
+                xfxs=[pdf.unpolarized_pdf, pdf.unpolarized_pdf],
+            ),
+            fk_mod.convolve(
+                pdg_convs=fk_mod.convolutions,
+                xfxs=[pdf.unpolarized_pdf, pdf.unpolarized_pdf],
+            ),
+        )
 
     def test_unpolarized_convolution(
         self,

pineappl_py/tests/test_grid.py

@@ -427,7 +427,21 @@ def test_grid_rotations(
         # check that the FK table can be loaded properly
         path = f"{tmp_path}/grid_merged_factors.pineappl.lz4"
         g.write_lz4(path)
-        _ = Grid.read(path)
+        g_mod = Grid.read(path)
+
+        # check that the convolutions are the same
+        np.testing.assert_allclose(
+            g.convolve(
+                pdg_convs=g.convolutions,
+                xfxs=[pdf.polarized_pdf],
+                alphas=pdf.alphasQ,
+            ),
+            g_mod.convolve(
+                pdg_convs=g_mod.convolutions,
+                xfxs=[pdf.polarized_pdf],
+                alphas=pdf.alphasQ,
+            ),
+        )
 
     def test_unpolarized_convolution(
         self,