def test_model_MDE(self):
model = SliceViewerModel(self.ws_MDE_3D)
self.assertNotEqual(model.get_ws((0,0,0), (1,1,1)), self.ws_MDE_3D)
self.assertEqual(model._get_ws(), self.ws_MDE_3D)
self.assertEqual(model.get_ws_type(), WS_TYPE.MDE)
dim_info = model.get_dim_info(0)
self.assertEqual(dim_info['minimum'], -3)
self.assertEqual(dim_info['maximum'], 3)
self.assertEqual(dim_info['number_of_bins'], 1)
self.assertAlmostEqual(dim_info['width'], 6)
self.assertEqual(dim_info['name'], 'h')
self.assertEqual(dim_info['units'], 'rlu')
self.assertEqual(dim_info['type'], 'MDE')
dim_infos = model.get_dimensions_info()
self.assertEqual(len(dim_infos), 3)
dim_info = dim_infos[2]
self.assertEqual(dim_info['minimum'], -5)
self.assertEqual(dim_info['maximum'], 5)
self.assertEqual(dim_info['number_of_bins'], 1)
self.assertAlmostEqual(dim_info['width'], 10)
self.assertEqual(dim_info['name'], 'l')
self.assertEqual(dim_info['units'], 'rlu')
self.assertEqual(dim_info['type'], 'MDE')
mdh = model.get_ws((None,0,None), (3,0.001,3))
assert_allclose(mdh.getSignalArray().squeeze(), [[0, 0, 0],
[0, 692.237618, 0],
[0, 118.362777, 0]])
d0 = mdh.getDimension(0)
d1 = mdh.getDimension(1)
d2 = mdh.getDimension(2)
self.assertEqual(d0.name, 'h')
self.assertEqual(d0.getNBins(), 3)
self.assertEqual(d0.getMinimum(), -3)
self.assertEqual(d0.getMaximum(), 3)
self.assertEqual(d1.name, 'k')
self.assertEqual(d1.getNBins(), 1)
self.assertAlmostEqual(d1.getMinimum(), -0.0005)
self.assertAlmostEqual(d1.getMaximum(), 0.0005)
self.assertEqual(d2.name, 'l')
self.assertEqual(d2.getNBins(), 3)
self.assertEqual(d2.getMinimum(), -5)
self.assertEqual(d2.getMaximum(), 5)
assert_allclose(model.get_data((None,0,None), (3,0.001,3)), [[0, 0, 0],
[0, 692.237618, 0],
[0, 118.362777, 0]])
assert_allclose(model.get_data((None,0,None), (3,0.001,3), transpose=True), [[0, 0, 0],
[0, 692.237618, 118.362777],
[0, 0, 0]])
def test_model_MDE(self):
model = SliceViewerModel(self.ws_MDE_3D)
self.assertNotEqual(model.get_ws((0, 0, 0), (1, 1, 1)), self.ws_MDE_3D)
self.assertEqual(model._get_ws(), self.ws_MDE_3D)
self.assertEqual(model.get_ws_type(), WS_TYPE.MDE)
dim_info = model.get_dim_info(0)
self.assertEqual(dim_info['minimum'], -3)
self.assertEqual(dim_info['maximum'], 3)
self.assertEqual(dim_info['number_of_bins'], 1)
self.assertAlmostEqual(dim_info['width'], 6)
self.assertEqual(dim_info['name'], 'h')
self.assertEqual(dim_info['units'], 'rlu')
self.assertEqual(dim_info['type'], 'MDE')
dim_infos = model.get_dimensions_info()
self.assertEqual(len(dim_infos), 3)
dim_info = dim_infos[2]
self.assertEqual(dim_info['minimum'], -5)
self.assertEqual(dim_info['maximum'], 5)
self.assertEqual(dim_info['number_of_bins'], 1)
self.assertAlmostEqual(dim_info['width'], 10)
self.assertEqual(dim_info['name'], 'l')
self.assertEqual(dim_info['units'], 'rlu')
self.assertEqual(dim_info['type'], 'MDE')
mdh = model.get_ws((None, 0, None), (3, 0.001, 3))
assert_allclose(mdh.getSignalArray().squeeze(),
[[0, 0, 0], [0, 692.237618, 0], [0, 118.362777, 0]])
d0 = mdh.getDimension(0)
d1 = mdh.getDimension(1)
d2 = mdh.getDimension(2)
self.assertEqual(d0.name, 'h')
self.assertEqual(d0.getNBins(), 3)
self.assertEqual(d0.getMinimum(), -3)
self.assertEqual(d0.getMaximum(), 3)
self.assertEqual(d1.name, 'k')
self.assertEqual(d1.getNBins(), 1)
self.assertAlmostEqual(d1.getMinimum(), -0.0005)
self.assertAlmostEqual(d1.getMaximum(), 0.0005)
self.assertEqual(d2.name, 'l')
self.assertEqual(d2.getNBins(), 3)
self.assertEqual(d2.getMinimum(), -5)
self.assertEqual(d2.getMaximum(), 5)
assert_allclose(model.get_data((None, 0, None), (3, 0.001, 3)),
[[0, 0, 0], [0, 692.237618, 0], [0, 118.362777, 0]])
assert_allclose(
model.get_data((None, 0, None), (3, 0.001, 3), transpose=True),
[[0, 0, 0], [0, 692.237618, 118.362777], [0, 0, 0]])
def test_model_MDH(self):
model = SliceViewerModel(self.ws_MD_3D)
self.assertEqual(model.get_ws(), self.ws_MD_3D)
self.assertEqual(model.get_ws_type(), WS_TYPE.MDH)
assert_equal(model.get_data((None, 2, 2)), range(90, 95))
assert_equal(model.get_data((1, 2, None)), range(18, 118, 25))
assert_equal(model.get_data((None, None, 0)),
np.reshape(range(50, 75), (5, 5)).T)
dim_info = model.get_dim_info(0)
self.assertEqual(dim_info['minimum'], -3)
self.assertEqual(dim_info['maximum'], 3)
self.assertEqual(dim_info['number_of_bins'], 5)
self.assertAlmostEqual(dim_info['width'], 1.2)
self.assertEqual(dim_info['name'], 'Dim1')
self.assertEqual(dim_info['units'], 'MomentumTransfer')
self.assertEqual(dim_info['type'], 'MDH')
dim_infos = model.get_dimensions_info()
self.assertEqual(len(dim_infos), 3)
dim_info = dim_infos[2]
self.assertEqual(dim_info['minimum'], -1)
self.assertEqual(dim_info['maximum'], 1)
self.assertEqual(dim_info['number_of_bins'], 4)
self.assertAlmostEqual(dim_info['width'], 0.5)
self.assertEqual(dim_info['name'], 'Dim3')
self.assertEqual(dim_info['units'], 'Angstrom')
self.assertEqual(dim_info['type'], 'MDH')
def test_model_matrix(self):
model = SliceViewerModel(self.ws2d_histo)
self.assertEqual(model.get_ws(), self.ws2d_histo)
self.assertEqual(model.get_ws_type(), WS_TYPE.MATRIX)
dim_info = model.get_dim_info(0)
self.assertEqual(dim_info['minimum'], 10)
self.assertEqual(dim_info['maximum'], 30)
self.assertEqual(dim_info['number_of_bins'], 2)
self.assertAlmostEqual(dim_info['width'], 10)
self.assertEqual(dim_info['name'], 'Wavelength')
self.assertEqual(dim_info['units'], 'Angstrom')
self.assertEqual(dim_info['type'], 'MATRIX')
dim_infos = model.get_dimensions_info()
self.assertEqual(len(dim_infos), 2)
dim_info = dim_infos[1]
self.assertEqual(dim_info['minimum'], 4)
self.assertEqual(dim_info['maximum'], 8)
self.assertEqual(dim_info['number_of_bins'], 2)
self.assertAlmostEqual(dim_info['width'], 2)
self.assertEqual(dim_info['name'], 'Energy transfer')
self.assertEqual(dim_info['units'], 'meV')
self.assertEqual(dim_info['type'], 'MATRIX')
def test_model_MDH(self):
model = SliceViewerModel(self.ws_MD_3D)
self.assertEqual(model.get_ws(), self.ws_MD_3D)
self.assertEqual(model.get_ws_type(), WS_TYPE.MDH)
signal = np.arange(100).reshape(5, 5, 4)
assert_equal(model.get_data((None, 2, 2)), signal[(slice(None), 3, 3)])
assert_equal(model.get_data((1, 2, None)), range(72, 76))
assert_equal(model.get_data((None, None, 0)), signal[(slice(None), slice(None), 2)])
dim_info = model.get_dim_info(0)
self.assertEqual(dim_info['minimum'], -3)
self.assertEqual(dim_info['maximum'], 3)
self.assertEqual(dim_info['number_of_bins'], 5)
self.assertAlmostEqual(dim_info['width'], 1.2)
self.assertEqual(dim_info['name'], 'Dim1')
self.assertEqual(dim_info['units'], 'MomentumTransfer')
self.assertEqual(dim_info['type'], 'MDH')
self.assertEqual(dim_info['qdim'], False)
dim_infos = model.get_dimensions_info()
self.assertEqual(len(dim_infos), 3)
dim_info = dim_infos[2]
self.assertEqual(dim_info['minimum'], -1)
self.assertEqual(dim_info['maximum'], 1)
self.assertEqual(dim_info['number_of_bins'], 4)
self.assertAlmostEqual(dim_info['width'], 0.5)
self.assertEqual(dim_info['name'], 'Dim3')
self.assertEqual(dim_info['units'], 'Angstrom')
self.assertEqual(dim_info['type'], 'MDH')
self.assertEqual(dim_info['qdim'], False)
def test_model_MDH(self):
model = SliceViewerModel(self.ws_MD_3D)
self.assertEqual(model.get_ws(), self.ws_MD_3D)
self.assertEqual(model.get_ws_type(), WS_TYPE.MDH)
assert_equal(model.get_data((None, 2, 2)), range(90,95))
assert_equal(model.get_data((1, 2, None)), range(18,118,25))
assert_equal(model.get_data((None, None, 0)), np.reshape(range(50,75), (5,5)).T)
dim_info = model.get_dim_info(0)
self.assertEqual(dim_info['minimum'], -3)
self.assertEqual(dim_info['maximum'], 3)
self.assertEqual(dim_info['number_of_bins'], 5)
self.assertAlmostEqual(dim_info['width'], 1.2)
self.assertEqual(dim_info['name'], 'Dim1')
self.assertEqual(dim_info['units'], 'MomentumTransfer')
self.assertEqual(dim_info['type'], 'MDH')
dim_infos = model.get_dimensions_info()
self.assertEqual(len(dim_infos), 3)
dim_info = dim_infos[2]
self.assertEqual(dim_info['minimum'], -1)
self.assertEqual(dim_info['maximum'], 1)
self.assertEqual(dim_info['number_of_bins'], 4)
self.assertAlmostEqual(dim_info['width'], 0.5)
self.assertEqual(dim_info['name'], 'Dim3')
self.assertEqual(dim_info['units'], 'Angstrom')
self.assertEqual(dim_info['type'], 'MDH')
def test_model_matrix(self):
model = SliceViewerModel(self.ws2d_histo)
self.assertEqual(model.get_ws(), self.ws2d_histo)
self.assertEqual(model.get_ws_type(), WS_TYPE.MATRIX)
dim_info = model.get_dim_info(0)
self.assertEqual(dim_info['minimum'], 10)
self.assertEqual(dim_info['maximum'], 30)
self.assertEqual(dim_info['number_of_bins'], 2)
self.assertAlmostEqual(dim_info['width'], 10)
self.assertEqual(dim_info['name'], 'Wavelength')
self.assertEqual(dim_info['units'], 'Angstrom')
self.assertEqual(dim_info['type'], 'MATRIX')
dim_infos = model.get_dimensions_info()
self.assertEqual(len(dim_infos), 2)
dim_info = dim_infos[1]
self.assertEqual(dim_info['minimum'], 4)
self.assertEqual(dim_info['maximum'], 8)
self.assertEqual(dim_info['number_of_bins'], 2)
self.assertAlmostEqual(dim_info['width'], 2)
self.assertEqual(dim_info['name'], 'Energy transfer')
self.assertEqual(dim_info['units'], 'meV')
self.assertEqual(dim_info['type'], 'MATRIX')
def test_model_MDE_basis_vectors_not_normalised_when_HKL(self, mock_binmd):
ws = _create_mock_mdeventworkspace(ndims=3,
coords=SpecialCoordinateSystem.HKL,
extents=(-3, 3, -4, 4, -5, 5),
names=('h', 'k', 'l'),
units=('r.l.u.', 'r.l.u.',
'r.l.u.'),
isq=(True, True, True))
model = SliceViewerModel(ws)
mock_binmd.return_value = self.ws_MD_3D # different workspace
self.assertNotEqual(model.get_ws((None, None, 0), (1, 2, 4)), ws)
mock_binmd.assert_called_once_with(
AxisAligned=False,
NormalizeBasisVectors=False,
BasisVector0='h,r.l.u.,1.0,0.0,0.0',
BasisVector1='k,r.l.u.,0.0,1.0,0.0',
BasisVector2='l,r.l.u.,0.0,0.0,1.0',
EnableLogging=False,
InputWorkspace=ws,
OutputBins=[1, 2, 1],
OutputExtents=[-3, 3, -4, 4, -2.0, 2.0],
OutputWorkspace=ws.name() + '_svrebinned')
mock_binmd.reset_mock()
def test_get_ws_MDE_with_limits_uses_limits_over_dimension_extents(self, mock_binmd):
model = SliceViewerModel(self.ws_MDE_3D)
mock_binmd.return_value = self.ws_MD_3D
self.assertNotEqual(model.get_ws((None, None, 0), (1, 2, 4), ((-2, 2), (-1, 1))),
self.ws_MDE_3D)
call_params = dict(InputWorkspace=self.ws_MDE_3D,
OutputWorkspace='ws_MDE_3D_svrebinned',
AlignedDim0='h,-2,2,1',
AlignedDim1='k,-1,1,2',
AlignedDim2='l,-2.0,2.0,1',
EnableLogging=False)
mock_binmd.assert_called_once_with(**call_params)
mock_binmd.reset_mock()
model.get_data((None, None, 0), (1, 2, 4), ((-2, 2), (-1, 1)))
mock_binmd.assert_called_once_with(**call_params)
def test_model_MDE(self, mock_binmd):
model = SliceViewerModel(self.ws_MDE_3D)
mock_binmd.return_value = self.ws_MD_3D
self.assertNotEqual(model.get_ws((None, None, 0), (1, 2, 4)),
self.ws_MDE_3D)
mock_binmd.assert_called_once_with(
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',
EnableLogging=False,
InputWorkspace=self.ws_MDE_3D,
OutputBins=[1, 2, 1],
OutputExtents=[-3, 3, -4, 4, -2.0, 2.0],
OutputWorkspace='ws_MDE_3D_svrebinned')
mock_binmd.reset_mock()
self.assertEqual(model._get_ws(), self.ws_MDE_3D)
self.assertEqual(model.get_ws_type(), WS_TYPE.MDE)
dim_info = model.get_dim_info(0)
self.assertEqual(dim_info['minimum'], -3)
self.assertEqual(dim_info['maximum'], 3)
self.assertEqual(dim_info['number_of_bins'], 1)
self.assertAlmostEqual(dim_info['width'], 6)
self.assertEqual(dim_info['name'], 'h')
self.assertEqual(dim_info['units'], 'rlu')
self.assertEqual(dim_info['type'], 'MDE')
self.assertEqual(dim_info['qdim'], False)
dim_infos = model.get_dimensions_info()
self.assertEqual(len(dim_infos), 3)
dim_info = dim_infos[2]
self.assertEqual(dim_info['minimum'], -5)
self.assertEqual(dim_info['maximum'], 5)
self.assertEqual(dim_info['number_of_bins'], 1)
self.assertAlmostEqual(dim_info['width'], 10)
self.assertEqual(dim_info['name'], 'l')
self.assertEqual(dim_info['units'], 'rlu')
self.assertEqual(dim_info['type'], 'MDE')
self.assertEqual(dim_info['qdim'], False)
def test_get_ws_mde_sets_minimum_width_on_data_limits(self, mock_binmd):
model = SliceViewerModel(self.ws_MDE_3D)
mock_binmd.return_value = self.ws_MD_3D
xmin = -5e-8
xmax = 5e-8
self.assertNotEqual(
model.get_ws((None, None, 0), (1, 2, 4), ((xmin, xmax), (-1, 1)),
[0, 1, None]), self.ws_MDE_3D)
call_params = dict(
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',
EnableLogging=False,
InputWorkspace=self.ws_MDE_3D,
OutputBins=[1, 2, 1],
OutputExtents=[xmin, xmin + MIN_WIDTH, -1, 1, -2.0, 2.0],
OutputWorkspace='ws_MDE_3D_svrebinned')
mock_binmd.assert_called_once_with(**call_params)
mock_binmd.reset_mock()
def test_get_ws_MDE_with_limits_uses_limits_over_dimension_extents(
self, mock_binmd):
model = SliceViewerModel(self.ws_MDE_3D)
mock_binmd.return_value = self.ws_MD_3D
self.assertNotEqual(
model.get_ws((None, None, 0), (1, 2, 4), ((-2, 2), (-1, 1)),
[0, 1, None]), self.ws_MDE_3D)
call_params = dict(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',
EnableLogging=False,
InputWorkspace=self.ws_MDE_3D,
OutputBins=[1, 2, 1],
OutputExtents=[-2, 2, -1, 1, -2.0, 2.0],
OutputWorkspace='ws_MDE_3D_svrebinned')
mock_binmd.assert_called_once_with(**call_params)
mock_binmd.reset_mock()
model.get_data((None, None, 0), (1, 2, 4), [0, 1, None],
((-2, 2), (-1, 1)))
mock_binmd.assert_called_once_with(**call_params)
