def test_can_workspaces_returned_if_save_can_selected_no_merged(self):
reduction_packages = [self.reduction_package]
names_to_save = get_all_names_to_save(reduction_packages, True)
self.assertEqual(names_to_save, {'lab_workspace', 'hab_workspace',
'reduced_lab_can', 'reduced_hab_can',
'reduced_lab_sample', 'reduced_hab_sample'})
def test_hab_and_lab_workspaces_returned_if_merged_workspace_not_present(
self):
state = mock.MagicMock()
workspaces = ['Sample', 'Transmission', 'Direct']
monitors = ['monitor1']
reduction_package = ReductionPackage(state, workspaces, monitors)
lab_workspace = CreateSampleWorkspace(Function='Flat background',
NumBanks=1,
BankPixelWidth=1,
NumEvents=1,
XMin=1,
XMax=14,
BinWidth=2)
hab_workspace = CreateSampleWorkspace(Function='Flat background',
NumBanks=1,
BankPixelWidth=1,
NumEvents=1,
XMin=1,
XMax=14,
BinWidth=2)
reduction_package.reduced_lab = lab_workspace
reduction_package.reduced_hab = hab_workspace
reduction_packages = [reduction_package]
names_to_save = get_all_names_to_save(reduction_packages)
self.assertEqual(names_to_save, set(['lab_workspace',
'hab_workspace']))
def test_can_workspaces_returned_if_save_can_selected(self):
names_to_save = get_all_names_to_save([self.reduction_package_merged], True)
names = {'merged_workspace', 'lab_workspace', 'hab_workspace',
'reduced_lab_can', 'reduced_hab_can',
'reduced_lab_sample', 'reduced_hab_sample'}
names_expected = [([name], [], []) for name in names]
self.assertEqual(sorted(names_to_save), sorted(names_expected))
def test_hab_and_lab_workspaces_returned_if_merged_workspace_not_present(
self):
reduction_packages = [self.reduction_package]
names_to_save = get_all_names_to_save(reduction_packages,
save_can=False)
self.assertEqual(names_to_save, {'lab_workspace', 'hab_workspace'})
def test_transmission_names_added_to_correct_workspaces_when_not_saving_can(self):
reduction_packages = [self.reduction_package_transmissions]
names_to_save = get_all_names_to_save(reduction_packages, False)
names_expected = {('lab_workspace', 'transmission', 'transmission_can'),
('hab_workspace', 'transmission', 'transmission_can')}
self.assertEqual(names_to_save, names_expected)
def test_can_workspaces_returned_if_save_can_selected(self):
reduction_packages = [self.reduction_package_merged]
names_to_save = get_all_names_to_save(reduction_packages, True)
names = {'merged_workspace', 'lab_workspace', 'hab_workspace',
'reduced_lab_can', 'reduced_hab_can',
'reduced_lab_sample', 'reduced_hab_sample'}
names_expected = set([(name, '', '') for name in names])
self.assertEqual(names_to_save, names_expected)
def test_transmission_names_added_to_correct_workspaces_when_not_saving_can(self):
reduction_packages = [self.reduction_package_transmissions]
names_to_save = get_all_names_to_save(reduction_packages, False)
names_expected = [(['lab_workspace'], ['transmission'], ['transmission_can']),
(['hab_workspace'], ['transmission'], ['transmission_can'])]
self.assertEqual(names_to_save, names_expected)
def test_can_workspaces_returned_if_save_can_selected_no_merged(self):
reduction_packages = [self.reduction_package]
names_to_save = get_all_names_to_save(reduction_packages, True)
self.assertEqual(
names_to_save, {
'lab_workspace', 'hab_workspace', 'reduced_lab_can',
'reduced_hab_can', 'reduced_lab_sample', 'reduced_hab_sample'
})
def test_can_workspaces_returned_if_save_can_selected_no_merged(self):
reduction_packages = [self.reduction_package]
names_to_save = get_all_names_to_save(reduction_packages, True)
names = {
'lab_workspace', 'hab_workspace', 'reduced_lab_can',
'reduced_hab_can', 'reduced_lab_sample', 'reduced_hab_sample'
}
names_expected = set([(name, '', '') for name in names])
self.assertEqual(names_to_save, names_expected)
def test_transmission_names_added_to_unsubtracted_can_and_sample(self):
reduction_packages = [self.reduction_package_transmissions]
names_to_save = get_all_names_to_save(reduction_packages, True)
names_expected = {('lab_workspace', 'transmission', 'transmission_can'),
('hab_workspace', 'transmission', 'transmission_can'),
('reduced_lab_can', '', 'transmission_can'),
('reduced_hab_can', '', 'transmission_can'),
('reduced_lab_sample', 'transmission', ''),
('reduced_hab_sample', 'transmission', '')}
self.assertEqual(names_to_save, names_expected)
def test_transmission_names_added_to_unsubtracted_can_and_sample(self):
reduction_packages = [self.reduction_package_transmissions]
names_to_save = get_all_names_to_save(reduction_packages, True)
names_expected = [(['lab_workspace'], ['transmission'], ['transmission_can']),
(['hab_workspace'], ['transmission'], ['transmission_can']),
(['reduced_lab_can'], [], ['transmission_can']),
(['reduced_hab_can'], [], ['transmission_can']),
(['reduced_lab_sample'], ['transmission'], []),
(['reduced_hab_sample'], ['transmission'], [])]
self.assertEqual(names_to_save, names_expected)
def test_returns_merged_name_if_present(self):
reduction_packages = [self.reduction_package_merged]
names_to_save = get_all_names_to_save(reduction_packages,
save_can=False)
self.assertEqual(names_to_save, {('merged_workspace', '', '')})
def test_hab_and_lab_workspaces_returned_if_merged_workspace_not_present(self):
reduction_packages = [self.reduction_package]
names_to_save = get_all_names_to_save(reduction_packages, save_can=False)
self.assertEqual(names_to_save, {'lab_workspace', 'hab_workspace'})
def test_returns_merged_name_if_present(self):
reduction_packages = [self.reduction_package_merged]
names_to_save = get_all_names_to_save(reduction_packages, save_can=False)
self.assertEqual(names_to_save, {'merged_workspace'})
