Skip to content
Snippets Groups Projects
Commit 9d876b2d authored by Matthew Andrew's avatar Matthew Andrew
Browse files

Removed magic numbers Re #23642

parent 384697bc
No related branches found
No related tags found
No related merge requests found
...@@ -48,6 +48,8 @@ class GroupingTabPresenter(object): ...@@ -48,6 +48,8 @@ class GroupingTabPresenter(object):
self.guessAlphaObserver = GroupingTabPresenter.GuessAlphaObserver(self) self.guessAlphaObserver = GroupingTabPresenter.GuessAlphaObserver(self)
self.pairing_table_widget.guessAlphaNotifier.add_subscriber(self.guessAlphaObserver) self.pairing_table_widget.guessAlphaNotifier.add_subscriber(self.guessAlphaObserver)
self.update_thread = self.create_update_thread()
def show(self): def show(self):
self._view.show() self._view.show()
...@@ -129,8 +131,6 @@ class GroupingTabPresenter(object): ...@@ -129,8 +131,6 @@ class GroupingTabPresenter(object):
self._model.show_all_groups_and_pairs() self._model.show_all_groups_and_pairs()
def handle_update_all_clicked(self): def handle_update_all_clicked(self):
self.update_thread = self.create_update_thread()
self.update_thread.threadWrapperSetUp(self.disable_editing, self.update_thread.threadWrapperSetUp(self.disable_editing,
self.enable_editing, self.enable_editing,
self._view.display_warning_box) self._view.display_warning_box)
......
...@@ -67,10 +67,8 @@ class GroupingTabView(QtGui.QWidget): ...@@ -67,10 +67,8 @@ class GroupingTabView(QtGui.QWidget):
self.vertical_layout.setObjectName("verticalLayout") self.vertical_layout.setObjectName("verticalLayout")
self.vertical_layout.addItem(self.horizontal_layout) self.vertical_layout.addItem(self.horizontal_layout)
self.vertical_layout.addItem(self.horizontal_layout_description) self.vertical_layout.addItem(self.horizontal_layout_description)
if self._grouping_table: self.vertical_layout.addWidget(self._grouping_table)
self.vertical_layout.addWidget(self._grouping_table) self.vertical_layout.addWidget(self._pairing_table)
if self._pairing_table:
self.vertical_layout.addWidget(self._pairing_table)
self.vertical_layout.addItem(self.horizontal_layout_base) self.vertical_layout.addItem(self.horizontal_layout_base)
self.setLayout(self.vertical_layout) self.setLayout(self.vertical_layout)
......
...@@ -6,6 +6,7 @@ from Muon.GUI.Common.muon_pair import MuonPair ...@@ -6,6 +6,7 @@ from Muon.GUI.Common.muon_pair import MuonPair
from Muon.GUI.Common.utilities.run_string_utils import valid_name_regex, valid_alpha_regex from Muon.GUI.Common.utilities.run_string_utils import valid_name_regex, valid_alpha_regex
from Muon.GUI.Common.observer_pattern import Observable from Muon.GUI.Common.observer_pattern import Observable
pair_columns = ['pair_name', 'group_1', 'group_2', 'alpha']
class PairingTablePresenter(object): class PairingTablePresenter(object):
...@@ -54,15 +55,15 @@ class PairingTablePresenter(object): ...@@ -54,15 +55,15 @@ class PairingTablePresenter(object):
table = self._view.get_table_contents() table = self._view.get_table_contents()
changed_item = table[row][col] changed_item = table[row][col]
update_model = True update_model = True
if col == 0 and not self.validate_pair_name(changed_item): if pair_columns[col] == 'pair_name' and not self.validate_pair_name(changed_item):
update_model = False update_model = False
if col == 1: if pair_columns[col] == 'group_1':
if changed_item == self._view.get_table_item_text(row, 2): if changed_item == self._view.get_table_item_text(row, pair_columns.index('group_2')):
table[row][2] = self._model.pairs[row].forward_group table[row][pair_columns.index('group_2')] = self._model.pairs[row].forward_group
if col == 2: if pair_columns[col] == 'group_2':
if changed_item == self._view.get_table_item_text(row, 1): if changed_item == self._view.get_table_item_text(row, pair_columns.index('group_1')):
table[row][1] = self._model.pairs[row].backward_group table[row][pair_columns.index('group_1')] = self._model.pairs[row].backward_group
if col == 3: if pair_columns[col] == 'alpha':
if not self.validate_alpha(changed_item): if not self.validate_alpha(changed_item):
update_model = False update_model = False
else: else:
......
...@@ -68,6 +68,7 @@ class GroupingTabPresenterTest(unittest.TestCase): ...@@ -68,6 +68,7 @@ class GroupingTabPresenterTest(unittest.TestCase):
# TESTS # TESTS
# ------------------------------------------------------------------------------------------------------------------ # ------------------------------------------------------------------------------------------------------------------
def test_context_menu_add_pair_adds_pair_if_two_groups_selected(self): def test_context_menu_add_pair_adds_pair_if_two_groups_selected(self):
self.assertEqual(self.pairing_table_view.num_rows(), 2)
self.grouping_table_view._get_selected_row_indices = mock.Mock(return_value=[0, 1]) self.grouping_table_view._get_selected_row_indices = mock.Mock(return_value=[0, 1])
self.grouping_table_view.contextMenuEvent(0) self.grouping_table_view.contextMenuEvent(0)
self.grouping_table_view.add_pair_action.triggered.emit(True) self.grouping_table_view.add_pair_action.triggered.emit(True)
...@@ -153,11 +154,13 @@ class GroupingTabPresenterTest(unittest.TestCase): ...@@ -153,11 +154,13 @@ class GroupingTabPresenterTest(unittest.TestCase):
self.assertEqual(mock_save.call_args[0][-1], "grouping.xml") self.assertEqual(mock_save.call_args[0][-1], "grouping.xml")
def test_update_all_calculates_groups_and_pairs(self): def test_update_all_calculates_groups_and_pairs(self):
self.presenter.thread_manager = mock.MagicMock() self.presenter.update_thread = mock.MagicMock()
self.view.update_button.clicked.emit(True) self.view.update_button.clicked.emit(True)
self.presenter.thread_manager.process.assert_called_once_with(self.presenter.listener, self.presenter.update_thread.threadWrapperSetUp.assert_called_once_with(self.presenter.disable_editing,
self.presenter.calculate_all_data, 0, [1]) self.presenter.enable_editing,
self.view.display_warning_box)
self.presenter.update_thread.start.assert_called_once_with()
def test_removing_group_removes_linked_pairs(self): def test_removing_group_removes_linked_pairs(self):
self.add_three_groups() self.add_three_groups()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment