Add unit tests for exporting pixel cuts from different workspaces types

Author: MohamedAlmaki

docs/source/release/v6.13.0/Workbench/SliceViewer/Bugfixes/36084.rst

@@ -0,0 +1 @@
+- Sliceviwer no longer raises an error when exporting a pixel cut from a MDHistoWorkspace

qt/python/mantidqt/mantidqt/widgets/sliceviewer/test/test_sliceviewer_model.py

@@ -677,6 +677,176 @@ def assert_call_as_expected(mock_ws, transpose, export_type, is_spectra, is_ragg
             mock_ws.name.return_value = "mock_ws"
             assert_call_as_expected(mock_ws, transpose=False, export_type=export_type, is_spectra=is_spectra, is_ragged=is_ragged)
 
+    @patch("mantidqt.widgets.sliceviewer.models.model.IntegrateMDHistoWorkspace")
+    @patch("mantidqt.widgets.sliceviewer.models.model.TransposeMD")
+    def test_export_pixel_cut_to_workspace_mdhisto(self, mock_transpose, mock_intMD):
+        slicepoint = [None, None, 0.5]
+        bin_params = [100, 100, 0.1]
+        dimension_indices = [0, 1, None]
+        xpos, ypos = 0.0, 2.0
+
+        dim0_delta = self.ws_MD_3D.getDimension(0).getBinWidth()
+        dim1_delta = self.ws_MD_3D.getDimension(1).getBinWidth()
+
+        xmin, xmax = xpos - 0.5 * dim0_delta, xpos + 0.5 * dim0_delta
+        ymin, ymax = ypos - 0.5 * dim1_delta, ypos + 0.5 * dim1_delta
+        zmin, zmax = 0.45, 0.55
+
+        def assert_call_as_expected(transpose, cut_type):
+            model = SliceViewerModel(self.ws_MD_3D)
+
+            help_msg = model.export_pixel_cut_to_workspace(slicepoint, bin_params, (xpos, ypos), transpose, dimension_indices, cut_type)
+
+            if not transpose:
+                if cut_type == "x":
+                    # X-cut: integrate along X, keep Y dimension
+                    xcut_params = dict(
+                        InputWorkspace=self.ws_MD_3D,
+                        P1Bin=[xmin, 0, xmax],
+                        P2Bin=[ymin, ymax],
+                        P3Bin=[zmin, zmax],
+                        OutputWorkspace="ws_MD_3D_cut_x",
+                    )
+                    mock_intMD.assert_called_once_with(**xcut_params)
+                    self.assertEqual("Cut along X created: ws_MD_3D_cut_x", help_msg)
+                elif cut_type == "y":
+                    # Y-cut: integrate along Y, keep X dimension
+                    ycut_params = dict(
+                        InputWorkspace=self.ws_MD_3D,
+                        P1Bin=[xmin, xmax],
+                        P2Bin=[ymin, 0, ymax],
+                        P3Bin=[zmin, zmax],
+                        OutputWorkspace="ws_MD_3D_cut_y",
+                    )
+                    mock_intMD.assert_called_once_with(**ycut_params)
+                    self.assertEqual("Cut along Y created: ws_MD_3D_cut_y", help_msg)
+            else:
+                # In transposed mode, dimensions are swapped
+                if cut_type == "x":
+                    # X-cut: integrate along Y, keep X dimension (transposed)
+                    xcut_params = dict(
+                        InputWorkspace=self.ws_MD_3D,
+                        P1Bin=[ymin, ymax],
+                        P2Bin=[xmin, 0, xmax],
+                        P3Bin=[zmin, zmax],
+                        OutputWorkspace="ws_MD_3D_cut_x",
+                    )
+                    mock_intMD.assert_called_once_with(**xcut_params)
+                    self.assertEqual("Cut along X created: ws_MD_3D_cut_x", help_msg)
+                elif cut_type == "y":
+                    # Y-cut: integrate along X, keep Y dimension (transposed)
+                    ycut_params = dict(
+                        InputWorkspace=self.ws_MD_3D,
+                        P1Bin=[ymin, 0, ymax],
+                        P2Bin=[xmin, xmax],
+                        P3Bin=[zmin, zmax],
+                        OutputWorkspace="ws_MD_3D_cut_y",
+                    )
+                    mock_intMD.assert_called_once_with(**ycut_params)
+                    self.assertEqual("Cut along Y created: ws_MD_3D_cut_y", help_msg)
+
+            mock_intMD.reset_mock()
+
+        for cut_type in ("x", "y"):
+            assert_call_as_expected(transpose=False, cut_type=cut_type)
+            assert_call_as_expected(transpose=True, cut_type=cut_type)
+
+    @patch("mantidqt.widgets.sliceviewer.models.model.BinMD")
+    @patch("mantidqt.widgets.sliceviewer.models.model.TransposeMD")
+    def test_export_pixel_cut_to_workspace_mdevent(self, mock_transpose, mock_binmd):
+        slicepoint = [None, None, 0.5]
+        bin_params = [100, 100, 0.1]
+        dimension_indices = [0, 1, None]
+        xpos, ypos = 0.0, 2.0
+
+        dim0_delta = self.ws_MDE_3D.getDimension(0).getBinWidth()
+        dim1_delta = self.ws_MDE_3D.getDimension(1).getBinWidth()
+
+        xmin, xmax = xpos - 0.5 * dim0_delta, xpos + 0.5 * dim0_delta
+        ymin, ymax = ypos - 0.5 * dim1_delta, ypos + 0.5 * dim1_delta
+        zmin, zmax = 0.45, 0.55
+
+        def assert_call_as_expected(transpose, cut_type):
+            model = SliceViewerModel(self.ws_MDE_3D)
+
+            help_msg = model.export_pixel_cut_to_workspace_mdevent(
+                slicepoint, bin_params, (xpos, ypos), transpose, dimension_indices, cut_type
+            )
+
+            common_params = dict(
+                InputWorkspace=self.ws_MDE_3D,
+                AxisAligned=False,
+                BasisVector0="h,rlu,1.0,0.0,0.0",
+                BasisVector1="k,rlu,0.0,1.0,0.0",
+                BasisVector2="l,rlu,0.0,0.0,1.0",
+            )
+
+            if not transpose:
+                extents = [xmin, xmax, ymin, ymax, zmin, zmax]
+
+                if cut_type == "x":
+                    expected_bins = [100, 1, 1]
+                    expected_ws = "ws_MDE_3D_cut_x"
+                    expected_msg = "Cut along X created: ws_MDE_3D_cut_x"
+                else:
+                    expected_bins = [1, 100, 1]
+                    expected_ws = "ws_MDE_3D_cut_y"
+                    expected_msg = "Cut along Y created: ws_MDE_3D_cut_y"
+            else:
+                extents = [ymin, ymax, xmin, xmax, zmin, zmax]
+
+                if cut_type == "x":
+                    expected_bins = [1, 100, 1]
+                    expected_ws = "ws_MDE_3D_cut_x"
+                    expected_msg = "Cut along X created: ws_MDE_3D_cut_x"
+                else:
+                    expected_bins = [100, 1, 1]
+                    expected_ws = "ws_MDE_3D_cut_y"
+                    expected_msg = "Cut along Y created: ws_MDE_3D_cut_y"
+
+            mock_binmd.assert_called_once_with(
+                **common_params, OutputExtents=extents, OutputBins=expected_bins, OutputWorkspace=expected_ws
+            )
+
+            self.assertEqual(expected_msg, help_msg)
+            mock_binmd.reset_mock()
+
+        for cut_type in ("x", "y"):
+            assert_call_as_expected(transpose=False, cut_type=cut_type)
+            assert_call_as_expected(transpose=True, cut_type=cut_type)
+
+    @patch("mantidqt.widgets.sliceviewer.models.roi.extract_roi_matrix")
+    def test_export_pixel_cut_to_workspace_matrix(self, mock_extract_roi):
+        slicepoint = [None, None]
+        bin_params = [100, 100]
+        dimension_indices = [0, 1]
+        xpos, ypos = 1.5, 3.0
+
+        def assert_call_as_expected(transpose, cut_type):
+            model = SliceViewerModel(self.ws2d_histo)
+            mock_extract_roi.reset_mock()
+
+            help_msg = model.export_pixel_cut_to_workspace(slicepoint, bin_params, (xpos, ypos), transpose, dimension_indices, cut_type)
+
+            if not transpose:
+                if cut_type == "x":
+                    mock_extract_roi.assert_called_once_with(self.ws2d_histo, None, None, ypos, ypos, False, "ws2d_histo_cut_x")
+                    self.assertEqual("Cut along X created: ws2d_histo_cut_x", help_msg)
+                elif cut_type == "y":
+                    mock_extract_roi.assert_called_once_with(self.ws2d_histo, xpos, xpos, None, None, True, "ws2d_histo_cut_y")
+                    self.assertEqual("Cut along Y created: ws2d_histo_cut_y", help_msg)
+            else:
+                if cut_type == "x":
+                    mock_extract_roi.assert_called_once_with(self.ws2d_histo, ypos, ypos, None, None, True, "ws2d_histo_cut_y")
+                    self.assertEqual("Cut along X created: ws2d_histo_cut_y", help_msg)
+                elif cut_type == "y":
+                    mock_extract_roi.assert_called_once_with(self.ws2d_histo, None, None, xpos, xpos, False, "ws2d_histo_cut_x")
+                    self.assertEqual("Cut along Y created: ws2d_histo_cut_x", help_msg)
+
+            for cut_type in ("x", "y"):
+                assert_call_as_expected(transpose=False, cut_type=cut_type)
+                assert_call_as_expected(transpose=False, cut_type=cut_type)
+
     @patch("mantidqt.widgets.sliceviewer.models.model.TransposeMD")
     @patch("mantidqt.widgets.sliceviewer.models.model.IntegrateMDHistoWorkspace")
     def test_export_region_for_mdhisto_workspace(self, mock_intMD, mock_transposemd):