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)