def test_window_title(self, mock_qappthread): mock_qappthread.return_value = mock_qappthread fig = MagicMock() fig.bbox.max = [1, 1] canvas = MantidFigureCanvas(fig) fig_mgr = FigureManagerWorkbench(canvas, 1) self.assertEqual(fig_mgr.get_window_title(), "Figure 1")
def test_reverse_axes_lines(self): fig, ax = plt.subplots(subplot_kw={'projection': 'mantid'}) spec2 = ax.plot(self.ws2d_histo, 'rs', specNum=2, linewidth=6) spec3 = ax.plot(self.ws2d_histo, 'rs', specNum=3, linewidth=6) spec1 = ax.plot(self.ws2d_histo, 'rs', specNum=1, linewidth=6) self.assertEqual(ax.get_lines()[0], spec2[0]) self.assertEqual(ax.get_lines()[1], spec3[0]) self.assertEqual(ax.get_lines()[2], spec1[0]) FigureManagerWorkbench._reverse_axis_lines(ax) self.assertEqual(ax.get_lines()[0], spec1[0]) self.assertEqual(ax.get_lines()[1], spec3[0]) self.assertEqual(ax.get_lines()[2], spec2[0])
def test_construction(self, mock_qappthread): mock_qappthread.return_value = mock_qappthread fig = MagicMock() fig.bbox.max = [1, 1] canvas = MantidFigureCanvas(fig) fig_mgr = FigureManagerWorkbench(canvas, 1) self.assertNotEqual(fig_mgr, None)
def _is_button_enabled(cls, fig, button): """ Create the figure manager and check whether its toolbar is toggled on or off for the given figure. We have to explicitly call set_button_visibility() here, which would otherwise be called within the show() function. """ canvas = MantidFigureCanvas(fig) fig_manager = FigureManagerWorkbench(canvas, 1) # This is only called when show() is called on the figure manager, so we have to manually call it here. fig_manager.toolbar.set_buttons_visibility(fig) return fig_manager.toolbar._actions[button].isEnabled()
def test_construction(self, mock_qappthread): mock_qappthread.return_value = mock_qappthread fig = MagicMock() canvas = FigureCanvasQTAgg(fig) fig_mgr = FigureManagerWorkbench(canvas, 1) self.assertNotEqual(fig_mgr, None)