def test_sample_is_set_correctly(self):
sample_details = SampleDetails(height=4.0, radius=0.25, center=[0., 0., 0.], shape="cylinder")
sample_details.set_material(chemical_formula="V")
ws = mantid.CreateSampleWorkspace(Function='Flat background', NumBanks=1, BankPixelWidth=1, XMax=10, BinWidth=1)
ws = absorb_corrections.run_cylinder_absorb_corrections(ws_to_correct=ws, multiple_scattering=False,
sample_details_obj=sample_details, is_vanadium=True)
self.assertAlmostEqual(ws.dataY(0)[2], 1.16864808, delta=1e-8)
self.assertAlmostEqual(ws.dataY(0)[5], 1.16872761, delta=1e-8)
self.assertAlmostEqual(ws.dataY(0)[9], 1.16883365, delta=1e-8)
def test_sample_is_set_correctly(self):
sample_details = SampleDetails(height=4.0, radius=0.25, center=[0., 0., 0.], shape="cylinder")
sample_details.set_material(chemical_formula="V")
ws = mantid.CreateSampleWorkspace(Function='Flat background', NumBanks=1, BankPixelWidth=1, XMax=10, BinWidth=1)
ws = absorb_corrections.run_cylinder_absorb_corrections(ws_to_correct=ws, multiple_scattering=False,
sample_details_obj=sample_details, is_vanadium=True)
self.assertAlmostEqual(ws.dataY(0)[2], 1.16864808, delta=1e-8)
self.assertAlmostEqual(ws.dataY(0)[5], 1.16872761, delta=1e-8)
self.assertAlmostEqual(ws.dataY(0)[9], 1.16883365, delta=1e-8)
def test_generate_sample_material(self):
# Create mock SampleDetails
sample_details = SampleDetails(height=1.0, radius=1.0,
center=[0.0, 0.0, 0.0])
sample_details.set_material(chemical_formula='Si', number_density=1.5)
sample_details.set_material_properties(absorption_cross_section=123,
scattering_cross_section=456)
# Run test
result = common.generate_sample_material(sample_details)
# Validate
expected = {'ChemicalFormula': 'Si',
'SampleNumberDensity': 1.5,
'AttenuationXSection': 123.0,
'ScatteringXSection': 456.0}
self.assertEquals(result, expected)
def test_generate_sample_geometry(self):
# Create mock SampleDetails
sample_details = SampleDetails(height=4.0, radius=3.0,
center=[0.5, 1.0, -3.2], shape='cylinder')
# Run test
result = common.generate_sample_geometry(sample_details)
# Validate result
expected = {'Shape': 'Cylinder',
'Height': 4.0,
'Radius': 3.0,
'Center': [0.5, 1.0, -3.2]}
self.assertEqual(result, expected)
def test_generate_sample_material(self):
# Create mock SampleDetails
sample_details = SampleDetails(height=1.0, radius=1.0,
center=[0.0, 0.0, 0.0])
sample_details.set_material(chemical_formula='Si', number_density=1.5)
sample_details.set_material_properties(absorption_cross_section=123,
scattering_cross_section=456)
# Run test
result = common.generate_sample_material(sample_details)
# Validate
expected = {'ChemicalFormula': 'Si',
'SampleNumberDensity': 1.5,
'AttenuationXSection': 123.0,
'ScatteringXSection': 456.0}
self.assertEqual(result, expected)