def test_missing_property_is_detected(self):
sample_properties = {
"cylinder_sample_height": 4.0,
"cylinder_sample_radius": 0.25,
"cylinder_position": [0., 0., 0.],
"chemical_formula": "V"
}
# Test each key one at a time
for blacklisted_key in iterkeys(sample_properties):
# Force python to make a shallow copy
modified_dict = sample_properties.copy()
modified_dict.pop(blacklisted_key)
# Check that is raises an error
with assertRaisesRegex(
self, KeyError,
"The following key was not found in the advanced configuration"
):
absorb_corrections.create_vanadium_sample_details_obj(
config_dict=modified_dict)
# Then check the error actually has the key name in it
with assertRaisesRegex(self, KeyError, blacklisted_key):
absorb_corrections.create_vanadium_sample_details_obj(
config_dict=modified_dict)
def calculate_van_absorb_corrections(ws_to_correct, multiple_scattering):
absorb_dict = hrpd_advanced_config.absorption_correction_params
sample_details_obj = absorb_corrections.create_vanadium_sample_details_obj(config_dict=absorb_dict)
ws_to_correct = absorb_corrections.run_cylinder_absorb_corrections(
ws_to_correct=ws_to_correct, multiple_scattering=multiple_scattering, sample_details_obj=sample_details_obj,
is_vanadium=True)
return ws_to_correct
def calculate_van_absorb_corrections(ws_to_correct, multiple_scattering):
absorb_dict = hrpd_advanced_config.absorption_correction_params
sample_details_obj = absorb_corrections.create_vanadium_sample_details_obj(config_dict=absorb_dict)
ws_to_correct = absorb_corrections.run_cylinder_absorb_corrections(
ws_to_correct=ws_to_correct, multiple_scattering=multiple_scattering, sample_details_obj=sample_details_obj,
is_vanadium=True)
return ws_to_correct
def calculate_van_absorb_corrections(ws_to_correct, multiple_scattering, is_vanadium):
mantid.MaskDetectors(ws_to_correct, SpectraList=list(range(1, 55)))
absorb_dict = polaris_advanced_config.absorption_correction_params
sample_details_obj = absorb_corrections.create_vanadium_sample_details_obj(config_dict=absorb_dict)
ws_to_correct = absorb_corrections.run_cylinder_absorb_corrections(
ws_to_correct=ws_to_correct, multiple_scattering=multiple_scattering, sample_details_obj=sample_details_obj,
is_vanadium=is_vanadium)
return ws_to_correct
def calculate_van_absorb_corrections(ws_to_correct, multiple_scattering, is_vanadium):
mantid.MaskDetectors(ws_to_correct, SpectraList=list(range(1, 55)))
absorb_dict = polaris_advanced_config.absorption_correction_params
sample_details_obj = absorb_corrections.create_vanadium_sample_details_obj(config_dict=absorb_dict)
ws_to_correct = absorb_corrections.run_cylinder_absorb_corrections(
ws_to_correct=ws_to_correct, multiple_scattering=multiple_scattering, sample_details_obj=sample_details_obj,
is_vanadium=is_vanadium)
return ws_to_correct
def calculate_van_absorb_corrections(ws_to_correct, multiple_scattering, is_vanadium):
# First 100 detectors are monitors or not connected to DAE
mantid.MaskDetectors(ws_to_correct, SpectraList=range(1, 101))
absorb_dict = gem_advanced_config.absorption_correction_params
sample_details_obj = absorb_corrections.create_vanadium_sample_details_obj(config_dict=absorb_dict)
ws_to_correct = absorb_corrections.run_cylinder_absorb_corrections(
ws_to_correct=ws_to_correct, multiple_scattering=multiple_scattering, sample_details_obj=sample_details_obj,
is_vanadium=True)
return ws_to_correct
def test_missing_property_is_detected(self):
sample_properties = {
"cylinder_sample_height": 4.0,
"cylinder_sample_radius": 0.25,
"cylinder_position": [0., 0., 0.],
"chemical_formula": "V"
}
ws = mantid.CreateSampleWorkspace(Function='Flat background', NumBanks=1, BankPixelWidth=1, XMax=2, BinWidth=1)
# Test each key one at a time
for blacklisted_key in iterkeys(sample_properties):
# Force python to make a shallow copy
modified_dict = sample_properties.copy()
modified_dict.pop(blacklisted_key)
# Check that is raises an error
with assertRaisesRegex(self, KeyError, "The following key was not found in the advanced configuration"):
ws = absorb_corrections.create_vanadium_sample_details_obj(config_dict=modified_dict)
# Then check the error actually has the key name in it
with assertRaisesRegex(self, KeyError, blacklisted_key):
ws = absorb_corrections.create_vanadium_sample_details_obj(config_dict=modified_dict)
