def setUp(self): self.load = mock.create_autospec(load_utils.LoadUtils, spec_set=True) self.load.getCurrentWS = mock.Mock( return_value=["TEST00000001", ["fwd", "bkwd"]]) self.load.getRunName = mock.MagicMock(return_value="MUSR00023456") self.load.hasDataChanged = mock.MagicMock(return_value=False) self.view = mock.create_autospec(fft_view.FFTView, spec_set=True) # signals self.view.tableClickSignal = mock.Mock(return_value=[3, 1]) self.view.phaseCheckSignal = mock.Mock(return_value=True) # needed for connect in presenter self.view.buttonSignal = mock.Mock() self.view.changed = mock.MagicMock() self.view.changedHideUnTick = mock.MagicMock() self.view.initFFTInput = mock.Mock(return_value={ "InputWorkspace": "testWS", "OutputWorkspace": "muon" }) self.view.addFFTComplex = mock.Mock( return_value={"InputImWorkspace": "MuonFFT"}) self.view.addFFTShift = mock.Mock() self.view.addRaw = mock.Mock() self.view.RePhaseAdvanced = mock.Mock() self.view.getFFTRePhase = mock.Mock() self.view.getFFTImPhase = mock.Mock() self.view.getWS = mock.Mock(return_value="MUSR00023456") self.view.getFirstGoodData = mock.Mock(return_value=0.1) self.view.getLastGoodData = mock.Mock(return_value=15.) self.view.getImBoxRow = mock.Mock(return_value=3) self.view.getShiftBoxRow = mock.Mock(return_value=5) self.view.isRaw = mock.Mock(return_value=True) self.view.isComplex = mock.Mock(return_value=True) self.view.isAutoShift = mock.Mock(return_value=True) self.view.setPhaseBox = mock.Mock() self.view.isNewPhaseTable = mock.Mock(return_value=True) self.view.activateButton = mock.Mock() self.view.deactivateButton = mock.Mock() # setup model self.model1 = mock.create_autospec(fft_model.FFTModel, spec_set=False) self.model = mock.create_autospec(fft_model.FFTWrapper, spec_set=False) # set presenter self.presenter = fft_presenter.FFTPresenter(self.view, self.model, self.load) # mock thread self.thread = mock.create_autospec(thread_model.ThreadModel) self.thread.start = mock.Mock() self.thread.started = mock.Mock() self.thread.finished = mock.Mock() self.thread.setInputs = mock.Mock() self.thread.loadData = mock.Mock() self.presenter.createThread = mock.Mock(return_value=self.thread)
def __init__(self, view, model): self.view = view self.load = load_utils.LoadUtils() if not self.load.MuonAnalysisExists(): return # create presenters for the views self.FFTPresenter = fft_presenter.FFTPresenter( self.view.getView("FFT"), model.getModel("FFT"), self.load) self.MaxEntPresenter = maxent_presenter.MaxEntPresenter( self.view.getView("MaxEnt"), model.getModel("MaxEnt"), self.load) # get the transform selection view self.selectionPresenter = transform_selection_presenter.TransformSelectionPresenter( self.view.selection) # gets a list of the views/methods self.methodsList = self.view.getMethods() # sets the transform selection view to have the correct options self.selectionPresenter.setMethodsCombo(self.methodsList) # connect self.view.selection.changeMethodSignal.connect(self.updateDisplay)