def test_has_fignum(self): mock_manager = MockFigureManager(0) GlobalFigureManager.set_active(mock_manager) self.assertTrue(GlobalFigureManager.has_fignum(0)) num = 3131313 mock_manager = MockFigureManager(num) GlobalFigureManager.set_active(mock_manager) self.assertTrue(GlobalFigureManager.has_fignum(num))
def test_destroy(self, mock_gc_collect): num = 0 mock_manager = MockFigureManager(num) GlobalFigureManager.set_active(mock_manager) self.assertEqual(1, len(GlobalFigureManager._activeQue)) self.assertEqual(1, len(GlobalFigureManager.figs)) with patch.object(GlobalFigureManager, 'notify_observers') as mock_notify_observers: GlobalFigureManager.destroy(num) self.assertEqual(0, len(GlobalFigureManager._activeQue)) self.assertEqual(0, len(GlobalFigureManager.figs)) mock_gc_collect.assert_called_once_with(1) mock_notify_observers.assert_has_calls( [call(FigureAction.Closed, num), call(FigureAction.OrderChanged, -1)])
def test_destroy_all(self): num = 0 self.add_manager(num) other_mock_fig = Mock() other_mock_manager = MockFigureManager(num + 1) other_mock_manager.canvas.figure = other_mock_fig GlobalFigureManager.set_active(other_mock_manager) self.assertEqual(2, len(GlobalFigureManager._activeQue)) self.assertEqual(2, len(GlobalFigureManager.figs)) GlobalFigureManager.destroy_all() self.assertEqual(0, len(GlobalFigureManager._activeQue)) self.assertEqual(0, len(GlobalFigureManager.figs))
def _window_activated(self): GlobalFigureManager.set_active(self)
def test_get_fig_manager(self): mock_manager = MockFigureManager(0) GlobalFigureManager.set_active(mock_manager) manager = GlobalFigureManager.get_fig_manager(0) self.assertIsNotNone(manager) self.assertEqual(mock_manager, manager)
def add_manager(self, num=0): mock_manager = MockFigureManager(num) mock_fig = Mock() mock_manager.canvas.figure = mock_fig GlobalFigureManager.set_active(mock_manager) return mock_manager, mock_fig
def test_get_active(self): mock_manager = MockFigureManager(0) GlobalFigureManager.set_active(mock_manager) self.assertEqual(mock_manager, GlobalFigureManager.get_active())
def test_set_active(self): self.assertEqual(0, len(GlobalFigureManager._activeQue)) mock_manager = MockFigureManager(0) GlobalFigureManager.set_active(mock_manager) self.assertEqual(1, len(GlobalFigureManager._activeQue))