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)