def test_single_peak_selection(self, mock_peaks_list_presenter):
mock_view = MagicMock()
name = 'ws1'
mock_model = create_mock_model(name)
mock_view.selected_index = 0
presenter = PeaksViewerPresenter(mock_model, mock_view)
presenter.notify(PeaksViewerPresenter.Event.PeakSelected)
mock_model.zoom_to.assert_called_once_with(0)
def test_single_peak_selection(self, mock_peaks_list_presenter):
name = 'ws1'
mock_model = create_mock_model(name)
viewlimits = (-1, 1), (-2, 2)
mock_model.viewlimits.return_value = viewlimits
self.mock_view.selected_index = 0
presenter = PeaksViewerPresenter(mock_model, self.mock_view)
presenter.notify(PeaksViewerPresenter.Event.PeakSelected)
mock_model.viewlimits.assert_called_once_with(0)
self.mock_view.set_axes_limits.assert_called_once_with(*viewlimits, auto_transform=False)
def test_single_peak_selection_if_peaks_not_drawn(
self, mock_peaks_list_presenter):
# peaks not drawn if one fo viewing axes non-Q
name = 'ws1'
mock_model = create_mock_model(name)
mock_model.has_representations_drawn.return_value = False
self.mock_view.selected_index = 0
presenter = PeaksViewerPresenter(mock_model, self.mock_view)
presenter.notify(PeaksViewerPresenter.Event.PeakSelected)
mock_model.viewlimits.assert_not_called()
self.mock_view.set_axes_limits.assert_not_called()
def test_clear_removes_painted_peaks(self, mock_peaks_list_presenter):
centers = ((1, 2, 3), (4, 5, 3.01))
slice_info = create_slice_info(centers, slice_value=3, slice_width=5)
test_model = create_peaks_viewer_model(centers, fg_color="r")
painter = MagicMock()
test_model.draw_peaks(slice_info, painter)
mock_view = MagicMock()
mock_view.painter = painter
presenter = PeaksViewerPresenter(test_model, mock_view)
presenter.notify(PeaksViewerPresenter.Event.ClearPeaks)
self.assertEqual(2, mock_view.painter.remove.call_count)
def test_clear_removes_painted_peaks(self, mock_peaks_list_presenter):
centers = ((1, 2, 3), (4, 5, 3.01))
slice_info = create_slice_info(centers, slice_value=3, slice_width=5)
test_model = create_peaks_viewer_model(centers, fg_color="r")
painter, axes = MagicMock(), MagicMock()
axes.get_xlim.return_value = (-1, 1)
painter.axes = axes
test_model.draw_peaks(slice_info, painter, SpecialCoordinateSystem.QSample)
self.mock_view.painter = painter
presenter = PeaksViewerPresenter(test_model, self.mock_view)
presenter.notify(PeaksViewerPresenter.Event.ClearPeaks)
self.assertEqual(2, self.mock_view.painter.remove.call_count)
def test_slice_point_changed_clears_old_peaks_and_overlays_visible(
self, mock_peaks_list_presenter):
centers = ((1, 2, 3), (4, 5, 3.01))
slice_info = create_slice_info(centers, slice_value=3, slice_width=5)
test_model = create_peaks_viewer_model(centers, fg_color="r")
# draw some peaks first so we can test clearing them
painter, axes = MagicMock(), MagicMock()
axes.get_xlim.return_value = (-1, 1)
painter.axes = axes
test_model.draw_peaks(slice_info, painter, SpecialCoordinateSystem.QSample)
# clear draw calls
painter.cross.reset_mock()
self.mock_view.painter = painter
self.mock_view.sliceinfo = create_slice_info(centers, slice_value=3, slice_width=5)
presenter = PeaksViewerPresenter(test_model, self.mock_view)
presenter.notify(PeaksViewerPresenter.Event.SlicePointChanged)
self.assertEqual(2, self.mock_view.painter.remove.call_count)
self.assertEqual(2, self.mock_view.painter.cross.call_count)
def test_slice_point_changed_clears_old_peaks_and_overlays_visible(
self, mock_peaks_list_presenter):
centers = ((1, 2, 3), (4, 5, 3.01))
slice_info = create_slice_info(centers, slice_value=3, slice_width=5)
test_model = create_peaks_viewer_model(centers, fg_color="r")
# draw some peaks first so we can test clearing them
painter = MagicMock()
test_model.draw_peaks(slice_info, painter)
# clear draw calls
painter.cross.reset_mock()
mock_view = MagicMock()
mock_view.painter = painter
mock_view.sliceinfo = create_slice_info(centers,
slice_value=3,
slice_width=5)
presenter = PeaksViewerPresenter(test_model, mock_view)
presenter.notify(PeaksViewerPresenter.Event.SlicePointChanged)
self.assertEqual(2, mock_view.painter.remove.call_count)
self.assertEqual(2, mock_view.painter.cross.call_count)