def _assert_supports_dynamic_rebinning(self, expectation, ws_type, has_original_workspace=None): ws = _create_mock_workspace(ws_type, coords=SpecialCoordinateSystem.QLab, has_oriented_lattice=False, ndims=3) if ws_type == MatrixWorkspace: ws.hasOriginalWorkspace.return_value = False elif has_original_workspace is not None: ws.hasOriginalWorkspace.return_value = has_original_workspace if has_original_workspace: ws.getOriginalWorkspace.return_value = self.ws_MDE_3D model = SliceViewerModel(ws) self.assertEqual(expectation, model.can_support_dynamic_rebinning())
def _assert_supports_dynamic_rebinning(self, expectation, ws_type, ndims=3, has_original_workspace=None, original_ws_ndims=None): ws = _create_mock_workspace(ws_type, coords=SpecialCoordinateSystem.QLab, has_oriented_lattice=False, ndims=ndims) if ws_type == MatrixWorkspace: ws.hasOriginalWorkspace.side_effect = lambda index: False elif has_original_workspace is not None: ws.hasOriginalWorkspace.side_effect = lambda index: has_original_workspace if not original_ws_ndims: original_ws_ndims = ndims orig_ws = _create_mock_workspace( ws_type, coords=SpecialCoordinateSystem.QLab, has_oriented_lattice=False, ndims=original_ws_ndims) ws.getOriginalWorkspace.side_effect = lambda index: orig_ws model = SliceViewerModel(ws) self.assertEqual(expectation, model.can_support_dynamic_rebinning())