provide non regression test

Author: kif

src/pyFAI/test/test_bug_regression.py

@@ -36,7 +36,7 @@
 __contact__ = "Jerome.Kieffer@esrf.fr"
 __license__ = "MIT"
 __copyright__ = "2015-2025 European Synchrotron Radiation Facility, Grenoble, France"
-__date__ = "20/11/2025"
+__date__ = "17/12/2025"
 
 import sys
 import os
@@ -45,15 +45,16 @@
 import subprocess
 import copy
 import logging
+from math import pi
 from .utilstest import UtilsTest
 from ..utils import mathutil
 import fabio
 from .. import load
 from ..integrator.azimuthal import AzimuthalIntegrator
 from .. import detectors
 from .. import units
-from math import pi
 from ..opencl import ocl
+from ..ext import splitPixel
 logger = logging.getLogger(__name__)
 
 try:
@@ -691,6 +692,42 @@ def test_bug_2697(self):
         res = numpy.isnan(ai.integrate1d(fimg.data, 300, method=("no","csr", "cython"), error_model="azimuth").sum_variance).sum()
         self.assertEqual(res, 0, "No NaNs found in Cython")
 
+    def test_bug_2736(self):
+        """Checks the pixel splitting with boundaries, see tutorial entitled PixelSplitting"""
+        ai = load({"dist":1,
+                 "poni1":2.2e-3,
+                 "poni2":2.8e-3,
+                 "rot3":-0.3,
+                 "detector":"Detector",
+                 "detector_config":{"pixel1":1e-3,
+                                    "pixel2":1e-3,
+                                    "max_shape":(5,5)}
+                })
+        ai.setChiDiscAtPi()
+        pos = numpy.array(ai.array_from_unit(typ="corner", unit="r_mm", scale=True), dtype="float64")
+        self.assertEqual((pos[..., 1]< -pi).sum(), 0, "No pixels below pi azim angle")
+        self.assertEqual((pos[..., 1]> +pi).sum(), 0, "No pixels above pi azim angle")
+        for i0 in range(pos.shape[0]):
+            for i1 in range(pos.shape[1]):
+                p = pos[i0, i1]
+                splitPixel.recenter(p, chiDiscAtPi=True)
+        self.assertEqual((pos[..., 1]< -pi).sum(), 5, "5 pixels below pi azim angle")
+        self.assertEqual((pos[..., 1]> +pi).sum(), 1, "1 pixels above pi azim angle")
+
+        # Change origin to zero
+        ai.setChiDiscAtZero()
+        pos = numpy.array(ai.array_from_unit(typ="corner", unit="r_mm", scale=True), dtype="float64")
+        self.assertEqual((pos[..., 1]< 0).sum(), 0, "No pixels below 0 azim angle")
+        self.assertEqual((pos[..., 1]> 2*pi).sum(), 0, "No pixels above 2pi azim angle")
+        for i0 in range(pos.shape[0]):
+            for i1 in range(pos.shape[1]):
+                p = pos[i0, i1]
+                splitPixel.recenter(p, chiDiscAtPi=False)
+        self.assertEqual((pos[..., 1]< 0).sum(), 1, "1 pixels below pi azim angle")
+        self.assertEqual((pos[..., 1]> 2*pi).sum(), 3, "3 pixels above pi azim angle")
+
+
+
 class TestBug1703(unittest.TestCase):
     """
     Check the normalization affect properly the propagated errors/intensity