def test_sliceviewer_MDE(self):
self.model.get_ws_type = mock.Mock(return_value=WS_TYPE.MDE)
presenter = SliceViewer(None, model=self.model, view=self.view)
# setup calls
self.assertEqual(self.model.get_dimensions_info.call_count, 0)
self.assertEqual(self.model.get_ws.call_count, 1)
self.assertEqual(
self.view.data_view.dimensions.get_slicepoint.call_count, 1)
self.assertEqual(
self.view.data_view.dimensions.get_bin_params.call_count, 1)
self.assertEqual(self.view.data_view.plot_MDH.call_count, 1)
# new_plot
self.model.reset_mock()
self.view.reset_mock()
presenter.new_plot()
self.assertEqual(self.model.get_ws.call_count, 1)
self.assertEqual(
self.view.data_view.dimensions.get_slicepoint.call_count, 1)
self.assertEqual(
self.view.data_view.dimensions.get_bin_params.call_count, 1)
self.assertEqual(self.view.data_view.plot_MDH.call_count, 1)
# update_plot_data
self.model.reset_mock()
self.view.reset_mock()
presenter.update_plot_data()
self.assertEqual(self.model.get_data.call_count, 1)
self.assertEqual(
self.view.data_view.dimensions.get_slicepoint.call_count, 1)
self.assertEqual(
self.view.data_view.dimensions.get_bin_params.call_count, 1)
self.assertEqual(self.view.data_view.update_plot_data.call_count, 1)
def test_update_plot_data_updates_axes_limits_when_orthog_data_tranposed(
self):
pres = SliceViewer(self.hkl_ws)
# not transpose
pres.view.data_view.dimensions.transpose = False
pres.update_plot_data()
extent = pres.view.data_view.image.get_extent()
assert_allclose(extent, [-10.0, 10.0, -9.0, 9.0])
# transpose
pres.view.data_view.dimensions.transpose = True
pres.update_plot_data()
extent = pres.view.data_view.image.get_extent()
# Should be the same as before as transposition is handled in the model
assert_allclose(extent, [-10.0, 10.0, -9.0, 9.0])
assert_allclose(extent[0:2], list(pres.view.data_view.ax.get_xlim()))
assert_allclose(extent[2:], list(pres.view.data_view.ax.get_ylim()))
pres.view.close()
def runTest(self):
HelperTestingClass.__init__(self)
hkl_ws = create_hkl_ws()
pres = SliceViewer(hkl_ws)
# not transpose
pres.view.data_view.dimensions.transpose = False
pres.update_plot_data()
extent = pres.view.data_view.image.get_extent()
assert_allclose(extent, [-10.0, 10.0, -9.0, 9.0])
# transpose
pres.view.data_view.dimensions.transpose = True
pres.update_plot_data()
extent = pres.view.data_view.image.get_extent()
# Should be the same as before as transposition is handled in the model
assert_allclose(extent, [-10.0, 10.0, -9.0, 9.0])
assert_allclose(extent[0:2], list(pres.view.data_view.ax.get_xlim()))
assert_allclose(extent[2:], list(pres.view.data_view.ax.get_ylim()))
pres.view.close()