class MuonContextTest(unittest.TestCase):
@classmethod
def setUpClass(cls):
pass
def setUp(self):
AnalysisDataService.clear()
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.data_context = MuonDataContext(self.loaded_data)
self.gui_context = MuonGuiContext()
self.group_pair_context = MuonGroupPairContext()
self.gui_context.update({'RebinType': 'None'})
self.context = MuonContext(muon_data_context=self.data_context,
muon_gui_context=self.gui_context,
muon_group_context=self.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', '')
def populate_ADS(self):
self.context.calculate_all_groups()
self.context.show_all_groups()
self.context.calculate_all_pairs()
self.context.show_all_pairs()
workspace = CreateWorkspace([0], [0], StoreInADS=False)
self.context.phase_context.add_phase_quad(
MuonWorkspaceWrapper(workspace,
'EMU19489; PhaseQuad; PhaseTable EMU19489'))
def test_reset_groups_and_pairs_to_default(self):
self.assertEquals(self.group_pair_context.group_names, ['fwd', 'bwd'])
self.assertEquals(self.group_pair_context.pair_names, ['long'])
def test_calculate_group_calculates_group_for_given_run(self):
counts_workspace, asymmetry_workspace = self.context.calculate_group(
'fwd', run=[19489])
self.assertEquals(type(counts_workspace), Workspace2D)
self.assertEquals(type(counts_workspace), Workspace2D)
def test_calculate_pair_calculates_pair_for_given_run(self):
pair_asymmetry = self.context.calculate_pair('long', run=[19489])
self.assertEquals(type(pair_asymmetry), Workspace2D)
def test_show_all_groups_calculates_and_shows_all_groups(self):
self.context.show_all_groups()
self.assertEquals(AnalysisDataService.getObjectNames(), [
'EMU19489', 'EMU19489 Groups',
'EMU19489; Group; bwd; Asymmetry; #1',
'EMU19489; Group; bwd; Counts; #1',
'EMU19489; Group; fwd; Asymmetry; #1',
'EMU19489; Group; fwd; Counts; #1', 'Muon Data'
])
def test_that_show_all_calculates_and_shows_all_groups_with_rebin(self):
self.gui_context['RebinType'] = 'Fixed'
self.gui_context['RebinFixed'] = 2
self.context.show_all_groups()
self.assertEquals(AnalysisDataService.getObjectNames(), [
'EMU19489', 'EMU19489 Groups',
'EMU19489; Group; bwd; Asymmetry; #1',
'EMU19489; Group; bwd; Asymmetry; Rebin; #1',
'EMU19489; Group; bwd; Counts; #1',
'EMU19489; Group; bwd; Counts; Rebin; #1',
'EMU19489; Group; fwd; Asymmetry; #1',
'EMU19489; Group; fwd; Asymmetry; Rebin; #1',
'EMU19489; Group; fwd; Counts; #1',
'EMU19489; Group; fwd; Counts; Rebin; #1', 'Muon Data'
])
def test_show_all_pairs_calculates_and_shows_all_pairs(self):
self.context.show_all_pairs()
self.assertEquals(AnalysisDataService.getObjectNames(), [
'EMU19489', 'EMU19489 Pairs', 'EMU19489; Pair Asym; long; #1',
'Muon Data'
])
def test_that_show_all_calculates_and_shows_all_pairs_with_rebin(self):
self.gui_context['RebinType'] = 'Fixed'
self.gui_context['RebinFixed'] = 2
self.context.show_all_pairs()
self.assertEquals(AnalysisDataService.getObjectNames(), [
'EMU19489', 'EMU19489 Pairs', 'EMU19489; Pair Asym; long; #1',
'EMU19489; Pair Asym; long; Rebin; #1', 'Muon Data'
])
def test_update_current_data_sets_current_run_in_data_context(self):
self.context.update_current_data()
self.assertEquals(self.data_context.current_data, self.load_result)
def test_update_current_data_sets_groups_and_pairs(self):
self.context.update_current_data()
self.assertEquals(self.group_pair_context.pair_names, ['long'])
self.assertEquals(self.group_pair_context.group_names, ['fwd', 'bwd'])
def test_show_raw_data_puts_raw_data_into_the_ADS(self):
self.context.show_raw_data()
self.assertEquals(AnalysisDataService.getObjectNames(), [
'EMU19489', 'EMU19489 Raw Data', 'EMU19489_raw_data', 'Muon Data'
])
def test_that_first_good_data_returns_correctly_when_from_file_chosen_option(
self):
self.gui_context.update({'FirstGoodDataFromFile': True})
first_good_data = self.context.first_good_data([19489])
self.assertEquals(first_good_data, 0.11)
def test_first_good_data_returns_correctly_when_manually_specified_used(
self):
self.gui_context.update({
'FirstGoodDataFromFile': False,
'FirstGoodData': 5
})
first_good_data = self.context.first_good_data([19489])
self.assertEquals(first_good_data, 5)
def test_that_last_good_data_returns_correctly_when_from_file_chosen_option(
self):
self.gui_context.update({'LastGoodDataFromFile': True})
last_good_data = self.context.last_good_data([19489])
self.assertEquals(last_good_data, 31.76)
def test_last_good_data_returns_correctly_when_manually_specified_used(
self):
self.gui_context.update({
'LastGoodDataFromFile': False,
'LastGoodData': 5
})
last_good_data = self.context.last_good_data([19489])
self.assertEquals(last_good_data, 5)
def test_that_dead_time_table_from_ADS_returns_table_name(self):
self.gui_context.update({
'DeadTimeSource': 'FromADS',
'DeadTimeTable': 'deadtime_table_name'
})
deadtime_table = self.context.dead_time_table([19489])
self.assertEquals(deadtime_table, 'deadtime_table_name')
def test_get_workspace_names_returns_all_stored_workspaces_if_all_selected(
self):
self.populate_ADS()
workspace_list = self.context.get_names_of_workspaces_to_fit(
'19489', 'fwd, bwd, long', True)
self.assertEqual(workspace_list, [
'EMU19489; Group; fwd; Asymmetry; #1',
'EMU19489; Group; bwd; Asymmetry; #1',
'EMU19489; Pair Asym; long; #1',
'EMU19489; PhaseQuad; PhaseTable EMU19489'
])
def test_get_workspace_names_returns_nothing_if_no_parameters_passed(self):
self.populate_ADS()
workspace_list = self.context.get_names_of_workspaces_to_fit()
self.assertEqual(workspace_list, [])
def test_get_workspaces_names_copes_with_bad_groups(self):
self.populate_ADS()
workspace_list = self.context.get_names_of_workspaces_to_fit(
'19489', 'fwd, bwd, long, random, wrong', True)
self.assertEqual(workspace_list, [
'EMU19489; Group; fwd; Asymmetry; #1',
'EMU19489; Group; bwd; Asymmetry; #1',
'EMU19489; Pair Asym; long; #1',
'EMU19489; PhaseQuad; PhaseTable EMU19489'
])
def test_get_workspaces_names_copes_with_non_existent_runs(self):
self.populate_ADS()
workspace_list = self.context.get_names_of_workspaces_to_fit(
'19489, 22222', 'fwd, bwd, long', True)
self.assertEqual(workspace_list, [
'EMU19489; Group; fwd; Asymmetry; #1',
'EMU19489; Group; bwd; Asymmetry; #1',
'EMU19489; Pair Asym; long; #1',
'EMU19489; PhaseQuad; PhaseTable EMU19489'
])
def test_that_run_ranged_correctly_parsed(self):
self.populate_ADS()
workspace_list = self.context.get_names_of_workspaces_to_fit(
'19489-95', 'fwd, bwd, long', True)
self.assertEqual(workspace_list, [
'EMU19489; Group; fwd; Asymmetry; #1',
'EMU19489; Group; bwd; Asymmetry; #1',
'EMU19489; Pair Asym; long; #1',
'EMU19489; PhaseQuad; PhaseTable EMU19489'
])
class MuonGroupPairContextTest(unittest.TestCase):
def setUp(self):
self.context = MuonGroupPairContext()
def test_can_be_created(self):
self.assertTrue(self.context)
def test_groups_and_pairs_initially_empty(self):
self.assertEqual(self.context.groups, [])
self.assertEqual(self.context.pairs, [])
self.assertEqual(self.context.diffs, [])
def test_group_can_be_added(self):
group = MuonGroup('group_1', [1, 3, 5, 7, 9])
self.context.add_group(group)
self.assertEqual(self.context['group_1'], group)
def test_non_group_cannot_be_added(self):
pair = MuonPair('pair_1')
diff = MuonDiff('diff_1', 'positive', 'negative')
self.assertRaises(AssertionError, self.context.add_group, pair)
self.assertRaises(AssertionError, self.context.add_group, diff)
def test_cannot_add_group_with_duplicate_name(self):
group_1 = MuonGroup('group_1', [1, 3, 5, 7, 9])
group_2 = MuonGroup('group_1', [1, 3, 5, 7, 9])
self.context.add_group(group_1)
self.assertRaises(ValueError, self.context.add_group, group_2)
def test_pair_can_be_added(self):
pair = MuonPair('pair_1')
self.context.add_pair(pair)
self.assertEqual(self.context['pair_1'], pair)
def test_non_pair_cannot_be_added(self):
group = MuonGroup('group_1', [1, 3, 5, 7, 9])
diff = MuonDiff('diff_1', 'positive', 'negative')
# Value error as cannot assert isinstance in code since could add MuonPair or MuonBasePair
self.assertRaises(ValueError, self.context.add_pair, group)
self.assertRaises(ValueError, self.context.add_pair, diff)
def test_cannot_add_pair_with_duplicate_name(self):
pair_1 = MuonPair('pair')
pair_2 = MuonPair('pair')
self.context.add_pair(pair_1)
self.assertRaises(ValueError, self.context.add_pair, pair_2)
def test_diff_can_be_added(self):
diff = MuonDiff('diff_1', 'positive', 'negative')
self.context.add_diff(diff)
self.assertEqual(self.context['diff_1'], diff)
def test_non_diff_cannot_be_added(self):
group = MuonGroup('group_1', [1, 3, 5, 7, 9])
pair = MuonPair('pair_1')
self.assertRaises(AssertionError, self.context.add_diff, group)
self.assertRaises(AssertionError, self.context.add_diff, pair)
def test_cannot_add_diff_with_duplicate_name(self):
diff_1 = MuonDiff('diff', 'positive', 'negative')
diff_2 = MuonDiff('diff', 'positive', 'negative')
self.context.add_diff(diff_1)
self.assertRaises(ValueError, self.context.add_diff, diff_2)
def test_groups_pairs_diffs_should_all_have_unique_names(self):
group_1 = MuonGroup('group', [1, 3, 5, 7, 9])
group_2 = MuonGroup('pair', [1, 3, 5, 7, 9])
group_3 = MuonGroup('diff', [1, 3, 5, 7, 9])
pair_1 = MuonPair('pair')
pair_2 = MuonPair('group')
pair_3 = MuonPair('diff')
diff_1 = MuonDiff('diff', 'positive', 'negative')
diff_2 = MuonDiff('group', 'positive', 'negative')
diff_3 = MuonDiff('pair', 'positive', 'negative')
# Add correct group, pair and diff
self.context.add_group(group_1)
self.context.add_pair(pair_1)
self.context.add_diff(diff_1)
# Now check cannot duplicate names
self.assertRaises(ValueError, self.context.add_group, group_2)
self.assertRaises(ValueError, self.context.add_group, group_3)
self.assertRaises(ValueError, self.context.add_pair, pair_2)
self.assertRaises(ValueError, self.context.add_pair, pair_3)
self.assertRaises(ValueError, self.context.add_diff, diff_2)
self.assertRaises(ValueError, self.context.add_diff, diff_3)
def test_group_names_returns_ordered_list_of_names(self):
group_1 = MuonGroup('group_1', [1, 3, 5, 7, 9])
group_2 = MuonGroup('group_2', [1, 3, 4, 7, 9])
group_3 = MuonGroup('group_3', [1, 3, 4, 7, 9])
self.context.add_group(group_1)
self.context.add_group(group_3)
self.context.add_group(group_2)
self.assertEqual(self.context.group_names,
['group_1', 'group_3', 'group_2'])
def test_pair_names_returns_ordered_list_of_names(self):
pair_1 = MuonPair('pair_1')
pair_2 = MuonPair('pair_2')
pair_3 = MuonPair('pair_3')
self.context.add_pair(pair_1)
self.context.add_pair(pair_2)
self.context.add_pair(pair_3)
self.assertEqual(self.context.pair_names,
['pair_1', 'pair_2', 'pair_3'])
def test_diff_names_returns_ordered_list_of_names(self):
diff_1 = MuonDiff('diff_1', 'positive', 'negative')
diff_2 = MuonDiff('diff_2', 'positive', 'negative')
diff_3 = MuonDiff('diff_3', 'positive', 'negative')
self.context.add_diff(diff_1)
self.context.add_diff(diff_2)
self.context.add_diff(diff_3)
self.assertEqual(self.context.diff_names,
['diff_1', 'diff_2', 'diff_3'])
def test_can_remove_groups_as_expected(self):
group_1 = MuonGroup('group_1', [1, 3, 5, 7, 9])
group_2 = MuonGroup('group_2', [1, 3, 4, 7, 9])
group_3 = MuonGroup('group_3', [1, 3, 4, 7, 9])
self.context.add_group(group_1)
self.context.add_group(group_2)
self.context.add_group(group_3)
self.context.remove_group('group_1')
self.assertEqual(self.context.group_names, ['group_2', 'group_3'])
def test_can_remove_pairs_as_expected(self):
pair_1 = MuonPair('pair_1')
pair_2 = MuonPair('pair_2')
pair_3 = MuonPair('pair_3')
self.context.add_pair(pair_1)
self.context.add_pair(pair_2)
self.context.add_pair(pair_3)
self.context.remove_pair('pair_2')
self.assertEqual(self.context.pair_names, ['pair_1', 'pair_3'])
def test_can_remove_diffs_as_expected(self):
diff_1 = MuonDiff('diff_1', 'positive', 'negative')
diff_2 = MuonDiff('diff_2', 'positive', 'negative')
diff_3 = MuonDiff('diff_3', 'positive', 'negative')
self.context.add_diff(diff_1)
self.context.add_diff(diff_2)
self.context.add_diff(diff_3)
self.context.remove_diff('diff_2')
self.assertEqual(self.context.diff_names, ['diff_1', 'diff_3'])
def test_get_group_workspace_names_returns_correct_workspace_names(self):
group = create_group_populated_by_two_workspace()
self.context.add_group(group)
workspace_list = self.context.get_group_workspace_names([[33333]],
['group1'],
False)
self.assertEqual(workspace_list, ['asymmetry_name_33333'])
def test_that_reset_to_default_groups_creates_correct_groups_and_pairs_for_single_period_data(
self):
workspace = CreateSampleWorkspace()
LoadInstrument(workspace, InstrumentName="EMU", RewriteSpectraMap=True)
self.context.reset_group_and_pairs_to_default(workspace, 'EMU',
'longitudanal', 1)
self.assertEquals(self.context.group_names, ['fwd', 'bwd'])
self.assertEquals(self.context.pair_names, ['long'])
self.assertEquals(self.context.diff_names, [])
for group in self.context.groups:
self.assertEquals(group.periods, [1])
def test_that_reset_to_default_groups_creates_correct_groups_and_pairs_for_multi_period_data(
self):
workspace = CreateSampleWorkspace()
LoadInstrument(workspace, InstrumentName="EMU", RewriteSpectraMap=True)
self.context.reset_group_and_pairs_to_default(workspace, 'EMU',
'longitudanal', 2)
self.assertEquals(self.context.group_names,
['fwd1', 'bwd1', 'fwd2', 'bwd2'])
self.assertEquals(self.context.pair_names, ['long1', 'long2'])
self.assertEquals(self.context.diff_names, ['pair_diff1'])
self.assertEquals(self.context.groups[0].periods, [1])
self.assertEquals(self.context.groups[1].periods, [1])
self.assertEquals(self.context.groups[2].periods, [2])
self.assertEquals(self.context.pairs[0].forward_group, 'fwd1')
self.assertEquals(self.context.pairs[0].backward_group, 'bwd1')
self.assertEquals(self.context.pairs[1].forward_group, 'fwd2')
self.assertEquals(self.context.pairs[1].backward_group, 'bwd2')
self.assertEquals(self.context.diffs[0].positive, 'long1')
self.assertEquals(self.context.diffs[0].negative, 'long2')
self.assertEquals(self.context.selected, 'long1')
def test_get_group_pair_name_and_run_from_workspace_name(self):
group_1 = MuonGroup('group_1', [1, 3, 5, 7, 9])
group_2 = MuonGroup('group_2', [1, 3, 4, 7, 9])
group_3 = MuonGroup('group_3', [1, 3, 4, 7, 9])
self.context.add_group(group_1)
self.context.add_group(group_2)
self.context.add_group(group_3)
group_1.update_counts_workspace(MuonRun([62260]), 'group_1_counts',
False)
group_1.update_asymmetry_workspace(MuonRun([62260]), 'group_1_asym',
'group_1_asym_unorm', False)
workspace_name_list = self.context.get_group_workspace_names(
runs=[[62260]], groups=['group_1'], rebin=False)
group_name, run = self.context.get_group_pair_name_and_run_from_workspace_name(
workspace_name_list[0])
self.assertEqual(group_name, 'group_1')
self.assertEqual(run, '62260')
def test_get_group_pair_name_and_run_works_for_co_added_runs(self):
group_1 = MuonGroup('group_1', [1, 3, 5, 7, 9])
group_2 = MuonGroup('group_2', [1, 3, 4, 7, 9])
group_3 = MuonGroup('group_3', [1, 3, 4, 7, 9])
self.context.add_group(group_1)
self.context.add_group(group_2)
self.context.add_group(group_3)
group_1.update_counts_workspace(MuonRun([62260, 62261]),
'group_1_counts', False)
group_1.update_asymmetry_workspace(MuonRun([62260,
62261]), 'group_1_asym',
'group_1_asym_unorm', False)
workspace_name_list = self.context.get_group_workspace_names(
runs=[[62260, 62261]], groups=['group_1'], rebin=False)
group_name, run = self.context.get_group_pair_name_and_run_from_workspace_name(
workspace_name_list[0])
self.assertEqual(group_name, 'group_1')
self.assertEqual(run, '62260-62261')
def test_that_get_group_pair_name_and_run_works_for_fit_workspace_names_containing_original_worspace(
self):
group_1 = MuonGroup('group_1', [1, 3, 5, 7, 9])
group_2 = MuonGroup('group_2', [1, 3, 4, 7, 9])
group_3 = MuonGroup('group_3', [1, 3, 4, 7, 9])
self.context.add_group(group_1)
self.context.add_group(group_2)
self.context.add_group(group_3)
group_1.update_counts_workspace(MuonRun([62260, 62261]),
'group_1_counts', False)
group_1.update_asymmetry_workspace(MuonRun([62260,
62261]), 'group_1_asym',
'group_1_asym_unorm', False)
workspace_name_list = self.context.get_group_workspace_names(
runs=[[62260, 62261]], groups=['group_1'], rebin=False)
group_name, run = self.context.get_group_pair_name_and_run_from_workspace_name(
workspace_name_list[0] + '; Fit Seq Flatbackground')
self.assertEqual(group_name, 'group_1')
self.assertEqual(run, '62260-62261')
def test_add_phasequad(self):
phasequad = MuonPhasequad("test", "table")
self.assertEqual(len(self.context._phasequad), 0)
self.assertEqual(len(self.context._pairs), 0)
self.context.add_phasequad(phasequad)
self.assertEqual(len(self.context._phasequad), 1)
self.assertEqual(len(self.context._pairs), 2)
self.assertEqual(self.context._phasequad[0].name, "test")
self.assertEqual(self.context._pairs[0].name, "test_Re_")
self.assertEqual(self.context._pairs[1].name, "test_Im_")
def test_rm_phasequad(self):
phasequad = MuonPhasequad("test", "table")
phasequad2 = MuonPhasequad("test2", "table2")
self.context.add_phasequad(phasequad)
self.context.add_phasequad(phasequad2)
self.assertEqual(len(self.context._phasequad), 2)
self.assertEqual(len(self.context._pairs), 4)
self.context.remove_phasequad(phasequad)
self.assertEqual(len(self.context._phasequad), 1)
self.assertEqual(len(self.context._pairs), 2)
self.assertEqual(self.context._phasequad[0].name, "test2")
self.assertEqual(self.context._pairs[0].name, "test2_Re_")
self.assertEqual(self.context._pairs[1].name, "test2_Im_")
class MuonGroupPairContextTest(unittest.TestCase):
def setUp(self):
self.context = MuonGroupPairContext()
def test_can_be_created(self):
self.assertTrue(self.context)
def test_groups_and_pairs_initially_empty(self):
self.assertEqual(self.context.groups, [])
self.assertEqual(self.context.pairs, [])
def test_group_can_be_added(self):
group = MuonGroup('group_1', [1,3,5,7,9])
self.context.add_group(group)
self.assertEqual(self.context['group_1'], group)
def test_non_group_cannot_be_added(self):
pair = MuonPair('pair_1')
self.assertRaises(AssertionError, self.context.add_group, pair)
def test_cannot_add_group_with_duplicate_name(self):
group_1 = MuonGroup('group_1', [1, 3, 5, 7, 9])
group_2 = MuonGroup('group_1', [1, 3, 5, 7, 9])
self.context.add_group(group_1)
self.assertRaises(ValueError, self.context.add_group, group_2)
def test_pair_can_be_added(self):
pair = MuonPair('pair_1')
self.context.add_pair(pair)
self.assertEqual(self.context['pair_1'], pair)
def test_non_pair_cannot_be_added(self):
pair = MuonPair('pair_1')
self.assertRaises(AssertionError, self.context.add_group, pair)
def test_cannot_add_pair_with_duplicate_name(self):
pair_1 = MuonPair('pair')
pair_2 = MuonPair('pair')
self.context.add_pair(pair_1)
self.assertRaises(ValueError, self.context.add_pair, pair_2)
def test_group_names_returns_ordered_list_of_names(self):
group_1 = MuonGroup('group_1', [1, 3, 5, 7, 9])
group_2 = MuonGroup('group_2', [1, 3, 4, 7, 9])
group_3 = MuonGroup('group_3', [1, 3, 4, 7, 9])
self.context.add_group(group_1)
self.context.add_group(group_3)
self.context.add_group(group_2)
self.assertEqual(self.context.group_names, ['group_1', 'group_3', 'group_2'])
def test_pair_names_returns_ordered_list_of_names(self):
pair_1 = MuonPair('pair_1')
pair_2 = MuonPair('pair_2')
pair_3 = MuonPair('pair_3')
self.context.add_pair(pair_1)
self.context.add_pair(pair_2)
self.context.add_pair(pair_3)
self.assertEqual(self.context.pair_names, ['pair_1', 'pair_2', 'pair_3'])
def test_can_remove_groups_as_expected(self):
group_1 = MuonGroup('group_1', [1, 3, 5, 7, 9])
group_2 = MuonGroup('group_2', [1, 3, 4, 7, 9])
group_3 = MuonGroup('group_3', [1, 3, 4, 7, 9])
self.context.add_group(group_1)
self.context.add_group(group_2)
self.context.add_group(group_3)
self.context.remove_group('group_1')
self.assertEqual(self.context.group_names, ['group_2', 'group_3'])
def test_can_remove_pairs_as_expected(self):
pair_1 = MuonPair('pair_1')
pair_2 = MuonPair('pair_2')
pair_3 = MuonPair('pair_3')
self.context.add_pair(pair_1)
self.context.add_pair(pair_2)
self.context.add_pair(pair_3)
self.context.remove_pair('pair_2')
self.assertEqual(self.context.pair_names, ['pair_1', 'pair_3'])
def test_get_group_workspace_names_returns_correct_workspace_names(self):
group = create_group_populated_by_two_workspace()
self.context.add_group(group)
workspace_list = self.context.get_group_workspace_names([[33333]], ['group1'], False)
self.assertEqual(workspace_list, ['asymmetry_name_33333'])
def test_that_reset_to_default_groups_creates_correct_groups_and_pairs_for_single_period_data(self):
workspace = CreateSampleWorkspace()
LoadInstrument(workspace, InstrumentName="EMU", RewriteSpectraMap=True)
self.context.reset_group_and_pairs_to_default(workspace, 'EMU', 'longitudanal', 1)
self.assertEquals(self.context.group_names, ['fwd', 'bwd'])
self.assertEquals(self.context.pair_names, ['long'])
for group in self.context.groups:
self.assertEquals(group.periods, [1])
def test_that_reset_to_default_groups_creates_correct_groups_and_pairs_for_multi_period_data(self):
workspace = CreateSampleWorkspace()
LoadInstrument(workspace, InstrumentName="EMU", RewriteSpectraMap=True)
self.context.reset_group_and_pairs_to_default(workspace, 'EMU', 'longitudanal', 2)
self.assertEquals(self.context.group_names, ['fwd1', 'bwd1', 'fwd2', 'bwd2'])
self.assertEquals(self.context.pair_names, ['long1', 'long2'])
self.assertEquals(self.context.groups[0].periods, [1])
self.assertEquals(self.context.groups[1].periods, [1])
self.assertEquals(self.context.groups[2].periods, [2])
self.assertEquals(self.context.pairs[0].forward_group, 'fwd1')
self.assertEquals(self.context.pairs[0].backward_group, 'bwd1')
self.assertEquals(self.context.pairs[1].forward_group, 'fwd2')
self.assertEquals(self.context.pairs[1].backward_group, 'bwd2')
self.assertEquals(self.context.selected, 'long1')
def test_get_group_pair_name_and_run_from_workspace_name(self):
group_1 = MuonGroup('group_1', [1, 3, 5, 7, 9])
group_2 = MuonGroup('group_2', [1, 3, 4, 7, 9])
group_3 = MuonGroup('group_3', [1, 3, 4, 7, 9])
self.context.add_group(group_1)
self.context.add_group(group_2)
self.context.add_group(group_3)
group_1.update_workspaces([62260], 'group_1_counts', 'group_1_asym', 'group_1_asym_unorm', False)
workspace_name_list = self.context.get_group_workspace_names(runs = [[62260]], groups=['group_1'], rebin=False)
group_name, run = self.context.get_group_pair_name_and_run_from_workspace_name(workspace_name_list[0])
self.assertEqual(group_name, 'group_1')
self.assertEqual(run, '62260')
def test_get_group_pair_name_and_run_works_for_co_added_runs(self):
group_1 = MuonGroup('group_1', [1, 3, 5, 7, 9])
group_2 = MuonGroup('group_2', [1, 3, 4, 7, 9])
group_3 = MuonGroup('group_3', [1, 3, 4, 7, 9])
self.context.add_group(group_1)
self.context.add_group(group_2)
self.context.add_group(group_3)
group_1.update_workspaces([62260, 62261], 'group_1_counts', 'group_1_asym', 'group_1_asym_unorm', False)
workspace_name_list = self.context.get_group_workspace_names(runs = [[62260, 62261]], groups=['group_1'], rebin=False)
group_name, run = self.context.get_group_pair_name_and_run_from_workspace_name(workspace_name_list[0])
self.assertEqual(group_name, 'group_1')
self.assertEqual(run, '62260-62261')
def test_that_get_group_pair_name_and_run_works_for_fit_workspace_names_containing_original_worspace(self):
group_1 = MuonGroup('group_1', [1, 3, 5, 7, 9])
group_2 = MuonGroup('group_2', [1, 3, 4, 7, 9])
group_3 = MuonGroup('group_3', [1, 3, 4, 7, 9])
self.context.add_group(group_1)
self.context.add_group(group_2)
self.context.add_group(group_3)
group_1.update_workspaces([62260, 62261], 'group_1_counts', 'group_1_asym', 'group_1_asym_unorm', False)
workspace_name_list = self.context.get_group_workspace_names(runs = [[62260, 62261]], groups=['group_1'], rebin=False)
group_name, run = self.context.get_group_pair_name_and_run_from_workspace_name(workspace_name_list[0] + '; Fit Seq Flatbackground')
self.assertEqual(group_name, 'group_1')
self.assertEqual(run, '62260-62261')
class MuonContextTest(unittest.TestCase):
@classmethod
def setUpClass(cls):
pass
def setUp(self):
AnalysisDataService.clear()
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.data_context = MuonDataContext(self.loaded_data)
self.gui_context = MuonGuiContext()
self.group_pair_context = MuonGroupPairContext()
self.gui_context.update({'RebinType': 'None'})
self.context = MuonContext(muon_data_context=self.data_context, muon_gui_context=self.gui_context, muon_group_context=self.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', '')
def test_reset_groups_and_pairs_to_default(self):
self.assertEquals(self.group_pair_context.group_names, ['fwd', 'bwd'])
self.assertEquals(self.group_pair_context.pair_names, ['long'])
def test_calculate_group_calculates_group_for_given_run(self):
counts_workspace, asymmetry_workspace = self.context.calculate_group('fwd', run=[19489])
self.assertEquals(type(counts_workspace), Workspace2D)
self.assertEquals(type(counts_workspace), Workspace2D)
def test_calculate_pair_calculates_pair_for_given_run(self):
pair_asymmetry = self.context.calculate_pair('long', run=[19489])
self.assertEquals(type(pair_asymmetry), Workspace2D)
def test_show_all_groups_calculates_and_shows_all_groups(self):
self.context.show_all_groups()
self.assertEquals(AnalysisDataService.getObjectNames(), ['EMU19489', 'EMU19489 Groups', 'EMU19489; Group; bwd; Asymmetry; #1',
'EMU19489; Group; bwd; Counts; #1', 'EMU19489; Group; fwd; Asymmetry; #1',
'EMU19489; Group; fwd; Counts; #1', 'Muon Data'])
def test_that_show_all_calculates_and_shows_all_groups_with_rebin(self):
self.gui_context['RebinType'] = 'Fixed'
self.gui_context['RebinFixed'] = 2
self.context.show_all_groups()
self.assertEquals(AnalysisDataService.getObjectNames(),
['EMU19489', 'EMU19489 Groups', 'EMU19489; Group; bwd; Asymmetry; #1', 'EMU19489; Group; bwd; Asymmetry; Rebin; #1',
'EMU19489; Group; bwd; Counts; #1', 'EMU19489; Group; bwd; Counts; Rebin; #1',
'EMU19489; Group; fwd; Asymmetry; #1', 'EMU19489; Group; fwd; Asymmetry; Rebin; #1',
'EMU19489; Group; fwd; Counts; #1', 'EMU19489; Group; fwd; Counts; Rebin; #1', 'Muon Data'])
def test_show_all_pairs_calculates_and_shows_all_pairs(self):
self.context.show_all_pairs()
self.assertEquals(AnalysisDataService.getObjectNames(), ['EMU19489', 'EMU19489 Pairs', 'EMU19489; Pair Asym; long; #1', 'Muon Data'])
def test_that_show_all_calculates_and_shows_all_pairs_with_rebin(self):
self.gui_context['RebinType'] = 'Fixed'
self.gui_context['RebinFixed'] = 2
self.context.show_all_pairs()
self.assertEquals(AnalysisDataService.getObjectNames(),
['EMU19489', 'EMU19489 Pairs', 'EMU19489; Pair Asym; long; #1', 'EMU19489; Pair Asym; long; Rebin; #1', 'Muon Data'])
def test_update_current_data_sets_current_run_in_data_context(self):
self.context.update_current_data()
self.assertEquals(self.data_context.current_data, self.load_result)
def test_update_current_data_sets_groups_and_pairs(self):
self.context.update_current_data()
self.assertEquals(self.group_pair_context.pair_names, ['long'])
self.assertEquals(self.group_pair_context.group_names, ['fwd', 'bwd'])
def test_show_raw_data_puts_raw_data_into_the_ADS(self):
self.context.show_raw_data()
self.assertEquals(AnalysisDataService.getObjectNames(), ['EMU19489', 'EMU19489 Raw Data', 'EMU19489_raw_data', 'Muon Data'])
def test_that_first_good_data_returns_correctly_when_from_file_chosen_option(self):
self.gui_context.update({'FirstGoodDataFromFile': True})
first_good_data = self.context.first_good_data([19489])
self.assertEquals(first_good_data, 0.11)
def test_first_good_data_returns_correctly_when_manually_specified_used(self):
self.gui_context.update({'FirstGoodDataFromFile': False, 'FirstGoodData': 5})
first_good_data = self.context.first_good_data([19489])
self.assertEquals(first_good_data, 5)
def test_that_last_good_data_returns_correctly_when_from_file_chosen_option(self):
self.gui_context.update({'LastGoodDataFromFile': True})
last_good_data = self.context.last_good_data([19489])
self.assertEquals(last_good_data, 31.76)
def test_last_good_data_returns_correctly_when_manually_specified_used(self):
self.gui_context.update({'LastGoodDataFromFile': False, 'LastGoodData': 5})
last_good_data = self.context.last_good_data([19489])
self.assertEquals(last_good_data, 5)
def test_that_dead_time_table_from_ADS_returns_table_name(self):
self.gui_context.update({'DeadTimeSource': 'FromADS', 'DeadTimeTable': 'deadtime_table_name'})
deadtime_table = self.context.dead_time_table([19489])
self.assertEquals(deadtime_table, 'deadtime_table_name')