def test_get_pair_phasequad_name(self): AnalysisDataService.clear() ConfigService['MantidOptions.InvisibleWorkspaces'] = 'True' filepath = FileFinder.findRuns('EMU00019489.nxs')[0] load_result, run_number, filename, psi_data = load_workspace_from_filename( filepath) context = setup_context() context.gui_context.update({'RebinType': 'None'}) context.data_context.instrument = 'EMU' context.data_context._loaded_data.add_data(workspace=load_result, run=[run_number], filename=filename, instrument='EMU') context.data_context.current_runs = [[run_number]] context.data_context.update_current_data() self.assertEqual( "EMU19489; PhaseQuad; test_Re; MA", get_pair_phasequad_name(context, "test_Re", "19489", False)) self.assertEqual( "EMU19489; PhaseQuad; test_Re; Rebin; MA", get_pair_phasequad_name(context, "test_Re", "19489", True))
def execute(self): failed_files = [] for filename in self._filenames: try: ws, run, filename, psi_data = load_utils.load_workspace_from_filename( filename) except Exception as error: failed_files += [(filename, error)] continue if not psi_data: instrument_from_workspace = ws['OutputWorkspace'][ 0].workspace.getInstrument().getName() else: # Load another instrument first instrument_from_workspace = "PSI" self._loaded_data_store.remove_data(run=[run]) self._loaded_data_store.add_data( run=[run], workspace=ws, filename=filename, instrument=instrument_from_workspace) if failed_files: message = load_utils.exception_message_for_failed_files( failed_files) raise ValueError(message)
def setUp(self): self.context = setup_context(True) self.context.data_context.instrument = 'MUSR' self.context.data_context.num_periods = mock.Mock(return_value=1) self.context.gui_context.update({'RebinType': 'None'}) self.view = maxent_view.MaxEntView() self.presenter = maxent_presenter.MaxEntPresenter( self.view, self.context) file_path = FileFinder.findRuns('MUSR00022725.nxs')[0] ws, run, filename, _ = load_utils.load_workspace_from_filename( file_path) self.context.data_context._loaded_data.remove_data(run=run) self.context.data_context._loaded_data.add_data(run=[run], workspace=ws, filename=filename, instrument='MUSR') self.context.data_context.current_runs = [[22725]] self.context.update_current_data() test_pair = MuonPair('test_pair', 'top', 'bottom', alpha=0.75) self.context.group_pair_context.add_pair(pair=test_pair) self._count = 0 self.view.warning_popup = mock.MagicMock()
def setUp(self): AnalysisDataService.clear() ConfigService['MantidOptions.InvisibleWorkspaces'] = 'True' self.filepath = FileFinder.findRuns('EMU00019489.nxs')[0] self.load_result, self.run_number, self.filename, psi_data = load_workspace_from_filename( self.filepath) self.assert_(not psi_data) self.context = setup_context() self.context.gui_context.update({'RebinType': 'None'}) self.loaded_data = self.context.data_context._loaded_data self.data_context = self.context.data_context self.gui_context = self.context.gui_context self.group_pair_context = self.context.group_pair_context self.data_context.instrument = 'EMU' self.loaded_data.add_data(workspace=self.load_result, run=[self.run_number], filename=self.filename, instrument='EMU') self.data_context.current_runs = [[self.run_number]] self.data_context.update_current_data() self.group_pair_context.reset_group_and_pairs_to_default( self.load_result['OutputWorkspace'][0].workspace, 'EMU', '', 1) self.run_list = [19489] self.groups = [MuonGroup("bwd"), MuonGroup("fwd")] self.rebins = [False, False] self.pairs = [MuonPair("long", "bwd", "fwd")]
def test_runinfo_correct(self): file_path = FileFinder.findRuns('MUSR00022725.nxs')[0] ws, run, filename, _ = load_utils.load_workspace_from_filename( file_path) self.data_context._loaded_data.remove_data(run=run) self.data_context._loaded_data.add_data(run=[run], workspace=ws, filename=filename, instrument='MUSR') self.data_context.current_runs = [[22725]] self.context.update_current_data() test_pair = MuonPair('test_pair', 'top', 'bottom', alpha=0.75) self.group_context.add_pair(pair=test_pair) self.presenter.update_view_from_model() expected = [ call("Instrument : MUSR"), call("Run : 22725"), call("Title : FeTeSe T=1 F=100"), call("Comment : FC first sample"), call("Start : 2009-03-24T04:18:58"), call("End : 2009-03-24T04:56:26"), call("Counts (MEv) : 20.076704"), call("Good Frames : 88540"), call("Counts per Good Frame : 226.753"), call("Counts per Good Frame per det : 3.543"), call("Average Temperature (K) : 19.69992"), call("Sample Temperature (K) : 1.0"), call("Sample Magnetic Field (G) : 100.0"), call("Number of DAQ Periods : 1") ] self.assertEqual(self.view.add_text_line.call_args_list, expected)
def test_load_workspace_from_filename_for_existing_file(self): ConfigService.Instance().setString("default.facility", "ISIS") filename = 'MUSR00022725.nsx' load_result, run, filename, _ = utils.load_workspace_from_filename(filename) self.assertEqual(load_result['DeadTimeTable'], None) self.assertEqual(load_result['FirstGoodData'], 0.106) self.assertEqual(load_result['MainFieldDirection'], 'Transverse') self.assertAlmostEqual(load_result['TimeZero'], 0.55000, 5) self.assertEqual(run, 22725) ConfigService.Instance().setString("default.facility", " ")
def setUp(self): self.filepath = FileFinder.findRuns('EMU00019489.nxs')[0] self.load_result, self.run_number, self.filename, _ = load_workspace_from_filename( self.filepath) self.loaded_data = MuonLoadData() self.context = MuonDataContext(load_data=self.loaded_data) self.context.instrument = 'EMU' self.loaded_data.add_data(workspace=self.load_result, run=[self.run_number], filename=self.filename, instrument='EMU') self.context.current_runs = [[self.run_number]] self.context.update_current_data()
def setUp(self): self.filepath = FileFinder.findRuns('EMU00019489.nxs')[0] self.load_result, self.run_number, self.filename, _ = load_workspace_from_filename(self.filepath) self.context = setup_context() self.context.gui_context.update({'RebinType': 'None'}) self.model = CorrectionsModel(self.context) self.runs = [[84447], [84448], [84449]] self.coadd_runs = [[84447, 84448, 84449]] self.context.data_context.instrument = 'EMU' self.context.data_context._loaded_data.add_data(workspace=self.load_result, run=[self.run_number], filename=self.filename, instrument='EMU') self.context.data_context.current_runs = [[self.run_number]] self.context.data_context.update_current_data() self.context.group_pair_context.reset_group_and_pairs_to_default( self.load_result['OutputWorkspace'][0].workspace, 'EMU', '', 1)
def perform_musr_file_finder(self): ConfigService['default.instrument'] = 'MUSR' file_path = FileFinder.findRuns('MUSR00022725.nxs')[0] ws, run, filename, psi_data = load_utils.load_workspace_from_filename( file_path) self.assert_(not psi_data) self.data_context._loaded_data.remove_data(run=run) self.data_context._loaded_data.add_data(run=[run], workspace=ws, filename=filename, instrument='MUSR') self.data_context.current_runs = [[22725]] self.context.data_context._instrument = "MUSR" self.context.update_current_data() test_pair = MuonPair('test_pair', 'top', 'bottom', alpha=0.75) self.group_context.add_pair(pair=test_pair) self.presenter.update_view_from_model()
def execute(self): failed_files = [] for filename in self._filenames: try: ws, run, filename, _ = load_utils.load_workspace_from_filename( filename) except ValueError as error: failed_files += [(filename, error)] continue self._loaded_data_store.remove_data(run=[run]) self._loaded_data_store.add_data( run=[run], workspace=ws, filename=filename, instrument=self._data_context.instrument) if failed_files: message = "The requested run could not be found. This could be due to: \n - The run does not yet exist." \ "\n - The file was not found locally (please check the user directories)." raise ValueError(message)
def setUp(self): self.context = setup_context(True) self.context.data_context.instrument = 'MUSR' self.context.gui_context.update({'RebinType': 'None'}) self.view = fft_view.FFTView() self.model1 = fft_model.FFTModel() self.model = fft_model.FFTWrapper self.run_list = [22725] self.groups = [MuonGroup(group) for group in GROUP_LIST] self.rebins = [False] * len(self.groups) self.pairs = [MuonPair(EXAMPLE_PAIR, 'top', 'bottom', alpha=0.75)] self.presenter = fft_presenter.FFTPresenter(self.view, self.model, self.context) file_path = FileFinder.findRuns('MUSR00022725.nxs')[0] ws, run, filename, _ = load_utils.load_workspace_from_filename( file_path) self.context.data_context._loaded_data.remove_data(run=run) self.context.data_context._loaded_data.add_data(run=[run], workspace=ws, filename=filename, instrument='MUSR') self.context.data_context.current_runs = [[22725]] self.context.update_current_data() self.context.group_pair_context.add_pair(pair=self.pairs[0]) self._calculate_all_data() self.context.group_pair_context._selected_groups = GROUP_LIST self.context.group_pair_context._selected_pairs = [EXAMPLE_PAIR] self.view.warning_popup = mock.MagicMock()