def test_categorizing_of_uncategorized_plot(self, mock_figure_class):
mock_figures = [mock.Mock(), mock.Mock(), mock.Mock()]
fig1_mock_manager = mock.Mock()
# This manager is used to compare the relative order of calls of two differebc functions
fig1_mock_manager.attach_mock(mock_figures[0].flag_as_kept, 'fig1_kept')
fig1_mock_manager.attach_mock(mock_figures[0].flag_as_current, 'fig1_current')
mock_figure_class.side_effect = mock_figures
cat1 = '1d'
cat2 = '2d'
cat1_get_active_figure = set_category(cat1)(GlobalFigureManager.get_active_figure)
cat2_get_active_figure = set_category(cat2)(GlobalFigureManager.get_active_figure)
# test is an arbitrary method just to make sure the correct figures are returned
cat1_get_active_figure().test(1) # create a figure of category 1
cat2_get_active_figure().test(2) # create a figure of category 2
GlobalFigureManager.set_figure_as_kept(2) # now there is no active figure
GlobalFigureManager.get_active_figure().test(3) # create an uncategorized figure
cat1_get_active_figure().test(4) # the previously uncategorized figure should now be categorized as cat1
mock_figures[0].test.assert_has_calls([call(1)])
mock_figures[1].test.assert_has_calls([call(2)])
mock_figures[2].test.assert_has_calls([call(3), call(4)])
self.assertTrue(fig1_mock_manager.mock_calls[-1] == call.fig1_kept()) # assert final status of fig1 is kept
self.assertTrue(GlobalFigureManager._active_figure == 3)
def test_category_switching(self,mock_figure_class):
mock_figures = [mock.Mock(),mock.Mock()]
mock_figure_class.side_effect = mock_figures
cat1 = '1d'
cat2 = '2d'
cat1_get_active_figure = set_category(cat1)(GlobalFigureManager.get_active_figure)
cat2_get_active_figure = set_category(cat2)(GlobalFigureManager.get_active_figure)
# test is an arbitrary method just to make sure the correct figures are returned
cat1_get_active_figure().test(1)
cat2_get_active_figure().test(2)
cat1_get_active_figure().test(3)
mock_figures[0].test.assert_has_calls([call(1), call(3)])
mock_figures[1].test.assert_has_calls([call(2)])
self.assertTrue(GlobalFigureManager._active_figure == 1)
def test_make_current_with_single_category(self, mock_figure_class):
mock_figures = [mock.Mock(), mock.Mock()]
# These manager is used to compare the relative order of calls of two different functions
mock_managers = [mock.Mock(), mock.Mock()]
for i in range(len(mock_figures)):
mock_managers[i].attach_mock(mock_figures[i].flag_as_kept, 'fig_kept')
mock_managers[i].attach_mock(mock_figures[i].flag_as_current, 'fig_current')
mock_figure_class.side_effect = mock_figures
cat1 = '1d'
cat1_get_active_figure = set_category(cat1)(GlobalFigureManager.get_active_figure)
# test is an arbitrary method just to make sure the correct figures are returned
cat1_get_active_figure().test(1) # create a figure of category 1
GlobalFigureManager.set_figure_as_kept(1) # now there is no active figure
cat1_get_active_figure().test(2) # this command should go to a new figure
self.assertTrue(mock_managers[0].mock_calls[-1] == call.fig_kept()) # assert fig1 now displays kept
self.assertTrue(mock_managers[1].mock_calls[-1] == call.fig_current()) # assert fig2 now displays current
GlobalFigureManager.set_figure_as_current(1)
self.assertTrue(mock_managers[0].mock_calls[-1] == call.fig_current()) # assert fig1 now displays current
self.assertTrue(mock_managers[1].mock_calls[-1] == call.fig_kept()) # assert fig2 now displays kept
cat1_get_active_figure().test(3) # This should go to fig1
GlobalFigureManager.get_active_figure().test(4) # This should go to fig1 as well
mock_figures[0].test.assert_has_calls([call(1), call(3), call(4)])
mock_figures[1].test.assert_has_calls([call(2)])
self.assertTrue(GlobalFigureManager._active_figure == 1)
def test_category_switching(self, mock_figure_class):
mock_figures = [mock.Mock(), mock.Mock()]
mock_figure_class.side_effect = mock_figures
cat1 = '1d'
cat2 = '2d'
cat1_get_active_figure = set_category(cat1)(
GlobalFigureManager.get_active_figure)
cat2_get_active_figure = set_category(cat2)(
GlobalFigureManager.get_active_figure)
# test is an arbitrary method just to make sure the correct figures are returned
cat1_get_active_figure().test(1)
cat2_get_active_figure().test(2)
cat1_get_active_figure().test(3)
mock_figures[0].test.assert_has_calls([call(1), call(3)])
mock_figures[1].test.assert_has_calls([call(2)])
self.assertTrue(GlobalFigureManager._active_figure == 1)
def test_create_single_categorised_figure(self,mock_figure_class):
mock_figures = [mock.Mock()]
mock_figure_class.side_effect = mock_figures
category = '1d'
# The following line is equivalent to applying the decorator setcategory with the parameter category
# to function GlobalFigureManager.get_active_figure
categorised_get_active_figure = set_category(category)(GlobalFigureManager.get_active_figure)
fig = categorised_get_active_figure()
self.assertTrue(fig == mock_figures[0]) # Assert Figure object came from right place
self.assertTrue(GlobalFigureManager.get_category(1) == category)
self.assertTrue(GlobalFigureManager.get_active_figure() == mock_figures[0]) # Check that it is set as the active figure
def test_create_categorised_figure_then_uncategorised_figure(self,mock_figure_class):
mock_figures = [mock.Mock(), mock.Mock()]
mock_figure_class.side_effect = mock_figures
category = '1d'
categorised_get_active_figure = set_category(category)(GlobalFigureManager.get_active_figure)
fig1 = categorised_get_active_figure()
fig2 = GlobalFigureManager.get_figure_number()
fig1_number = GlobalFigureManager.number_of_figure(fig1)
fig2_number = GlobalFigureManager.number_of_figure(fig2)
self.assertTrue(GlobalFigureManager.get_active_figure() == fig2)
self.assertTrue(GlobalFigureManager.get_category(fig1_number) == category)
self.assertRaises(KeyError, GlobalFigureManager.get_category, fig2_number)
self.assertTrue( fig1_number == 1 and fig2_number == 2)
def test_categorizing_of_uncategorized_plot(self, mock_figure_class):
mock_figures = [mock.Mock(), mock.Mock(), mock.Mock()]
fig1_mock_manager = mock.Mock()
# This manager is used to compare the relative order of calls of two differebc functions
fig1_mock_manager.attach_mock(mock_figures[0].flag_as_kept,
'fig1_kept')
fig1_mock_manager.attach_mock(mock_figures[0].flag_as_current,
'fig1_current')
mock_figure_class.side_effect = mock_figures
cat1 = '1d'
cat2 = '2d'
cat1_get_active_figure = set_category(cat1)(
GlobalFigureManager.get_active_figure)
cat2_get_active_figure = set_category(cat2)(
GlobalFigureManager.get_active_figure)
# test is an arbitrary method just to make sure the correct figures are returned
cat1_get_active_figure().test(1) # create a figure of category 1
cat2_get_active_figure().test(2) # create a figure of category 2
GlobalFigureManager.set_figure_as_kept(
2) # now there is no active figure
GlobalFigureManager.get_active_figure().test(
3) # create an uncategorized figure
cat1_get_active_figure().test(
4
) # the previously uncategorized figure should now be categorized as cat1
mock_figures[0].test.assert_has_calls([call(1)])
mock_figures[1].test.assert_has_calls([call(2)])
mock_figures[2].test.assert_has_calls([call(3), call(4)])
self.assertTrue(
fig1_mock_manager.mock_calls[-1] ==
call.fig1_kept()) # assert final status of fig1 is kept
self.assertTrue(GlobalFigureManager._active_figure == 3)
def test_create_single_categorised_figure(self, mock_figure_class):
mock_figures = [mock.Mock()]
mock_figure_class.side_effect = mock_figures
category = '1d'
# The following line is equivalent to applying the decorator setcategory with the parameter category
# to function GlobalFigureManager.get_active_figure
categorised_get_active_figure = set_category(category)(
GlobalFigureManager.get_active_figure)
fig = categorised_get_active_figure()
self.assertTrue(
fig ==
mock_figures[0]) # Assert Figure object came from right place
self.assertTrue(GlobalFigureManager.get_category(1) == category)
self.assertTrue(
GlobalFigureManager.get_active_figure() ==
mock_figures[0]) # Check that it is set as the active figure
def test_create_categorised_figure_then_uncategorised_figure(
self, mock_figure_class):
mock_figures = [mock.Mock(), mock.Mock()]
mock_figure_class.side_effect = mock_figures
category = '1d'
categorised_get_active_figure = set_category(category)(
GlobalFigureManager.get_active_figure)
fig1 = categorised_get_active_figure()
fig2 = GlobalFigureManager.get_figure_number()
fig1_number = GlobalFigureManager.number_of_figure(fig1)
fig2_number = GlobalFigureManager.number_of_figure(fig2)
self.assertTrue(GlobalFigureManager.get_active_figure() == fig2)
self.assertTrue(
GlobalFigureManager.get_category(fig1_number) == category)
self.assertRaises(KeyError, GlobalFigureManager.get_category,
fig2_number)
self.assertTrue(fig1_number == 1 and fig2_number == 2)
def test_make_current_with_single_category(self, mock_figure_class):
mock_figures = [mock.Mock(), mock.Mock()]
# These manager is used to compare the relative order of calls of two different functions
mock_managers = [mock.Mock(), mock.Mock()]
for i in range(len(mock_figures)):
mock_managers[i].attach_mock(mock_figures[i].flag_as_kept,
'fig_kept')
mock_managers[i].attach_mock(mock_figures[i].flag_as_current,
'fig_current')
mock_figure_class.side_effect = mock_figures
cat1 = '1d'
cat1_get_active_figure = set_category(cat1)(
GlobalFigureManager.get_active_figure)
# test is an arbitrary method just to make sure the correct figures are returned
cat1_get_active_figure().test(1) # create a figure of category 1
GlobalFigureManager.set_figure_as_kept(
1) # now there is no active figure
cat1_get_active_figure().test(
2) # this command should go to a new figure
self.assertTrue(mock_managers[0].mock_calls[-1] ==
call.fig_kept()) # assert fig1 now displays kept
self.assertTrue(mock_managers[1].mock_calls[-1] ==
call.fig_current()) # assert fig2 now displays current
GlobalFigureManager.set_figure_as_current(1)
self.assertTrue(mock_managers[0].mock_calls[-1] ==
call.fig_current()) # assert fig1 now displays current
self.assertTrue(mock_managers[1].mock_calls[-1] ==
call.fig_kept()) # assert fig2 now displays kept
cat1_get_active_figure().test(3) # This should go to fig1
GlobalFigureManager.get_active_figure().test(
4) # This should go to fig1 as well
mock_figures[0].test.assert_has_calls([call(1), call(3), call(4)])
mock_figures[1].test.assert_has_calls([call(2)])
self.assertTrue(GlobalFigureManager._active_figure == 1)