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'})